MediaCrawler/mac-agent-os-main/05_tools/10_dashboard/routes/v2_accounts.py

"""
v2_accounts.py — V2 统一账号管理 API 路由

基于 AccountService 提供联邦账号中心的统一数据视图。
旧 API（matrix.py /api/matrix/accounts）保留兼容。

前缀: /api/v2
"""
import json
import logging
import os
import sys
import time
from pathlib import Path
from fastapi import APIRouter, HTTPException

logger = logging.getLogger("dashboard.routes.v2_accounts")

_THIS_DIR = Path(__file__).resolve().parent.parent
AGENT_SYNC = Path(os.environ.get("AGENT_SYNC", str(Path.home() / "workbuddy-agent-os" / "agent-sync")))
AGENT_LOCAL = Path(os.environ.get("AGENT_LOCAL", str(Path.home() / "workbuddy-agent-os" / "agent-local")))

router = APIRouter(prefix="/api/v2", tags=["v2_accounts"])


def _get_svc():
    """获取 AccountService 实例（延迟导入，单例缓存）"""
    if not hasattr(_get_svc, "_instance"):
        sys.path.insert(0, str(_THIS_DIR))
        from services.account_service import AccountService
        _get_svc._instance = AccountService()
    return _get_svc._instance


# ═══════════════════════════════════════════════════════════
# 账号列表与详情
# ═══════════════════════════════════════════════════════════

@router.get("/accounts")
def api_v2_accounts(
    machine: str = "",
    platform: str = "",
    status: str = "",
    q: str = "",
):
    """获取所有账号的统一视图"""
    try:
        svc = _get_svc()
        accounts = svc.get_all_accounts()

        if machine:
            accounts = [a for a in accounts if a.get("owner_machine") == machine]
        if platform:
            accounts = [a for a in accounts if a.get("platform") == platform]
        if status:
            accounts = [a for a in accounts if a.get("login_status") == status]
        if q:
            lq = q.lower()
            accounts = [a for a in accounts if lq in a["id"].lower() or lq in a.get("phone", "").lower()
                        or lq in a.get("nickname", "").lower() or lq in a.get("identity_dir", "").lower()
                        or lq in a.get("owner_machine", "").lower()]

        return {"accounts": accounts, "total": len(accounts), "generated_at": time.time()}
    except Exception as e:
        logger.error(f"v2/accounts 查询失败: {e}", exc_info=True)
        raise HTTPException(500, detail=str(e))


@router.get("/accounts/{account_id}")
def api_v2_account_detail(account_id: str):
    """获取单个账号详情"""
    try:
        svc = _get_svc()
        acct = svc.get_account_detail(account_id)
        if not acct:
            raise HTTPException(404, detail=f"账号 {account_id} 不存在")
        return acct
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(500, detail=str(e))


@router.get("/status-summary")
def api_v2_status_summary():
    """获取账号状态汇总"""
    try:
        svc = _get_svc()
        accounts = svc.get_all_accounts()
        summary = {"total": len(accounts), "logged_in": 0, "cookie_expiring": 0,
                    "no_cookie": 0, "banned": 0, "disabled": 0, "unknown": 0}
        for a in accounts:
            s = a.get("login_status", "unknown")
            if s in summary:
                summary[s] += 1
            else:
                summary["unknown"] += 1
        return summary
    except Exception as e:
        raise HTTPException(500, detail=str(e))


@router.patch("/accounts/{account_id}")
def api_v2_update_notes(account_id: str, data: dict):
    """更新账号备注/标签

    - 本机账号：写入本机 accounts.yaml（MatrixManager）
    - 远程账号：本地找不到时，tags 写入集中标签文件（agent-local/data/account_tags_cache.json）
    """
    try:
        import yaml
        sys.path.insert(0, str(AGENT_SYNC / "05_tools" / "07_matrix" / "scripts"))
        from matrix_mgmt import MatrixManager
        mgr = MatrixManager()
        update_data = {}
        if "notes" in data:
            update_data["notes"] = data["notes"]
        if "tags" in data:
            update_data["tags"] = data["tags"]
        if not update_data:
            return {"status": "ok"}
        try:
            mgr.update_account(account_id, update_data)
        except ValueError:
            # 远程账号（本机 MatrixManager 无此账号）→ tags 写入集中标签文件
            if "tags" in update_data and update_data["tags"] is not None:
                _save_remote_tags(account_id, update_data["tags"])
            else:
                raise HTTPException(404, detail=f"账号 {account_id} 不存在")
        return {"status": "ok"}
    except HTTPException:
        raise
    except Exception as e:
        raise HTTPException(500, detail=str(e))


def _save_remote_tags(account_id: str, tags: list) -> None:
    """远程账号标签 → 集中标签文件（agent-local/data/account_tags_cache.json）

    读取侧 account_service._load_tags_cache 已从此文件读远程 tags，写入即可生效。
    """
    import json as _json
    _CACHE_PATH = AGENT_LOCAL / "data" / "account_tags_cache.json"
    data = {}
    if _CACHE_PATH.exists():
        try:
            data = _json.loads(_CACHE_PATH.read_text())
        except Exception:
            data = {}
    data[account_id] = [str(t) for t in (tags or [])]
    _CACHE_PATH.parent.mkdir(parents=True, exist_ok=True)
    _CACHE_PATH.write_text(_json.dumps(data, ensure_ascii=False, indent=1))
    logger.info("  🏷️ 远程账号 %s 标签已保存到集中文件 (%d 条)", account_id, len(data[account_id]))


# ═══════════════════════════════════════════════════════════
# 批量操作
# ═══════════════════════════════════════════════════════════

@router.post("/accounts/batch")
def api_v2_accounts_batch(data: dict):
    """批量操作选中的账号"""
    account_ids = data.get("account_ids", [])
    action = data.get("action", "")
    params = data.get("params", {})

    if not account_ids:
        raise HTTPException(400, detail="account_ids 必填")
    if not action:
        raise HTTPException(400, detail="action 必填")

    valid_actions = ("collect", "login", "nurture", "comment", "record")
    if action not in valid_actions:
        raise HTTPException(400, detail=f"action 必须是 {valid_actions} 之一")

    try:
        from services.command_bus import CommandBus
        svc = _get_svc()
        all_accounts = svc.get_all_accounts()
        acct_map = {a["id"]: a for a in all_accounts}
        machine_groups = {}
        for aid in account_ids:
            a = acct_map.get(aid)
            if not a:
                raise HTTPException(400, detail=f"账号 {aid} 不存在")
            m = a.get("owner_machine", "")
            if m not in machine_groups:
                machine_groups[m] = []
            machine_groups[m].append(aid)

        results = []
        for machine, ids in machine_groups.items():
            r = CommandBus.dispatch(action, ids, params)
            results.append(r)
        return {"status": "ok", "results": results}
    except HTTPException:
        raise
    except Exception as e:
        logger.error(f"v2/accounts/batch 失败: {e}")
        raise HTTPException(500, detail=str(e))


# ═══════════════════════════════════════════════════════════
# 状态刷新
# ═══════════════════════════════════════════════════════════

@router.post("/accounts/refresh")
def api_v2_accounts_refresh(data: dict):
    """强制刷新账号状态"""
    try:
        svc = _get_svc()
        account_ids = data.get("account_ids")
        result = svc.batch_refresh_status(account_ids)
        return {"status": "ok", "refreshed": len(result), "accounts": result}
    except Exception as e:
        raise HTTPException(500, detail=str(e))