@iducky/media-agent 1.1.0

package/src/media_agent/commands.sh

@@ -0,0 +1,436 @@
+#!/bin/bash
+# 社交媒体账号管理 CLI v3.3
+set -e
+
+SCRIPT_DIR="${MEDIA_AGENT_HOME:-$HOME/.media-agent/social-accounts}"
+export MEDIA_AGENT_HOME="$SCRIPT_DIR"
+PYTHON="${MEDIA_AGENT_PYTHON:-python3}"
+run_script() { "$PYTHON" -m media_agent.cli run-script "$@"; }
+
+require_profile() {
+    [[ "$1" =~ ^[A-Za-z0-9_-]+$ ]] || { echo "错误: 无效 profile_id"; exit 1; }
+    if [ -z "$1" ]; then echo "错误: 缺少 profile_id"; echo "用法: media-agent $CMD <profile_id> [args]"; exit 1; fi
+}
+
+check_capability() {
+    local pid="$1" cap="$2"
+    local has
+    has=$("$PYTHON" - "$SCRIPT_DIR/profiles/$pid/config.json" "$cap" <<'PY_CAP'
+import json, sys
+try:
+    cfg = json.load(open(sys.argv[1]))
+    caps = cfg.get('account', {}).get('capabilities', [])
+    if not caps:
+        atype = cfg.get('account', {}).get('account_type', '')
+        if atype == 'personal': caps = ['creator_index_collect']
+        elif atype == 'enterprise': caps = ['leads_export', 'leads_rankings', 'leads_taxonomy']
+    print('ok' if sys.argv[2] in caps else 'missing')
+except (OSError, ValueError):
+    print('missing')
+PY_CAP
+)
+    if [ "$has" != "ok" ]; then echo "错误: $pid 缺少 capability: $cap"; exit 4; fi
+}
+
+require_toutiao_profile() {
+    local pid="$1" config="$SCRIPT_DIR/profiles/$1/config.json"
+    [ -f "$config" ] || { echo "错误: Profile 不存在: $pid"; exit 5; }
+    "$PYTHON" - "$config" <<'PY'
+import json, sys
+cfg=json.load(open(sys.argv[1]))
+raise SystemExit(0 if cfg.get('account',{}).get('platform','').lower() == 'toutiao' else 4)
+PY
+    rc=$?
+    [ $rc -eq 0 ] || { echo "错误: $pid 不是 toutiao Profile"; exit $rc; }
+}
+
+CMD="${1:---help}"; shift 2>/dev/null || true
+
+case "$CMD" in
+  enterprise-login)
+    require_profile "$1"
+    [ -f "$SCRIPT_DIR/profiles/$1/config.json" ] || { echo "错误: Profile 不存在: $1"; exit 5; }
+    run_script enterprise_login.py "$@"
+    ;;
+  validate-rankings) run_script validate_rankings.py "$@" ;;
+  validate-industry-taxonomy) run_script validate_industry_taxonomy.py "$@" ;;
+
+  init)           run_script account_manager.py init "$@" ;;
+  list|ls)        run_script account_manager.py list ;;
+  status|st)      require_profile "$1"; run_script account_manager.py status "$1" ;;
+  locks)          run_script account_manager.py locks ;;
+
+  login-start)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" &
+    disown
+    sleep 3
+    run_script login_controller.py "$PROFILE_ID" --send '{"action":"status"}'
+    ;;
+
+  login-status)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" --send '{"action":"status"}'
+    ;;
+
+  login-scan-done)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" --send '{"action":"scan_done"}'
+    ;;
+
+  login-submit-code)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" --submit-code-stdin
+    ;;
+
+  login-close)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" --send '{"action":"close"}'
+    sleep 3
+    rm -f "$SCRIPT_DIR/locks/${PROFILE_ID}.lock"
+    ;;
+
+  login-refresh-qr)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" --send '{"action":"refresh_qr"}'
+    ;;
+
+  login-resend-code)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script login_controller.py "$PROFILE_ID" --send '{"action":"resend_code"}'
+    ;;
+
+  check-login)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    SITE="all"
+    for i in $(seq 1 $#); do
+        if [ "${!i}" = "--site" ]; then idx=$((i+1)); SITE="${!idx}"; break; fi
+    done
+    run_script check_login.py "$PROFILE_ID" --site "$SITE"
+    ;;
+
+  toutiao-login-start)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"
+    mkdir -p "$SCRIPT_DIR/logs"
+    run_script toutiao_login_controller.py "$PROFILE_ID" \
+      >> "$SCRIPT_DIR/logs/${PROFILE_ID}_toutiao_controller.log" 2>&1 &
+    disown
+    for _ in $(seq 1 45); do
+        [ -S "$SCRIPT_DIR/controllers/${PROFILE_ID}.sock" ] && break
+        sleep 1
+    done
+    run_script toutiao_login_ipc.py "$PROFILE_ID" status --base "$SCRIPT_DIR"
+    ;;
+
+  toutiao-login-status)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"
+    run_script toutiao_login_ipc.py "$PROFILE_ID" status --base "$SCRIPT_DIR"
+    ;;
+
+  toutiao-login-scan-done)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"
+    shift
+    EXPECTED=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in --expected-account-name) EXPECTED="$2"; shift 2 ;; *) shift ;; esac
+    done
+    if [ -n "$EXPECTED" ]; then
+        run_script toutiao_login_ipc.py "$PROFILE_ID" scan-done --expected-account-name "$EXPECTED" --base "$SCRIPT_DIR"
+    else
+        run_script toutiao_login_ipc.py "$PROFILE_ID" scan-done --base "$SCRIPT_DIR"
+    fi
+    ;;
+
+  toutiao-login-verify-account)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"; shift
+    [ "$1" = "--expected-account-name" ] && [ -n "$2" ] || { echo "错误: 缺少 --expected-account-name"; exit 1; }
+    run_script toutiao_login_ipc.py "$PROFILE_ID" verify-account --expected-account-name "$2" --base "$SCRIPT_DIR"
+    ;;
+
+  toutiao-login-refresh-qr)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"
+    run_script toutiao_login_ipc.py "$PROFILE_ID" refresh-qr --base "$SCRIPT_DIR"
+    ;;
+
+  toutiao-login-close)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"
+    run_script toutiao_login_ipc.py "$PROFILE_ID" close --base "$SCRIPT_DIR"
+    ;;
+
+  toutiao-check-login)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; require_toutiao_profile "$PROFILE_ID"
+    run_script toutiao_check_login.py "$PROFILE_ID" --base "$SCRIPT_DIR"
+    ;;
+
+  close)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    echo "关闭: $PROFILE_ID"
+    touch "$SCRIPT_DIR/locks/${PROFILE_ID}.close"
+    # Wait for browser to actually close (check with lsof)
+    for i in $(seq 1 30); do
+        if ! lsof -n 2>/dev/null | grep -q "$PROFILE_ID/browser_data"; then
+            break
+        fi
+        sleep 2
+    done
+    rm -f "$SCRIPT_DIR/locks/${PROFILE_ID}.lock" "$SCRIPT_DIR/locks/${PROFILE_ID}.close"
+    echo "已关闭并释放锁"
+    ;;
+
+  collect-douyin-index)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_index_collect"
+    shift; LISTS="realtime,rising"; PAGES=3
+    while [[ $# -gt 0 ]]; do
+        case "$1" in --lists) LISTS="$2"; shift 2 ;; --pages) PAGES="$2"; shift 2 ;; *) shift ;; esac
+    done
+    run_script check_login.py "$PROFILE_ID" --site creator > /dev/null 2>&1
+    if [ $? -ne 0 ]; then echo "状态: LOGIN_REQUIRED"; exit 2; fi
+    run_script douyin_hotspot_v2.py
+    ;;
+
+  export-short-video)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "leads_export"
+    shift; DATE=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "export-short-video <profile_id> [--date YYYY-MM-DD]"; exit 0 ;;
+            --dry-run) echo "[DRY-RUN] export-short-video $PROFILE_ID"; exit 0 ;;
+            --date) DATE="$2"; shift 2 ;; *) shift ;;
+        esac
+    done
+    [ -z "$DATE" ] && DATE=$(date -v-1d +%Y-%m-%d 2>/dev/null || "$PYTHON" -c "from datetime import date,timedelta; print((date.today()-timedelta(1)).isoformat())")
+    run_script export_short_video.py "$PROFILE_ID" "$DATE"
+    ;;
+
+  collect-video-rankings)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "leads_rankings"
+    shift; INDUSTRY_L1=""; INDUSTRY_L2=""; DATE=""; PAGES=2
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "collect-video-rankings <profile_id> --industry-l1 <L1> --industry-l2 <L2>"; exit 0 ;;
+            --dry-run) echo "[DRY-RUN] collect-video-rankings $PROFILE_ID"; exit 0 ;;
+            --industry-l1) INDUSTRY_L1="$2"; shift 2 ;; --industry-l2) INDUSTRY_L2="$2"; shift 2 ;;
+            --date) DATE="$2"; shift 2 ;; --pages) PAGES="$2"; shift 2 ;; *) shift ;;
+        esac
+    done
+    [ -z "$DATE" ] && DATE=$(date -v-1d +%Y-%m-%d 2>/dev/null || "$PYTHON" -c "from datetime import date,timedelta; print((date.today()-timedelta(1)).isoformat())")
+    [ -z "$INDUSTRY_L1" ] || [ -z "$INDUSTRY_L2" ] && { echo "WAIT_INDUSTRY_SELECTION"; exit 6; }
+    run_script collect_video_rankings.py
+    ;;
+
+  collect-industry-taxonomy)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "leads_taxonomy"
+    shift
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "collect-industry-taxonomy <profile_id> (candidate-only)"; exit 0 ;;
+            --dry-run) echo "[DRY-RUN] collect-industry-taxonomy $PROFILE_ID (candidate-only)"; exit 0 ;;
+            *) shift ;;
+        esac
+    done
+    run_script collect_industry_taxonomy.py
+    ;;
+
+  publish-dry-run)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    shift; MEDIA_PATH=""; SCHEDULED=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "publish-dry-run <profile_id> --media-path <path> [--scheduled-at <ISO>]"; exit 0 ;;
+            --media-path) MEDIA_PATH="$2"; shift 2 ;;
+            --scheduled-at) SCHEDULED="$2"; shift 2 ;;
+            *) shift ;;
+        esac
+    done
+    [ -z "$MEDIA_PATH" ] && { echo "错误: 缺少 --media-path"; exit 1; }
+    if [ -n "$SCHEDULED" ]; then
+        run_script douyin_publish.py --dry-run --profile-id "$PROFILE_ID" --media-path "$MEDIA_PATH" --scheduled-at "$SCHEDULED"
+    else
+        run_script douyin_publish.py --dry-run --profile-id "$PROFILE_ID" --media-path "$MEDIA_PATH"
+    fi
+    ;;
+
+  publish-prepare)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    shift; MEDIA_PATH=""; SCHEDULED=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "publish-prepare <profile_id> --media-path <path> [--scheduled-at <ISO>]"; exit 0 ;;
+            --media-path) MEDIA_PATH="$2"; shift 2 ;;
+            --scheduled-at) SCHEDULED="$2"; shift 2 ;;
+            *) shift ;;
+        esac
+    done
+    [ -z "$MEDIA_PATH" ] && { echo "错误: 缺少 --media-path"; exit 1; }
+    if [ -n "$SCHEDULED" ]; then
+        run_script douyin_publish.py --prepare --profile-id "$PROFILE_ID" --media-path "$MEDIA_PATH" --scheduled-at "$SCHEDULED"
+    else
+        run_script douyin_publish.py --prepare --profile-id "$PROFILE_ID" --media-path "$MEDIA_PATH"
+    fi
+    ;;
+
+  publish-confirm)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    run_script douyin_publish.py --confirm-submit --profile-id "$PROFILE_ID"
+    ;;
+
+  publish-status)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    shift; TASK_ID=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "publish-status <profile_id> [--task-id <id>]"; exit 0 ;;
+            --task-id) TASK_ID="$2"; shift 2 ;;
+            *) shift ;;
+        esac
+    done
+    if [ -n "$TASK_ID" ]; then
+        run_script douyin_publish.py --status --profile-id "$PROFILE_ID" --task-id "$TASK_ID"
+    else
+        run_script douyin_publish.py --status --profile-id "$PROFILE_ID"
+    fi
+    ;;
+
+  publish-close)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    run_script douyin_publish.py --close --profile-id "$PROFILE_ID"
+    ;;
+
+  xhs-login-start)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    rm -f "$SCRIPT_DIR/signals/${PROFILE_ID}_response.json"
+    mkdir -p "$SCRIPT_DIR/logs"
+    run_script xiaohongshu_login_controller.py "$PROFILE_ID" \
+      >> "$SCRIPT_DIR/logs/${PROFILE_ID}_xiaohongshu_controller.log" 2>&1 &
+    disown
+    run_script xiaohongshu_login_ipc.py "$PROFILE_ID" wait-start
+    ;;
+
+  xhs-login-status)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script xiaohongshu_login_ipc.py "$PROFILE_ID" status
+    ;;
+
+  xhs-login-scan-done)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script xiaohongshu_login_ipc.py "$PROFILE_ID" scan-done
+    ;;
+
+  xhs-login-refresh-qr)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script xiaohongshu_login_ipc.py "$PROFILE_ID" refresh-qr
+    ;;
+
+  xhs-login-close)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script xiaohongshu_login_ipc.py "$PROFILE_ID" close
+    ;;
+
+  xhs-check-login)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"
+    run_script xiaohongshu_check_login.py "$PROFILE_ID"
+    ;;
+
+  xhs-publish-dry-run)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    shift; MEDIA_PATH=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "xhs-publish-dry-run <profile_id> --media-path <path>"; exit 0 ;;
+            --media-path) MEDIA_PATH="$2"; shift 2 ;;
+            *) shift ;;
+        esac
+    done
+    [ -z "$MEDIA_PATH" ] && { echo "错误: 缺少 --media-path"; exit 1; }
+    run_script xiaohongshu_publish.py --dry-run --profile-id "$PROFILE_ID" --media-path "$MEDIA_PATH"
+    ;;
+
+  xhs-publish-prepare)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    shift; MEDIA_PATH=""; TITLE=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "xhs-publish-prepare <profile_id> --media-path <path> --title <标题>"; exit 0 ;;
+            --media-path) MEDIA_PATH="$2"; shift 2 ;;
+            --title) TITLE="$2"; shift 2 ;;
+            *) shift ;;
+        esac
+    done
+    [ -z "$MEDIA_PATH" ] && { echo "错误: 缺少 --media-path"; exit 1; }
+    [ -z "$TITLE" ] && { echo "错误: 缺少 --title"; exit 1; }
+    run_script xiaohongshu_publish.py --prepare --profile-id "$PROFILE_ID" --media-path "$MEDIA_PATH" --title "$TITLE"
+    ;;
+
+  xhs-publish-confirm)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    run_script xiaohongshu_publish.py --confirm-submit --profile-id "$PROFILE_ID"
+    ;;
+
+  xhs-publish-status)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    shift; TASK_ID=""
+    while [[ $# -gt 0 ]]; do
+        case "$1" in
+            --help) echo "xhs-publish-status <profile_id> [--task-id <id>]"; exit 0 ;;
+            --task-id) TASK_ID="$2"; shift 2 ;;
+            *) shift ;;
+        esac
+    done
+    if [ -n "$TASK_ID" ]; then
+        run_script xiaohongshu_publish.py --status --profile-id "$PROFILE_ID" --task-id "$TASK_ID"
+    else
+        run_script xiaohongshu_publish.py --status --profile-id "$PROFILE_ID"
+    fi
+    ;;
+
+  xhs-publish-close)
+    PROFILE_ID="$1"; require_profile "$PROFILE_ID"; check_capability "$PROFILE_ID" "creator_publish"
+    run_script xiaohongshu_publish.py --close --profile-id "$PROFILE_ID"
+    ;;
+
+  help|--help|-h)
+    echo "media-agent - 社媒账号管理命令"
+    echo "  login-start <id>           启动持久化登录控制器 (daemon)"
+    echo "  login-status <id>          查询控制器状态"
+    echo "  login-scan-done <id>       通知控制器扫码完成"
+    echo "  login-refresh-qr <id>      刷新过期二维码 (当前页面内)"
+    echo "  login-submit-code <id>      提交验证码 (从stdin读取, 不落盘)"
+    echo "  login-resend-code <id>      重新发送短信验证码 (保留当前会话)"
+    echo "  login-close <id>           正常关闭控制器并释放锁"
+    echo "  check-login <id> [--site] 检查登录状态"
+    echo "  enterprise-login <id> --site leads"
+    echo "  validate-rankings <task_dir>"
+    echo "  validate-industry-taxonomy <candidate.yaml>"
+    echo "  close <id>                 关闭浏览器并释放锁"
+    echo "  collect-douyin-index <id>"
+    echo "  export-short-video <id>"
+    echo "  collect-video-rankings <id>"
+    echo "  collect-industry-taxonomy <id>"
+    echo "  publish-dry-run <id> --media-path <path>"
+    echo "  publish-prepare <id> --media-path <path> [--scheduled-at <ISO>]"
+    echo "  publish-confirm <id>"
+    echo "  publish-status <id> [--task-id <id>]"
+    echo "  publish-close <id>"
+    echo "  toutiao-login-start <id>       启动今日头条持久化登录控制器"
+    echo "  toutiao-login-status <id>      查询今日头条控制器状态"
+    echo "  toutiao-login-scan-done <id> [--expected-account-name <昵称>]"
+    echo "  toutiao-login-verify-account <id> --expected-account-name <昵称>"
+    echo "  toutiao-login-refresh-qr <id>  在原二维码容器刷新"
+    echo "  toutiao-login-close <id>       由原控制器正常关闭"
+    echo "  toutiao-check-login <id>       独立只读检查登录状态"
+    echo "  xhs-login-start <id>         小红书登录控制器 (daemon)"
+    echo "  xhs-login-status <id>        小红书控制器状态"
+    echo "  xhs-login-scan-done <id>     小红书扫码完成"
+    echo "  xhs-login-refresh-qr <id>    小红书刷新二维码"
+    echo "  xhs-login-close <id>         小红书关闭控制器"
+    echo "  xhs-check-login <id>         小红书登录状态检查"
+    echo "  xhs-publish-dry-run <id> --media-path <path>"
+    echo "  xhs-publish-prepare <id> --media-path <path> --title <标题>"
+    echo "  xhs-publish-confirm <id>"
+    echo "  xhs-publish-status <id> [--task-id <id>]"
+    echo "  xhs-publish-close <id>"
+    echo "Exit: 0=OK 1=err 2=LOGIN_REQUIRED 3=PROFILE_LOCKED 4=NO_CAPABILITY 5=NOT_FOUND 6=WAIT_INDUSTRY 7=indeterminate 8=DUPLICATE_BLOCKED"
+    ;;
+
+  *) echo "未知命令: $CMD"; echo "使用: media-agent help"; exit 1 ;;
+esac

package/src/media_agent/platforms/__init__.py

@@ -0,0 +1 @@
+"""Media Agent capabilities."""

package/src/media_agent/platforms/douyin/__init__.py

@@ -0,0 +1 @@
+"""Media Agent capabilities."""

package/src/media_agent/platforms/douyin/check_login.py

@@ -0,0 +1,196 @@
+"""Legacy operation, executed only through main()."""
+
+def main():
+    #!/usr/bin/env python3
+    """Real page login check. Usage: python3 check_login.py <profile_id> [--site creator|leads|all]"""
+    from media_agent.runtime.paths import runtime_home
+    import json, time, os, sys
+    from pathlib import Path
+    from datetime import datetime, timezone
+    from media_agent.runtime.browser import launch_persistent_context
+
+    BASE = runtime_home()
+    ALIAS = sys.argv[1] if len(sys.argv) > 1 else sys.exit(1)
+    SITE = 'all'  # default
+    for i, a in enumerate(sys.argv):
+        if a == '--site' and i + 1 < len(sys.argv):
+            SITE = sys.argv[i + 1]; break
+
+    PROFILE_DIR = BASE / 'profiles' / ALIAS
+    LOCK_FILE = BASE / 'locks' / f'{ALIAS}.lock'
+
+    if not PROFILE_DIR.exists():
+        print(json.dumps({'profile_id': ALIAS, 'login_status': 'unknown', 'error': 'PROFILE_NOT_FOUND'}, indent=2))
+        sys.exit(5)
+
+    cfg = json.loads((PROFILE_DIR / 'config.json').read_text())
+    state = json.loads((PROFILE_DIR / 'state.json').read_text()) if (PROFILE_DIR / 'state.json').exists() else {}
+
+    # Determine sites to check
+    sites_cfg = cfg.get('sites', {})
+    if not sites_cfg:
+        # Backward compat: single site
+        target = cfg.get('account', {}).get('target_url', 'https://creator.douyin.com/')
+        sites_cfg = {'default': {'target_url': target}}
+
+    if SITE == 'creator':
+        sites_to_check = {'creator': sites_cfg.get('creator', {})}
+    elif SITE == 'leads':
+        sites_to_check = {'leads': sites_cfg.get('leads', {})}
+    else:
+        sites_to_check = {k: v for k, v in sites_cfg.items() if k in ('creator', 'leads')}
+        if not sites_to_check:
+            sites_to_check = sites_cfg  # backward compat
+
+    LOCK_FILE.parent.mkdir(parents=True, exist_ok=True)
+    if LOCK_FILE.exists():
+        try:
+            os.kill(json.loads(LOCK_FILE.read_text()).get('pid', 0), 0)
+            print(json.dumps({'profile_id': ALIAS, 'login_status': 'locked', 'lock_status': 'locked'}, indent=2))
+            sys.exit(3)
+        except:
+            LOCK_FILE.unlink()
+    LOCK_FILE.write_text(json.dumps({'task_id': 'check-login', 'pid': os.getpid(), 'time': datetime.now().isoformat(), 'site': SITE}, indent=2))
+
+    try:
+        ctx = launch_persistent_context(
+            user_data_dir=str(PROFILE_DIR / 'browser_data'),
+            headless=False, stealth_args=True,
+            viewport=cfg['fingerprint']['viewport'],
+            locale=cfg['fingerprint']['locale'],
+            timezone=cfg['fingerprint']['timezone'],
+            humanize=False, geoip=False
+        )
+        page = ctx.pages[0] if ctx.pages else ctx.new_page()
+
+        results = {}
+        for site_name, site_cfg in sites_to_check.items():
+            target_url = site_cfg.get('target_url', '')
+            page.goto(target_url, wait_until='load', timeout=60000)
+            time.sleep(5)
+
+            url = page.url
+            title = page.title()
+
+            # DOM-based detection
+            dom = page.evaluate('''() => {
+                const result = {};
+                const all = document.querySelectorAll('a, div, span, li');
+                // Creator-specific: more specific dashboard indicators
+            const isCreator = '${site_name}' === 'creator';
+            const dashboardKeywords = isCreator
+                ? ['发布视频', '内容管理', '作品管理', '创作灵感', '互动管理']
+                : ['运营工作台', '数据分析', '线索管理', '昨日数据概览', '线索版'];
+            // Also check for creator-micro in URL (creator only)
+            const hasCreatorMicro = isCreator && window.location.href.includes('creator-micro');
+                const foundDashboard = [];
+                const seenDash = new Set();
+                for (const el of all) {
+                    if (el.offsetParent === null) continue;
+                    const t = el.textContent.trim();
+                    for (const kw of dashboardKeywords) {
+                        if (t.includes(kw) && !seenDash.has(kw)) { seenDash.add(kw); foundDashboard.push(kw); }
+                    }
+                }
+                result.dashboard_detected = foundDashboard.length >= 2 || (isCreator && hasCreatorMicro);
+                result.dashboard_evidence = foundDashboard;
+                if (isCreator && hasCreatorMicro) result.dashboard_evidence.push('creator-micro URL');
+
+                const loginKeywords = ['扫码登录', '验证码登录', '密码登录', '手机号登录', '其他方式'];
+                const foundLogin = []; const seenLogin = new Set();
+                for (const el of all) {
+                    if (el.offsetParent === null) continue;
+                    const t = el.textContent.trim();
+                    for (const kw of loginKeywords) {
+                        if (t.includes(kw) && !seenLogin.has(kw)) { seenLogin.add(kw); foundLogin.push(kw); }
+                    }
+                }
+                const loginInputs = document.querySelectorAll('input[type="tel"], input[placeholder*="手机"], input[placeholder*="验证码"]');
+                const hasLoginInputs = Array.from(loginInputs).some(el => el.offsetParent !== null);
+                result.login_form_detected = foundLogin.length > 0 || hasLoginInputs;
+                result.login_form_evidence = foundLogin;
+                if (hasLoginInputs) result.login_form_evidence.push('login_input_visible');
+
+                const qrImgs = document.querySelectorAll('img[src*="qr" i], canvas');
+                let hasQrImg = false;
+                for (const el of qrImgs) {
+                    if (el.offsetParent === null) continue;
+                    const rect = el.getBoundingClientRect();
+                    if (Math.abs(rect.width - rect.height) < 20 && rect.width >= 100 && rect.width < 400) { hasQrImg = true; break; }
+                }
+                const qrDivs = document.querySelectorAll('div');
+                let hasQrText = false;
+                for (const el of qrDivs) {
+                    if (el.offsetParent === null) continue;
+                    const t = el.textContent.trim();
+                    if ((t === '扫码登录' || t.includes('二维码')) && t.length < 20) { hasQrText = true; break; }
+                }
+                result.qr_detected = hasQrImg || hasQrText;
+                result.qr_evidence = [];
+                if (hasQrImg) result.qr_evidence.push('qr_image_visible');
+                if (hasQrText) result.qr_evidence.push('qr_text_visible');
+                return result;
+            }''')
+
+            dashboard = dom['dashboard_detected']
+            login_form = dom['login_form_detected']
+            qr = dom['qr_detected']
+
+            if dashboard and not login_form and not qr:
+                status = 'logged_in'
+            elif (login_form or qr) and not dashboard:
+                status = 'login_required'
+            elif dashboard and (login_form or qr):
+                status = 'indeterminate'
+            else:
+                status = 'indeterminate'
+
+            results[site_name] = {
+                'target_url': target_url, 'current_url': url[:120], 'page_title': title,
+                'dashboard_detected': dashboard, 'dashboard_evidence': dom['dashboard_evidence'],
+                'login_form_detected': login_form, 'login_form_evidence': dom['login_form_evidence'],
+                'qr_detected': qr, 'qr_evidence': dom['qr_evidence'],
+                'login_status': status,
+            }
+
+        ctx.close()
+        LOCK_FILE.unlink(missing_ok=True)
+
+        # Update state
+        for site_name, r in results.items():
+            if site_name in ('creator', 'leads', 'default'):
+                key = site_name if site_name != 'default' else 'default'
+                if key in state:
+                    state[key]['login_status'] = r['login_status']
+                    state[key]['checked_at'] = datetime.now(timezone.utc).isoformat()
+                    state[key]['dashboard_detected'] = r['dashboard_detected']
+                    state[key]['login_form_detected'] = r['login_form_detected']
+                    state[key]['qr_detected'] = r['qr_detected']
+                    state[key]['evidence'] = r['login_status']
+        (PROFILE_DIR / 'state.json').write_text(json.dumps(state, indent=2))
+
+        # Compute overall status
+        statuses = [r['login_status'] for r in results.values()]
+        if all(s == 'logged_in' for s in statuses):
+            overall = 'logged_in'; exit_code = 0
+        elif any(s == 'login_required' for s in statuses):
+            overall = 'login_required'; exit_code = 2
+        elif any(s == 'indeterminate' for s in statuses):
+            overall = 'indeterminate'; exit_code = 7
+        else:
+            overall = 'indeterminate'; exit_code = 7
+
+        output = {
+            'profile_id': ALIAS, 'site': SITE, 'overall_status': overall,
+            'sites': results,
+            'browser_closed': True, 'lock_released': True,
+        }
+        print(json.dumps(output, indent=2, ensure_ascii=False))
+        sys.exit(exit_code)
+    except Exception as e:
+        LOCK_FILE.unlink(missing_ok=True)
+        print(json.dumps({'profile_id': ALIAS, 'login_status': 'check_failed', 'error': str(e)[:100]}, indent=2))
+        sys.exit(1)
+
+if __name__ == '__main__':
+    main()