twblue/src/wxUI/dialogs/blueski/showUserProfile.py

# -*- coding: utf-8 -*-
import wx
import logging
import languageHandler
import builtins
from threading import Thread

_ = getattr(builtins, "_", lambda s: s)

logger = logging.getLogger(__name__)

class ShowUserProfileDialog(wx.Dialog):
    def __init__(self, parent, session, user_identifier: str): # user_identifier can be DID or handle
        super(ShowUserProfileDialog, self).__init__(parent, title=_("User Profile"), style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER)

        self.session = session
        self.user_identifier = user_identifier
        self.profile_data = None # Will store the formatted profile dict
        self.target_user_did = None # Will store the resolved DID of the profile being viewed

        self._init_ui()
        self.SetMinSize((400, 300))
        self.CentreOnParent()

        Thread(target=self.load_profile_data, daemon=True).start()

    def _init_ui(self):
        panel = wx.Panel(self)
        main_sizer = wx.BoxSizer(wx.VERTICAL)

        # Profile Info Section (StaticTexts for labels and values)
        self.info_grid_sizer = wx.FlexGridSizer(cols=2, vgap=5, hgap=5)
        self.info_grid_sizer.AddGrowableCol(1, 1)

        fields = [
            (_("&Name:"), "displayName"), (_("&Handle:"), "handle"), (_("&DID:"), "did"),
            (_("&Followers:"), "followersCount"), (_("&Following:"), "followsCount"), (_("&Posts:"), "postsCount"),
            (_("&Bio:"), "description")
        ]
        self.profile_field_ctrls = {}

        for label_text, data_key in fields:
            lbl = wx.StaticText(panel, label=label_text)
            style = wx.TE_READONLY | wx.TE_PROCESS_TAB
            if data_key == "description":
                style |= wx.TE_MULTILINE
            else:
                style |= wx.BORDER_NONE
            val_ctrl = wx.TextCtrl(panel, style=style)
            if data_key != "description": # Make it look like a label
                 val_ctrl.SetBackgroundColour(panel.GetBackgroundColour())
            val_ctrl.AcceptsFocusFromKeyboard = lambda: True

            self.info_grid_sizer.Add(lbl, 0, wx.ALIGN_RIGHT | wx.ALIGN_TOP | wx.ALL, 2)
            self.info_grid_sizer.Add(val_ctrl, 1, wx.EXPAND | wx.ALL, 2)
            self.profile_field_ctrls[data_key] = val_ctrl

        # Avatar and Banner (placeholders for now)
        self.avatar_text = wx.StaticText(panel, label=_("Avatar URL: ") + _("N/A"))
        self.info_grid_sizer.Add(self.avatar_text, 0, wx.ALIGN_RIGHT | wx.ALIGN_TOP | wx.ALL, 2)
        self.banner_text = wx.StaticText(panel, label=_("Banner URL: ") + _("N/A"))
        self.info_grid_sizer.Add(self.banner_text, 0, wx.ALIGN_RIGHT | wx.ALIGN_TOP | wx.ALL, 2)


        main_sizer.Add(self.info_grid_sizer, 1, wx.EXPAND | wx.ALL, 10)

        # Action Buttons
        actions_sizer = wx.BoxSizer(wx.HORIZONTAL)
        # Placeholders, enable/disable logic will be in load_profile_data
        self.follow_btn = wx.Button(panel, label=_("Follow"))
        self.unfollow_btn = wx.Button(panel, label=_("Unfollow"))
        self.mute_btn = wx.Button(panel, label=_("Mute"))
        self.unmute_btn = wx.Button(panel, label=_("Unmute"))
        self.block_btn = wx.Button(panel, label=_("Block"))
        # Unblock might be more complex if it needs block URI or is shown conditionally

        self.follow_btn.Bind(wx.EVT_BUTTON, lambda evt, cmd="follow_user": self.on_user_action(evt, cmd))
        self.unfollow_btn.Bind(wx.EVT_BUTTON, lambda evt, cmd="unfollow_user": self.on_user_action(evt, cmd))
        self.mute_btn.Bind(wx.EVT_BUTTON, lambda evt, cmd="mute_user": self.on_user_action(evt, cmd))
        self.unmute_btn.Bind(wx.EVT_BUTTON, lambda evt, cmd="unmute_user": self.on_user_action(evt, cmd))
        self.block_btn.Bind(wx.EVT_BUTTON, lambda evt, cmd="block_user": self.on_user_action(evt, cmd))
        self.unblock_btn = wx.Button(panel, label=_("Unblock")) # Added unblock button
        self.unblock_btn.Bind(wx.EVT_BUTTON, lambda evt, cmd="unblock_user": self.on_user_action(evt, cmd))


        actions_sizer.Add(self.follow_btn, 0, wx.ALL, 3)
        actions_sizer.Add(self.unfollow_btn, 0, wx.ALL, 3)
        actions_sizer.Add(self.mute_btn, 0, wx.ALL, 3)
        actions_sizer.Add(self.unmute_btn, 0, wx.ALL, 3)
        actions_sizer.Add(self.block_btn, 0, wx.ALL, 3)
        actions_sizer.Add(self.unblock_btn, 0, wx.ALL, 3) # Added unblock button
        main_sizer.Add(actions_sizer, 0, wx.ALIGN_CENTER | wx.TOP | wx.BOTTOM, 10)

        # Close Button
        close_btn = wx.Button(panel, wx.ID_CANCEL, _("&Close"))
        close_btn.SetDefault() # Allow Esc to close
        main_sizer.Add(close_btn, 0, wx.ALIGN_RIGHT | wx.ALL, 10)
        self.SetEscapeId(close_btn.GetId())

        panel.SetSizer(main_sizer)
        self.Fit() # Fit dialog to content

    def load_profile_data(self):
        wx.CallAfter(self.SetStatusText, _("Loading profile..."))
        for ctrl in self.profile_field_ctrls.values():
            wx.CallAfter(ctrl.SetValue, _("Loading..."))

        # Initially hide all action buttons until state is known
        wx.CallAfter(self.follow_btn.Hide)
        wx.CallAfter(self.unfollow_btn.Hide)
        wx.CallAfter(self.mute_btn.Hide)
        wx.CallAfter(self.unmute_btn.Hide)
        wx.CallAfter(self.block_btn.Hide)
        wx.CallAfter(self.unblock_btn.Hide)

        try:
            api = self.session._ensure_client()
            try:
                raw_profile = api.app.bsky.actor.get_profile({"actor": self.user_identifier})
            except Exception:
                raw_profile = None
            wx.CallAfter(self._apply_profile_data, raw_profile)

        except Exception as e:
            logger.error(f"Error loading profile for {self.user_identifier}: {e}", exc_info=True)
            wx.CallAfter(self._apply_profile_error, e)

    def _apply_profile_data(self, raw_profile):
        if raw_profile:
            self.profile_data = self._format_profile_data(raw_profile)
            self.target_user_did = self.profile_data.get("did")
            self.user_identifier = self.target_user_did or self.user_identifier

            self.update_ui_fields()
            self.update_action_buttons_state()
            self.SetTitle(_("Profile: {handle}").format(handle=self.profile_data.get("handle", "")))
            self.SetStatusText(_("Profile loaded."))
        else:
            for ctrl in self.profile_field_ctrls.values():
                ctrl.SetValue(_("Not found."))
            self.SetStatusText(_("Profile not found for '{ident}'.").format(ident=self.user_identifier))
            wx.MessageBox(_("User profile for '{ident}' not found.").format(ident=self.user_identifier), _("Error"), wx.OK | wx.ICON_ERROR, self)
        self.Layout()

    def _apply_profile_error(self, err):
        for ctrl in self.profile_field_ctrls.values():
            ctrl.SetValue(_("Error loading."))
        self.SetStatusText(_("Error loading profile."))
        wx.MessageBox(_("Error loading profile: {error}").format(error=str(err)), _("Error"), wx.OK | wx.ICON_ERROR, self)
        self.Layout()

    def update_ui_fields(self):
        if not self.profile_data:
            return

        for key, ctrl in self.profile_field_ctrls.items():
            value = self.profile_data.get(key) # _format_profile_data should provide values or None/empty
            if key == "description" and value: # Make bio multi-line if content exists
                ctrl.SetMinSize((-1, 60)) # Allow some height for bio

            if isinstance(value, (int, float)):
                ctrl.SetValue(str(value))
            else: # String or None
                ctrl.SetValue(value or _("N/A"))

        # For URLs, could make them clickable or add a "Copy URL" button
        avatar_url = self.profile_data.get("avatar") or _("N/A")
        banner_url = self.profile_data.get("banner") or _("N/A")
        self.avatar_text.SetLabel(_("Avatar URL: ") + avatar_url)
        self.avatar_text.SetToolTip(avatar_url if avatar_url != _("N/A") else "")
        self.banner_text.SetLabel(_("Banner URL: ") + banner_url)
        self.banner_text.SetToolTip(banner_url if banner_url != _("N/A") else "")
        self.Layout()

    def update_action_buttons_state(self):
        if not self.profile_data or not self.target_user_did or self.target_user_did == self._get_own_did():
            self.follow_btn.Hide()
            self.unfollow_btn.Hide()
            self.mute_btn.Hide()
            self.unmute_btn.Hide()
            self.block_btn.Hide()
            self.unblock_btn.Hide()
            self.Layout()
            return

        viewer_state = self.profile_data.get("viewer", {})
        is_following = bool(viewer_state.get("following"))
        is_muted = bool(viewer_state.get("muted"))
        # 'blocking' in viewer state is the URI of *our* block record, if we are blocking them.
        is_blocking_them = bool(viewer_state.get("blocking"))
        # 'blockedBy' means *they* are blocking us. If true, most actions might fail or be hidden.
        is_blocked_by_them = bool(viewer_state.get("blockedBy"))

        if is_blocked_by_them: # If they block us, we can't do much.
            self.follow_btn.Hide()
            self.unfollow_btn.Hide()
            self.mute_btn.Hide()
            self.unmute_btn.Hide()
            # We can still block them, or unblock them if we previously did.
            self.block_btn.Show(not is_blocking_them)
            self.unblock_btn.Show(is_blocking_them)
            self.Layout()
            return

        self.follow_btn.Show(not is_following and not is_blocking_them)
        self.unfollow_btn.Show(is_following and not is_blocking_them)

        self.mute_btn.Show(not is_muted and not is_blocking_them)
        self.unmute_btn.Show(is_muted and not is_blocking_them)

        self.block_btn.Show(not is_blocking_them) # Show block if we are not currently blocking them (even if they block us)
        self.unblock_btn.Show(is_blocking_them)   # Show unblock if we are currently blocking them

        self.Layout() # Refresh sizer to show/hide buttons correctly


    def on_user_action(self, event, command: str):
        if not self.target_user_did: # Should be set by load_profile_data
            wx.MessageBox(_("User identifier (DID) not available for this action."), _("Error"), wx.OK | wx.ICON_ERROR)
            return

        # Confirmation for sensitive actions
        confirmation_map = {
            "unfollow_user": _("Are you sure you want to unfollow @{handle}?").format(handle=self.profile_data.get("handle","this user")),
            "block_user": _("Are you sure you want to block @{handle}? This will prevent them from interacting with you and hide their content.").format(handle=self.profile_data.get("handle","this user")),
            # Unblock usually doesn't need confirmation, but can be added if desired.
        }
        if command in confirmation_map:
            dlg = wx.MessageDialog(self, confirmation_map[command], _("Confirm Action"), wx.YES_NO | wx.ICON_QUESTION)
            if dlg.ShowModal() != wx.ID_YES:
                dlg.Destroy()
                return
            dlg.Destroy()

        wx.BeginBusyCursor()
        self.SetStatusText(_("Performing action: {action}...").format(action=command))
        action_button = event.GetEventObject()
        if action_button:
            action_button.Disable()

        try:
            if command == "block_user" and hasattr(self.session, "block_user"):
                ok = self.session.block_user(self.target_user_did)
                if not ok:
                    raise RuntimeError(_("Failed to block user."))
            elif command == "unblock_user" and hasattr(self.session, "unblock_user"):
                viewer_state = self.profile_data.get("viewer", {}) if self.profile_data else {}
                block_uri = viewer_state.get("blocking")
                if not block_uri:
                    raise RuntimeError(_("Block information not available."))
                ok = self.session.unblock_user(block_uri)
                if not ok:
                    raise RuntimeError(_("Failed to unblock user."))
            else:
                raise RuntimeError(_("This action is not supported yet."))

            wx.EndBusyCursor()
            wx.MessageBox(_("Action completed."), _("Success"), wx.OK | wx.ICON_INFORMATION, self)
            wx.CallAfter(asyncio.create_task, self.load_profile_data())
        except Exception as e:
            wx.EndBusyCursor()
            if action_button:
                action_button.Enable()
            self.SetStatusText(_("Action failed."))
            wx.MessageBox(str(e), _("Error"), wx.OK | wx.ICON_ERROR, self)

    def _get_own_did(self):
        if isinstance(self.session.db, dict):
            did = self.session.db.get("user_id")
            if did:
                return did
        try:
            api = self.session._ensure_client()
            if getattr(api, "me", None):
                return api.me.did
        except Exception:
            pass
        return None

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

        return {
            "did": g(profile_model, "did"),
            "handle": g(profile_model, "handle"),
            "displayName": g(profile_model, "displayName") or g(profile_model, "display_name") or g(profile_model, "handle"),
            "description": g(profile_model, "description"),
            "avatar": g(profile_model, "avatar"),
            "banner": g(profile_model, "banner"),
            "followersCount": g(profile_model, "followersCount"),
            "followsCount": g(profile_model, "followsCount"),
            "postsCount": g(profile_model, "postsCount"),
            "viewer": g(profile_model, "viewer") or {},
        }

    def SetStatusText(self, text): # Simple status text for dialog title
        self.SetTitle(f"{_('User Profile')} - {text}")