"use client"; import { RegistryContext, useAtomSet, useAtomValue } from "@effect/atom-react";import { squashAtomCommandFailure } from "@t3tools/client-runtime/state/runtime";import {  FILL_PREVIEW_VIEWPORT,  PREVIEW_AUTOMATION_OPERATIONS,  type EnvironmentId,  type PreviewAutomationNavigateInput,  type PreviewAutomationOpenInput,  type PreviewAutomationResizeInput,  type PreviewAutomationResizeResult,  type PreviewAutomationSetColorSchemeInput,  type PreviewAutomationSetColorSchemeResult,  type PreviewAutomationHost as PreviewAutomationHostState,  type PreviewAutomationRequest,  type PreviewAutomationStatus,  type PreviewRenderedViewportSize,  type PreviewViewportSetting,  type ScopedThreadRef,} from "@t3tools/contracts";import { resolvePreviewViewport } from "@t3tools/shared/previewViewport";import { useCallback, useContext, useEffect, useMemo, useRef, useState } from "react";import { Atom } from "effect/unstable/reactivity"; import {  applyPreviewServerSnapshot,  readThreadPreviewState,  reconcilePreviewServerSessions,  updatePreviewServerSnapshot,} from "~/previewStateStore";import { selectThreadPreviewMiniPlayer, usePreviewMiniPlayerStore } from "~/previewMiniPlayerStore";import { resolveBrowserNavigationTarget } from "~/browser/browserTargetResolver";import {  readActiveBrowserRecordingTargets,  startBrowserRecording,  stopBrowserRecording,  stopBrowserRecordingForUpload,} from "~/browser/browserRecording";import { resolveBrowserRecordingStopTarget } from "~/browser/browserRecordingScope";import { uploadBrowserRecording } from "~/browser/browserRecordingUpload";import {  acquireBrowserSurfaceActivity,  useBrowserSurfaceStore,} from "~/browser/browserSurfaceStore";import {  browserDefaultOpenProfileId,  browserDefaultOpenViewport,  resolveBrowserDefaults,} from "~/browser/browserDefaults";import { runBrowserViewportMutation } from "~/browser/browserViewportActions";import { previewRuntimeTabId } from "~/browser/previewRuntimeTabId";import { isElectron } from "~/env";import { useEnvironments } from "~/state/environments";import { previewEnvironment } from "~/state/preview";import { useAtomQueryRunner } from "~/state/use-atom-query-runner";import { useAtomCommand } from "~/state/use-atom-command"; import { previewBridge } from "./previewBridge";import {  PreviewAutomationOperationError,  PreviewAutomationOverlayTimeoutError,  PreviewAutomationRecordingNotActiveError,  PreviewAutomationTargetUnavailableError,  PreviewAutomationViewportTimeoutError,} from "./previewAutomationErrors";import {  explicitlySuppressesPreviewMiniPlayer,  previewAutomationDefaultViewport,  previewAutomationOpenNeedsOverlay,  shouldAutoShowPreviewForAutomationUse,  shouldOpenPreviewMiniPlayer,} from "./previewAutomationOpenReadiness";import {  assertPreviewRuntimeCurrent,  waitForNavigationReadiness,} from "./previewNavigationReadiness";import { createPreviewAutomationRequestConsumerAtom } from "./previewAutomationRequestConsumer";import { createPreviewAutomationClientId } from "./previewAutomationClientId";import {  needsPreviewAutomationSessionSync,  resolvePreviewAutomationOpenTab,  resolvePreviewAutomationTarget,} from "./previewAutomationTarget";import { resolveHostWaitBudgetMs, waitForHostReadiness } from "./previewAutomationHostBudget";import { isPreviewViewportReady } from "./previewViewportReadiness";import { shouldRollbackPreviewViewport } from "./previewViewportRollback"; const PREVIEW_PRESENTATION_SETTLE_TIMEOUT_MS = 500; const waitForPreviewPresentation = async (runtimeTabId: string): Promise<void> => {  const deadline = Date.now() + PREVIEW_PRESENTATION_SETTLE_TIMEOUT_MS;  while (Date.now() <= deadline) {    if (useBrowserSurfaceStore.getState().byTabId[runtimeTabId]?.visible) return;    await new Promise<void>((resolve) => window.setTimeout(resolve, 16));  }}; const waitForDesktopOverlay = async (  threadRef: ScopedThreadRef,  requestId: string,  tabId: string,  runtimeTabId: string,  operation: PreviewAutomationRequest["operation"],  deadlineMs: number,): Promise<void> => {  const waitBudgetMs = Math.max(0, deadlineMs - Date.now());  const ready = await waitForHostReadiness(deadlineMs, async () => {    const state = assertPreviewRuntimeCurrent(threadRef, tabId, runtimeTabId, {      operation,      requestId,    });    if (state.desktopByTabId[tabId] && previewBridge && isPreviewWebviewRendering(runtimeTabId)) {      const status = await previewBridge.automation.status(runtimeTabId);      return status.available;    }    return false;  });  if (ready) return;  throw new PreviewAutomationOverlayTimeoutError({    requestId,    environmentId: threadRef.environmentId,    threadId: threadRef.threadId,    timeoutMs: waitBudgetMs,  });}; interface ExecutablePreviewWebview extends Element {  readonly executeJavaScript: (code: string, userGesture?: boolean) => Promise<unknown>;} const findPreviewWebview = (tabId: string): ExecutablePreviewWebview | null =>  Array.from(document.querySelectorAll<ExecutablePreviewWebview>("webview[data-preview-tab]")).find(    (candidate) => candidate.getAttribute("data-preview-tab") === tabId,  ) ?? null; const isPreviewWebviewRendering = (runtimeTabId: string): boolean => {  const wrapper = findPreviewWebview(runtimeTabId)?.closest<HTMLElement>("[data-preview-viewport]");  return wrapper?.getAttribute("data-preview-rendering") === "active";}; const readWebviewViewport = async (  webview: ExecutablePreviewWebview,): Promise<PreviewRenderedViewportSize | null> => {  const value = await webview.executeJavaScript(    "({ width: window.innerWidth, height: window.innerHeight })",  );  if (typeof value !== "object" || value === null) return null;  const { width, height } = value as { readonly width?: unknown; readonly height?: unknown };  return typeof width === "number" &&    Number.isInteger(width) &&    width > 0 &&    typeof height === "number" &&    Number.isInteger(height) &&    height > 0    ? { width, height }    : null;}; const readRenderedViewport = async (  runtimeTabId: string,): Promise<PreviewRenderedViewportSize | null> => {  const webview = findPreviewWebview(runtimeTabId);  if (!webview) return null;  return await readWebviewViewport(webview);}; const readDeclaredViewport = (  webview: ExecutablePreviewWebview | null,): PreviewRenderedViewportSize | null => {  const width = Number(webview?.getAttribute("data-preview-css-width"));  const height = Number(webview?.getAttribute("data-preview-css-height"));  return Number.isInteger(width) && width > 0 && Number.isInteger(height) && height > 0    ? { width, height }    : null;}; const waitForRenderedViewport = async (  threadRef: ScopedThreadRef,  tabId: string,  runtimeTabId: string,  setting: PreviewViewportSetting,  timeoutMs: number,  context: {    readonly requestId: PreviewAutomationRequest["requestId"];    readonly operation: PreviewAutomationRequest["operation"];    readonly environmentId: EnvironmentId;    readonly threadId: PreviewAutomationRequest["threadId"];  },): Promise<PreviewRenderedViewportSize> => {  const deadline = Date.now() + timeoutMs;  while (Date.now() <= deadline) {    assertPreviewRuntimeCurrent(threadRef, tabId, runtimeTabId, context);    try {      const webview = findPreviewWebview(runtimeTabId);      const appliedSettingKey = webview?.getAttribute("data-preview-viewport-key") ?? null;      const declaredViewport = readDeclaredViewport(webview);      const renderedViewport = webview ? await readWebviewViewport(webview) : null;      if (        renderedViewport &&        isPreviewViewportReady({          setting,          appliedSettingKey,          declaredViewport,          renderedViewport,        })      ) {        return renderedViewport;      }    } catch {      // Registration and navigation can transiently replace the guest while      // React applies the server snapshot. Retry until the operation deadline.    }    await new Promise<void>((resolve) => window.setTimeout(resolve, 50));  }  throw new PreviewAutomationViewportTimeoutError({    ...context,    tabId,    timeoutMs,  });}; const currentStatus = async (  threadRef: ScopedThreadRef,  requestedTabId: string | null,): Promise<PreviewAutomationStatus> => {  const state = readThreadPreviewState(threadRef);  const { snapshot, tabId } = resolvePreviewAutomationTarget(state, requestedTabId);  const runtimeTabId = tabId ? previewRuntimeTabId(threadRef, state.serverEpoch, tabId) : null;  const visible = runtimeTabId    ? (useBrowserSurfaceStore.getState().byTabId[runtimeTabId]?.visible ?? false)    : false;  const renderingActive = runtimeTabId ? isPreviewWebviewRendering(runtimeTabId) : false;  const viewportSetting = snapshot ? (snapshot.viewport ?? FILL_PREVIEW_VIEWPORT) : undefined;  const viewport =    runtimeTabId && renderingActive      ? await readRenderedViewport(runtimeTabId).catch(() => null)      : null;  const viewportStatus = {    ...(viewportSetting === undefined ? {} : { viewportSetting }),    ...(viewport === null ? {} : { viewport }),  };  if (runtimeTabId && tabId && previewBridge && state.desktopByTabId[tabId]) {    const status = await previewBridge.automation.status(runtimeTabId);    return { ...status, tabId, visible, ...viewportStatus };  }  const navStatus = snapshot?.navStatus;  return {    available: Boolean(previewBridge?.automation),    visible,    tabId,    url: navStatus && navStatus._tag !== "Idle" ? navStatus.url : null,    title: navStatus && navStatus._tag !== "Idle" ? navStatus.title : null,    loading: navStatus?._tag === "Loading",    ...viewportStatus,  };}; const raiseAtomCommandFailure = (result: Parameters<typeof squashAtomCommandFailure>[0]): never => {  throw squashAtomCommandFailure(result);}; const raisePreviewAutomationHostError = (  error: PreviewAutomationRecordingNotActiveError,): never => {  throw error;}; export function PreviewAutomationHosts() {  const { environments } = useEnvironments();  if (!isElectron || !previewBridge?.automation) return null;  return (    <>      {/*       * Host lifetime follows the desktop runtime's environment connections,       * not the routed thread. This keeps background threads automatable and       * lets the subscription runtime own reconnects for every saved target.       */}      {environments.map((environment) => (        <PreviewAutomationHost          key={environment.environmentId}          environmentId={environment.environmentId}        />      ))}    </>  );} function PreviewAutomationHost(props: { readonly environmentId: EnvironmentId }) {  const { environmentId } = props;  const registry = useContext(RegistryContext);  const [automationClientId] = useState(createPreviewAutomationClientId);  const initialAutomationHost = useMemo<PreviewAutomationHostState>(    () => ({      clientId: automationClientId,      environmentId,      supportedOperations: [...PREVIEW_AUTOMATION_OPERATIONS],    }),    [automationClientId, environmentId],  );  const automationRequestsAtom = previewEnvironment.automationRequests({    environmentId,    input: initialAutomationHost,  });  const listPreviews = useAtomQueryRunner(previewEnvironment.list, {    reportFailure: false,  });  const open = useAtomCommand(previewEnvironment.open, {    reportFailure: false,  });  const resize = useAtomCommand(previewEnvironment.resize, {    reportFailure: false,  });  const respondToAutomation = useAtomCommand(    previewEnvironment.respondToAutomation,    "preview automation response",  );  const focusAutomationHost = useAtomCommand(    previewEnvironment.focusAutomationHost,    "preview automation host focus",  );  const [automationConnectionAtom] = useState(() => Atom.make<string | null>(null));  const automationConnectionId = useAtomValue(automationConnectionAtom);  const presentationSuppressedRuntimeTabsRef = useRef(new Map<string, Set<string>>());   const handleRequest = useCallback(    async (request: PreviewAutomationRequest): Promise<unknown> => {      // Session sync and tab creation consume the same budget as overlay registration.      const hostDeadlineMs = Date.now() + resolveHostWaitBudgetMs(request.timeoutMs);      const threadRef: ScopedThreadRef = {        environmentId,        threadId: request.threadId,      };      let tabId = request.tabId ?? null;      const browserActivity = { release: null as (() => void) | null };      try {        let state = readThreadPreviewState(threadRef);        const needsSessionSync = needsPreviewAutomationSessionSync(state, request.tabId);        if (needsSessionSync) {          const listTarget = {            environmentId,            input: { threadId: request.threadId },          } as const;          registry.refresh(previewEnvironment.list(listTarget));          const result = await listPreviews(listTarget);          if (result._tag === "Failure") {            return raiseAtomCommandFailure(result);          }          reconcilePreviewServerSessions(threadRef, result.value);          state = readThreadPreviewState(threadRef);        }        tabId = request.tabId ?? state.snapshot?.tabId ?? null;        const unavailableTarget = {          requestId: request.requestId,          operation: request.operation,          environmentId,          threadId: request.threadId,          tabId,          bridgeAvailable: Boolean(previewBridge),        };        const requireReadyTab = async () => {          const bridge = previewBridge;          const readyTabId = tabId;          if (!bridge || !readyTabId) {            throw new PreviewAutomationTargetUnavailableError(unavailableTarget);          }          const readyState = readThreadPreviewState(threadRef);          const runtimeTabId = previewRuntimeTabId(threadRef, readyState.serverEpoch, readyTabId);          if (request.operation !== "open") {            const { autoShowFloatingPreview } = await resolveBrowserDefaults();            if (              shouldAutoShowPreviewForAutomationUse({                operation: request.operation,                autoShowFloatingPreview,                presentationSuppressed:                  presentationSuppressedRuntimeTabsRef.current                    .get(request.threadId)                    ?.has(runtimeTabId) ?? false,              })            ) {              usePreviewMiniPlayerStore.getState().open(threadRef, readyTabId);            }          }          browserActivity.release ??= acquireBrowserSurfaceActivity(runtimeTabId);          await waitForDesktopOverlay(            threadRef,            request.requestId,            readyTabId,            runtimeTabId,            request.operation,            hostDeadlineMs,          );          return {            bridge,            tabId: readyTabId,            runtimeTabId,          };        };        switch (request.operation) {          case "status":            return await currentStatus(threadRef, tabId);          case "open": {            const input = request.input as PreviewAutomationOpenInput;            const resolvedInputUrl = input.url              ? resolveBrowserNavigationTarget(environmentId, {                  kind: "url",                  url: input.url,                }).resolvedUrl              : undefined;            let activeTabId = resolvePreviewAutomationOpenTab(              state,              request.tabId,              input.reuseExistingTab ?? true,            );            let activeSnapshot = activeTabId              ? (state.sessions[activeTabId] ?? state.snapshot ?? undefined)              : undefined;            const reusedExistingTab = activeTabId !== null;            tabId = activeTabId;            if (!activeTabId) {              const defaults = await resolveBrowserDefaults();              const result = await open({                environmentId,                input: {                  threadId: request.threadId,                  ...(resolvedInputUrl ? { url: resolvedInputUrl } : {}),                  // An agent that didn't state a size gets the user's                  // configured default, same as a hand-opened tab.                  viewport: browserDefaultOpenViewport(defaults),                  profileId: browserDefaultOpenProfileId(defaults),                },              });              if (result._tag === "Failure") {                return raiseAtomCommandFailure(result);              }              const snapshot = result.value;              applyPreviewServerSnapshot(threadRef, snapshot);              activeTabId = snapshot.tabId;              activeSnapshot = snapshot;              tabId = activeTabId;            }            const activeRuntimeTabId = previewRuntimeTabId(              threadRef,              readThreadPreviewState(threadRef).serverEpoch,              activeTabId,            );            if (activeSnapshot) {              const defaultViewport = previewAutomationDefaultViewport(                reusedExistingTab,                activeSnapshot,              );              if (defaultViewport) {                const resizeResult = await runBrowserViewportMutation(                  activeRuntimeTabId,                  async () => {                    assertPreviewRuntimeCurrent(                      threadRef,                      activeTabId,                      activeRuntimeTabId,                      request,                    );                    return await resize({                      environmentId,                      input: {                        threadId: request.threadId,                        tabId: activeTabId,                        viewport: defaultViewport,                      },                    });                  },                );                if (resizeResult._tag === "Failure") {                  return raiseAtomCommandFailure(resizeResult);                }                activeSnapshot = resizeResult.value;                updatePreviewServerSnapshot(threadRef, resizeResult.value);              }            }            const shouldPresentPreview = shouldOpenPreviewMiniPlayer(              input,              (await resolveBrowserDefaults()).autoShowFloatingPreview,            );            const explicitlySuppressed = explicitlySuppressesPreviewMiniPlayer(input);            const suppressedTabs = presentationSuppressedRuntimeTabsRef.current.get(              request.threadId,            );            if (explicitlySuppressed) {              if (suppressedTabs) {                suppressedTabs.add(activeRuntimeTabId);              } else {                presentationSuppressedRuntimeTabsRef.current.set(                  request.threadId,                  new Set([activeRuntimeTabId]),                );              }              const miniPlayer = selectThreadPreviewMiniPlayer(                usePreviewMiniPlayerStore.getState().byThreadKey,                threadRef,              );              if (miniPlayer?.tabId === activeTabId) {

Lines 1–500 of 819. Load more