@agentrhq/webcmd 0.1.2 → 0.2.0

package/clis/deepseek/utils.js

@@ -1,487 +0,0 @@
-import { ArgumentError } from '@agentrhq/webcmd/errors';
-
-export const DEEPSEEK_DOMAIN = 'chat.deepseek.com';
-export const DEEPSEEK_URL = 'https://chat.deepseek.com/';
-export const TEXTAREA_SELECTOR = 'textarea[placeholder*="DeepSeek"]';
-export const MESSAGE_SELECTOR = '.ds-message';
-const CONVERSATION_ID_RE = /^[a-f0-9]{8}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{4}-[a-f0-9]{12}$/i;
-
-/**
- * Normalize a DeepSeek conversation ID. Accepts a bare UUID or any URL that
- * embeds one (`/a/chat/s/<id>` or full chat URL).
- *
- * Throws ArgumentError when the input does not contain a UUID-shaped id, so
- * `detail` fails before any browser navigation happens.
- */
-export function parseDeepSeekConversationId(input) {
-    const raw = String(input ?? '').trim();
-    if (!raw) {
-        throw new ArgumentError('id', 'must be a non-empty conversation ID or URL');
-    }
-    const urlMatch = raw.match(/\/a\/chat\/s\/([a-f0-9-]+)/i);
-    const candidate = urlMatch ? urlMatch[1] : raw;
-    if (!CONVERSATION_ID_RE.test(candidate)) {
-        throw new ArgumentError(
-            'id',
-            `not a valid DeepSeek conversation ID (got "${input}"); expected a UUID like "749e6bbd-6a45-4440-beaa-ae5238bf06d8" or a full /a/chat/s/<id> URL`,
-        );
-    }
-    return candidate.toLowerCase();
-}
-
-export async function isOnDeepSeek(page) {
-    const url = await page.evaluate('window.location.href').catch(() => '');
-    if (typeof url !== 'string' || !url) return false;
-    try {
-        const h = new URL(url).hostname;
-        return h === 'deepseek.com' || h.endsWith('.deepseek.com');
-    } catch {
-        return false;
-    }
-}
-
-export async function ensureOnDeepSeek(page) {
-    if (await isOnDeepSeek(page)) return false;
-    await page.goto(DEEPSEEK_URL);
-    // Wait for the composer textarea instead of a fixed 3 s sleep. On the login
-    // page it never mounts; swallow the timeout so callers (status / read /
-    // history) can still inspect page state.
-    try {
-        await page.wait({ selector: TEXTAREA_SELECTOR, timeout: 8 });
-    } catch {
-        // Login or error page — downstream will see hasTextarea=false / empty results.
-    }
-    return true;
-}
-
-export async function getPageState(page) {
-    return page.evaluate(`(() => {
-        const url = window.location.href;
-        const title = document.title;
-        const textarea = document.querySelector('${TEXTAREA_SELECTOR}');
-        const avatar = document.querySelector('img[src*="user-avatar"]');
-        return {
-            url,
-            title,
-            hasTextarea: !!textarea,
-            isLoggedIn: !!avatar,
-        };
-    })()`);
-}
-
-export async function selectModel(page, modelName) {
-    return page.evaluate(`(() => {
-        var radios = document.querySelectorAll('div[role="radio"]');
-        if (radios.length === 0) return { ok: false };
-        var name = '${modelName}'.toLowerCase();
-        var index = name === 'instant' ? 0 : name === 'expert' ? 1 : name === 'vision' ? 2 : -1;
-        if (index < 0 || index >= radios.length) return { ok: false };
-        var target = radios[index];
-        var alreadySelected = target.getAttribute('aria-checked') === 'true';
-        if (!alreadySelected) target.click();
-        return { ok: true, toggled: !alreadySelected };
-    })()`);
-}
-
-export async function setFeature(page, featureName, enabled) {
-    // Match by position: DeepThink is the first toggle, Search is the second
-    var index = featureName === 'DeepThink' ? 0 : 1;
-    return page.evaluate(`(() => {
-        var toggles = Array.from(document.querySelectorAll('.ds-toggle-button'));
-        var btn = toggles[${index}];
-        if (!btn) return { ok: false };
-        var isActive = btn.classList.contains('ds-toggle-button--selected');
-        if (${enabled} !== isActive) btn.click();
-        return { ok: true, toggled: ${enabled} !== isActive };
-    })()`);
-}
-
-export async function sendMessage(page, prompt) {
-    const promptJson = JSON.stringify(prompt);
-    return page.evaluate(`(async () => {
-        const box = document.querySelector('${TEXTAREA_SELECTOR}');
-        if (!box) return { ok: false, reason: 'textarea not found' };
-
-        box.focus();
-        box.value = '';
-        document.execCommand('selectAll');
-        document.execCommand('insertText', false, ${promptJson});
-        await new Promise(r => setTimeout(r, 800));
-
-        // Find the send button: last non-toggle button in the textarea's container
-        var container = box.parentElement;
-        while (container && !container.querySelector('div[role="button"]')) {
-            container = container.parentElement;
-        }
-        if (container) {
-            var btns = container.querySelectorAll('div[role="button"]:not(.ds-toggle-button)');
-            var sendBtn = btns[btns.length - 1];
-            if (sendBtn && sendBtn.getAttribute('aria-disabled') === 'false'
-                && sendBtn.querySelectorAll('svg').length > 0) {
-                sendBtn.click();
-                return { ok: true };
-            }
-        }
-
-        box.dispatchEvent(new KeyboardEvent('keydown', { key: 'Enter', code: 'Enter', keyCode: 13, bubbles: true }));
-        return { ok: true, method: 'enter' };
-    })()`);
-}
-
-export async function getBubbleCount(page) {
-    const count = await page.evaluate(`(() => {
-        return document.querySelectorAll('${MESSAGE_SELECTOR}').length;
-    })()`);
-    return count || 0;
-}
-
-// Parse thinking response using text as a fallback when DOM-level extraction
-// is not available.  Does NOT split on \n\n — that heuristic silently corrupts
-// multi-paragraph thinking or multi-paragraph answers.  Instead, everything
-// after the header is treated as thinking content, and `response` stays empty
-// until the caller provides a DOM-separated answer.
-export function parseThinkingResponse(rawText) {
-    if (!rawText) return null;
-
-    // Match thinking header patterns: "Thought for X seconds" or "已思考（用时 X 秒）"
-    const thinkHeaderMatch = rawText.match(/^(Thought for ([\d.]+) seconds?|已思考（用时 ([\d.]+) 秒）)\s*/);
-
-    if (!thinkHeaderMatch) {
-        // No thinking section found, return plain response
-        return { response: rawText, thinking: null, thinking_time: null };
-    }
-
-    const thinkingTime = thinkHeaderMatch[2] || thinkHeaderMatch[3];
-    const afterHeader = rawText.slice(thinkHeaderMatch[0].length);
-
-    // Treat everything after the header as thinking.  The response will be
-    // populated by the DOM-level extraction in waitForResponse().
-    return {
-        response: '',
-        thinking: afterHeader.trim(),
-        thinking_time: thinkingTime,
-    };
-}
-
-export async function waitForResponse(page, baselineCount, prompt, timeoutMs, parseThinking = false) {
-    const startTime = Date.now();
-    let lastText = '';
-    let stableCount = 0;
-
-    while (Date.now() - startTime < timeoutMs) {
-        await page.wait(3);
-
-        let result;
-        try {
-            result = await page.evaluate(`(() => {
-                const bubbles = document.querySelectorAll('${MESSAGE_SELECTOR}');
-                const texts = Array.from(bubbles).map(b => (b.innerText || '').trim()).filter(Boolean);
-                var last = texts[texts.length - 1] || '';
-
-                // DOM-level thinking/response separation.
-                // DeepSeek renders thinking in a collapsible container with a
-                // distinct class (e.g. .ds-markdown--think or similar) and the
-                // final answer in the main .ds-markdown region.  By querying
-                // these separately we avoid any text-heuristic split.
-                var thinkEl = null, answerEl = null, thinkTime = null;
-                if (${parseThinking} && bubbles.length > 0) {
-                    var lastBubble = bubbles[bubbles.length - 1];
-                    // Thinking container — DeepSeek uses various class names;
-                    // try common selectors.
-                    thinkEl = lastBubble.querySelector('.ds-markdown--think')
-                           || lastBubble.querySelector('[class*="think"]');
-                    // Final answer container — the main markdown block that is
-                    // NOT the thinking section.
-                    var markdownEls = lastBubble.querySelectorAll('.ds-markdown');
-                    for (var i = 0; i < markdownEls.length; i++) {
-                        if (markdownEls[i] !== thinkEl
-                            && !(thinkEl && thinkEl.contains(markdownEls[i]))
-                            && !markdownEls[i].classList.contains('ds-markdown--think')) {
-                            answerEl = markdownEls[i];
-                        }
-                    }
-                    // Thinking time from the toggle/header element
-                    var timeEl = lastBubble.querySelector('[class*="think"] ~ *')
-                              || lastBubble.querySelector('.ds-thinking-header');
-                    if (!timeEl) {
-                        // Fallback: parse from raw text header
-                        var m = last.match(/^(?:Thought for ([\\d.]+) seconds?|已思考（用时 ([\\d.]+) 秒）)/);
-                        if (m) thinkTime = m[1] || m[2];
-                    } else {
-                        var tm = (timeEl.textContent || '').match(/([\\d.]+)/);
-                        if (tm) thinkTime = tm[1];
-                    }
-                }
-
-                return {
-                    count: texts.length,
-                    last: last,
-                    // DOM-separated fields (null when not available)
-                    thinkText: thinkEl ? (thinkEl.innerText || '').trim() : null,
-                    answerText: answerEl ? (answerEl.innerText || '').trim() : null,
-                    thinkTime: thinkTime,
-                };
-            })()`);
-        } catch {
-            continue;
-        }
-
-        if (!result) continue;
-
-        const candidate = result.last;
-        if (candidate && result.count > baselineCount && candidate !== prompt.trim()) {
-            if (candidate === lastText) {
-                stableCount++;
-                if (stableCount >= 3) {
-                    if (parseThinking) {
-                        // Prefer DOM-level separation
-                        if (result.thinkText != null || result.answerText != null) {
-                            return {
-                                thinking: result.thinkText || '',
-                                response: result.answerText || '',
-                                thinking_time: result.thinkTime || null,
-                            };
-                        }
-                        // Fallback to text-header parsing (no \n\n split)
-                        return parseThinkingResponse(candidate);
-                    }
-                    return candidate;
-                }
-            } else {
-                stableCount = 0;
-            }
-            lastText = candidate;
-        }
-    }
-
-    if (parseThinking && lastText) {
-        return parseThinkingResponse(lastText);
-    }
-    return lastText || null;
-}
-
-export async function getVisibleMessages(page) {
-    const result = await page.evaluate(`(() => {
-        const msgs = document.querySelectorAll('${MESSAGE_SELECTOR}');
-        return Array.from(msgs).map(m => {
-            // User messages carry an extra hash-class alongside ds-message
-            const isUser = m.className.split(/\\s+/).length > 2;
-            return {
-                Role: isUser ? 'user' : 'assistant',
-                Text: (m.innerText || '').trim(),
-            };
-        }).filter(m => m.Text);
-    })()`);
-    return Array.isArray(result) ? result : [];
-}
-
-export async function getConversationList(page) {
-    await ensureOnDeepSeek(page);
-    // Expand sidebar if collapsed
-    await page.evaluate(`(() => {
-        if (document.querySelectorAll('a[href*="/a/chat/s/"]').length === 0) {
-            const btn = document.querySelector('div[tabindex="0"][role="button"]');
-            if (btn) btn.click();
-        }
-    })()`);
-    for (let attempt = 0; attempt < 5; attempt++) {
-        await page.wait(2);
-        const items = await page.evaluate(`(() => {
-            const items = [];
-            const links = document.querySelectorAll('a[href*="/a/chat/s/"]');
-            links.forEach((link, i) => {
-                const title = (link.innerText || '').trim().split('\\n')[0].trim();
-                const href = link.getAttribute('href') || '';
-                const idMatch = href.match(/\\/s\\/([a-f0-9-]+)/);
-                items.push({
-                    Index: i + 1,
-                    Id: idMatch ? idMatch[1] : href,
-                    Title: title || '(untitled)',
-                    Url: 'https://chat.deepseek.com' + href,
-                });
-            });
-            return items;
-        })()`);
-        if (Array.isArray(items) && items.length > 0) return items;
-    }
-    return [];
-}
-
-/**
- * Pick the URL of the most recent non-pinned conversation, or the first overall
- * if every visible conversation is pinned.
- *
- * Used by `ask` when the workspace was recycled and we need to resume an
- * existing thread instead of opening a new chat. Polls the sidebar for up to
- * 10s and returns null if no conversation links surface in time, so callers
- * can fail fast instead of silently navigating to a fresh page.
- *
- * Pinned detection is text-based on the section header ("置顶" / "Pinned"),
- * because DeepSeek's CSS-module class names are randomized per build.
- */
-export async function pickResumeUrl(page) {
-    await page.evaluate(`(() => {
-        if (document.querySelectorAll('a[href*="/a/chat/s/"]').length === 0) {
-            const btn = document.querySelector('div[tabindex="0"][role="button"]');
-            if (btn) btn.click();
-        }
-    })()`);
-    for (let attempt = 0; attempt < 5; attempt++) {
-        await page.wait(2);
-        const url = await page.evaluate(`(() => {
-            const links = document.querySelectorAll('a[href*="/a/chat/s/"]');
-            if (links.length === 0) return null;
-            const PINNED_HEADER = /^\\s*(置\\s*顶|Pinned)\\s*$/i;
-            const isPinned = (link) => {
-                const section = link.parentElement;
-                const header = section && section.firstElementChild;
-                if (!header || header === link) return false;
-                return PINNED_HEADER.test((header.innerText || header.textContent || '').trim());
-            };
-            const target = Array.from(links).find((l) => !isPinned(l)) || links[0];
-            const href = target.getAttribute('href') || '';
-            return href ? 'https://chat.deepseek.com' + href : null;
-        })()`);
-        if (url) return url;
-    }
-    return null;
-}
-
-async function waitForFilePreview(page, fileName) {
-    for (let attempt = 0; attempt < 8; attempt++) {
-        await page.wait(2);
-        const ready = await page.evaluate(`(() => {
-            var name = ${JSON.stringify(fileName)};
-            var hasFileName = Array.from(document.querySelectorAll('div'))
-                .some(function(el) { return el.children.length === 0 && (el.textContent || '').trim() === name; });
-            if (hasFileName) return true;
-            // Vision mode shows an image thumbnail, not filename text. Require
-            // a preview-like node here; send-button readiness is checked later.
-            var box = document.querySelector('${TEXTAREA_SELECTOR}');
-            if (!box) return false;
-            var c = box.parentElement;
-            while (c && !c.querySelector('div[role="button"]')) c = c.parentElement;
-            if (!c) return false;
-            return !!c.querySelector('img[src], canvas, video, [style*="background-image"], [class*="preview"], [class*="upload"]');
-        })()`);
-        if (ready) return true;
-    }
-    return false;
-}
-
-export async function sendWithFile(page, filePath, prompt) {
-    const fs = await import('node:fs');
-    const path = await import('node:path');
-    const absPath = path.default.resolve(filePath);
-
-    if (!fs.default.existsSync(absPath)) {
-        return { ok: false, reason: `File not found: ${absPath}` };
-    }
-
-    const stats = fs.default.statSync(absPath);
-    if (stats.size > 100 * 1024 * 1024) {
-        return { ok: false, reason: `File too large (${(stats.size / 1024 / 1024).toFixed(1)} MB). Max: 100 MB` };
-    }
-
-    const fileName = path.default.basename(absPath);
-
-    // Collapse sidebar to keep DOM simple for send button matching
-    await page.evaluate(`(() => {
-        if (document.querySelectorAll('a[href*="/a/chat/s/"]').length > 0) {
-            const btn = document.querySelector('div[tabindex="0"][role="button"]');
-            if (btn) btn.click();
-        }
-    })()`);
-    await page.wait(0.5);
-
-    let uploaded = false;
-    if (page.setFileInput) {
-        try {
-            await page.setFileInput([absPath], 'input[type="file"]');
-            uploaded = true;
-        } catch (err) {
-            const msg = String(err?.message || err);
-            if (!msg.includes('Unknown action') && !msg.includes('not supported') && !msg.includes('Not allowed')) {
-                throw err;
-            }
-        }
-    }
-
-    if (!uploaded) {
-        const content = fs.default.readFileSync(absPath);
-        const base64 = content.toString('base64');
-        const fallbackResult = await page.evaluate(`(async () => {
-            var binary = atob('${base64}');
-            var bytes = new Uint8Array(binary.length);
-            for (var i = 0; i < binary.length; i++) bytes[i] = binary.charCodeAt(i);
-
-            var file = new File([bytes], ${JSON.stringify(fileName)});
-            var dt = new DataTransfer();
-            dt.items.add(file);
-
-            var inp = document.querySelector('input[type="file"]');
-            if (!inp) return { ok: false, reason: 'file input not found' };
-
-            var propsKey = Object.keys(inp).find(function(k) { return k.startsWith('__reactProps$'); });
-            if (!propsKey || typeof inp[propsKey].onChange !== 'function') {
-                return { ok: false, reason: 'React onChange not found' };
-            }
-
-            inp.files = dt.files;
-            // Use inp.files, not dt.files; assignment transfers ownership
-            inp[propsKey].onChange({ target: { files: inp.files } });
-            return { ok: true };
-        })()`);
-        if (fallbackResult && !fallbackResult.ok) return fallbackResult;
-    }
-
-    const ready = await waitForFilePreview(page, fileName);
-    if (!ready) return { ok: false, reason: 'file preview did not appear' };
-
-    // File preview appears immediately but send button stays disabled until
-    // the server upload finishes. Wait for it.
-    let sendEnabled = false;
-    for (let tick = 0; tick < 15; tick++) {
-        const enabled = await page.evaluate(`(() => {
-            var box = document.querySelector('${TEXTAREA_SELECTOR}');
-            if (!box) return false;
-            var c = box.parentElement;
-            while (c && !c.querySelector('div[role="button"]')) c = c.parentElement;
-            if (!c) return false;
-            var btns = c.querySelectorAll('div[role="button"]:not(.ds-toggle-button)');
-            var last = btns[btns.length - 1];
-            return !!(last && last.getAttribute('aria-disabled') === 'false');
-        })()`);
-        if (enabled) {
-            sendEnabled = true;
-            break;
-        }
-        await page.wait(1);
-    }
-    if (!sendEnabled) {
-        return { ok: false, reason: 'send button did not enable after upload' };
-    }
-
-    return sendMessage(page, prompt);
-}
-
-// Retries on CDP "Promise was collected" errors caused by DeepSeek's SPA router transitions.
-export async function withRetry(fn, retries = 2) {
-    for (let i = 0; i <= retries; i++) {
-        try {
-            return await fn();
-        } catch (err) {
-            const msg = String(err?.message || err);
-            if (i < retries && msg.includes('Promise was collected')) {
-                await new Promise(r => setTimeout(r, 2000));
-                continue;
-            }
-            throw err;
-        }
-    }
-}
-
-export function parseBoolFlag(value) {
-    if (typeof value === 'boolean') return value;
-    return String(value ?? '').trim().toLowerCase() === 'true';
-}