@agenticmail/enterprise 0.5.326 → 0.5.328

package/src/browser/server-context.ts

@@ -1,688 +0,0 @@
-import fs from "node:fs";
-
-import { fetchJson, fetchOk } from "./cdp.helpers.js";
-import { appendCdpPath, createTargetViaCdp, normalizeCdpWsUrl } from "./cdp.js";
-import {
-  isChromeCdpReady,
-  isChromeReachable,
-  launchAgenticMailChrome,
-  resolveAgenticMailUserDataDir,
-  stopAgenticMailChrome,
-} from "./chrome.js";
-import type { ResolvedBrowserProfile } from "./config.js";
-import { resolveProfile } from "./config.js";
-import {
-  ensureChromeExtensionRelayServer,
-  stopChromeExtensionRelayServer,
-} from "./extension-relay.js";
-import {
-  assertBrowserNavigationAllowed,
-  InvalidBrowserNavigationUrlError,
-  withBrowserNavigationPolicy,
-} from "./navigation-guard.js";
-import type { PwAiModule } from "./pw-ai-module.js";
-import { getPwAiModule } from "./pw-ai-module.js";
-import {
-  refreshResolvedBrowserConfigFromDisk,
-  resolveBrowserProfileWithHotReload,
-} from "./resolved-config-refresh.js";
-import type {
-  BrowserServerState,
-  BrowserRouteContext,
-  BrowserTab,
-  ContextOptions,
-  ProfileContext,
-  ProfileRuntimeState,
-  ProfileStatus,
-} from "./server-context.types.js";
-import { resolveTargetIdFromTabs } from "./target-id.js";
-import { movePathToTrash } from "./trash.js";
-import { SsrFBlockedError } from "./enterprise-compat.js";
-
-export type {
-  BrowserRouteContext,
-  BrowserServerState,
-  BrowserTab,
-  ProfileContext,
-  ProfileRuntimeState,
-  ProfileStatus,
-} from "./server-context.types.js";
-
-export function listKnownProfileNames(state: BrowserServerState): string[] {
-  const names = new Set(Object.keys(state.resolved.profiles));
-  for (const name of state.profiles.keys()) {
-    names.add(name);
-  }
-  return [...names];
-}
-
-/**
- * Normalize a CDP WebSocket URL to use the correct base URL.
- */
-function normalizeWsUrl(raw: string | undefined, cdpBaseUrl: string): string | undefined {
-  if (!raw) {
-    return undefined;
-  }
-  try {
-    return normalizeCdpWsUrl(raw, cdpBaseUrl);
-  } catch {
-    return raw;
-  }
-}
-
-/**
- * Create a profile-scoped context for browser operations.
- */
-function createProfileContext(
-  opts: ContextOptions,
-  profile: ResolvedBrowserProfile,
-): ProfileContext {
-  const state = () => {
-    const current = opts.getState();
-    if (!current) {
-      throw new Error("Browser server not started");
-    }
-    return current;
-  };
-
-  const getProfileState = (): ProfileRuntimeState => {
-    const current = state();
-    let profileState = current.profiles.get(profile.name);
-    if (!profileState) {
-      profileState = { profile, running: null, lastTargetId: null };
-      current.profiles.set(profile.name, profileState);
-    }
-    return profileState;
-  };
-
-  const setProfileRunning = (running: ProfileRuntimeState["running"]) => {
-    const profileState = getProfileState();
-    profileState.running = running;
-  };
-
-  const listTabs = async (): Promise<BrowserTab[]> => {
-    // For remote profiles, use Playwright's persistent connection to avoid ephemeral sessions
-    if (!profile.cdpIsLoopback) {
-      const mod = await getPwAiModule({ mode: "strict" });
-      const listPagesViaPlaywright = (mod as Partial<PwAiModule> | null)?.listPagesViaPlaywright;
-      if (typeof listPagesViaPlaywright === "function") {
-        const pages = await listPagesViaPlaywright({ cdpUrl: profile.cdpUrl });
-        return pages.map((p) => ({
-          targetId: p.targetId,
-          title: p.title,
-          url: p.url,
-          type: p.type,
-        }));
-      }
-    }
-
-    const raw = await fetchJson<
-      Array<{
-        id?: string;
-        title?: string;
-        url?: string;
-        webSocketDebuggerUrl?: string;
-        type?: string;
-      }>
-    >(appendCdpPath(profile.cdpUrl, "/json/list"));
-    return raw
-      .map((t) => ({
-        targetId: t.id ?? "",
-        title: t.title ?? "",
-        url: t.url ?? "",
-        wsUrl: normalizeWsUrl(t.webSocketDebuggerUrl, profile.cdpUrl),
-        type: t.type,
-      }))
-      .filter((t) => Boolean(t.targetId));
-  };
-
-  const openTab = async (url: string): Promise<BrowserTab> => {
-    const ssrfPolicyOpts = withBrowserNavigationPolicy(state().resolved.ssrfPolicy);
-    await assertBrowserNavigationAllowed({ url, ...ssrfPolicyOpts });
-
-    // For remote profiles, use Playwright's persistent connection to create tabs
-    // This ensures the tab persists beyond a single request
-    if (!profile.cdpIsLoopback) {
-      const mod = await getPwAiModule({ mode: "strict" });
-      const createPageViaPlaywright = (mod as Partial<PwAiModule> | null)?.createPageViaPlaywright;
-      if (typeof createPageViaPlaywright === "function") {
-        const page = await createPageViaPlaywright({
-          cdpUrl: profile.cdpUrl,
-          url,
-          ...ssrfPolicyOpts,
-          navigationChecked: true,
-        });
-        const profileState = getProfileState();
-        profileState.lastTargetId = page.targetId;
-        return {
-          targetId: page.targetId,
-          title: page.title,
-          url: page.url,
-          type: page.type,
-        };
-      }
-    }
-
-    const createdViaCdp = await createTargetViaCdp({
-      cdpUrl: profile.cdpUrl,
-      url,
-      ...ssrfPolicyOpts,
-      navigationChecked: true,
-    })
-      .then((r) => r.targetId)
-      .catch(() => null);
-
-    if (createdViaCdp) {
-      const profileState = getProfileState();
-      profileState.lastTargetId = createdViaCdp;
-      const deadline = Date.now() + 2000;
-      while (Date.now() < deadline) {
-        const tabs = await listTabs().catch(() => [] as BrowserTab[]);
-        const found = tabs.find((t) => t.targetId === createdViaCdp);
-        if (found) {
-          return found;
-        }
-        await new Promise((r) => setTimeout(r, 100));
-      }
-      return { targetId: createdViaCdp, title: "", url, type: "page" };
-    }
-
-    const encoded = encodeURIComponent(url);
-    type CdpTarget = {
-      id?: string;
-      title?: string;
-      url?: string;
-      webSocketDebuggerUrl?: string;
-      type?: string;
-    };
-
-    const endpointUrl = new URL(appendCdpPath(profile.cdpUrl, "/json/new"));
-    const endpoint = endpointUrl.search
-      ? (() => {
-          endpointUrl.searchParams.set("url", url);
-          return endpointUrl.toString();
-        })()
-      : `${endpointUrl.toString()}?${encoded}`;
-    const created = await fetchJson<CdpTarget>(endpoint, 1500, {
-      method: "PUT",
-    }).catch(async (err) => {
-      if (String(err).includes("HTTP 405")) {
-        return await fetchJson<CdpTarget>(endpoint, 1500);
-      }
-      throw err;
-    });
-
-    if (!created.id) {
-      throw new Error("Failed to open tab (missing id)");
-    }
-    const profileState = getProfileState();
-    profileState.lastTargetId = created.id;
-    return {
-      targetId: created.id,
-      title: created.title ?? "",
-      url: created.url ?? url,
-      wsUrl: normalizeWsUrl(created.webSocketDebuggerUrl, profile.cdpUrl),
-      type: created.type,
-    };
-  };
-
-  const resolveRemoteHttpTimeout = (timeoutMs: number | undefined) => {
-    if (profile.cdpIsLoopback) {
-      return timeoutMs ?? 300;
-    }
-    const resolved = state().resolved;
-    if (typeof timeoutMs === "number" && Number.isFinite(timeoutMs)) {
-      return Math.max(Math.floor(timeoutMs), resolved.remoteCdpTimeoutMs);
-    }
-    return resolved.remoteCdpTimeoutMs;
-  };
-
-  const resolveRemoteWsTimeout = (timeoutMs: number | undefined) => {
-    if (profile.cdpIsLoopback) {
-      const base = timeoutMs ?? 300;
-      return Math.max(200, Math.min(2000, base * 2));
-    }
-    const resolved = state().resolved;
-    if (typeof timeoutMs === "number" && Number.isFinite(timeoutMs)) {
-      return Math.max(Math.floor(timeoutMs) * 2, resolved.remoteCdpHandshakeTimeoutMs);
-    }
-    return resolved.remoteCdpHandshakeTimeoutMs;
-  };
-
-  const isReachable = async (timeoutMs?: number) => {
-    const httpTimeout = resolveRemoteHttpTimeout(timeoutMs);
-    const wsTimeout = resolveRemoteWsTimeout(timeoutMs);
-    return await isChromeCdpReady(profile.cdpUrl, httpTimeout, wsTimeout);
-  };
-
-  const isHttpReachable = async (timeoutMs?: number) => {
-    const httpTimeout = resolveRemoteHttpTimeout(timeoutMs);
-    return await isChromeReachable(profile.cdpUrl, httpTimeout);
-  };
-
-  const attachRunning = (running: NonNullable<ProfileRuntimeState["running"]>) => {
-    setProfileRunning(running);
-    running.proc.on("exit", () => {
-      // Guard against server teardown (e.g., SIGUSR1 restart)
-      if (!opts.getState()) {
-        return;
-      }
-      const profileState = getProfileState();
-      if (profileState.running?.pid === running.pid) {
-        setProfileRunning(null);
-      }
-    });
-  };
-
-  const ensureBrowserAvailable = async (): Promise<void> => {
-    const current = state();
-    const remoteCdp = !profile.cdpIsLoopback;
-    const isExtension = profile.driver === "extension";
-    const profileState = getProfileState();
-    const httpReachable = await isHttpReachable();
-
-    if (isExtension && remoteCdp) {
-      throw new Error(
-        `Profile "${profile.name}" uses driver=extension but cdpUrl is not loopback (${profile.cdpUrl}).`,
-      );
-    }
-
-    if (isExtension) {
-      if (!httpReachable) {
-        await ensureChromeExtensionRelayServer({ cdpUrl: profile.cdpUrl });
-        if (await isHttpReachable(1200)) {
-          // continue: we still need the extension to connect for CDP websocket.
-        } else {
-          throw new Error(
-            `Chrome extension relay for profile "${profile.name}" is not reachable at ${profile.cdpUrl}.`,
-          );
-        }
-      }
-
-      if (await isReachable(600)) {
-        return;
-      }
-      // Relay server is up, but no attached tab yet. Prompt user to attach.
-      throw new Error(
-        `Chrome extension relay is running, but no tab is connected. Click the AgenticMail Chrome extension icon on a tab to attach it (profile "${profile.name}").`,
-      );
-    }
-
-    if (!httpReachable) {
-      if ((current.resolved.attachOnly || remoteCdp) && opts.onEnsureAttachTarget) {
-        await opts.onEnsureAttachTarget(profile);
-        if (await isHttpReachable(1200)) {
-          return;
-        }
-      }
-      if (current.resolved.attachOnly || remoteCdp) {
-        throw new Error(
-          remoteCdp
-            ? `Remote CDP for profile "${profile.name}" is not reachable at ${profile.cdpUrl}.`
-            : `Browser attachOnly is enabled and profile "${profile.name}" is not running.`,
-        );
-      }
-      const launched = await launchAgenticMailChrome(current.resolved, profile);
-      attachRunning(launched);
-      return;
-    }
-
-    // Port is reachable - check if we own it
-    if (await isReachable()) {
-      return;
-    }
-
-    // HTTP responds but WebSocket fails - port in use by something else
-    if (!profileState.running) {
-      throw new Error(
-        `Port ${profile.cdpPort} is in use for profile "${profile.name}" but not by agenticmail. ` +
-          `Run action=reset-profile profile=${profile.name} to kill the process.`,
-      );
-    }
-
-    // We own it but WebSocket failed - restart
-    if (current.resolved.attachOnly || remoteCdp) {
-      if (opts.onEnsureAttachTarget) {
-        await opts.onEnsureAttachTarget(profile);
-        if (await isReachable(1200)) {
-          return;
-        }
-      }
-      throw new Error(
-        remoteCdp
-          ? `Remote CDP websocket for profile "${profile.name}" is not reachable.`
-          : `Browser attachOnly is enabled and CDP websocket for profile "${profile.name}" is not reachable.`,
-      );
-    }
-
-    await stopAgenticMailChrome(profileState.running);
-    setProfileRunning(null);
-
-    const relaunched = await launchAgenticMailChrome(current.resolved, profile);
-    attachRunning(relaunched);
-
-    if (!(await isReachable(600))) {
-      throw new Error(
-        `Chrome CDP websocket for profile "${profile.name}" is not reachable after restart.`,
-      );
-    }
-  };
-
-  const ensureTabAvailable = async (targetId?: string): Promise<BrowserTab> => {
-    await ensureBrowserAvailable();
-    const profileState = getProfileState();
-    const tabs1 = await listTabs();
-    if (tabs1.length === 0) {
-      if (profile.driver === "extension") {
-        throw new Error(
-          `tab not found (no attached Chrome tabs for profile "${profile.name}"). ` +
-            "Click the AgenticMail Browser Relay toolbar icon on the tab you want to control (badge ON).",
-        );
-      }
-      await openTab("about:blank");
-    }
-
-    const tabs = await listTabs();
-    // For remote profiles using Playwright's persistent connection, we don't need wsUrl
-    // because we access pages directly through Playwright, not via individual WebSocket URLs.
-    const candidates =
-      profile.driver === "extension" || !profile.cdpIsLoopback
-        ? tabs
-        : tabs.filter((t) => Boolean(t.wsUrl));
-
-    const resolveById = (raw: string) => {
-      const resolved = resolveTargetIdFromTabs(raw, candidates);
-      if (!resolved.ok) {
-        if (resolved.reason === "ambiguous") {
-          return "AMBIGUOUS" as const;
-        }
-        return null;
-      }
-      return candidates.find((t) => t.targetId === resolved.targetId) ?? null;
-    };
-
-    const pickDefault = () => {
-      const last = profileState.lastTargetId?.trim() || "";
-      const lastResolved = last ? resolveById(last) : null;
-      if (lastResolved && lastResolved !== "AMBIGUOUS") {
-        return lastResolved;
-      }
-      // Prefer a real page tab first (avoid service workers/background targets).
-      const page = candidates.find((t) => (t.type ?? "page") === "page");
-      return page ?? candidates.at(0) ?? null;
-    };
-
-    let chosen = targetId ? resolveById(targetId) : pickDefault();
-    if (!chosen && profile.driver === "extension" && candidates.length === 1) {
-      // If an agent passes a stale/foreign targetId but we only have a single attached tab,
-      // recover by using that tab instead of failing hard.
-      chosen = candidates[0] ?? null;
-    }
-
-    if (chosen === "AMBIGUOUS") {
-      throw new Error("ambiguous target id prefix");
-    }
-    if (!chosen) {
-      throw new Error("tab not found");
-    }
-    profileState.lastTargetId = chosen.targetId;
-    return chosen;
-  };
-
-  const focusTab = async (targetId: string): Promise<void> => {
-    const tabs = await listTabs();
-    const resolved = resolveTargetIdFromTabs(targetId, tabs);
-    if (!resolved.ok) {
-      if (resolved.reason === "ambiguous") {
-        throw new Error("ambiguous target id prefix");
-      }
-      throw new Error("tab not found");
-    }
-
-    if (!profile.cdpIsLoopback) {
-      const mod = await getPwAiModule({ mode: "strict" });
-      const focusPageByTargetIdViaPlaywright = (mod as Partial<PwAiModule> | null)
-        ?.focusPageByTargetIdViaPlaywright;
-      if (typeof focusPageByTargetIdViaPlaywright === "function") {
-        await focusPageByTargetIdViaPlaywright({
-          cdpUrl: profile.cdpUrl,
-          targetId: resolved.targetId,
-        });
-        const profileState = getProfileState();
-        profileState.lastTargetId = resolved.targetId;
-        return;
-      }
-    }
-
-    await fetchOk(appendCdpPath(profile.cdpUrl, `/json/activate/${resolved.targetId}`));
-    const profileState = getProfileState();
-    profileState.lastTargetId = resolved.targetId;
-  };
-
-  const closeTab = async (targetId: string): Promise<void> => {
-    const tabs = await listTabs();
-    const resolved = resolveTargetIdFromTabs(targetId, tabs);
-    if (!resolved.ok) {
-      if (resolved.reason === "ambiguous") {
-        throw new Error("ambiguous target id prefix");
-      }
-      throw new Error("tab not found");
-    }
-
-    // For remote profiles, use Playwright's persistent connection to close tabs
-    if (!profile.cdpIsLoopback) {
-      const mod = await getPwAiModule({ mode: "strict" });
-      const closePageByTargetIdViaPlaywright = (mod as Partial<PwAiModule> | null)
-        ?.closePageByTargetIdViaPlaywright;
-      if (typeof closePageByTargetIdViaPlaywright === "function") {
-        await closePageByTargetIdViaPlaywright({
-          cdpUrl: profile.cdpUrl,
-          targetId: resolved.targetId,
-        });
-        return;
-      }
-    }
-
-    await fetchOk(appendCdpPath(profile.cdpUrl, `/json/close/${resolved.targetId}`));
-  };
-
-  const stopRunningBrowser = async (): Promise<{ stopped: boolean }> => {
-    if (profile.driver === "extension") {
-      const stopped = await stopChromeExtensionRelayServer({
-        cdpUrl: profile.cdpUrl,
-      });
-      return { stopped };
-    }
-    const profileState = getProfileState();
-    if (!profileState.running) {
-      return { stopped: false };
-    }
-    await stopAgenticMailChrome(profileState.running);
-    setProfileRunning(null);
-    return { stopped: true };
-  };
-
-  const resetProfile = async () => {
-    if (profile.driver === "extension") {
-      await stopChromeExtensionRelayServer({ cdpUrl: profile.cdpUrl }).catch(() => {});
-      return { moved: false, from: profile.cdpUrl };
-    }
-    if (!profile.cdpIsLoopback) {
-      throw new Error(
-        `reset-profile is only supported for local profiles (profile "${profile.name}" is remote).`,
-      );
-    }
-    const userDataDir = resolveAgenticMailUserDataDir(profile.name);
-    const profileState = getProfileState();
-
-    const httpReachable = await isHttpReachable(300);
-    if (httpReachable && !profileState.running) {
-      // Port in use but not by us - kill it
-      try {
-        const mod = await import("./pw-ai.js");
-        await mod.closePlaywrightBrowserConnection();
-      } catch {
-        // ignore
-      }
-    }
-
-    if (profileState.running) {
-      await stopRunningBrowser();
-    }
-
-    try {
-      const mod = await import("./pw-ai.js");
-      await mod.closePlaywrightBrowserConnection();
-    } catch {
-      // ignore
-    }
-
-    if (!fs.existsSync(userDataDir)) {
-      return { moved: false, from: userDataDir };
-    }
-
-    const moved = await movePathToTrash(userDataDir);
-    return { moved: true, from: userDataDir, to: moved };
-  };
-
-  return {
-    profile,
-    ensureBrowserAvailable,
-    ensureTabAvailable,
-    isHttpReachable,
-    isReachable,
-    listTabs,
-    openTab,
-    focusTab,
-    closeTab,
-    stopRunningBrowser,
-    resetProfile,
-  };
-}
-
-export function createBrowserRouteContext(opts: ContextOptions): BrowserRouteContext {
-  const refreshConfigFromDisk = opts.refreshConfigFromDisk === true;
-
-  const state = () => {
-    const current = opts.getState();
-    if (!current) {
-      throw new Error("Browser server not started");
-    }
-    return current;
-  };
-
-  const forProfile = (profileName?: string): ProfileContext => {
-    const current = state();
-    const name = profileName ?? current.resolved.defaultProfile;
-    const profile = resolveBrowserProfileWithHotReload({
-      current,
-      refreshConfigFromDisk,
-      name,
-    });
-
-    if (!profile) {
-      const available = Object.keys(current.resolved.profiles).join(", ");
-      throw new Error(`Profile "${name}" not found. Available profiles: ${available || "(none)"}`);
-    }
-    return createProfileContext(opts, profile);
-  };
-
-  const listProfiles = async (): Promise<ProfileStatus[]> => {
-    const current = state();
-    refreshResolvedBrowserConfigFromDisk({
-      current,
-      refreshConfigFromDisk,
-      mode: "cached",
-    });
-    const result: ProfileStatus[] = [];
-
-    for (const name of Object.keys(current.resolved.profiles)) {
-      const profileState = current.profiles.get(name);
-      const profile = resolveProfile(current.resolved, name);
-      if (!profile) {
-        continue;
-      }
-
-      let tabCount = 0;
-      let running = false;
-
-      if (profileState?.running) {
-        running = true;
-        try {
-          const ctx = createProfileContext(opts, profile);
-          const tabs = await ctx.listTabs();
-          tabCount = tabs.filter((t) => t.type === "page").length;
-        } catch {
-          // Browser might not be responsive
-        }
-      } else {
-        // Check if something is listening on the port
-        try {
-          const reachable = await isChromeReachable(profile.cdpUrl, 200);
-          if (reachable) {
-            running = true;
-            const ctx = createProfileContext(opts, profile);
-            const tabs = await ctx.listTabs().catch(() => []);
-            tabCount = tabs.filter((t) => t.type === "page").length;
-          }
-        } catch {
-          // Not reachable
-        }
-      }
-
-      result.push({
-        name,
-        cdpPort: profile.cdpPort,
-        cdpUrl: profile.cdpUrl,
-        color: profile.color,
-        running,
-        tabCount,
-        isDefault: name === current.resolved.defaultProfile,
-        isRemote: !profile.cdpIsLoopback,
-      });
-    }
-
-    return result;
-  };
-
-  // Create default profile context for backward compatibility
-  const getDefaultContext = () => forProfile();
-
-  const mapTabError = (err: unknown) => {
-    if (err instanceof SsrFBlockedError) {
-      return { status: 400, message: err.message };
-    }
-    if (err instanceof InvalidBrowserNavigationUrlError) {
-      return { status: 400, message: err.message };
-    }
-    const msg = String(err);
-    if (msg.includes("ambiguous target id prefix")) {
-      return { status: 409, message: "ambiguous target id prefix" };
-    }
-    if (msg.includes("tab not found")) {
-      return { status: 404, message: msg };
-    }
-    if (msg.includes("not found")) {
-      return { status: 404, message: msg };
-    }
-    return null;
-  };
-
-  return {
-    state,
-    forProfile,
-    listProfiles,
-    // Legacy methods delegate to default profile
-    ensureBrowserAvailable: () => getDefaultContext().ensureBrowserAvailable(),
-    ensureTabAvailable: (targetId) => getDefaultContext().ensureTabAvailable(targetId),
-    isHttpReachable: (timeoutMs) => getDefaultContext().isHttpReachable(timeoutMs),
-    isReachable: (timeoutMs) => getDefaultContext().isReachable(timeoutMs),
-    listTabs: () => getDefaultContext().listTabs(),
-    openTab: (url) => getDefaultContext().openTab(url),
-    focusTab: (targetId) => getDefaultContext().focusTab(targetId),
-    closeTab: (targetId) => getDefaultContext().closeTab(targetId),
-    stopRunningBrowser: () => getDefaultContext().stopRunningBrowser(),
-    resetProfile: () => getDefaultContext().resetProfile(),
-    mapTabError,
-  };
-}