@danceiny/gotry 0.0.1-rc.15 → 0.0.1-rc.16

package/dist/capabilities/session/extension-bridge.js

@@ -0,0 +1,345 @@
+import { randomUUID } from 'node:crypto';
+import { createServer } from 'node:http';
+export const BRIDGE_PORTS = [
+    8791,
+    8792,
+    8793,
+    8794,
+    8795
+];
+export const BRIDGE_PROTOCOL = 'session-bridge.v1';
+export const EXTENSION_ID = 'olpgkofjhhiiiahdkkbcninhjmegghfe';
+export const EXTENSION_ORIGIN = `chrome-extension://${EXTENSION_ID}`;
+export const EXTENSION_CONNECTED_WINDOW_MS = 45_000;
+export const JOBS_LONG_POLL_MS = 20_000;
+export const DEFAULT_EXTENSION_WAIT_MS = 6_000;
+export const MAX_RESULT_BODY_BYTES = 8 * 1024 * 1024;
+function readBody(req, cap) {
+    return new Promise((resolve)=>{
+        const chunks = [];
+        let size = 0;
+        let done = false;
+        req.on('data', (c)=>{
+            if (done) return;
+            size += c.length;
+            if (size > cap) {
+                done = true;
+                req.removeAllListeners('data');
+                resolve({
+                    err: `body 超 ${cap} 字节上限(防失控回包)`
+                });
+                return;
+            }
+            chunks.push(c);
+        });
+        req.on('end', ()=>{
+            if (!done) {
+                done = true;
+                resolve(Buffer.concat(chunks).toString('utf8'));
+            }
+        });
+        req.on('error', ()=>{
+            if (!done) {
+                done = true;
+                resolve({
+                    err: 'body 读取失败'
+                });
+            }
+        });
+    });
+}
+export async function createSessionBridge(opts = {}) {
+    const ports = opts.ports ?? [
+        ...BRIDGE_PORTS
+    ];
+    const extensionOrigin = opts.extensionOrigin ?? EXTENSION_ORIGIN;
+    const now = opts.now ?? Date.now;
+    const queue = [];
+    const inFlight = new Map();
+    const parked = [];
+    let lastSeenAt = 0;
+    let closed = false;
+    let port = 0;
+    const extensionConnected = ()=>lastSeenAt > 0 && now() - lastSeenAt < EXTENSION_CONNECTED_WINDOW_MS;
+    function takeQueued() {
+        const next = queue.shift();
+        if (next) inFlight.set(next.job.jobId, next);
+        return next ?? null;
+    }
+    const server = createServer((req, res)=>{
+        if (closed) {
+            res.statusCode = 503;
+            res.end();
+            return;
+        }
+        const urlPath = (req.url ?? '').split('?')[0];
+        const origin = req.headers.origin;
+        const isPost = req.method === 'POST';
+        if (isPost && origin !== extensionOrigin) {
+            res.statusCode = 403;
+            res.setHeader('content-type', 'application/json');
+            res.end(JSON.stringify({
+                ok: false,
+                error: 'origin 不在桥白名单'
+            }));
+            return;
+        }
+        if (origin === extensionOrigin) {
+            lastSeenAt = now();
+        }
+        const finish = (code, body)=>{
+            res.statusCode = code;
+            res.setHeader('content-type', 'application/json');
+            res.end(JSON.stringify(body));
+        };
+        if (req.method === 'GET' && urlPath === '/health') {
+            finish(200, {
+                ok: true,
+                service: 'gotry-session-bridge',
+                protocol: BRIDGE_PROTOCOL
+            });
+            return;
+        }
+        if (req.method === 'GET' && urlPath === '/status') {
+            finish(200, {
+                ok: true,
+                port,
+                protocol: BRIDGE_PROTOCOL,
+                extensionConnected: extensionConnected(),
+                lastSeenMsAgo: lastSeenAt > 0 ? now() - lastSeenAt : null,
+                queued: queue.length,
+                inFlight: inFlight.size
+            });
+            return;
+        }
+        if (isPost && urlPath === '/jobs') {
+            const next = takeQueued();
+            if (next) {
+                finish(200, {
+                    job: next.job
+                });
+                return;
+            }
+            const parkTimer = setTimeout(()=>{
+                const i = parked.findIndex((p)=>p.res === res);
+                if (i >= 0) parked.splice(i, 1);
+                finish(200, {
+                    job: null
+                });
+            }, JOBS_LONG_POLL_MS);
+            parked.push({
+                res,
+                timer: parkTimer
+            });
+            res.on('close', ()=>{
+                const i = parked.findIndex((p)=>p.res === res);
+                if (i >= 0) parked.splice(i, 1);
+                clearTimeout(parkTimer);
+            });
+            return;
+        }
+        if (isPost && urlPath.startsWith('/results/')) {
+            const jobId = decodeURIComponent(urlPath.slice('/results/'.length));
+            void readBody(req, MAX_RESULT_BODY_BYTES).then((body)=>{
+                if (typeof body !== 'string') {
+                    finish(400, {
+                        ok: false,
+                        error: body.err
+                    });
+                    return;
+                }
+                let parsed;
+                try {
+                    parsed = JSON.parse(body);
+                } catch  {
+                    finish(400, {
+                        ok: false,
+                        error: '回包不是 JSON'
+                    });
+                    return;
+                }
+                finish(200, {
+                    ok: true
+                });
+                resolveJob(jobId, parsed);
+            });
+            return;
+        }
+        finish(404, {
+            ok: false,
+            error: 'unknown endpoint'
+        });
+    });
+    function resolveJob(jobId, parsed) {
+        const inFlightHit = inFlight.get(jobId);
+        if (inFlightHit) {
+            inFlight.delete(jobId);
+            if (inFlightHit.timer) clearTimeout(inFlightHit.timer);
+            inFlightHit.resolve({
+                ok: true,
+                result: parsed
+            });
+            return;
+        }
+        const idx = queue.findIndex((q)=>q.job.jobId === jobId);
+        if (idx < 0) return;
+        const [entry] = queue.splice(idx, 1);
+        if (entry.timer) clearTimeout(entry.timer);
+        entry.resolve({
+            ok: true,
+            result: parsed
+        });
+    }
+    function dispatchToParked(entry) {
+        const p = parked.shift();
+        if (!p) return false;
+        clearTimeout(p.timer);
+        inFlight.set(entry.job.jobId, entry);
+        p.res.statusCode = 200;
+        p.res.setHeader('content-type', 'application/json');
+        p.res.end(JSON.stringify({
+            job: entry.job
+        }));
+        return true;
+    }
+    const bridge = {
+        port: 0,
+        extensionConnected,
+        submit (job, submitOpts = {}) {
+            const extensionWaitMs = submitOpts.extensionWaitMs ?? DEFAULT_EXTENSION_WAIT_MS;
+            const timeoutMs = submitOpts.timeoutMs ?? job.timeoutMs ?? 30_000;
+            const full = {
+                jobId: randomUUID(),
+                ...job
+            };
+            return new Promise((resolve)=>{
+                if (closed) {
+                    resolve({
+                        ok: false,
+                        reason: 'bridge-unavailable',
+                        summary: '扩展桥已关闭'
+                    });
+                    return;
+                }
+                const entry = {
+                    job: full,
+                    resolve: null,
+                    timer: null
+                };
+                const settle = (outcome)=>{
+                    if (entry.timer) clearTimeout(entry.timer);
+                    const i = queue.findIndex((q)=>q.job.jobId === full.jobId);
+                    if (i >= 0) queue.splice(i, 1);
+                    inFlight.delete(full.jobId);
+                    resolve(outcome);
+                };
+                entry.resolve = settle;
+                entry.timer = setTimeout(()=>{
+                    settle({
+                        ok: false,
+                        reason: 'timeout',
+                        summary: `扩展未在 ${timeoutMs}ms 内回包(标签页无嗅探命中或扩展已停用)`
+                    });
+                }, timeoutMs);
+                if (!dispatchToParked(entry)) queue.push(entry);
+                let waited = 0;
+                const waitTimer = setInterval(()=>{
+                    waited += 250;
+                    if (bridge.extensionConnected()) {
+                        clearInterval(waitTimer);
+                        return;
+                    }
+                    if (waited >= extensionWaitMs) {
+                        clearInterval(waitTimer);
+                        settle({
+                            ok: false,
+                            reason: 'extension-not-connected',
+                            summary: 'GoTry Session Bridge 扩展未连接(未安装或已停用)。一次性安装:npx gotry setup 查看指引 → chrome://extensions 开发者模式「加载已解压的扩展程序」指向 ~/.gotry/extension'
+                        });
+                    }
+                }, 250);
+            });
+        },
+        close: ()=>new Promise((resolve)=>{
+                closed = true;
+                const voided = {
+                    ok: false,
+                    reason: 'bridge-unavailable',
+                    summary: '扩展桥关闭,任务作废(零花费)'
+                };
+                for (const q of queue.splice(0)){
+                    if (q.timer) clearTimeout(q.timer);
+                    q.resolve(voided);
+                }
+                for (const [, q] of inFlight){
+                    if (q.timer) clearTimeout(q.timer);
+                    q.resolve(voided);
+                }
+                inFlight.clear();
+                for (const p of parked.splice(0)){
+                    clearTimeout(p.timer);
+                    p.res.destroy();
+                }
+                server.close(()=>resolve());
+                server.closeAllConnections?.();
+            })
+    };
+    const listenAt = (p)=>new Promise((resolve, reject)=>{
+            const onError = (e)=>{
+                server.off('listening', onListening);
+                reject(e);
+            };
+            const onListening = ()=>{
+                server.off('error', onError);
+                resolve();
+            };
+            server.once('error', onError);
+            server.once('listening', onListening);
+            server.listen(p, '127.0.0.1');
+        });
+    for (const p of ports){
+        try {
+            await listenAt(p);
+            if (!opts.keepBridge) {
+                server.unref();
+            }
+            port = server.address().port;
+            bridge.port = port;
+            return {
+                ok: true,
+                bridge
+            };
+        } catch  {}
+    }
+    return {
+        ok: false,
+        summary: `扩展桥端口池 ${ports.join('/')} 全部被占(多为并行 gotry 实例);不影响官方通道 gotry_flyai_search。关闭多余实例后重试。`
+    };
+}
+let singleton = null;
+export async function getOrCreateSessionBridge(opts = {}) {
+    if (singleton) return singleton;
+    const created = await createSessionBridge(opts);
+    const result = created.ok ? {
+        ok: true,
+        bridge: created.bridge
+    } : {
+        ok: false,
+        summary: created.summary
+    };
+    singleton = result;
+    return result;
+}
+export async function __resetSessionBridgeForTest() {
+    if (singleton?.ok) await singleton.bridge.close().catch(()=>{});
+    singleton = null;
+}
+export function __setSessionBridgeForTest(bridge) {
+    singleton = bridge ? {
+        ok: true,
+        bridge
+    } : null;
+}
+
+
+//# sourceURL=/Users/bytedance/work/gotry-publish/ts/capabilities/session/extension-bridge.ts

package/dist/capabilities/session/extension-channel.js

@@ -0,0 +1,93 @@
+import { getOrCreateSessionBridge } from './extension-bridge.js';
+export const NEEDS_EXTENSION_HINT = '需要一次性安装 GoTry Session Bridge 浏览器扩展(约 30 秒,装完零弹窗):跑 npx gotry setup 拿指引,' + '或在 chrome://extensions 开发者模式「加载已解压的扩展程序」指向 ~/.gotry/extension';
+export function classifyBridgeFailure(kind) {
+    return kind === 'extension-not-connected' ? 'needs-extension' : 'error';
+}
+function bridgeReady(b) {
+    return b.ok === true;
+}
+export async function extensionCookieNames(q) {
+    const bridge = await getOrCreateSessionBridge();
+    if (!bridgeReady(bridge)) return {
+        ok: false,
+        kind: 'bridge-unavailable',
+        summary: bridge.summary
+    };
+    const outcome = await bridge.bridge.submit({
+        kind: 'cookie-names',
+        site: q.site,
+        timeoutMs: q.timeoutMs ?? 8_000
+    }, {
+        timeoutMs: q.timeoutMs ?? 8_000
+    });
+    if (!outcome.ok) return {
+        ok: false,
+        kind: outcome.reason,
+        summary: outcome.summary
+    };
+    const names = Array.isArray(outcome.result.names) ? outcome.result.names.filter((n)=>typeof n === 'string') : [];
+    return {
+        ok: true,
+        tickets: names
+    };
+}
+export async function extensionOpenLogin(q) {
+    const bridge = await getOrCreateSessionBridge();
+    if (!bridgeReady(bridge)) return {
+        ok: false,
+        kind: 'bridge-unavailable',
+        summary: bridge.summary
+    };
+    const outcome = await bridge.bridge.submit({
+        kind: 'open-login',
+        site: q.site,
+        url: q.url
+    }, {
+        timeoutMs: q.timeoutMs ?? 15_000
+    });
+    if (!outcome.ok) return {
+        ok: false,
+        kind: outcome.reason,
+        summary: outcome.summary
+    };
+    if (!outcome.result.opened) return {
+        ok: false,
+        kind: 'job-error',
+        summary: outcome.result.error ?? '登录入口打开失败'
+    };
+    return {
+        ok: true,
+        opened: true
+    };
+}
+export async function extensionSearchJob(q) {
+    const bridge = await getOrCreateSessionBridge();
+    if (!bridgeReady(bridge)) return {
+        ok: false,
+        kind: 'bridge-unavailable',
+        summary: bridge.summary
+    };
+    const timeoutMs = q.timeoutMs ?? 30_000;
+    const outcome = await bridge.bridge.submit({
+        kind: 'search',
+        site: q.site,
+        url: q.url,
+        timeoutMs
+    }, {
+        timeoutMs: timeoutMs + 10_000
+    });
+    if (!outcome.ok) return {
+        ok: false,
+        kind: outcome.reason,
+        summary: outcome.summary
+    };
+    return {
+        ok: true,
+        body: typeof outcome.result.body === 'string' ? outcome.result.body : '',
+        title: typeof outcome.result.title === 'string' ? outcome.result.title : '',
+        timedOut: outcome.result.timeout === true
+    };
+}
+
+
+//# sourceURL=/Users/bytedance/work/gotry-publish/ts/capabilities/session/extension-channel.ts

package/dist/capabilities/session/extract.js

@@ -37,4 +37,4 @@ export function nameAffinity(a, b) {
 }
 
 
-//# sourceURL=/Users/bytedance/work/gotry/ts/capabilities/session/extract.ts
+//# sourceURL=/Users/bytedance/work/gotry-publish/ts/capabilities/session/extract.ts

package/dist/capabilities/session/health-watch.js

@@ -0,0 +1,216 @@
+import { request } from 'node:http';
+import { BRIDGE_PORTS, EXTENSION_CONNECTED_WINDOW_MS } from './extension-bridge.js';
+export async function combinedProbe(opts, businessProbe) {
+    if (!await defaultProbe(opts)) return false;
+    if (!businessProbe) return true;
+    try {
+        const channel = await import('./extension-channel.js');
+        const bridge = await import('./extension-bridge.js');
+        bridge.getOrCreateSessionBridge({
+            keepBridge: opts.keepBridge === true
+        });
+        const cookie = await channel.extensionCookieNames({
+            site: 'ctrip-flight',
+            domain: 'ctrip.com',
+            ticketNames: [
+                'cticket',
+                'uid'
+            ],
+            timeoutMs: 8_000
+        });
+        return cookie.ok;
+    } catch  {
+        return false;
+    }
+}
+export async function defaultProbe(opts) {
+    const ports = opts.ports ?? [
+        ...BRIDGE_PORTS
+    ];
+    for (const port of ports){
+        const ok = await new Promise((resolve)=>{
+            const req = request({
+                host: '127.0.0.1',
+                port,
+                path: '/status',
+                method: 'GET',
+                timeout: 2_000
+            }, (res)=>{
+                if (!res.statusCode || res.statusCode >= 400) {
+                    resolve(false);
+                    return;
+                }
+                let buf = '';
+                res.setEncoding('utf8');
+                res.on('data', (c)=>{
+                    buf += c;
+                });
+                res.on('end', ()=>{
+                    try {
+                        const payload = JSON.parse(buf);
+                        resolve(payload.extensionConnected === true);
+                    } catch  {
+                        resolve(false);
+                    }
+                });
+                res.on('error', ()=>resolve(false));
+            });
+            req.on('error', ()=>resolve(false));
+            req.on('timeout', ()=>{
+                req.destroy();
+                resolve(false);
+            });
+            req.end();
+        });
+        if (ok) return true;
+    }
+    return false;
+}
+export function startExtensionHealthWatch(opts = {}) {
+    const timeoutMs = opts.timeoutMs ?? 120_000;
+    const intervalMs = opts.intervalMs ?? 5_000;
+    const maxRetries = opts.maxRetries ?? 1;
+    const businessProbe = opts.businessProbe ?? true;
+    const probe = opts.probe ?? (async ()=>{
+        const bridge = await import('./extension-bridge.js');
+        await bridge.getOrCreateSessionBridge({
+            keepBridge: opts.keepBridge === true
+        });
+        return combinedProbe(opts, businessProbe);
+    });
+    const now = opts.now ?? Date.now;
+    const state = {
+        cancelled: false,
+        readyCount: 0,
+        resolveOutcome: null
+    };
+    const finish = (outcome)=>{
+        if (state.resolveOutcome) {
+            const resolve = state.resolveOutcome;
+            state.resolveOutcome = null;
+            resolve(outcome);
+        }
+    };
+    const onAbort = ()=>{
+        state.cancelled = true;
+        finish({
+            ready: false,
+            reason: 'cancelled',
+            attempts: state.readyCount,
+            waitedMs: 0
+        });
+    };
+    opts.signal?.addEventListener('abort', onAbort, {
+        once: true
+    });
+    const waitReady = ()=>new Promise((resolve)=>{
+            if (state.cancelled) {
+                resolve({
+                    ready: false,
+                    reason: 'cancelled',
+                    attempts: 0,
+                    waitedMs: 0
+                });
+                return;
+            }
+            if (timeoutMs === 0) {
+                resolve({
+                    ready: false,
+                    reason: 'timeout',
+                    attempts: 0,
+                    waitedMs: 0
+                });
+                return;
+            }
+            state.resolveOutcome = resolve;
+            const deadline = now() + timeoutMs;
+            let attempts = 0;
+            let waitedMs = 0;
+            const tick = async ()=>{
+                if (state.cancelled || state.resolveOutcome === null) return;
+                attempts += 1;
+                try {
+                    const ready = await probe();
+                    if (ready && !state.cancelled) {
+                        state.readyCount += 1;
+                        waitedMs = timeoutMs - Math.max(0, deadline - now());
+                        if (state.readyCount <= maxRetries) {
+                            finish({
+                                ready: true,
+                                attempts,
+                                waitedMs: waitedMs > 0 ? waitedMs : 0
+                            });
+                            return;
+                        }
+                    }
+                } catch  {}
+                waitedMs = timeoutMs - Math.max(0, deadline - now());
+                if (now() >= deadline) {
+                    finish({
+                        ready: false,
+                        reason: 'timeout',
+                        attempts,
+                        waitedMs
+                    });
+                    return;
+                }
+                const remaining = deadline - now();
+                const delay = Math.min(intervalMs, Math.max(0, remaining));
+                if (delay <= 0) {
+                    finish({
+                        ready: false,
+                        reason: 'timeout',
+                        attempts,
+                        waitedMs
+                    });
+                    return;
+                }
+                setTimeout(tick, delay);
+            };
+            void tick();
+        });
+    return {
+        waitReady,
+        cancel: ()=>onAbort(),
+        get readyCount () {
+            return state.readyCount;
+        }
+    };
+}
+export async function withAutoRetryOnExtension(args) {
+    const first = await args.fetch();
+    if (!args.isExtensionNeeded(first)) {
+        return {
+            result: first,
+            retried: false,
+            attempts: 1,
+            outcome: {
+                ready: true,
+                attempts: 0,
+                waitedMs: 0
+            }
+        };
+    }
+    const watch = startExtensionHealthWatch(args.watch ?? {});
+    const outcome = await watch.waitReady();
+    if (!outcome.ready) {
+        return {
+            result: first,
+            retried: false,
+            attempts: 1,
+            outcome
+        };
+    }
+    const retriedResult = await (args.retry ?? args.fetch)();
+    return {
+        result: retriedResult,
+        retried: true,
+        attempts: 2,
+        outcome
+    };
+}
+export const HEALTH_WATCH_INTERVAL_FLOOR_MS = 1_000;
+export const HEALTH_WATCH_INTERVAL_CEIL_MS = EXTENSION_CONNECTED_WINDOW_MS - 1_000;
+
+
+//# sourceURL=/Users/bytedance/work/gotry-publish/ts/capabilities/session/health-watch.ts

package/dist/capabilities/session/read-guard.js

@@ -96,4 +96,4 @@ export async function attachReadGuardPuppeteer(browser, auditPath) {
 }
 
 
-//# sourceURL=/Users/bytedance/work/gotry/ts/capabilities/session/read-guard.ts
+//# sourceURL=/Users/bytedance/work/gotry-publish/ts/capabilities/session/read-guard.ts

package/dist/capabilities/session/transport.js

@@ -113,4 +113,4 @@ export async function openSession(opts = {}) {
 }
 
 
-//# sourceURL=/Users/bytedance/work/gotry/ts/capabilities/session/transport.ts
+//# sourceURL=/Users/bytedance/work/gotry-publish/ts/capabilities/session/transport.ts