paseo/packages/app/src/components/terminal-emulator-webview.native.tsx

import {
  useCallback,
  useEffect,
  useImperativeHandle,
  useMemo,
  useRef,
  useState,
  type ComponentProps,
} from "react";
import {
  Keyboard,
  StyleSheet,
  View,
  type GestureResponderEvent,
  type StyleProp,
  type ViewStyle,
} from "react-native";
import { WebView, type WebViewMessageEvent } from "react-native-webview";
import type { ITheme } from "@xterm/xterm";
import type { TerminalState } from "@getpaseo/protocol/messages";
import type { TerminalInputModeState } from "@getpaseo/protocol/terminal-input-mode";
import type { TerminalOutputData } from "../terminal/runtime/terminal-emulator-runtime";
import type {
  TerminalLocalFileLinkSource,
  TerminalLocalFileLinkTarget,
} from "../terminal/local-links/terminal-local-link-provider";
import { terminalEmulatorWebViewHtml } from "../terminal/webview/terminal-emulator-webview-html";
import type { PendingTerminalModifiers } from "../utils/terminal-keys";
import { openExternalUrl } from "../utils/open-external-url";
import type { TerminalEmulatorHandle, TerminalEmulatorProps } from "./terminal-emulator-contract";
import { shouldHandleTerminalBridgeMessage } from "./terminal-webview-message-routing";
import { shouldResetForMissingRendererReady } from "./terminal-webview-readiness";

type BridgeInboundMessage =
  | {
      type: "mount";
      streamKey: string;
      initialSnapshot: TerminalState | null;
      scrollbackLines: number;
      theme: ITheme;
      fontFamily?: string;
      fontSize?: number;
      pendingModifiers: PendingTerminalModifiers;
      swipeGesturesEnabled: boolean;
    }
  | { type: "unmount"; streamKey: string }
  | { type: "writeOutput"; streamKey: string; text: string }
  | { type: "restoreOutput"; streamKey: string; text: string }
  | { type: "renderSnapshot"; streamKey: string; state: TerminalState | null }
  | { type: "paste"; streamKey: string; text: string }
  | { type: "clear"; streamKey: string }
  | { type: "focus"; streamKey: string; forceRefocus?: boolean }
  | { type: "resize"; streamKey: string; forceClaim: boolean; shouldClaim?: boolean }
  | { type: "setTheme"; streamKey: string; theme: ITheme }
  | { type: "setScrollback"; streamKey: string; lines: number }
  | { type: "setFont"; streamKey: string; fontFamily?: string; fontSize?: number }
  | { type: "setPendingModifiers"; streamKey: string; pendingModifiers: PendingTerminalModifiers }
  | { type: "setSwipeGesturesEnabled"; streamKey: string; enabled: boolean }
  | {
      type: "resolveLocalFileLinkResponse";
      streamKey: string;
      requestId: number;
      target: TerminalLocalFileLinkTarget | null;
    };

type BridgeOutboundMessage =
  | { type: "bridgeReady" }
  | { type: "rendererReady"; streamKey: string; isReady: boolean }
  | { type: "input"; streamKey: string; data: string }
  | {
      type: "resize";
      streamKey: string;
      rows: number;
      cols: number;
      shouldClaim?: boolean;
      forceClaim?: boolean;
    }
  | {
      type: "terminalKey";
      streamKey: string;
      key: string;
      ctrl: boolean;
      shift: boolean;
      alt: boolean;
      meta: boolean;
    }
  | { type: "pendingModifiersConsumed"; streamKey: string }
  | { type: "inputModeChange"; streamKey: string; state: TerminalInputModeState }
  | { type: "openExternalUrl"; streamKey: string; url: string }
  | {
      type: "resolveLocalFileLink";
      streamKey: string;
      requestId: number;
      source: TerminalLocalFileLinkSource;
    }
  | {
      type: "openLocalFileLink";
      streamKey: string;
      target: TerminalLocalFileLinkTarget;
      disposition: "main" | "side";
    }
  | { type: "swipeLeft"; streamKey: string }
  | { type: "swipeRight"; streamKey: string }
  | { type: "debug"; message: string; details?: unknown };

const TERMINAL_WEBVIEW_SOURCE = { html: terminalEmulatorWebViewHtml };
const TERMINAL_WEBVIEW_ORIGIN_WHITELIST = ["*"];
const BRIDGE_READY_TIMEOUT_MS = 2_500;
const RENDERER_READY_TIMEOUT_MS = 2_500;
const TERMINAL_TAP_MOVE_TOLERANCE_PX = 8;
type WebViewProps = ComponentProps<typeof WebView>;
interface PendingTerminalTap {
  startX: number;
  startY: number;
  moved: boolean;
}

function buildThemeKey(theme: ITheme): string {
  return JSON.stringify(theme);
}

function serializeForInjectedJavaScript(message: BridgeInboundMessage): string {
  return JSON.stringify(message).replace(/<\/script/gi, "<\\/script");
}

function createMountMessage(input: {
  streamKey: string;
  initialSnapshot: TerminalState | null;
  scrollbackLines: number;
  theme: ITheme;
  fontFamily?: string;
  fontSize?: number;
  pendingModifiers: PendingTerminalModifiers;
  swipeGesturesEnabled: boolean;
}): BridgeInboundMessage {
  return {
    type: "mount",
    streamKey: input.streamKey,
    initialSnapshot: input.initialSnapshot,
    scrollbackLines: input.scrollbackLines,
    theme: input.theme,
    fontFamily: input.fontFamily,
    fontSize: input.fontSize,
    pendingModifiers: input.pendingModifiers,
    swipeGesturesEnabled: input.swipeGesturesEnabled,
  };
}

export default function WebViewTerminalEmulator({
  ref,
  streamKey,
  testId = "terminal-surface",
  xtermTheme = {
    background: "#0b0b0b",
    foreground: "#e6e6e6",
    cursor: "#e6e6e6",
  },
  scrollbackLines,
  fontFamily,
  fontSize,
  swipeGesturesEnabled = false,
  onSwipeLeft,
  onSwipeRight,
  initialSnapshot = null,
  onInput,
  onFocus,
  onResize,
  onTerminalKey,
  onPendingModifiersConsumed,
  onInputModeChange,
  onResolveLocalFileLink,
  onOpenLocalFileLink,
  onRendererReadyChange,
  pendingModifiers = { ctrl: false, shift: false, alt: false },
  focusRequestToken = 0,
  resizeRequestToken = 0,
}: TerminalEmulatorProps) {
  const webViewRef = useRef<WebView>(null);
  const [webViewEpoch, setWebViewEpoch] = useState(0);
  const [bridgeReadyVersion, setBridgeReadyVersion] = useState(0);
  const bridgeReadyRef = useRef(false);
  const bridgeReadyVersionRef = useRef(0);
  const rendererReadyVersionRef = useRef(0);
  const pendingMessagesRef = useRef<BridgeInboundMessage[]>([]);
  const outputDecoderRef = useRef(new TextDecoder());
  const mountRequestedStreamKeyRef = useRef<string | null>(null);
  const rendererReadyStreamKeyRef = useRef<string | null>(null);
  const bridgeReadyTimeoutRef = useRef<ReturnType<typeof setTimeout> | null>(null);
  const rendererReadyTimeoutRef = useRef<ReturnType<typeof setTimeout> | null>(null);
  const pendingTapRef = useRef<PendingTerminalTap | null>(null);
  const mountConfigRef = useRef({
    streamKey,
    initialSnapshot,
    scrollbackLines,
    theme: xtermTheme,
    fontFamily,
    fontSize,
    pendingModifiers,
    swipeGesturesEnabled,
  });
  mountConfigRef.current = {
    streamKey,
    initialSnapshot,
    scrollbackLines,
    theme: xtermTheme,
    fontFamily,
    fontSize,
    pendingModifiers,
    swipeGesturesEnabled,
  };
  const callbacksRef = useRef({
    onInput,
    onFocus,
    onResize,
    onTerminalKey,
    onPendingModifiersConsumed,
    onInputModeChange,
    onRendererReadyChange,
    onResolveLocalFileLink,
    onOpenLocalFileLink,
    onSwipeLeft,
    onSwipeRight,
  });
  callbacksRef.current = {
    onInput,
    onFocus,
    onResize,
    onTerminalKey,
    onPendingModifiersConsumed,
    onInputModeChange,
    onRendererReadyChange,
    onResolveLocalFileLink,
    onOpenLocalFileLink,
    onSwipeLeft,
    onSwipeRight,
  };

  const clearBridgeReadyTimeout = useCallback(() => {
    if (bridgeReadyTimeoutRef.current === null) return;
    clearTimeout(bridgeReadyTimeoutRef.current);
    bridgeReadyTimeoutRef.current = null;
  }, []);

  const clearRendererReadyTimeout = useCallback(() => {
    if (rendererReadyTimeoutRef.current === null) return;
    clearTimeout(rendererReadyTimeoutRef.current);
    rendererReadyTimeoutRef.current = null;
  }, []);

  const resetWebViewDocument = useCallback(() => {
    clearBridgeReadyTimeout();
    clearRendererReadyTimeout();
    bridgeReadyRef.current = false;
    pendingMessagesRef.current = [];
    mountRequestedStreamKeyRef.current = null;
    rendererReadyStreamKeyRef.current = null;
    callbacksRef.current.onRendererReadyChange?.({ streamKey, isReady: false });
    setWebViewEpoch((value) => value + 1);
  }, [clearBridgeReadyTimeout, clearRendererReadyTimeout, streamKey]);

  const scheduleBridgeReadyWatchdog = useCallback(() => {
    clearBridgeReadyTimeout();
    const expectedBridgeReadyVersion = bridgeReadyVersionRef.current;
    bridgeReadyTimeoutRef.current = setTimeout(() => {
      bridgeReadyTimeoutRef.current = null;
      if (bridgeReadyVersionRef.current !== expectedBridgeReadyVersion || bridgeReadyRef.current) {
        return;
      }
      resetWebViewDocument();
    }, BRIDGE_READY_TIMEOUT_MS);
  }, [clearBridgeReadyTimeout, resetWebViewDocument]);

  const scheduleRendererReadyWatchdog = useCallback(() => {
    clearRendererReadyTimeout();
    const expectedRendererReadyVersion = rendererReadyVersionRef.current;
    rendererReadyTimeoutRef.current = setTimeout(() => {
      rendererReadyTimeoutRef.current = null;
      if (
        !shouldResetForMissingRendererReady({
          expectedReadyVersion: expectedRendererReadyVersion,
          currentReadyVersion: rendererReadyVersionRef.current,
          expectedStreamKey: streamKey,
          mountRequestedStreamKey: mountRequestedStreamKeyRef.current,
          rendererReadyStreamKey: rendererReadyStreamKeyRef.current,
        })
      ) {
        return;
      }
      resetWebViewDocument();
    }, RENDERER_READY_TIMEOUT_MS);
  }, [clearRendererReadyTimeout, resetWebViewDocument, streamKey]);

  const flushPendingMessages = useCallback(() => {
    if (!bridgeReadyRef.current || !webViewRef.current) return;
    const pending = pendingMessagesRef.current.splice(0);
    for (const message of pending) {
      const payload = serializeForInjectedJavaScript(message);
      webViewRef.current.injectJavaScript(
        `window.__PASEO_TERMINAL_WEBVIEW_RECEIVE__ && window.__PASEO_TERMINAL_WEBVIEW_RECEIVE__(${payload}); true;`,
      );
    }
  }, []);

  const sendToWebView = useCallback((message: BridgeInboundMessage) => {
    if (!bridgeReadyRef.current || !webViewRef.current) {
      pendingMessagesRef.current.push(message);
      return;
    }
    const payload = serializeForInjectedJavaScript(message);
    webViewRef.current.injectJavaScript(
      `window.__PASEO_TERMINAL_WEBVIEW_RECEIVE__ && window.__PASEO_TERMINAL_WEBVIEW_RECEIVE__(${payload}); true;`,
    );
  }, []);

  useImperativeHandle(
    ref,
    (): TerminalEmulatorHandle => ({
      writeOutput: (data: TerminalOutputData) => {
        const output = outputDecoderRef.current.decode(data, { stream: true });
        if (output.length === 0) {
          return;
        }
        sendToWebView({ type: "writeOutput", streamKey, text: output });
      },
      restoreOutput: (data: TerminalOutputData) => {
        outputDecoderRef.current.decode();
        const text = outputDecoderRef.current.decode(data, { stream: false });
        if (text.length === 0) {
          return;
        }
        sendToWebView({ type: "restoreOutput", streamKey, text });
      },
      renderSnapshot: (state: TerminalState | null) => {
        outputDecoderRef.current.decode();
        sendToWebView({ type: "renderSnapshot", streamKey, state });
      },
      paste: (text: string) => {
        sendToWebView({ type: "paste", streamKey, text });
      },
      copySelection: async () => "",
      clear: () => {
        outputDecoderRef.current.decode();
        sendToWebView({ type: "clear", streamKey });
      },
      claimSize: () => {
        sendToWebView({ type: "resize", streamKey, forceClaim: true });
      },
      showKeyboard: () => {
        sendToWebView({ type: "focus", streamKey, forceRefocus: true });
        webViewRef.current?.requestFocus();
      },
      blur: () => {
        webViewRef.current?.injectJavaScript(
          "window.__PASEO_TERMINAL_WEBVIEW_BLUR__ && window.__PASEO_TERMINAL_WEBVIEW_BLUR__(); true;",
        );
        Keyboard.dismiss();
      },
    }),
    [sendToWebView, streamKey],
  );

  useEffect(() => {
    outputDecoderRef.current.decode();
  }, [streamKey]);

  useEffect(() => {
    if (bridgeReadyVersion <= 0) return;
    const mountMessage = createMountMessage(mountConfigRef.current);
    mountRequestedStreamKeyRef.current = streamKey;
    rendererReadyStreamKeyRef.current = null;
    sendToWebView(mountMessage);
    flushPendingMessages();
    scheduleRendererReadyWatchdog();
  }, [
    bridgeReadyVersion,
    flushPendingMessages,
    scheduleRendererReadyWatchdog,
    sendToWebView,
    streamKey,
  ]);

  const themeKey = useMemo(() => buildThemeKey(xtermTheme), [xtermTheme]);
  useEffect(() => {
    if (!mountRequestedStreamKeyRef.current) return;
    sendToWebView({ type: "setTheme", streamKey, theme: xtermTheme });
  }, [sendToWebView, streamKey, themeKey, xtermTheme]);

  useEffect(() => {
    if (!mountRequestedStreamKeyRef.current) return;
    sendToWebView({ type: "setScrollback", streamKey, lines: scrollbackLines });
  }, [scrollbackLines, sendToWebView, streamKey]);

  useEffect(() => {
    if (!mountRequestedStreamKeyRef.current) return;
    sendToWebView({ type: "setFont", streamKey, fontFamily, fontSize });
  }, [fontFamily, fontSize, sendToWebView, streamKey]);

  useEffect(() => {
    if (!mountRequestedStreamKeyRef.current) return;
    sendToWebView({ type: "setPendingModifiers", streamKey, pendingModifiers });
  }, [pendingModifiers, sendToWebView, streamKey]);

  useEffect(() => {
    if (!mountRequestedStreamKeyRef.current) return;
    sendToWebView({ type: "setSwipeGesturesEnabled", streamKey, enabled: swipeGesturesEnabled });
  }, [sendToWebView, streamKey, swipeGesturesEnabled]);

  useEffect(() => {
    if (focusRequestToken <= 0) return;
    sendToWebView({ type: "resize", streamKey, forceClaim: true });
    sendToWebView({ type: "focus", streamKey });
    webViewRef.current?.requestFocus();
  }, [focusRequestToken, sendToWebView, streamKey]);

  useEffect(() => {
    if (resizeRequestToken <= 0) return;
    sendToWebView({ type: "resize", streamKey, forceClaim: false, shouldClaim: false });
  }, [resizeRequestToken, sendToWebView, streamKey]);

  useEffect(() => {
    return () => {
      if (mountRequestedStreamKeyRef.current) {
        const previousStreamKey = mountRequestedStreamKeyRef.current;
        callbacksRef.current.onRendererReadyChange?.({
          streamKey: previousStreamKey,
          isReady: false,
        });
        sendToWebView({ type: "unmount", streamKey: previousStreamKey });
      }
      bridgeReadyRef.current = false;
      pendingMessagesRef.current = [];
      mountRequestedStreamKeyRef.current = null;
      rendererReadyStreamKeyRef.current = null;
      clearBridgeReadyTimeout();
      clearRendererReadyTimeout();
    };
  }, [clearBridgeReadyTimeout, clearRendererReadyTimeout, sendToWebView]);

  const handleLifecycleMessage = useCallback(
    (message: BridgeOutboundMessage): boolean => {
      if (message.type === "bridgeReady") {
        bridgeReadyRef.current = true;
        bridgeReadyVersionRef.current += 1;
        clearBridgeReadyTimeout();
        setBridgeReadyVersion((value) => value + 1);
        return true;
      }
      if (message.type === "rendererReady") {
        if (message.streamKey === mountRequestedStreamKeyRef.current) {
          rendererReadyStreamKeyRef.current = message.isReady ? message.streamKey : null;
          if (message.isReady) {
            rendererReadyVersionRef.current += 1;
            clearRendererReadyTimeout();
          }
        }
        callbacksRef.current.onRendererReadyChange?.({
          streamKey: message.streamKey,
          isReady: message.isReady,
        });
        return true;
      }
      return false;
    },
    [clearBridgeReadyTimeout, clearRendererReadyTimeout],
  );

  const resolveLocalFileLink = useCallback(
    async (message: Extract<BridgeOutboundMessage, { type: "resolveLocalFileLink" }>) => {
      try {
        const target = await (callbacksRef.current.onResolveLocalFileLink?.(message.source) ??
          null);
        sendToWebView({
          type: "resolveLocalFileLinkResponse",
          streamKey: message.streamKey,
          requestId: message.requestId,
          target,
        });
      } catch {
        sendToWebView({
          type: "resolveLocalFileLinkResponse",
          streamKey: message.streamKey,
          requestId: message.requestId,
          target: null,
        });
      }
    },
    [sendToWebView],
  );

  const handleTerminalMessage = useCallback(
    (
      message: Exclude<BridgeOutboundMessage, { type: "bridgeReady" } | { type: "rendererReady" }>,
    ) => {
      if (message.type === "resolveLocalFileLink") {
        void resolveLocalFileLink(message);
        return;
      }
      if (message.type === "openLocalFileLink") {
        callbacksRef.current.onOpenLocalFileLink?.(message.target, message.disposition);
        return;
      }
      switch (message.type) {
        case "input":
          callbacksRef.current.onInput?.(message.data);
          break;
        case "resize":
          callbacksRef.current.onResize?.({
            rows: message.rows,
            cols: message.cols,
            shouldClaim: message.shouldClaim !== false,
            forceClaim: message.forceClaim,
          });
          break;
        case "terminalKey":
          callbacksRef.current.onTerminalKey?.({
            key: message.key,
            ctrl: message.ctrl,
            shift: message.shift,
            alt: message.alt,
            meta: message.meta,
          });
          break;
        case "pendingModifiersConsumed":
          callbacksRef.current.onPendingModifiersConsumed?.();
          break;
        case "inputModeChange":
          callbacksRef.current.onInputModeChange?.(message.state);
          break;
        case "openExternalUrl":
          void openExternalUrl(message.url);
          break;
        case "swipeLeft":
          callbacksRef.current.onSwipeLeft?.();
          break;
        case "swipeRight":
          callbacksRef.current.onSwipeRight?.();
          break;
        case "debug":
          break;
      }
    },
    [resolveLocalFileLink],
  );

  const handleMessage = useCallback(
    (event: WebViewMessageEvent) => {
      let message: BridgeOutboundMessage;
      try {
        message = JSON.parse(event.nativeEvent.data) as BridgeOutboundMessage;
      } catch {
        return;
      }

      if (message.type === "bridgeReady" || message.type === "rendererReady") {
        handleLifecycleMessage(message);
        return;
      }
      if (
        !shouldHandleTerminalBridgeMessage(message, {
          desiredStreamKey: streamKey,
          mountedStreamKey: mountRequestedStreamKeyRef.current,
        })
      ) {
        return;
      }
      handleTerminalMessage(message);
    },
    [handleLifecycleMessage, handleTerminalMessage, streamKey],
  );

  const handleLoadStart = useCallback<NonNullable<WebViewProps["onLoadStart"]>>(() => {
    bridgeReadyRef.current = false;
    mountRequestedStreamKeyRef.current = null;
    rendererReadyStreamKeyRef.current = null;
    scheduleBridgeReadyWatchdog();
  }, [scheduleBridgeReadyWatchdog]);

  const handleContentProcessDidTerminate = useCallback<
    NonNullable<WebViewProps["onContentProcessDidTerminate"]>
  >(() => {
    resetWebViewDocument();
  }, [resetWebViewDocument]);

  const handleRenderProcessGone = useCallback<
    NonNullable<WebViewProps["onRenderProcessGone"]>
  >(() => {
    resetWebViewDocument();
  }, [resetWebViewDocument]);

  const handleWebViewTouchStart = useCallback((event: GestureResponderEvent) => {
    pendingTapRef.current = {
      startX: event.nativeEvent.pageX,
      startY: event.nativeEvent.pageY,
      moved: false,
    };
  }, []);

  const handleWebViewTouchMove = useCallback((event: GestureResponderEvent) => {
    const pendingTap = pendingTapRef.current;
    if (!pendingTap) return;
    const dx = event.nativeEvent.pageX - pendingTap.startX;
    const dy = event.nativeEvent.pageY - pendingTap.startY;
    if (
      Math.abs(dx) > TERMINAL_TAP_MOVE_TOLERANCE_PX ||
      Math.abs(dy) > TERMINAL_TAP_MOVE_TOLERANCE_PX
    ) {
      pendingTap.moved = true;
    }
  }, []);

  const handleWebViewTouchEnd = useCallback(() => {
    const pendingTap = pendingTapRef.current;
    pendingTapRef.current = null;
    if (!pendingTap || pendingTap.moved) {
      return;
    }
    webViewRef.current?.requestFocus();
    callbacksRef.current.onFocus?.();
    sendToWebView({ type: "focus", streamKey, forceRefocus: true });
  }, [sendToWebView, streamKey]);

  const handleWebViewTouchCancel = useCallback(() => {
    pendingTapRef.current = null;
  }, []);

  const backgroundColor = xtermTheme.background ?? "#0b0b0b";
  const rootStyle = useMemo<StyleProp<ViewStyle>>(
    () => [styles.root, { backgroundColor }],
    [backgroundColor],
  );
  const webViewStyle = useMemo<StyleProp<ViewStyle>>(
    () => [styles.webView, { backgroundColor }],
    [backgroundColor],
  );
  const webViewContainerStyle = useMemo<StyleProp<ViewStyle>>(
    () => [styles.webViewContainer, { backgroundColor }],
    [backgroundColor],
  );

  return (
    <View style={rootStyle} testID={testId}>
      <WebView
        key={webViewEpoch}
        ref={webViewRef}
        source={TERMINAL_WEBVIEW_SOURCE}
        style={webViewStyle}
        containerStyle={webViewContainerStyle}
        originWhitelist={TERMINAL_WEBVIEW_ORIGIN_WHITELIST}
        scrollEnabled
        nestedScrollEnabled
        bounces={false}
        overScrollMode="never"
        keyboardDisplayRequiresUserAction={false}
        automaticallyAdjustContentInsets={false}
        contentInsetAdjustmentBehavior="never"
        textInteractionEnabled
        allowsLinkPreview={false}
        setSupportMultipleWindows={false}
        setBuiltInZoomControls={false}
        setDisplayZoomControls={false}
        textZoom={100}
        onTouchStart={handleWebViewTouchStart}
        onTouchMove={handleWebViewTouchMove}
        onTouchEnd={handleWebViewTouchEnd}
        onTouchCancel={handleWebViewTouchCancel}
        onMessage={handleMessage}
        onLoadStart={handleLoadStart}
        onContentProcessDidTerminate={handleContentProcessDidTerminate}
        onRenderProcessGone={handleRenderProcessGone}
      />
    </View>
  );
}

const styles = StyleSheet.create({
  root: {
    flex: 1,
    minHeight: 0,
    minWidth: 0,
    overflow: "hidden",
  },
  webView: {
    flex: 1,
  },
  webViewContainer: {
    flex: 1,
  },
});