ask-pro 0.1.0

package/dist/src/browser/config.js

@@ -0,0 +1,110 @@
+import { CHATGPT_URL, DEFAULT_MODEL_STRATEGY, DEFAULT_MODEL_TARGET } from "./constants.js";
+import { normalizeBrowserModelStrategy } from "./modelStrategy.js";
+import { defaultAskProBrowserProfileDir } from "./profilePaths.js";
+import { normalizeChatgptUrl } from "./utils.js";
+export const DEFAULT_BROWSER_CONFIG = {
+    chromeProfile: null,
+    chromePath: null,
+    chromeCookiePath: null,
+    attachRunning: false,
+    browserTabRef: null,
+    url: CHATGPT_URL,
+    chatgptUrl: CHATGPT_URL,
+    timeoutMs: 1_200_000,
+    debugPort: null,
+    inputTimeoutMs: 60_000,
+    assistantRecheckDelayMs: 0,
+    assistantRecheckTimeoutMs: 120_000,
+    profileLockTimeoutMs: 300_000,
+    autoReattachDelayMs: 0,
+    autoReattachIntervalMs: 0,
+    autoReattachTimeoutMs: 120_000,
+    cookieSync: true,
+    cookieNames: null,
+    cookieSyncWaitMs: 0,
+    inlineCookies: null,
+    inlineCookiesSource: null,
+    headless: false,
+    keepBrowser: false,
+    hideWindow: false,
+    startMinimized: false,
+    desiredModel: DEFAULT_MODEL_TARGET,
+    modelStrategy: DEFAULT_MODEL_STRATEGY,
+    debug: false,
+    allowCookieErrors: false,
+    remoteChrome: null,
+    remoteChromeBrowserWSEndpoint: null,
+    remoteChromeProfileRoot: null,
+    manualLogin: false,
+    manualLoginProfileDir: null,
+    manualLoginCookieSync: false,
+    manualLoginWaitMs: 20 * 60_000,
+    acceptLanguage: "en-US,en",
+};
+export function resolveBrowserConfig(config) {
+    const debugPortEnv = parseDebugPort(process.env.ASK_PRO_BROWSER_PORT ?? process.env.ASK_PRO_BROWSER_DEBUG_PORT);
+    const envAllowCookieErrors = (process.env.ASK_PRO_BROWSER_ALLOW_COOKIE_ERRORS ?? "").trim().toLowerCase() === "true" ||
+        (process.env.ASK_PRO_BROWSER_ALLOW_COOKIE_ERRORS ?? "").trim() === "1";
+    const rawUrl = config?.chatgptUrl ?? config?.url ?? DEFAULT_BROWSER_CONFIG.url;
+    const normalizedUrl = normalizeChatgptUrl(rawUrl ?? DEFAULT_BROWSER_CONFIG.url, DEFAULT_BROWSER_CONFIG.url);
+    const desiredModel = config?.desiredModel ?? DEFAULT_BROWSER_CONFIG.desiredModel ?? DEFAULT_MODEL_TARGET;
+    const modelStrategy = normalizeBrowserModelStrategy(config?.modelStrategy) ??
+        DEFAULT_BROWSER_CONFIG.modelStrategy ??
+        DEFAULT_MODEL_STRATEGY;
+    const isWindows = process.platform === "win32";
+    const manualLogin = config?.manualLogin ?? (isWindows ? true : DEFAULT_BROWSER_CONFIG.manualLogin);
+    const cookieSyncDefault = isWindows ? false : DEFAULT_BROWSER_CONFIG.cookieSync;
+    const resolvedProfileDir = config?.manualLoginProfileDir ??
+        process.env.ASK_PRO_BROWSER_PROFILE_DIR ??
+        defaultAskProBrowserProfileDir();
+    return {
+        ...DEFAULT_BROWSER_CONFIG,
+        ...config,
+        url: normalizedUrl,
+        chatgptUrl: normalizedUrl,
+        timeoutMs: config?.timeoutMs ?? DEFAULT_BROWSER_CONFIG.timeoutMs,
+        debugPort: config?.debugPort ?? debugPortEnv ?? DEFAULT_BROWSER_CONFIG.debugPort,
+        inputTimeoutMs: config?.inputTimeoutMs ?? DEFAULT_BROWSER_CONFIG.inputTimeoutMs,
+        assistantRecheckDelayMs: config?.assistantRecheckDelayMs ?? DEFAULT_BROWSER_CONFIG.assistantRecheckDelayMs,
+        assistantRecheckTimeoutMs: config?.assistantRecheckTimeoutMs ?? DEFAULT_BROWSER_CONFIG.assistantRecheckTimeoutMs,
+        profileLockTimeoutMs: config?.profileLockTimeoutMs ?? DEFAULT_BROWSER_CONFIG.profileLockTimeoutMs,
+        autoReattachDelayMs: config?.autoReattachDelayMs ?? DEFAULT_BROWSER_CONFIG.autoReattachDelayMs,
+        autoReattachIntervalMs: config?.autoReattachIntervalMs ?? DEFAULT_BROWSER_CONFIG.autoReattachIntervalMs,
+        autoReattachTimeoutMs: config?.autoReattachTimeoutMs ?? DEFAULT_BROWSER_CONFIG.autoReattachTimeoutMs,
+        cookieSync: config?.cookieSync ?? cookieSyncDefault,
+        cookieNames: config?.cookieNames ?? DEFAULT_BROWSER_CONFIG.cookieNames,
+        cookieSyncWaitMs: config?.cookieSyncWaitMs ?? DEFAULT_BROWSER_CONFIG.cookieSyncWaitMs,
+        inlineCookies: config?.inlineCookies ?? DEFAULT_BROWSER_CONFIG.inlineCookies,
+        inlineCookiesSource: config?.inlineCookiesSource ?? DEFAULT_BROWSER_CONFIG.inlineCookiesSource,
+        headless: config?.headless ?? DEFAULT_BROWSER_CONFIG.headless,
+        keepBrowser: config?.keepBrowser ?? DEFAULT_BROWSER_CONFIG.keepBrowser,
+        hideWindow: config?.hideWindow ?? DEFAULT_BROWSER_CONFIG.hideWindow,
+        startMinimized: config?.startMinimized ?? DEFAULT_BROWSER_CONFIG.startMinimized,
+        desiredModel,
+        modelStrategy,
+        chromeProfile: config?.chromeProfile ?? DEFAULT_BROWSER_CONFIG.chromeProfile,
+        chromePath: config?.chromePath ?? DEFAULT_BROWSER_CONFIG.chromePath,
+        chromeCookiePath: config?.chromeCookiePath ?? DEFAULT_BROWSER_CONFIG.chromeCookiePath,
+        attachRunning: config?.attachRunning ?? DEFAULT_BROWSER_CONFIG.attachRunning,
+        browserTabRef: config?.browserTabRef ?? DEFAULT_BROWSER_CONFIG.browserTabRef,
+        debug: config?.debug ?? DEFAULT_BROWSER_CONFIG.debug,
+        allowCookieErrors: config?.allowCookieErrors ?? envAllowCookieErrors ?? DEFAULT_BROWSER_CONFIG.allowCookieErrors,
+        remoteChromeBrowserWSEndpoint: config?.remoteChromeBrowserWSEndpoint ?? DEFAULT_BROWSER_CONFIG.remoteChromeBrowserWSEndpoint,
+        remoteChromeProfileRoot: config?.remoteChromeProfileRoot ?? DEFAULT_BROWSER_CONFIG.remoteChromeProfileRoot,
+        thinkingTime: config?.thinkingTime,
+        manualLogin,
+        manualLoginProfileDir: manualLogin ? resolvedProfileDir : null,
+        manualLoginCookieSync: config?.manualLoginCookieSync ?? DEFAULT_BROWSER_CONFIG.manualLoginCookieSync,
+        manualLoginWaitMs: config?.manualLoginWaitMs ?? DEFAULT_BROWSER_CONFIG.manualLoginWaitMs,
+        acceptLanguage: config?.acceptLanguage ?? DEFAULT_BROWSER_CONFIG.acceptLanguage,
+    };
+}
+function parseDebugPort(raw) {
+    if (!raw)
+        return null;
+    const value = Number.parseInt(raw, 10);
+    if (!Number.isFinite(value) || value <= 0 || value > 65535) {
+        return null;
+    }
+    return value;
+}

package/dist/src/browser/constants.js

@@ -0,0 +1,85 @@
+import { DEFAULT_CHATGPT_BROWSER_MODEL_LABEL } from "./chatgptModelCatalog.js";
+export const CHATGPT_URL = "https://chatgpt.com/";
+export const DEFAULT_MODEL_TARGET = DEFAULT_CHATGPT_BROWSER_MODEL_LABEL;
+export const DEFAULT_MODEL_STRATEGY = "select";
+export const COOKIE_URLS = [
+    "https://chatgpt.com",
+    "https://chat.openai.com",
+    "https://atlas.openai.com",
+];
+export const INPUT_SELECTORS = [
+    'textarea[data-id="prompt-textarea"]',
+    'textarea[placeholder*="Send a message"]',
+    'textarea[aria-label="Chat with ChatGPT"]',
+    'textarea[aria-label="Message ChatGPT"]',
+    "textarea:not([disabled])",
+    'textarea[name="prompt-textarea"]',
+    "#prompt-textarea",
+    ".ProseMirror",
+    '[contenteditable="true"][role="textbox"]',
+    '[contenteditable="true"][data-virtualkeyboard="true"]',
+];
+export const ANSWER_SELECTORS = [
+    'article[data-testid^="conversation-turn"][data-message-author-role="assistant"]',
+    'article[data-testid^="conversation-turn"][data-turn="assistant"]',
+    'article[data-testid^="conversation-turn"] [data-message-author-role="assistant"]',
+    'article[data-testid^="conversation-turn"] [data-turn="assistant"]',
+    'article[data-testid^="conversation-turn"] .markdown',
+    '[data-message-author-role="assistant"] .markdown',
+    '[data-turn="assistant"] .markdown',
+    '[data-message-author-role="assistant"]',
+    '[data-turn="assistant"]',
+];
+export const CONVERSATION_TURN_SELECTOR = 'article[data-testid^="conversation-turn"], div[data-testid^="conversation-turn"], section[data-testid^="conversation-turn"], ' +
+    "article[data-message-author-role], div[data-message-author-role], section[data-message-author-role], " +
+    "article[data-turn], div[data-turn], section[data-turn]";
+export const ASSISTANT_ROLE_SELECTOR = '[data-message-author-role="assistant"], [data-turn="assistant"]';
+export const CLOUDFLARE_SCRIPT_SELECTOR = 'script[src*="/challenge-platform/"]';
+export const CLOUDFLARE_TITLE = "just a moment";
+export const PROMPT_PRIMARY_SELECTOR = "#prompt-textarea";
+export const PROMPT_FALLBACK_SELECTOR = 'textarea[name="prompt-textarea"]';
+export const FILE_INPUT_SELECTORS = [
+    'form input[type="file"]:not([accept])',
+    'input[type="file"][multiple]:not([accept])',
+    'input[type="file"][multiple]',
+    'input[type="file"]:not([accept])',
+    'form input[type="file"][accept]',
+    'input[type="file"][accept]',
+    'input[type="file"]',
+    'input[type="file"][data-testid*="file"]',
+];
+// Legacy single selectors kept for compatibility with older call-sites
+export const FILE_INPUT_SELECTOR = FILE_INPUT_SELECTORS[0];
+export const GENERIC_FILE_INPUT_SELECTOR = FILE_INPUT_SELECTORS[3];
+export const MENU_CONTAINER_SELECTOR = '[role="menu"], [data-radix-collection-root]';
+export const MENU_ITEM_SELECTOR = 'button, [role="menuitem"], [role="menuitemradio"], [data-testid*="model-switcher-"]';
+export const UPLOAD_STATUS_SELECTORS = [
+    '[data-testid*="upload"]',
+    '[data-testid*="attachment"]',
+    '[data-testid*="progress"]',
+    '[data-state="loading"]',
+    '[data-state="uploading"]',
+    '[data-state="pending"]',
+    '[aria-live="polite"]',
+    '[aria-live="assertive"]',
+];
+export const STOP_BUTTON_SELECTOR = '[data-testid="stop-button"]';
+export const SEND_BUTTON_SELECTORS = [
+    'button[data-testid="send-button"]',
+    'button[data-testid*="composer-send"]',
+    'form button[type="submit"]',
+    'button[type="submit"][data-testid*="send"]',
+    'button[aria-label*="Send"]',
+];
+export const SEND_BUTTON_SELECTOR = SEND_BUTTON_SELECTORS[0];
+export const MODEL_BUTTON_SELECTOR = '[data-testid="model-switcher-dropdown-button"], button.__composer-pill[aria-haspopup="menu"], button.__composer-pill';
+export const COPY_BUTTON_SELECTORS = [
+    'button[data-testid="copy-turn-action-button"]',
+    'button[aria-label="Copy response"]',
+    'button[aria-label="Response copied"]',
+    'button[aria-label="Copy message"]',
+    'button[aria-label="Message copied"]',
+];
+export const COPY_BUTTON_SELECTOR = COPY_BUTTON_SELECTORS.join(", ");
+// Action buttons that only appear once a turn has finished rendering.
+export const FINISHED_ACTIONS_SELECTOR = `${COPY_BUTTON_SELECTOR}, button[data-testid="good-response-turn-action-button"], button[data-testid="bad-response-turn-action-button"], button[aria-label="Share"]`;

package/dist/src/browser/cookies.js

@@ -0,0 +1,191 @@
+import { COOKIE_URLS } from "./constants.js";
+import { delay } from "./utils.js";
+import { getCookies } from "@steipete/sweet-cookie";
+export class ChromeCookieSyncError extends Error {
+}
+export async function syncCookies(Network, url, profile, logger, options = {}) {
+    const { allowErrors = false, filterNames, inlineCookies, cookiePath, waitMs = 0 } = options;
+    try {
+        // Learned: inline cookies are the most deterministic (avoid Keychain + profile ambiguity).
+        const cookies = inlineCookies?.length
+            ? normalizeInlineCookies(inlineCookies, new URL(url).hostname)
+            : await readChromeCookiesWithWait(url, profile, filterNames ?? undefined, cookiePath ?? undefined, waitMs, logger);
+        if (!cookies.length) {
+            return 0;
+        }
+        let applied = 0;
+        for (const cookie of cookies) {
+            const cookieWithUrl = attachUrl(cookie, url);
+            try {
+                // Learned: CDP will silently drop cookies without a url; always attach one.
+                const result = await Network.setCookie(cookieWithUrl);
+                if (result?.success) {
+                    applied += 1;
+                }
+            }
+            catch (error) {
+                const message = error instanceof Error ? error.message : String(error);
+                logger(`Failed to set cookie ${cookie.name}: ${message}`);
+            }
+        }
+        return applied;
+    }
+    catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        if (allowErrors) {
+            logger(`Cookie sync failed (continuing with override): ${message}`);
+            return 0;
+        }
+        throw error instanceof ChromeCookieSyncError ? error : new ChromeCookieSyncError(message);
+    }
+}
+async function readChromeCookiesWithWait(url, profile, filterNames, cookiePath, waitMs, logger) {
+    if (waitMs <= 0) {
+        return readChromeCookies(url, profile, filterNames, cookiePath);
+    }
+    let cookies = [];
+    let firstError;
+    try {
+        cookies = await readChromeCookies(url, profile, filterNames, cookiePath);
+    }
+    catch (error) {
+        firstError = error;
+    }
+    if (cookies.length > 0 && !firstError) {
+        return cookies;
+    }
+    const waitLabel = waitMs >= 1000 ? `${Math.round(waitMs / 1000)}s` : `${waitMs}ms`;
+    const message = firstError instanceof Error ? firstError.message : String(firstError ?? "");
+    if (firstError) {
+        logger(`[cookies] Cookie read failed (${message}); waiting ${waitLabel} then retrying once.`);
+    }
+    else {
+        logger(`[cookies] No cookies found; waiting ${waitLabel} then retrying once.`);
+    }
+    await delay(waitMs);
+    return readChromeCookies(url, profile, filterNames, cookiePath);
+}
+async function readChromeCookies(url, profile, filterNames, cookiePath) {
+    const origins = Array.from(new Set([stripQuery(url), ...COOKIE_URLS]));
+    const chromeProfile = cookiePath ?? profile ?? undefined;
+    const timeoutMs = readDuration("ASK_PRO_COOKIE_LOAD_TIMEOUT_MS", 5_000);
+    // Learned: read from multiple origins to capture auth cookies that land on chat.openai.com + atlas.
+    const { cookies, warnings } = await getCookies({
+        url,
+        origins,
+        names: filterNames?.length ? filterNames : undefined,
+        browsers: ["chrome"],
+        mode: "merge",
+        chromeProfile,
+        timeoutMs,
+    });
+    if (process.env.ASK_PRO_DEBUG_COOKIES === "1" && warnings.length) {
+        // eslint-disable-next-line no-console
+        console.error(`[cookies] sweet-cookie warnings:\n- ${warnings.join("\n- ")}`);
+    }
+    const merged = new Map();
+    for (const cookie of cookies) {
+        const normalized = toCdpCookie(cookie);
+        if (!normalized)
+            continue;
+        const key = `${normalized.domain ?? ""}:${normalized.name}`;
+        if (!merged.has(key))
+            merged.set(key, normalized);
+    }
+    return Array.from(merged.values());
+}
+function normalizeInlineCookies(rawCookies, fallbackHost) {
+    const merged = new Map();
+    for (const cookie of rawCookies) {
+        if (!cookie?.name)
+            continue;
+        // Learned: inline cookies may omit url/domain; default to current host with a safe path.
+        const normalized = {
+            name: cookie.name,
+            value: cookie.value ?? "",
+            url: cookie.url,
+            domain: cookie.domain ?? fallbackHost,
+            path: cookie.path ?? "/",
+            expires: normalizeExpiration(cookie.expires),
+            secure: cookie.secure ?? true,
+            httpOnly: cookie.httpOnly ?? false,
+            sameSite: cookie.sameSite,
+        };
+        const key = `${normalized.domain ?? fallbackHost}:${normalized.name}`;
+        if (!merged.has(key)) {
+            merged.set(key, normalized);
+        }
+    }
+    return Array.from(merged.values());
+}
+function toCdpCookie(cookie) {
+    if (!cookie?.name)
+        return null;
+    const out = {
+        name: cookie.name,
+        value: cookie.value,
+        domain: cookie.domain,
+        path: cookie.path ?? "/",
+        secure: cookie.secure ?? true,
+        httpOnly: cookie.httpOnly ?? false,
+    };
+    if (typeof cookie.expires === "number")
+        out.expires = cookie.expires;
+    if (cookie.sameSite === "Lax" || cookie.sameSite === "Strict" || cookie.sameSite === "None") {
+        out.sameSite = cookie.sameSite;
+    }
+    return out;
+}
+function attachUrl(cookie, fallbackUrl) {
+    const cookieWithUrl = { ...cookie };
+    if (!cookieWithUrl.url) {
+        if (!cookieWithUrl.domain || cookieWithUrl.domain === "localhost") {
+            cookieWithUrl.url = fallbackUrl;
+        }
+        else if (!cookieWithUrl.domain.startsWith(".")) {
+            cookieWithUrl.url = `https://${cookieWithUrl.domain}`;
+        }
+    }
+    // When url is present, let Chrome derive the host from it; keeping domain can trigger CDP sanitization errors.
+    if (cookieWithUrl.url) {
+        // Learned: CDP rejects cookies with both url + domain in some cases; drop domain to avoid failures.
+        delete cookieWithUrl.domain;
+    }
+    return cookieWithUrl;
+}
+function stripQuery(url) {
+    try {
+        const parsed = new URL(url);
+        parsed.hash = "";
+        parsed.search = "";
+        return parsed.toString();
+    }
+    catch {
+        return url;
+    }
+}
+function normalizeExpiration(expires) {
+    if (!expires || Number.isNaN(expires)) {
+        return undefined;
+    }
+    const value = Number(expires);
+    if (value <= 0) {
+        return undefined;
+    }
+    if (value > 1_000_000_000_000) {
+        // Learned: Chrome may store WebKit microseconds since 1601; convert to Unix seconds.
+        return Math.round(value / 1_000_000 - 11644473600);
+    }
+    if (value > 1_000_000_000) {
+        // Likely milliseconds; normalize to seconds for CDP.
+        return Math.round(value / 1000);
+    }
+    return Math.round(value);
+}
+function readDuration(envKey, defaultValueMs) {
+    const raw = process.env[envKey];
+    if (!raw)
+        return defaultValueMs;
+    const value = Number.parseInt(raw, 10);
+    return Number.isFinite(value) && value > 0 ? value : defaultValueMs;
+}