@iducky/media-agent 1.1.0

package/src/media_agent/platforms/toutiao/toutiao_login_ipc.py

@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+"""Bounded request-correlated IPC client for the Toutiao controller."""
+
+from __future__ import annotations
+from media_agent.runtime.paths import runtime_home
+
+import argparse
+import json
+import socket
+import uuid
+from pathlib import Path
+
+
+def send_command(base: Path, profile_id: str, action: str, *, timeout: float = 10.0, **data) -> dict:
+    request_id = uuid.uuid4().hex
+    socket_path = base / "controllers" / f"{profile_id}.sock"
+    if not socket_path.exists():
+        return {"state": "CONTROLLER_NOT_FOUND", "request_id": request_id, "profile_id": profile_id}
+    request = {"profile_id": profile_id, "request_id": request_id, "action": action, **data}
+    client = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+    client.settimeout(timeout)
+    try:
+        client.connect(str(socket_path))
+        client.sendall(json.dumps(request, ensure_ascii=False).encode())
+        try:
+            client.shutdown(socket.SHUT_WR)
+        except OSError:
+            # 半关闭只是 EOF 提示；请求已完整发送，对端先关闭时（如 macOS 偶发
+            # ENOTCONN）不得把整个请求判为失败，继续读取已缓冲的响应。
+            pass
+        chunks = []
+        while True:
+            chunk = client.recv(4096)
+            if not chunk:
+                break
+            chunks.append(chunk)
+        if not chunks:
+            return {"state": "IPC_TIMEOUT", "request_id": request_id, "profile_id": profile_id}
+        response = json.loads(b"".join(chunks).decode())
+        if response.get("request_id") != request_id:
+            return {"state": "STALE_RESPONSE", "request_id": request_id, "profile_id": profile_id}
+        return response
+    except (socket.timeout, TimeoutError):
+        return {"state": "IPC_TIMEOUT", "request_id": request_id, "profile_id": profile_id}
+    except (ConnectionRefusedError, ConnectionResetError, BrokenPipeError):
+        return {"state": "CONTROLLER_DEAD", "request_id": request_id, "profile_id": profile_id}
+    except OSError as exc:
+        return {"state": "IPC_ERROR", "request_id": request_id, "profile_id": profile_id,
+                "error": str(exc)[:120]}
+    finally:
+        client.close()
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser()
+    parser.add_argument("profile_id")
+    parser.add_argument("action", choices=("status", "scan-done", "verify-account", "refresh-qr", "close"))
+    parser.add_argument("--expected-account-name")
+    parser.add_argument("--base", type=Path, default=runtime_home())
+    args = parser.parse_args()
+    payload = {}
+    if args.expected_account_name is not None:
+        payload["expected_account_name"] = args.expected_account_name
+    result = send_command(args.base, args.profile_id, args.action.replace("-", "_"), **payload)
+    print(json.dumps(result, ensure_ascii=False, indent=2))
+    return 0 if result.get("state") not in {"CONTROLLER_NOT_FOUND", "CONTROLLER_DEAD", "IPC_TIMEOUT", "IPC_ERROR"} else 1
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

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

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

package/src/media_agent/platforms/xiaohongshu/xiaohongshu_check_login.py

@@ -0,0 +1,189 @@
+"""Legacy operation, executed only through main()."""
+
+def main():
+    #!/usr/bin/env python3
+    """Xiaohongshu Creator login check. Usage: python3 xiaohongshu_check_login.py <profile_id>"""
+    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
+    from media_agent.platforms.xiaohongshu.xiaohongshu_login_evidence import classify_login_evidence
+
+    BASE = runtime_home()
+    ALIAS = sys.argv[1] if len(sys.argv) > 1 else sys.exit(1)
+
+    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 {}
+
+    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': 'xhs-check-login', 'pid': os.getpid(), 'time': datetime.now().isoformat()}, indent=2))
+
+    ctx = None
+    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()
+        page.goto('https://creator.xiaohongshu.com/', wait_until='load', timeout=60000)
+        time.sleep(5)
+
+        url = page.url
+        title = page.title()
+
+        dom = page.evaluate('''() => {
+            const result = {};
+            const all = document.querySelectorAll('a, div, span, li, nav, header, aside');
+
+            // Dashboard indicators
+            const dashboardKeywords = ['创作服务平台', '首页', '笔记管理', '数据看板', '创作中心', '发布笔记', '我的笔记'];
+            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); }
+                }
+            }
+            const hasHomeRoute = window.location.href.includes('/new/home') || window.location.href.includes('/creator');
+            const hasAuthRoute = !window.location.href.includes('/login');
+            result.dashboard_detected = foundDashboard.length >= 2 || (foundDashboard.length >= 1 && hasHomeRoute && hasAuthRoute);
+            result.dashboard_evidence = foundDashboard;
+            if (hasHomeRoute) result.dashboard_evidence.push('backend_route');
+
+            // Login form indicators
+            const loginKeywords = ['短信登录', 'APP扫一扫登录', '扫码登录', '手机号'];
+            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*="手机"]');
+            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');
+
+            // QR code detection
+            const qrImgs = document.querySelectorAll('img, 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) < 30 && rect.width >= 100 && rect.width < 400) { hasQrImg = true; break; }
+            }
+            let hasQrText = false;
+            for (const el of document.querySelectorAll('div, span')) {
+                if (el.offsetParent === null) continue;
+                const t = el.textContent.trim();
+                if ((t.includes('扫码') || t.includes('二维码') || t.includes('小红书扫码')) && t.length < 30) { 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');
+
+            // Account name: use structural selectors only; never hard-code an account.
+            let accountName = '';
+            const accountSelectors = [
+                '[class*="user-name"]',
+                '[class*="userName"]',
+                '[class*="nickname"]',
+                '[class*="nick-name"]',
+                '[class*="account-name"]',
+                'header [class*="name"]'
+            ];
+            for (const selector of accountSelectors) {
+                for (const el of document.querySelectorAll(selector)) {
+                    if (el.offsetParent === null) continue;
+                    const t = el.textContent.trim();
+                    if (t.length >= 2 && t.length <= 40 && !dashboardKeywords.includes(t)) {
+                        accountName = t;
+                        break;
+                    }
+                }
+                if (accountName) break;
+            }
+            result.account_name = accountName || 'unknown';
+
+            return result;
+        }''')
+
+        login_form = dom['login_form_detected']
+        qr = dom['qr_detected']
+        status = classify_login_evidence(
+            url=url,
+            dashboard_evidence=dom['dashboard_evidence'],
+            login_form_detected=login_form,
+            qr_detected=qr,
+        )
+        dashboard = status == 'logged_in' or dom['dashboard_detected']
+
+        ctx.close()
+        LOCK_FILE.unlink(missing_ok=True)
+
+        # Update state
+        site_state = state.setdefault('xiaohongshu', {})
+        site_state['login_status'] = status
+        site_state['checked_at'] = datetime.now(timezone.utc).isoformat()
+        site_state['dashboard_detected'] = dashboard
+        site_state['login_form_detected'] = login_form
+        site_state['qr_detected'] = qr
+        (PROFILE_DIR / 'state.json').write_text(json.dumps(state, indent=2))
+
+        exit_code = 0 if status == 'logged_in' else (2 if status == 'login_required' else 7)
+
+        output = {
+            'profile_id': ALIAS,
+            'site': 'xiaohongshu',
+            'login_status': status,
+            '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'],
+            'account_name': dom['account_name'],
+            'browser_closed': True,
+            'lock_released': True,
+        }
+        print(json.dumps(output, indent=2, ensure_ascii=False))
+        sys.exit(exit_code)
+    except Exception as e:
+        try:
+            if ctx:
+                ctx.close()
+        except Exception:
+            pass
+        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()