twblue/src/wxUI/buffers/atprotosocial/panels.py

# -*- coding: utf-8 -*-
import wx
import languageHandler  # Ensure _() is available
import logging
import wx
import config
from mysc.repeating_timer import RepeatingTimer
import arrow
import arrow
from datetime import datetime

from multiplatform_widgets import widgets

log = logging.getLogger("wxUI.buffers.atprotosocial.panels")


class ATProtoSocialHomeTimelinePanel(object):
    """Minimal Home timeline buffer for Bluesky.

    Exposes a .buffer wx.Panel with a List control and provides
    start_stream()/get_more_items() to fetch items from atproto.
    """

    def __init__(self, parent, name: str, session):
        super().__init__()
        self.session = session
        self.account = session.get_name()
        self.name = name
        self.type = "home_timeline"
        self.invisible = True
        self.needs_init = True
        self.buffer = _HomePanel(parent, name)
        self.buffer.session = session
        self.buffer.name = name
        # Ensure controller can resolve current account from the GUI panel
        self.buffer.account = self.account
        self.items = []  # list of dicts: {uri, author, handle, display_name, text, indexed_at}
        self.cursor = None
        self._auto_timer = None

    def start_stream(self, mandatory=False, play_sound=True):
        """Fetch newest items and render them."""
        try:
            count = self.session.settings["general"]["max_posts_per_call"] or 40
        except Exception:
            count = 40
        try:
            api = self.session._ensure_client()
            # The atproto SDK expects params, not raw kwargs
            try:
                from atproto import models as at_models  # type: ignore
                # Home: algorithmic/default timeline
                try:
                    params = at_models.AppBskyFeedGetTimeline.Params(limit=count)
                    res = api.app.bsky.feed.get_timeline(params)
                except Exception:
                    # Some SDKs may require explicit algorithm for home; try behavioral
                    params = at_models.AppBskyFeedGetTimeline.Params(limit=count, algorithm="behavioral")
                    res = api.app.bsky.feed.get_timeline(params)
            except Exception:
                # Fallback to plain dict params if typed models unavailable
                res = api.app.bsky.feed.get_timeline({"limit": count})
            feed = getattr(res, "feed", [])
            self.cursor = getattr(res, "cursor", None)
            self.items = []
            for it in feed:
                post = getattr(it, "post", None)
                if not post:
                    continue
                # No additional client-side filtering; server distinguishes timelines correctly
                record = getattr(post, "record", None)
                author = getattr(post, "author", None)
                text = getattr(record, "text", "") if record else ""
                handle = getattr(author, "handle", "") if author else ""
                display_name = (
                    getattr(author, "display_name", None)
                    or getattr(author, "displayName", None)
                    or ""
                ) if author else ""
                indexed_at = getattr(post, "indexed_at", None)
                item = {
                    "uri": getattr(post, "uri", ""),
                    "author": display_name or handle,
                    "handle": handle,
                    "display_name": display_name,
                    "text": text,
                    "indexed_at": indexed_at,
                }
                self._append_item(item, to_top=self._reverse())
            # Full rerender to ensure column widths and selection
            self._render_list(replace=True)
            return len(self.items)
        except Exception:
            log.exception("Failed to load Bluesky home timeline")
            self.buffer.list.clear()
            self.buffer.list.insert_item(False, _("Error"), _("Could not load timeline."), "")
            return 0

    def get_more_items(self):
        if not self.cursor:
            return 0
        try:
            api = self.session._ensure_client()
            try:
                from atproto import models as at_models  # type: ignore
                params = at_models.AppBskyFeedGetTimeline.Params(limit=40, cursor=self.cursor)
                res = api.app.bsky.feed.get_timeline(params)
            except Exception:
                res = api.app.bsky.feed.get_timeline({"limit": 40, "cursor": self.cursor})
            feed = getattr(res, "feed", [])
            self.cursor = getattr(res, "cursor", None)
            new_items = []
            for it in feed:
                post = getattr(it, "post", None)
                if not post:
                    continue
                # No additional client-side filtering
                record = getattr(post, "record", None)
                author = getattr(post, "author", None)
                text = getattr(record, "text", "") if record else ""
                handle = getattr(author, "handle", "") if author else ""
                display_name = (
                    getattr(author, "display_name", None)
                    or getattr(author, "displayName", None)
                    or ""
                ) if author else ""
                indexed_at = getattr(post, "indexed_at", None)
                new_items.append({
                    "uri": getattr(post, "uri", ""),
                    "author": display_name or handle,
                    "handle": handle,
                    "display_name": display_name,
                    "text": text,
                    "indexed_at": indexed_at,
                })
            if not new_items:
                return 0
            for it in new_items:
                self._append_item(it, to_top=self._reverse())
            # Render only the newly added slice
            self._render_list(replace=False, start=len(self.items) - len(new_items))
            return len(new_items)
        except Exception:
            log.exception("Failed to load more Bluesky timeline items")
            return 0

    # Alias to integrate with mainController expectations for ATProto
    def load_more_posts(self, *args, **kwargs):
        return self.get_more_items()

    def _reverse(self) -> bool:
        try:
            return bool(self.session.settings["general"].get("reverse_timelines", False))
        except Exception:
            return False

    def _append_item(self, item: dict, to_top: bool = False):
        if to_top:
            self.items.insert(0, item)
        else:
            self.items.append(item)

    def _render_list(self, replace: bool, start: int = 0):
        if replace:
            self.buffer.list.clear()
        for i in range(start, len(self.items)):
            it = self.items[i]
            dt = ""
            if it.get("indexed_at"):
                try:
                    # Mastodon-like date formatting: relative or full date
                    rel = False
                    try:
                        rel = bool(self.session.settings["general"].get("relative_times", False))
                    except Exception:
                        rel = False
                    ts = arrow.get(str(it["indexed_at"]))
                    if rel:
                        dt = ts.humanize(locale=languageHandler.curLang[:2])
                    else:
                        dt = ts.format(_("dddd, MMMM D, YYYY H:m:s"), locale=languageHandler.curLang[:2])
                except Exception:
                    try:
                        dt = str(it["indexed_at"])[:16].replace("T", " ")
                    except Exception:
                        dt = ""
            text = it.get("text", "").replace("\n", " ")
            if len(text) > 200:
                text = text[:197] + "..."
            # Display name and handle like Mastodon: "Display (@handle)"
            author_col = it.get("author", "")
            handle = it.get("handle", "")
            if handle and it.get("display_name"):
                author_col = f"{it.get('display_name')} (@{handle})"
            elif handle and not author_col:
                author_col = f"@{handle}"
            self.buffer.list.insert_item(False, author_col, text, dt)

    # For compatibility with controller expectations
    def save_positions(self):
        try:
            pos = self.buffer.list.get_selected()
            self.session.db[self.name + "_pos"] = pos
        except Exception:
            pass

    # Support actions that need a selected item identifier (e.g., reply)
    def get_selected_item_id(self):
        try:
            idx = self.buffer.list.get_selected()
            if idx is None or idx < 0:
                return None
            return self.items[idx].get("uri")
        except Exception:
            return None

    def get_message(self):
        try:
            idx = self.buffer.list.get_selected()
            if idx is None or idx < 0:
                return ""
            it = self.items[idx]
            author = it.get("display_name") or it.get("author") or ""
            handle = it.get("handle")
            if handle:
                author = f"{author} (@{handle})" if author else f"@{handle}"
            text = it.get("text", "").replace("\n", " ")
            dt = ""
            if it.get("indexed_at"):
                try:
                    dt = str(it["indexed_at"])[:16].replace("T", " ")
                except Exception:
                    dt = ""
            parts = [p for p in [author, text, dt] if p]
            return ", ".join(parts)
        except Exception:
            return ""

    # Auto-refresh support (polling) to simulate near real-time updates
    def _periodic_refresh(self):
        try:
            # Ensure UI updates happen on the main thread
            wx.CallAfter(self.start_stream, False, False)
        except Exception:
            pass

    def enable_auto_refresh(self, seconds: int | None = None):
        try:
            if self._auto_timer:
                return
            if seconds is None:
                # Use global update_period (minutes) → seconds; minimum 15s
                minutes = config.app["app-settings"].get("update_period", 2)
                seconds = max(15, int(minutes * 60))
            self._auto_timer = RepeatingTimer(seconds, self._periodic_refresh)
            self._auto_timer.start()
        except Exception:
            log.exception("Failed to enable auto refresh for Bluesky panel %s", self.name)

    def disable_auto_refresh(self):
        try:
            if self._auto_timer:
                self._auto_timer.stop()
                self._auto_timer = None
        except Exception:
            pass


class _HomePanel(wx.Panel):
    def __init__(self, parent, name):
        super().__init__(parent, name=name)
        self.name = name
        self.type = "home_timeline"
        sizer = wx.BoxSizer(wx.VERTICAL)
        self.list = widgets.list(self, _("Author"), _("Post"), _("Date"), style=wx.LC_REPORT | wx.LC_SINGLE_SEL)
        self.list.set_windows_size(0, 120)
        self.list.set_windows_size(1, 360)
        self.list.set_windows_size(2, 150)
        self.list.set_size()
        sizer.Add(self.list.list, 1, wx.EXPAND | wx.ALL, 0)
        self.SetSizer(sizer)


class ATProtoSocialFollowingTimelinePanel(ATProtoSocialHomeTimelinePanel):
    """Following-only timeline (reverse-chronological)."""

    def __init__(self, parent, name: str, session):
        super().__init__(parent, name, session)
        self.type = "following_timeline"
        # Make sure the underlying wx panel also reflects this type
        try:
            self.buffer.type = "following_timeline"
        except Exception:
            pass

    def start_stream(self, mandatory=False, play_sound=True):
        try:
            count = self.session.settings["general"]["max_posts_per_call"] or 40
        except Exception:
            count = 40
        try:
            api = self.session._ensure_client()
            # Following timeline via reverse-chronological algorithm on get_timeline
            # Use plain dict to avoid typed-model mismatches across SDK versions
            res = api.app.bsky.feed.get_timeline({"limit": count, "algorithm": "reverse-chronological"})
            feed = getattr(res, "feed", [])
            self.cursor = getattr(res, "cursor", None)
            self.items = []
            for it in feed:
                post = getattr(it, "post", None)
                if not post:
                    continue
                record = getattr(post, "record", None)
                author = getattr(post, "author", None)
                text = getattr(record, "text", "") if record else ""
                handle = getattr(author, "handle", "") if author else ""
                display_name = (
                    getattr(author, "display_name", None)
                    or getattr(author, "displayName", None)
                    or ""
                ) if author else ""
                indexed_at = getattr(post, "indexed_at", None)
                item = {
                    "uri": getattr(post, "uri", ""),
                    "author": display_name or handle,
                    "handle": handle,
                    "display_name": display_name,
                    "text": text,
                    "indexed_at": indexed_at,
                }
                self._append_item(item, to_top=self._reverse())
            self._render_list(replace=True)
            return len(self.items)
        except Exception:
            log.exception("Failed to load Bluesky following timeline")
            self.buffer.list.clear()
            self.buffer.list.insert_item(False, _("Error"), _("Could not load timeline."), "")
            return 0

    def get_more_items(self):
        if not self.cursor:
            return 0
        try:
            api = self.session._ensure_client()
            # Pagination via reverse-chronological algorithm on get_timeline
            res = api.app.bsky.feed.get_timeline({"limit": 40, "cursor": self.cursor, "algorithm": "reverse-chronological"})
            feed = getattr(res, "feed", [])
            self.cursor = getattr(res, "cursor", None)
            new_items = []
            for it in feed:
                post = getattr(it, "post", None)
                if not post:
                    continue
                record = getattr(post, "record", None)
                author = getattr(post, "author", None)
                text = getattr(record, "text", "") if record else ""
                handle = getattr(author, "handle", "") if author else ""
                display_name = (
                    getattr(author, "display_name", None)
                    or getattr(author, "displayName", None)
                    or ""
                ) if author else ""
                indexed_at = getattr(post, "indexed_at", None)
                new_items.append({
                    "uri": getattr(post, "uri", ""),
                    "author": display_name or handle,
                    "handle": handle,
                    "display_name": display_name,
                    "text": text,
                    "indexed_at": indexed_at,
                })
            if not new_items:
                return 0
            for it in new_items:
                self._append_item(it, to_top=self._reverse())
            self._render_list(replace=False, start=len(self.items) - len(new_items))
            return len(new_items)
        except Exception:
            log.exception("Failed to load more items for following timeline")
            return 0