twblue/src/controller/blueski/handler.py

from __future__ import annotations

import logging
import wx
import asyncio
import output
from mysc.thread_utils import call_threaded
from wxUI.dialogs.blueski.showUserProfile import ShowUserProfileDialog
from typing import Any
import languageHandler  # Ensure _() injection

logger = logging.getLogger(__name__)


class Handler:
    """Handler for Bluesky integration: creates minimal buffers."""

    def __init__(self):
        super().__init__()
        self.menus = dict(
            compose="&Post",
        )
        self.item_menu = "&Post"

    def create_buffers(self, session, createAccounts=True, controller=None):
        name = session.get_name()
        if createAccounts:
            from pubsub import pub
            pub.sendMessage("core.create_account", name=name, session_id=session.session_id, logged=session.logged)
        
        if not session.logged:
            logger.debug(f"Session {session.session_id} is not logged in, skipping timeline buffer creation.")
            return
        if name not in controller.accounts:
            controller.accounts.append(name)

        root_position = controller.view.search(name, name)
        # Discover/home timeline
        from pubsub import pub
        pub.sendMessage(
            "createBuffer",
            buffer_type="home_timeline",
            session_type="blueski",
            buffer_title=_("Discover"),
            parent_tab=root_position,
            start=True,
            kwargs=dict(parent=controller.view.nb, name="home_timeline", session=session)
        )
        # Home (Following-only timeline - reverse-chronological)
        pub.sendMessage(
            "createBuffer",
            buffer_type="following_timeline",
            session_type="blueski",
            buffer_title=_("Home"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="following_timeline", session=session)
        )
        # Mentions (replies, mentions, quotes)
        pub.sendMessage(
            "createBuffer",
            buffer_type="MentionsBuffer",
            session_type="blueski",
            buffer_title=_("Mentions"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="mentions", session=session)
        )
        # Notifications
        pub.sendMessage(
            "createBuffer",
            buffer_type="notifications",
            session_type="blueski",
            buffer_title=_("Notifications"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="notifications", session=session)
        )
        # Sent posts
        pub.sendMessage(
            "createBuffer",
            buffer_type="SentBuffer",
            session_type="blueski",
            buffer_title=_("Sent"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="sent", session=session)
        )
        # Likes
        pub.sendMessage(
            "createBuffer",
            buffer_type="likes",
            session_type="blueski",
            buffer_title=_("Likes"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="likes", session=session)
        )
        # Followers
        pub.sendMessage(
            "createBuffer",
            buffer_type="FollowersBuffer",
            session_type="blueski",
            buffer_title=_("Followers"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="followers", session=session)
        )
        # Followings (Users you follow)
        pub.sendMessage(
            "createBuffer",
            buffer_type="FollowingBuffer",
            session_type="blueski",
            buffer_title=_("Followings"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="following", session=session)
        )
        # Blocks
        pub.sendMessage(
            "createBuffer",
            buffer_type="BlocksBuffer",
            session_type="blueski",
            buffer_title=_("Blocked Users"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="blocked", session=session)
        )
        # Chats
        pub.sendMessage(
            "createBuffer",
            buffer_type="ConversationListBuffer",
            session_type="blueski",
            buffer_title=_("Chats"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="direct_messages", session=session)
        )

        # Timelines container
        pub.sendMessage(
            "createBuffer",
            buffer_type="EmptyBuffer",
            session_type="base",
            buffer_title=_("Timelines"),
            parent_tab=root_position,
            start=False,
            kwargs=dict(parent=controller.view.nb, name="timelines", account=name)
        )
        timelines_position = controller.view.search("timelines", name)

        # Saved user timelines
        try:
            timelines = session.settings["other_buffers"].get("timelines")
            if timelines is None:
                timelines = []
            if isinstance(timelines, str):
                timelines = [t for t in timelines.split(",") if t]
            for actor in timelines:
                handle = actor
                try:
                    if isinstance(actor, str) and actor.startswith("did:"):
                        profile = session.get_profile(actor)
                        if profile:
                            def g(obj, key, default=None):
                                if isinstance(obj, dict):
                                    return obj.get(key, default)
                                return getattr(obj, key, default)
                            handle = g(profile, "handle") or actor
                except Exception:
                    handle = actor
                title = _("Timeline for {user}").format(user=handle)
                pub.sendMessage(
                    "createBuffer",
                    buffer_type="UserTimeline",
                    session_type="blueski",
                    buffer_title=title,
                    parent_tab=timelines_position,
                    start=False,
                    kwargs=dict(parent=controller.view.nb, name=f"{handle}-timeline", session=session, actor=actor, handle=handle)
                )
        except Exception:
            logger.exception("Failed to restore Bluesky timeline buffers")

        # Saved followers/following timelines
        try:
            followers = session.settings["other_buffers"].get("followers_timelines")
            if followers is None:
                followers = []
            if isinstance(followers, str):
                followers = [t for t in followers.split(",") if t]
            for actor in followers:
                handle = actor
                try:
                    if isinstance(actor, str) and actor.startswith("did:"):
                        profile = session.get_profile(actor)
                        if profile:
                            def g(obj, key, default=None):
                                if isinstance(obj, dict):
                                    return obj.get(key, default)
                                return getattr(obj, key, default)
                            handle = g(profile, "handle") or actor
                except Exception:
                    handle = actor
                title = _("Followers for {user}").format(user=handle)
                pub.sendMessage(
                    "createBuffer",
                    buffer_type="FollowersBuffer",
                    session_type="blueski",
                    buffer_title=title,
                    parent_tab=timelines_position,
                    start=False,
                    kwargs=dict(parent=controller.view.nb, name=f"{handle}-followers", session=session, actor=actor, handle=handle)
                )
        except Exception:
            logger.exception("Failed to restore Bluesky followers buffers")

        try:
            following = session.settings["other_buffers"].get("following_timelines")
            if following is None:
                following = []
            if isinstance(following, str):
                following = [t for t in following.split(",") if t]
            for actor in following:
                handle = actor
                try:
                    if isinstance(actor, str) and actor.startswith("did:"):
                        profile = session.get_profile(actor)
                        if profile:
                            def g(obj, key, default=None):
                                if isinstance(obj, dict):
                                    return obj.get(key, default)
                                return getattr(obj, key, default)
                            handle = g(profile, "handle") or actor
                except Exception:
                    handle = actor
                title = _("Following for {user}").format(user=handle)
                pub.sendMessage(
                    "createBuffer",
                    buffer_type="FollowingBuffer",
                    session_type="blueski",
                    buffer_title=title,
                    parent_tab=timelines_position,
                    start=False,
                    kwargs=dict(parent=controller.view.nb, name=f"{handle}-following", session=session, actor=actor, handle=handle)
                )
        except Exception:
            logger.exception("Failed to restore Bluesky following buffers")

        # Start the background poller for real-time-like updates
        try:
            session.start_streaming()
        except Exception:
            logger.exception("Failed to start Bluesky streaming for session %s", name)

    def start_buffer(self, controller, buffer):
        """Start a newly created Bluesky buffer."""
        try:
            if hasattr(buffer, "start_stream"):
                buffer.start_stream(mandatory=True, play_sound=False)
            # Enable periodic auto-refresh to simulate real-time updates
            if hasattr(buffer, "enable_auto_refresh"):
                buffer.enable_auto_refresh()
        finally:
            # Ensure we won't try to start it again
            try:
                buffer.needs_init = False
            except Exception:
                pass

    def account_settings(self, buffer, controller):
        """Open a minimal account settings dialog for Bluesky."""
        try:
            current_mode = None
            try:
                current_mode = buffer.session.settings["general"].get("boost_mode")
            except Exception:
                current_mode = None
            ask_default = True if current_mode in (None, "ask") else False

            from wxUI.dialogs.blueski.configuration import AccountSettingsDialog
            dlg = AccountSettingsDialog(controller.view, ask_before_boost=ask_default)
            resp = dlg.ShowModal()
            if resp == wx.ID_OK:
                vals = dlg.get_values()
                boost_mode = "ask" if vals.get("ask_before_boost") else "direct"
                try:
                    buffer.session.settings["general"]["boost_mode"] = boost_mode
                    buffer.session.settings.write()
                except Exception:
                    logger.exception("Failed to persist Bluesky boost_mode setting")
            dlg.Destroy()
        except Exception:
            logger.exception("Error opening Bluesky account settings dialog")

    def user_details(self, buffer):
        """Show user profile dialog for the selected user/post."""
        session = getattr(buffer, "session", None)
        if not session:
            output.speak(_("No active session to view user details."), True)
            return

        item = buffer.get_item() if hasattr(buffer, "get_item") else None
        if not item:
            output.speak(_("No user selected or identified to view details."), True)
            return

        def g(obj, key, default=None):
            if isinstance(obj, dict):
                return obj.get(key, default)
            return getattr(obj, key, default)

        user_ident = None

        # If we're in a user list, the item itself is the user profile dict/model.
        if g(item, "did") or g(item, "handle"):
            user_ident = g(item, "did") or g(item, "handle")
        else:
            author = g(item, "author")
            if not author:
                post = g(item, "post") or g(item, "record")
                author = g(post, "author") if post else None
            if author:
                user_ident = g(author, "did") or g(author, "handle")

        if not user_ident:
            output.speak(_("No user selected or identified to view details."), True)
            return

        parent = getattr(buffer, "buffer", None) or wx.GetApp().GetTopWindow()
        dialog = ShowUserProfileDialog(parent, session, user_ident)
        dialog.ShowModal()
        dialog.Destroy()

    async def handle_action(self, action_name: str, user_id: str, payload: dict[str, Any]) -> dict[str, Any] | None:
        logger.debug("handle_action stub: %s %s %s", action_name, user_id, payload)
        return None

    async def handle_message_command(self, command: str, user_id: str, message_id: str, payload: dict[str, Any]) -> dict[str, Any] | None:
        logger.debug("handle_message_command stub: %s %s %s %s", command, user_id, message_id, payload)
        return None

    async def handle_user_command(self, command: str, user_id: str, target_user_id: str, payload: dict[str, Any]) -> dict[str, Any] | None:
        logger.debug("handle_user_command stub: %s %s %s %s", command, user_id, target_user_id, payload)
        return None

    def add_to_favourites(self, buffer):
        """Standard action for Alt+Win+F"""
        if hasattr(buffer, "add_to_favorites"):
            buffer.add_to_favorites()
        elif hasattr(buffer, "on_like"):
             # Fallback
             buffer.on_like(None)
    
    def remove_from_favourites(self, buffer):
        """Standard action for Alt+Shift+Win+F"""
        if hasattr(buffer, "remove_from_favorites"):
            buffer.remove_from_favorites()
        elif hasattr(buffer, "on_like"):
            buffer.on_like(None)
            
    def follow(self, buffer):
        """Standard action for Ctrl+Win+S"""
        session = getattr(buffer, "session", None)
        if not session:
            output.speak(_("No active session."), True)
            return

        def g(obj, key, default=None):
            if isinstance(obj, dict):
                return obj.get(key, default)
            return getattr(obj, key, default)

        user_ident = None
        item = buffer.get_item() if hasattr(buffer, "get_item") else None
        if item:
            if g(item, "handle") or g(item, "did"):
                user_ident = g(item, "handle") or g(item, "did")
            else:
                author = g(item, "author")
                if not author:
                    post = g(item, "post") or g(item, "record")
                    author = g(post, "author") if post else None
                if author:
                    user_ident = g(author, "handle") or g(author, "did")

        users = [user_ident] if user_ident else []
        from controller.blueski import userActions as user_actions_controller
        user_actions_controller.userActions(session, users)

    def open_conversation(self, controller, buffer):
        """Standard action for Control+Win+C"""
        item = buffer.get_item()
        if not item:
            return

        uri = None
        if hasattr(buffer, "get_selected_item_id"):
            uri = buffer.get_selected_item_id()
        if not uri:
            uri = getattr(item, "uri", None) or (item.get("post", {}).get("uri") if isinstance(item, dict) else None)
        if not uri: return
        
        # Buffer Title
        handle = None
        display_name = None
        if hasattr(buffer, "get_selected_item_author_details"):
            details = buffer.get_selected_item_author_details()
            if details:
                handle = details.get("handle")
        if not handle:
            def g(obj, key, default=None):
                if isinstance(obj, dict):
                    return obj.get(key, default)
                return getattr(obj, key, default)
            author = g(item, "author") or g(g(item, "post"), "author")
            if author:
                handle = g(author, "handle")
                display_name = g(author, "displayName") or g(author, "display_name")
        label = handle or display_name or _("Unknown")
        title = _("Conversation with {0}").format(label)
        
        from pubsub import pub
        pub.sendMessage(
            "createBuffer",
            buffer_type="conversation",
            session_type="blueski",
            buffer_title=title,
            parent_tab=controller.view.search(buffer.session.get_name(), buffer.session.get_name()) if hasattr(buffer.session, "get_name") else None,
            start=True,
            kwargs=dict(parent=controller.view.nb, name=title, session=buffer.session, uri=uri)
        )

    def open_timeline(self, controller, buffer, default="posts"):
        if not hasattr(buffer, "get_item"):
            return
        item = buffer.get_item()
        if not item:
            output.speak(_("No user selected."), True)
            return

        def g(obj, key, default=None):
            if isinstance(obj, dict):
                return obj.get(key, default)
            return getattr(obj, key, default)

        users = []
        handle = None
        if hasattr(buffer, "get_selected_item_author_details"):
            details = buffer.get_selected_item_author_details()
            if details:
                handle = details.get("handle") or details.get("did")
        if not handle:
            if g(item, "handle") or g(item, "did"):
                handle = g(item, "handle") or g(item, "did")
            else:
                author = g(item, "author") or g(g(item, "post"), "author")
                if author:
                    handle = g(author, "handle") or g(author, "did")

        if not handle:
            output.speak(_("No user selected."), True)
            return
        users.append(handle)

        # Add mentioned users if available (facets)
        record = g(g(item, "post"), "record") or g(item, "record")
        facets = g(record, "facets", []) if record else []
        handle_cache = {}

        def resolve_handle(did):
            if did in handle_cache:
                return handle_cache[did]
            try:
                profile = buffer.session.get_profile(did)
                if profile:
                    h = g(profile, "handle")
                    if h:
                        handle_cache[did] = h
                        return h
            except Exception:
                pass
            return None

        self_did = buffer.session.db.get("user_id")
        for facet in facets or []:
            features = g(facet, "features", []) or []
            for feat in features:
                ftype = g(feat, "$type") or g(feat, "py_type") or ""
                if "facet#mention" in ftype:
                    did = g(feat, "did")
                    if not did or did == self_did:
                        continue
                    h = resolve_handle(did)
                    if h and h not in users:
                        users.append(h)

        from wxUI.dialogs.mastodon import userTimeline as userTimelineDialog
        dlg = userTimelineDialog.UserTimeline(users=users, default=default)
        try:
            if hasattr(dlg, "autocompletion"):
                dlg.autocompletion.Enable(False)
        except Exception:
            pass
        if dlg.ShowModal() != wx.ID_OK:
            dlg.Destroy()
            return

        action = dlg.get_action()
        user = dlg.get_user().strip() or handle
        dlg.Destroy()

        if user.startswith("@"):
            user = user[1:]
        user_payload = {"handle": user}
        if action == "posts":
            result = self.open_user_timeline(main_controller=controller, session=buffer.session, user_payload=user_payload)
        elif action == "followers":
            result = self.open_followers_timeline(main_controller=controller, session=buffer.session, user_payload=user_payload)
        elif action == "following":
            result = self.open_following_timeline(main_controller=controller, session=buffer.session, user_payload=user_payload)
        else:
            return

        if asyncio.iscoroutine(result):
            call_threaded(asyncio.run, result)

    def open_followers_timeline(self, main_controller, session, user_payload=None):
        actor, handle = self._resolve_actor(session, user_payload)
        if not actor:
            output.speak(_("No user selected."), True)
            return
        self._open_user_list(main_controller, session, actor, handle, list_type="followers")

    def open_following_timeline(self, main_controller, session, user_payload=None):
        actor, handle = self._resolve_actor(session, user_payload)
        if not actor:
            output.speak(_("No user selected."), True)
            return
        self._open_user_list(main_controller, session, actor, handle, list_type="following")

    def open_user_timeline(self, main_controller, session, user_payload=None):
        """Open posts timeline for a user (Alt+Win+I)."""
        actor, handle = self._resolve_actor(session, user_payload)
        if not actor:
            output.speak(_("No user selected."), True)
            return

        actor, handle = self._resolve_actor(session, {"did": actor, "handle": handle})
        if not handle:
            handle = actor

        account_name = session.get_name()
        list_name = f"{handle}-timeline"
        if main_controller.search_buffer(list_name, account_name):
            index = main_controller.view.search(list_name, account_name)
            if index is not None:
                main_controller.view.change_buffer(index)
            return

        title = _("Timeline for {user}").format(user=handle)
        from pubsub import pub
        pub.sendMessage(
            "createBuffer",
            buffer_type="UserTimeline",
            session_type="blueski",
            buffer_title=title,
            parent_tab=main_controller.view.search("timelines", account_name),
            start=True,
            kwargs=dict(parent=main_controller.view.nb, name=list_name, session=session, actor=actor, handle=handle)
        )
        try:
            timelines = session.settings["other_buffers"].get("timelines")
            if timelines is None:
                timelines = []
            if isinstance(timelines, str):
                timelines = [t for t in timelines.split(",") if t]
            key = actor or handle
            if key in timelines:
                from wxUI import commonMessageDialogs
                commonMessageDialogs.timeline_exist()
                return
            if key:
                timelines.append(key)
                session.settings["other_buffers"]["timelines"] = timelines
                session.settings.write()
        except Exception:
            logger.exception("Failed to persist Bluesky timeline buffer")

    def _resolve_actor(self, session, user_payload):
        def g(obj, key, default=None):
            if isinstance(obj, dict):
                return obj.get(key, default)
            return getattr(obj, key, default)

        actor = None
        handle = None
        if user_payload:
            actor = g(user_payload, "did") or g(user_payload, "handle")
            handle = g(user_payload, "handle") or g(user_payload, "did")
        if isinstance(actor, str):
            actor = actor.strip()
            if actor.startswith("@"):
                actor = actor[1:]
        if isinstance(handle, str):
            handle = handle.strip()
            if handle.startswith("@"):
                handle = handle[1:]
        # Resolve handle -> DID when possible, and keep handle for titles
        try:
            if isinstance(actor, str) and not actor.startswith("did:"):
                profile = session.get_profile(actor)
                if profile:
                    did = g(profile, "did")
                    if did:
                        actor = did
                    if not handle:
                        handle = g(profile, "handle")
        except Exception:
            pass

        if not actor:
            actor = session.db.get("user_id") or session.db.get("user_name")
            handle = session.db.get("user_name") or actor

        if not handle and isinstance(actor, str):
            try:
                if actor.startswith("did:"):
                    profile = session.get_profile(actor)
                    if profile:
                        handle = g(profile, "handle")
            except Exception:
                pass

        return actor, handle

    def _open_user_list(self, main_controller, session, actor, handle, list_type):
        account_name = session.get_name()
        if not handle:
            handle = actor
        own_actor = session.db.get("user_id") or session.db.get("user_name")
        own_handle = session.db.get("user_name")
        if actor == own_actor or (own_handle and actor == own_handle) or (handle and own_handle and handle == own_handle):
            name = "followers" if list_type == "followers" else "following"
            index = main_controller.view.search(name, account_name)
            if index is not None:
                main_controller.view.change_buffer(index)
                return
        list_name = f"{handle}-{list_type}"
        if main_controller.search_buffer(list_name, account_name):
            index = main_controller.view.search(list_name, account_name)
            if index is not None:
                main_controller.view.change_buffer(index)
            return

        settings_key = "followers_timelines" if list_type == "followers" else "following_timelines"
        try:
            stored = session.settings["other_buffers"].get(settings_key)
            if stored is None:
                stored = []
            if isinstance(stored, str):
                stored = [t for t in stored.split(",") if t]
            key = actor or handle
            if key in stored:
                from wxUI import commonMessageDialogs
                commonMessageDialogs.timeline_exist()
                return
        except Exception:
            stored = None

        title = _("Followers for {user}").format(user=handle) if list_type == "followers" else _("Following for {user}").format(user=handle)
        from pubsub import pub
        pub.sendMessage(
            "createBuffer",
            buffer_type="FollowersBuffer" if list_type == "followers" else "FollowingBuffer",
            session_type="blueski",
            buffer_title=title,
            parent_tab=main_controller.view.search("timelines", account_name),
            start=True,
            kwargs=dict(parent=main_controller.view.nb, name=list_name, session=session, actor=actor, handle=handle)
        )
        try:
            if stored is None:
                stored = session.settings["other_buffers"].get(settings_key) or []
                if isinstance(stored, str):
                    stored = [t for t in stored.split(",") if t]
            key = actor or handle
            if key:
                stored.append(key)
                session.settings["other_buffers"][settings_key] = stored
                session.settings.write()
        except Exception:
            logger.exception("Failed to persist Bluesky %s buffer", list_type)

    def delete(self, buffer, controller):
        """Standard action for delete key / menu item"""
        item = buffer.get_item()
        if not item: return

        uri = getattr(item, "uri", None) or (item.get("post", {}).get("uri") if isinstance(item, dict) else None)
        if not uri: return

        import wx
        if wx.MessageBox(_("Are you sure you want to delete this post?"), _("Delete post"), wx.YES_NO | wx.ICON_QUESTION) == wx.YES:
            if buffer.session.delete_post(uri):
                import output
                output.speak(_("Post deleted."))
                # Refresh buffer
                if hasattr(buffer, "start_stream"):
                    buffer.start_stream(mandatory=True, play_sound=False)
            else:
                import output
                output.speak(_("Failed to delete post."))

    def search(self, controller, session):
        """Open search dialog and create search buffer for results."""
        dlg = wx.TextEntryDialog(
            controller.view,
            _("Enter search term:"),
            _("Search Bluesky")
        )
        if dlg.ShowModal() != wx.ID_OK:
            dlg.Destroy()
            return

        query = dlg.GetValue().strip()
        dlg.Destroy()

        if not query:
            return

        # Create unique buffer name for this search
        buffer_name = f"search_{query[:20]}"
        account_name = session.get_name()

        # Check if buffer already exists
        existing = controller.search_buffer(buffer_name, account_name)
        if existing:
            # Navigate to existing buffer
            index = controller.view.search(buffer_name, account_name)
            if index is not None:
                controller.view.change_buffer(index)
                # Refresh search
                existing.search_query = query
                existing.start_stream(mandatory=True, play_sound=False)
            return

        # Create new search buffer
        title = _("Search: {query}").format(query=query)
        from pubsub import pub
        pub.sendMessage(
            "createBuffer",
            buffer_type="SearchBuffer",
            session_type="blueski",
            buffer_title=title,
            parent_tab=controller.view.search(account_name, account_name),
            start=True,
            kwargs=dict(
                parent=controller.view.nb,
                name=buffer_name,
                session=session,
                query=query
            )
        )