@harness.farm/social-cli 0.1.0 → 0.1.2

package/dist/browser/runner.js

@@ -0,0 +1,75 @@
+/**
+ * Runner — orchestrates login flow + command execution for any adapter.
+ *
+ * Flow:
+ *   1. Connect to Chrome tab
+ *   2. Check if already logged in (via saved session or live cookies)
+ *   3. If not, navigate to loginUrl and wait for user to log in
+ *   4. Save session cookies
+ *   5. Run the requested command
+ *   6. Render output
+ */
+import { newTab, sleep } from './cdp.js';
+import { captureSession, restoreSession, hasSession } from './session.js';
+import { renderTable } from '../output/table.js';
+export async function run(adapter, opts) {
+    const port = opts.cdpPort ?? 9222;
+    console.log(`\n🔌 连接 Chrome CDP :${port}...`);
+    const client = await newTab(port);
+    // --- 登录流程 ---
+    let loggedIn = false;
+    // 先尝试恢复已保存的 session
+    if (hasSession(adapter.platform)) {
+        await restoreSession(client, adapter.platform);
+        // 导航到目标站验证 session 是否还有效
+        await client.navigate(adapter.loginUrl, 3000);
+        loggedIn = await adapter.isLoggedIn(client);
+        if (!loggedIn) {
+            console.log('⚠️  已保存的 session 已失效，需要重新登录');
+        }
+    }
+    // session 无效或没有 session → 引导用户登录
+    if (!loggedIn) {
+        await client.navigate(adapter.loginUrl, 2000);
+        console.log(`\n🔑 请在 Chrome 中登录 ${adapter.platform}，完成后按 Enter 继续...`);
+        await waitForEnter();
+        // 等待并轮询登录状态
+        for (let i = 0; i < 30; i++) {
+            loggedIn = await adapter.isLoggedIn(client);
+            if (loggedIn)
+                break;
+            process.stdout.write(`\r等待登录... ${(i + 1) * 2}s`);
+            await sleep(2000);
+        }
+        process.stdout.write('\n');
+        if (!loggedIn) {
+            client.close();
+            throw new Error('登录超时，请重试');
+        }
+        // 保存 session
+        await captureSession(client, adapter.platform);
+    }
+    console.log(`✅ 已登录 ${adapter.platform}`);
+    // --- 执行命令 ---
+    const handler = adapter.commands[opts.command];
+    if (!handler) {
+        const available = Object.keys(adapter.commands).join(', ');
+        client.close();
+        throw new Error(`未知命令 "${opts.command}"，可用: ${available}`);
+    }
+    console.log(`\n🔍 执行: ${adapter.platform} ${opts.command} ${(opts.args ?? []).join(' ')}\n`);
+    const result = await handler(client, opts.args ?? []);
+    client.close();
+    // --- 渲染 ---
+    renderTable(result.columns, result.rows);
+}
+function waitForEnter() {
+    return new Promise((resolve) => {
+        process.stdin.setRawMode?.(false);
+        process.stdin.resume();
+        process.stdin.once('data', () => {
+            process.stdin.pause();
+            resolve();
+        });
+    });
+}

package/dist/browser/session.js

@@ -0,0 +1,38 @@
+/**
+ * Session manager — save/load cookies to disk per platform.
+ */
+import fs from 'fs';
+import path from 'path';
+const SESSION_DIR = path.join(process.env.HOME ?? '.', '.cdp-scraper', 'sessions');
+function sessionPath(platform) {
+    return path.join(SESSION_DIR, `${platform}.json`);
+}
+export function saveSession(platform, cookies) {
+    fs.mkdirSync(SESSION_DIR, { recursive: true });
+    fs.writeFileSync(sessionPath(platform), JSON.stringify(cookies, null, 2));
+    console.log(`✅ Session saved → ${sessionPath(platform)} (${cookies.length} cookies)`);
+}
+export function loadSession(platform) {
+    const p = sessionPath(platform);
+    if (!fs.existsSync(p))
+        return null;
+    return JSON.parse(fs.readFileSync(p, 'utf-8'));
+}
+export function hasSession(platform) {
+    return fs.existsSync(sessionPath(platform));
+}
+/** 从 browser 读取 cookies 并保存 */
+export async function captureSession(client, platform) {
+    const cookies = await client.getAllCookies();
+    saveSession(platform, cookies);
+    return cookies;
+}
+/** 恢复已保存的 session 到 browser */
+export async function restoreSession(client, platform) {
+    const cookies = loadSession(platform);
+    if (!cookies)
+        return false;
+    await client.setCookies(cookies);
+    console.log(`✅ Session restored ← ${platform} (${cookies.length} cookies)`);
+    return true;
+}

package/dist/cli.js

@@ -0,0 +1,99 @@
+#!/usr/bin/env node
+/**
+ * CLI entry — supports both YAML adapters and TypeScript adapters.
+ *
+ * Usage:
+ *   tsx src/cli.ts <platform> <command> [args...]
+ *
+ * Adapter resolution order:
+ *   1. adapters/<platform>.yaml   ← YAML-first
+ *   2. src/adapters/<platform>.ts ← TypeScript fallback
+ *
+ * Examples:
+ *   tsx src/cli.ts xhs search 法律ai
+ *   tsx src/cli.ts xhs like "https://..."
+ *   tsx src/cli.ts xhs comment "https://..." "太棒了！"
+ *   tsx src/cli.ts xhs post --title "标题" --content "内容"
+ */
+import path from 'path';
+import fs from 'fs';
+import { fileURLToPath } from 'url';
+import { runYamlCommand } from './runner/yaml-runner.js';
+import { run } from './browser/runner.js';
+import { adapters } from './adapters/index.js';
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const ROOT = path.resolve(__dirname, '..');
+const [, , platform, command, ...rest] = process.argv;
+if (!platform || !command) {
+    printHelp();
+    process.exit(1);
+}
+// ── Resolve adapter ────────────────────────────────────────────────────────
+const yamlPath = path.join(ROOT, 'adapters', `${platform}.yaml`);
+const hasYaml = fs.existsSync(yamlPath);
+const tsAdapter = adapters[platform];
+if (!hasYaml && !tsAdapter) {
+    console.error(`❌ Unknown platform "${platform}"`);
+    console.error(`   YAML adapters: ${listYamlAdapters().join(', ') || '(none)'}`);
+    console.error(`   TS adapters:   ${Object.keys(adapters).join(', ')}`);
+    process.exit(1);
+}
+// ── Parse args ─────────────────────────────────────────────────────────────
+// Support both positional args and --key value flags
+const args = parseArgs(rest);
+// ── Run ───────────────────────────────────────────────────────────────────
+if (hasYaml) {
+    // YAML adapter: pass positional args in order
+    runYamlCommand(yamlPath, command, args.positional).catch(err => {
+        console.error('❌', err.message);
+        process.exit(1);
+    });
+}
+else {
+    // TypeScript adapter: pass flags as array (legacy)
+    run(tsAdapter, { command, args: rest }).catch(err => {
+        console.error('❌', err.message);
+        process.exit(1);
+    });
+}
+// ── Helpers ───────────────────────────────────────────────────────────────
+function parseArgs(argv) {
+    const positional = [];
+    const flags = {};
+    for (let i = 0; i < argv.length; i++) {
+        if (argv[i].startsWith('--')) {
+            flags[argv[i].slice(2)] = argv[++i] ?? '';
+        }
+        else {
+            positional.push(argv[i]);
+        }
+    }
+    return { positional, flags };
+}
+function listYamlAdapters() {
+    const dir = path.join(ROOT, 'adapters');
+    if (!fs.existsSync(dir))
+        return [];
+    return fs.readdirSync(dir)
+        .filter(f => f.endsWith('.yaml'))
+        .map(f => f.replace('.yaml', ''));
+}
+function printHelp() {
+    const yaml = listYamlAdapters();
+    const ts = Object.keys(adapters);
+    console.log('用法: tsx src/cli.ts <platform> <command> [args...]');
+    console.log('');
+    if (yaml.length) {
+        console.log('YAML 平台 (推荐):');
+        yaml.forEach(p => console.log(`  ${p}`));
+    }
+    if (ts.length) {
+        console.log('TS 平台:');
+        ts.forEach(p => console.log(`  ${p}`));
+    }
+    console.log('');
+    console.log('示例:');
+    console.log('  tsx src/cli.ts xhs search 法律ai');
+    console.log('  tsx src/cli.ts xhs like "https://www.xiaohongshu.com/explore/..."');
+    console.log('  tsx src/cli.ts xhs comment "https://..." "太棒了！"');
+}

package/dist/output/table.js

@@ -0,0 +1,43 @@
+/**
+ * CLI table renderer with CJK character width support.
+ */
+function dispWidth(s) {
+    let w = 0;
+    for (const c of s)
+        w += c.codePointAt(0) > 127 ? 2 : 1;
+    return w;
+}
+function truncate(s, maxW) {
+    let out = '', cur = 0;
+    for (const c of s) {
+        const cw = c.codePointAt(0) > 127 ? 2 : 1;
+        if (cur + cw > maxW - 1)
+            return out + '…';
+        out += c;
+        cur += cw;
+    }
+    return out;
+}
+function pad(s, width) {
+    return s + ' '.repeat(Math.max(0, width - dispWidth(s)));
+}
+function sep(cols, char = '-') {
+    return '+' + cols.map((c) => char.repeat(c.width + 2)).join('+') + '+';
+}
+function row(cols, values) {
+    return '|' + cols.map((c, i) => ` ${pad(truncate(String(values[i] ?? ''), c.width), c.width)} `).join('|') + '|';
+}
+export function renderTable(cols, data) {
+    const divider = sep(cols);
+    const doubleSep = sep(cols, '=');
+    console.log(divider);
+    console.log(row(cols, cols.map((c) => c.header)));
+    console.log(doubleSep);
+    data.forEach((item, i) => {
+        console.log(row(cols, cols.map((c) => String(item[c.key] ?? ''))));
+        if (i < data.length - 1)
+            console.log(divider);
+    });
+    console.log(divider);
+    console.log(`\n共 ${data.length} 条结果`);
+}

package/dist/runner/step-executor.js

@@ -0,0 +1,142 @@
+/**
+ * Step executor — maps YAML step types to agent-browser CLI calls.
+ *
+ * Each method runs `agent-browser <args>` and returns parsed JSON result.
+ */
+import { execSync } from 'child_process';
+export class StepExecutor {
+    wsUrl;
+    connected = false;
+    constructor(wsUrl) {
+        this.wsUrl = wsUrl;
+    }
+    /** Connect to Chrome tab (once per session) */
+    connect() {
+        if (this.connected)
+            return;
+        this.run(['connect', this.wsUrl]);
+        this.connected = true;
+    }
+    // ── Core step methods ──────────────────────────────────────────────────────
+    open(url) {
+        return this.run(['open', url]);
+    }
+    click(selector) {
+        return this.run(['click', selector]);
+    }
+    /** Click an element by its visible text (uses eval under the hood) */
+    clickText(text) {
+        const js = `(function(){
+      var el = [...document.querySelectorAll('*')].find(function(e){
+        return e.textContent.trim() === ${JSON.stringify(text)} && e.children.length === 0;
+      });
+      if(el){ el.click(); return true; }
+      return false;
+    })()`;
+        const r = this.eval(js);
+        if (!r.value)
+            return { ok: false, error: `Text not found: "${text}"` };
+        return { ok: true };
+    }
+    fill(selector, value) {
+        // Use eval to avoid shell-quoting issues with complex selectors
+        const js = `(function(){
+      var el = document.querySelector(${JSON.stringify(selector)});
+      if (!el) return false;
+      el.focus();
+      var nativeSet = Object.getOwnPropertyDescriptor(window.HTMLInputElement.prototype, 'value')
+                   || Object.getOwnPropertyDescriptor(window.HTMLTextAreaElement.prototype, 'value');
+      if (nativeSet && nativeSet.set) nativeSet.set.call(el, ${JSON.stringify(value)});
+      else el.value = ${JSON.stringify(value)};
+      el.dispatchEvent(new Event('input', { bubbles: true }));
+      el.dispatchEvent(new Event('change', { bubbles: true }));
+      return true;
+    })()`;
+        return this.eval(js);
+    }
+    type(selector, value) {
+        return this.run(['type', selector, value]);
+    }
+    /** Type into a contenteditable element via execCommand */
+    typeContentEditable(selector, value) {
+        const js = `(function(){
+      var el = document.querySelector(${JSON.stringify(selector)});
+      if(!el) return false;
+      el.focus();
+      document.execCommand('selectAll', false, null);
+      document.execCommand('insertText', false, ${JSON.stringify(value)});
+      return el.textContent || el.value || true;
+    })()`;
+        return this.eval(js);
+    }
+    /** Type via agent-browser's real keystroke simulation (works with Draft.js / React) */
+    typeKeys(selector, value) {
+        return this.run(['type', selector, value]);
+    }
+    wait(msOrSelector) {
+        if (typeof msOrSelector === 'number') {
+            return this.run(['wait', String(msOrSelector)]);
+        }
+        return this.run(['wait', msOrSelector]);
+    }
+    /** Press a key via agent-browser press (real key event, e.g. z, x, Enter, Control+Enter) */
+    pressKey(key) {
+        return this.run(['press', key]);
+    }
+    /** Insert text into the currently focused element via agent-browser keyboard type */
+    keyboardInsertText(text) {
+        // 'keyboard type' sends real key events char-by-char — works with Draft.js
+        return this.run(['keyboard', 'type', text]);
+    }
+    eval(js) {
+        const r = this.run(['eval', js]);
+        if (r.ok && r.value !== undefined) {
+            // agent-browser wraps eval result in data.result
+            const data = r.value;
+            return { ok: true, value: data.result ?? r.value };
+        }
+        return r;
+    }
+    screenshot(path) {
+        return this.run(path ? ['screenshot', path] : ['screenshot']);
+    }
+    upload(selector, filePath) {
+        return this.run(['upload', selector, filePath]);
+    }
+    getUrl() {
+        const r = this.run(['get', 'url']);
+        const data = r.value;
+        return data?.url ?? '';
+    }
+    snapshot() {
+        const r = this.run(['snapshot']);
+        return String(r.value ?? '');
+    }
+    // ── Internal runner ────────────────────────────────────────────────────────
+    run(args) {
+        try {
+            const cmd = `agent-browser ${args.map(a => this.shellQuote(a)).join(' ')}`;
+            const out = execSync(cmd, {
+                env: { ...process.env, AGENT_BROWSER_JSON: '1' },
+                timeout: 30000,
+                encoding: 'utf8',
+            });
+            // Find last JSON line (agent-browser may emit warnings before JSON)
+            const jsonLine = out.trim().split('\n').reverse().find(l => l.startsWith('{'));
+            if (!jsonLine)
+                return { ok: true };
+            const parsed = JSON.parse(jsonLine);
+            if (!parsed.success)
+                return { ok: false, error: parsed.error ?? 'unknown error' };
+            return { ok: true, value: parsed.data };
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            return { ok: false, error: msg };
+        }
+    }
+    shellQuote(s) {
+        // Wrap in single quotes, escape internal single quotes
+        return `'${s.replace(/'/g, "'\\''")}'`;
+    }
+}