from __future__ import annotations from typing import Any import httpx from yuxi.utils.logging_config import logger from .mp.client import MPClient from .wecom.client import WeComClient async def list_peers_mp(client: MPClient, http_client: httpx.AsyncClient) -> list[dict[str, Any]]: try: token = await client.get_access_token() url = "https://api.weixin.qq.com/cgi-bin/user/get" resp = await http_client.get(url, params={"access_token": token}) data = resp.json() if data.get("errcode") != 0: return [] openids = data.get("data", {}).get("openid", []) results: list[dict[str, Any]] = [] for openid in openids[:100]: results.append( { "id": f"wx:{openid}", "name": openid, } ) return results except Exception as e: logger.warning(f"[WeChat/Directory] list_peers_mp failed: {e}") return [] async def list_peers_wecom(client: WeComClient, http_client: httpx.AsyncClient) -> list[dict[str, Any]]: try: token = await client.get_access_token() url = "https://qyapi.weixin.qq.com/cgi-bin/user/list" resp = await http_client.get(url, params={"access_token": token, "department_id": 1, "fetch_child": 1}) data = resp.json() if data.get("errcode") != 0: return [] userlist = data.get("userlist", []) return [{"id": f"wx:{u.get('userid', '')}", "name": u.get("name", u.get("userid", ""))} for u in userlist] except Exception as e: logger.warning(f"[WeChat/Directory] list_peers_wecom failed: {e}") return [] async def list_peers_bridge(bridge_client, cfg: dict[str, Any]) -> list[dict[str, Any]]: try: contacts = cfg.get("contacts", []) return [{"id": f"wx:{c.get('id', '')}", "name": c.get("name", c.get("id", ""))} for c in contacts] except Exception as e: logger.warning(f"[WeChat/Directory] list_peers_bridge failed: {e}") return [] async def list_groups_wecom(client: WeComClient, http_client: httpx.AsyncClient) -> list[dict[str, Any]]: try: token = await client.get_access_token() url = "https://qyapi.weixin.qq.com/cgi-bin/appchat/list" resp = await http_client.get(url, params={"access_token": token}) data = resp.json() if data.get("errcode") != 0: return [] chat_list = data.get("chat_list", []) return [ {"id": f"wx:group:{c.get('chatid', '')}", "name": c.get("name", c.get("chatid", ""))} for c in chat_list ] except Exception as e: logger.warning(f"[WeChat/Directory] list_groups_wecom failed: {e}") return [] async def list_groups_mp(client: MPClient, http_client: httpx.AsyncClient) -> list[dict[str, Any]]: return [] async def list_groups_bridge(bridge_client, cfg: dict[str, Any]) -> list[dict[str, Any]]: try: groups_cfg = cfg.get("groups", {}) return [ {"id": f"wx:group:{chat_id}", "name": chat_config.get("name", chat_id)} for chat_id, chat_config in groups_cfg.items() ] except Exception as e: logger.warning(f"[WeChat/Directory] list_groups_bridge failed: {e}") return [] def format_target_display(target: dict[str, Any]) -> str: to = target.get("to", "unknown") chat_type = target.get("chatType", "direct") return f"[{chat_type}] {to}" def build_cross_context_presentation( source_ctx: dict[str, Any], target_ctx: dict[str, Any] | None = None, ) -> str: src_peer = source_ctx.get("peer_name", source_ctx.get("to", "unknown")) return f"[转发自 {src_peer}]"