4992 lines · 179.1 KB · TypeScript
RawDownload
import {  type ApprovalRequestId,  DEFAULT_MODEL,  defaultInstanceIdForDriver,  type EnvironmentId,  type MessageId,  type ModelSelection,  type ProjectScript,  type ProjectId,  type ProviderApprovalDecision,  ProviderInstanceId,  type ServerProvider,  type ResolvedKeybindingsConfig,  type ScopedThreadRef,  type ThreadId,  type TurnId,  type KeybindingCommand,  OrchestrationThreadActivity,  ProviderInteractionMode,  ProviderDriverKind,  RuntimeMode,  TerminalOpenInput,} from "@t3tools/contracts";import {  connectionStatusText,  type EnvironmentConnectionPresentation,} from "@t3tools/client-runtime/connection";import {  parseScopedThreadKey,  scopedThreadKey,  scopeProjectRef,  scopeThreadRef,} from "@t3tools/client-runtime/environment";import {  applyClaudePromptEffortPrefix,  createModelSelection,  resolvePromptInjectedEffort,} from "@t3tools/shared/model";import { projectScriptCwd, projectScriptRuntimeEnv } from "@t3tools/shared/projectScripts";import { truncate } from "@t3tools/shared/String";import { nextTerminalId, resolveTerminalSessionLabel } from "@t3tools/shared/terminalLabels";import { Debouncer } from "@tanstack/react-pacer";import { useAtomValue } from "@effect/atom-react";import { lazy, memo, Suspense, useCallback, useEffect, useMemo, useRef, useState } from "react";import { useNavigate } from "@tanstack/react-router";import { useShallow } from "zustand/react/shallow";import {  isAtomCommandInterrupted,  mapAtomCommandResult,  settlePromise,  squashAtomCommandFailure,  type AtomCommandResult,} from "@t3tools/client-runtime/state/runtime";import * as Cause from "effect/Cause";import { AsyncResult } from "effect/unstable/reactivity";import { isElectron } from "../env";import { readLocalApi } from "../localApi";import { useDiffPanelStore } from "../diffPanelStore";import {  collapseExpandedComposerCursor,  parseStandaloneComposerSlashCommand,} from "../composer-logic";import {  derivePendingApprovals,  derivePendingUserInputs,  derivePhase,  deriveTimelineEntries,  deriveActiveWorkStartedAt,  deriveActivePlanState,  findSidebarProposedPlan,  findLatestProposedPlan,  deriveWorkLogEntries,  hasActionableProposedPlan,  isLatestTurnSettled,} from "../session-logic";import { type LegendListRef } from "@legendapp/list/react";import {  buildPendingUserInputAnswers,  derivePendingUserInputProgress,  setPendingUserInputCustomAnswer,  togglePendingUserInputOptionSelection,  type PendingUserInputDraftAnswer,} from "../pendingUserInput";import { useUiStateStore } from "../uiStateStore";import {  buildPlanImplementationThreadTitle,  buildPlanImplementationPrompt,  resolvePlanFollowUpSubmission,} from "../proposedPlan";import {  DEFAULT_INTERACTION_MODE,  DEFAULT_RUNTIME_MODE,  DEFAULT_THREAD_TERMINAL_ID,  MAX_TERMINALS_PER_GROUP,  type ChatMessage,  type SessionPhase,  type Thread,  type TurnDiffSummary,} from "../types";import { useTheme } from "../hooks/useTheme";import { useTurnDiffSummaries } from "../hooks/useTurnDiffSummaries";import { isCommandPaletteOpen } from "../commandPaletteContext";import { buildTemporaryWorktreeBranchName } from "@t3tools/shared/git";import { useMediaQuery } from "../hooks/useMediaQuery";import { RIGHT_PANEL_INLINE_LAYOUT_MEDIA_QUERY } from "../rightPanelLayout";import {  selectActiveRightPanel,  selectActiveRightPanelSurface,  selectThreadRightPanelState,  type RightPanelSurface,  useRightPanelStore,} from "../rightPanelStore";import {  isPreviewSupportedInRuntime,  setActivePreviewTab,  useThreadPreviewState,} from "../previewStateStore";import { addBrowserSurface } from "./preview/addBrowserSurface";import { closePreviewSession } from "./preview/closePreviewSession";import { subscribePreviewAction } from "./preview/previewActionBus";import { getConfiguredPreviewUrls } from "./preview/previewEmptyStateLogic";import { PreviewAutomationOwner } from "./preview/PreviewAutomationOwner";import { RightPanelTabs } from "./RightPanelTabs";import { DiffWorkerPoolProvider } from "./DiffWorkerPoolProvider";import { BranchToolbar } from "./BranchToolbar";import { resolveShortcutCommand, shortcutLabelForCommand } from "../keybindings";import PlanSidebar from "./PlanSidebar";import ThreadTerminalDrawer from "./ThreadTerminalDrawer";import { ChevronDownIcon, TriangleAlertIcon, WifiOffIcon } from "lucide-react";import { cn, randomHex } from "~/lib/utils";import { stackedThreadToast, toastManager } from "./ui/toast";import { decodeProjectScriptKeybindingRule } from "~/lib/projectScriptKeybindings";import { type NewProjectScriptInput } from "./ProjectScriptsControl";import {  commandForProjectScript,  nextProjectScriptId,  projectScriptIdFromCommand,} from "~/projectScripts";import { newDraftId, newMessageId, newThreadId } from "~/lib/utils";import { getProviderModelCapabilities, resolveSelectableProvider } from "../providerModels";import { useEnvironmentSettings } from "../hooks/useSettings";import { resolveAppModelSelectionForInstance } from "../modelSelection";import { getTerminalFocusOwner } from "../lib/terminalFocus";import { resolveNewDraftStartFromOrigin } from "../lib/chatThreadActions";import {  deriveLogicalProjectKeyFromSettings,  selectProjectGroupingSettings,} from "../logicalProject";import { buildDraftThreadRouteParams } from "../threadRoutes";import {  type ComposerImageAttachment,  type DraftThreadEnvMode,  useComposerDraftStore,  type DraftId,} from "../composerDraftStore";import {  appendTerminalContextsToPrompt,  formatTerminalContextLabel,  type TerminalContextDraft,  type TerminalContextSelection,} from "../lib/terminalContext";import {  appendElementContextsToPrompt,  type ElementContextDraft,  formatElementContextLabel,} from "../lib/elementContext";import { appendPreviewAnnotationPrompt } from "../lib/previewAnnotation";import { appendReviewCommentsToPrompt, type ReviewCommentContext } from "../reviewCommentContext";import { environmentCatalog } from "../connection/catalog";import { selectThreadTerminalUiState, useTerminalUiStateStore } from "../terminalUiStateStore";import { useKnownTerminalSessions, useThreadRunningTerminalIds } from "../state/terminalSessions";import { projectEnvironment } from "../state/projects";import { useEnvironmentQuery } from "../state/query";import {  primaryServerAvailableEditorsAtom,  primaryServerKeybindingsAtom,  serverEnvironment,} from "../state/server";import { terminalEnvironment } from "../state/terminal";import { threadEnvironment } from "../state/threads";import { vcsEnvironment } from "../state/vcs";import { useEnvironments, usePrimaryEnvironment } from "../state/environments";import {  useProject,  useProjects,  useThread,  useThreadProposedPlans,  useThreadRefs,} from "../state/entities";import { environmentShell } from "../state/shell";import { ChatComposer, type ChatComposerHandle } from "./chat/ChatComposer";import { ExpandedImageDialog } from "./chat/ExpandedImageDialog";import { PullRequestThreadDialog } from "./PullRequestThreadDialog";import { MessagesTimeline } from "./chat/MessagesTimeline";import { ChatHeader } from "./chat/ChatHeader";import { PanelLayoutControls, RightPanelMaximizeControl } from "./chat/PanelLayoutControls";import { type ExpandedImagePreview } from "./chat/ExpandedImagePreview";import { NoActiveThreadState } from "./NoActiveThreadState";import { resolveEffectiveEnvMode } from "./BranchToolbar.logic";import { ProviderStatusBanner } from "./chat/ProviderStatusBanner";import { ThreadErrorBanner } from "./chat/ThreadErrorBanner";import { ComposerBannerStack, type ComposerBannerStackItem } from "./chat/ComposerBannerStack";import {  MAX_HIDDEN_MOUNTED_TERMINAL_THREADS,  buildExpiredTerminalContextToastCopy,  buildLocalDraftThread,  collectUserMessageBlobPreviewUrls,  createLocalDispatchSnapshot,  deriveComposerSendState,  hasServerAcknowledgedLocalDispatch,  getStartedThreadModelChangeBlockReason,  LAST_INVOKED_SCRIPT_BY_PROJECT_KEY,  LastInvokedScriptByProjectSchema,  type LocalDispatchSnapshot,  PullRequestDialogState,  cloneComposerImageForRetry,  deriveLockedProvider,  readFileAsDataUrl,  reconcileMountedTerminalThreadIds,  resolveSendEnvMode,  revokeBlobPreviewUrl,  revokeUserMessagePreviewUrls,  waitForStartedServerThread,} from "./ChatView.logic";import { useLocalStorage } from "~/hooks/useLocalStorage";import { useComposerHandleContext } from "../composerHandleContext";import { sanitizeThreadErrorMessage } from "~/rpc/transportError";import { RightPanelSheet } from "./RightPanelSheet";import { previewEnvironment } from "../state/preview";import { useAtomCommand } from "../state/use-atom-command";import { Button } from "./ui/button";import {  buildVersionMismatchDismissalKey,  dismissVersionMismatch,  isVersionMismatchDismissed,  resolveServerConfigVersionMismatch,} from "../versionSkew";import { useAssetUrls } from "../assets/assetUrls"; const IMAGE_ONLY_BOOTSTRAP_PROMPT =  "[User attached one or more images without additional text. Respond using the conversation context and the attached image(s).]";const EMPTY_ACTIVITIES: OrchestrationThreadActivity[] = [];const EMPTY_PROVIDERS: ServerProvider[] = [];const EMPTY_PROVIDER_SKILLS: ServerProvider["skills"] = [];const EMPTY_PENDING_USER_INPUT_ANSWERS: Record<string, PendingUserInputDraftAnswer> = {};const PreviewPanel = lazy(() =>  import("./preview/PreviewPanel").then((module) => ({ default: module.PreviewPanel })),);const DiffPanel = lazy(() => import("./DiffPanel"));const FilePreviewPanel = lazy(() => import("./files/FilePreviewPanel"));const EMPTY_PENDING_FILE_SURFACE_IDS: ReadonlySet<string> = new Set();const TYPE_TO_FOCUS_EDITABLE_SELECTOR = [  "input",  "textarea",  "select",  '[contenteditable="true"]',  '[contenteditable="plaintext-only"]',  '[role="textbox"]',].join(",");const TYPE_TO_FOCUS_INTERACTIVE_SELECTOR = [  "button",  "a[href]",  "summary",  '[role="button"]',  '[role="checkbox"]',  '[role="menuitem"]',  '[role="option"]',  '[role="radio"]',  '[role="switch"]',  '[role="tab"]',].join(",");const TYPE_TO_FOCUS_FLOATING_LAYER_SELECTOR = [  '[data-slot="dialog"]',  '[data-slot="menu-popup"]',  '[data-slot="select-popup"]',  '[data-slot="popover-popup"]',  '[data-slot="combobox-popup"]',  '[data-slot="autocomplete-popup"]',].join(","); type EnvironmentUnavailableState = {  readonly environmentId: EnvironmentId;  readonly label: string;  readonly connection: EnvironmentConnectionPresentation;}; type ThreadPlanCatalogEntry = Pick<Thread, "id" | "proposedPlans">; function eventPathContainsSelector(event: Event, selector: string): boolean {  const path = event.composedPath();  if (path.length === 0 && event.target) {    path.push(event.target);  }  return path.some((target) => target instanceof Element && target.closest(selector));} function shouldTypeToFocusComposer(event: KeyboardEvent): boolean {  if (event.defaultPrevented || event.isComposing) return false;  if (event.metaKey || event.ctrlKey || event.altKey) return false;  if (event.key.length !== 1) return false;   if (eventPathContainsSelector(event, TYPE_TO_FOCUS_EDITABLE_SELECTOR)) return false;  if (eventPathContainsSelector(event, TYPE_TO_FOCUS_INTERACTIVE_SELECTOR)) return false;  if (document.querySelector(TYPE_TO_FOCUS_FLOATING_LAYER_SELECTOR)) return false;   return true;} function formatOutgoingPrompt(params: {  provider: ProviderDriverKind;  model: string | null;  models: ReadonlyArray<ServerProvider["models"][number]>;  effort: string | null;  text: string;}): string {  const caps = getProviderModelCapabilities(params.models, params.model, params.provider);  const promptEffort = resolvePromptInjectedEffort(caps, params.effort);  return applyClaudePromptEffortPrefix(params.text, promptEffort);}const SCRIPT_TERMINAL_COLS = 120;const SCRIPT_TERMINAL_ROWS = 30; type ChatViewProps =  | {      environmentId: EnvironmentId;      threadId: ThreadId;      onDiffPanelOpen?: () => void;      reserveTitleBarControlInset?: boolean;      routeKind: "server";      draftId?: never;    }  | {      environmentId: EnvironmentId;      threadId: ThreadId;      onDiffPanelOpen?: () => void;      reserveTitleBarControlInset?: boolean;      routeKind: "draft";      draftId: DraftId;    }; interface TerminalLaunchContext {  threadId: ThreadId;  cwd: string;  worktreePath: string | null;} type PersistentTerminalLaunchContext = Pick<TerminalLaunchContext, "cwd" | "worktreePath">; function useLocalDispatchState(input: {  activeThread: Thread | undefined;  activeLatestTurn: Thread["latestTurn"] | null;  phase: SessionPhase;  activePendingApproval: ApprovalRequestId | null;  activePendingUserInput: ApprovalRequestId | null;  threadError: string | null | undefined;}) {  const [localDispatch, setLocalDispatch] = useState<LocalDispatchSnapshot | null>(null);   const resetLocalDispatch = useCallback(() => {    setLocalDispatch(null);  }, []);   const serverAcknowledgedLocalDispatch = useMemo(    () =>      hasServerAcknowledgedLocalDispatch({        localDispatch,        phase: input.phase,        latestTurn: input.activeLatestTurn,        session: input.activeThread?.session ?? null,        hasPendingApproval: input.activePendingApproval !== null,        hasPendingUserInput: input.activePendingUserInput !== null,        threadError: input.threadError,      }),    [      input.activeLatestTurn,      input.activePendingApproval,      input.activePendingUserInput,      input.activeThread?.session,      input.phase,      input.threadError,      localDispatch,    ],  );  const activeLocalDispatch = serverAcknowledgedLocalDispatch ? null : localDispatch;  const beginLocalDispatch = useCallback(    (options?: { preparingWorktree?: boolean }) => {      const preparingWorktree = Boolean(options?.preparingWorktree);      setLocalDispatch((current) => {        const active = serverAcknowledgedLocalDispatch ? null : current;        if (active) {          return active.preparingWorktree === preparingWorktree            ? active            : { ...active, preparingWorktree };        }        return createLocalDispatchSnapshot(input.activeThread, options);      });    },    [input.activeThread, serverAcknowledgedLocalDispatch],  );   return {    beginLocalDispatch,    resetLocalDispatch,    localDispatchStartedAt: activeLocalDispatch?.startedAt ?? null,    isPreparingWorktree: activeLocalDispatch?.preparingWorktree ?? false,    isSendBusy: activeLocalDispatch !== null,  };} /** Same terminal ids (order ignored) — avoids reconcile when only server session ordering differs. */function terminalIdListsEqual(left: readonly string[], right: readonly string[]): boolean {  if (left.length !== right.length) {    return false;  }  if (left.length === 0) {    return true;  }  const sortedLeft = left.toSorted((a, b) => a.localeCompare(b));  const sortedRight = right.toSorted((a, b) => a.localeCompare(b));  for (let index = 0; index < sortedLeft.length; index += 1) {    if (sortedLeft[index] !== sortedRight[index]) {      return false;    }  }  return true;} /** * Server knows about fewer sessions than the client, but every server id still exists locally. * Typical right after `terminal.open`: known-session list lags; reconciling would drop the new id * and later re-add it as a separate group (no split layout). */function serverTerminalIdsStrictSubsetOfClient(  serverIds: readonly string[],  clientIds: readonly string[],): boolean {  if (serverIds.length >= clientIds.length || clientIds.length === 0) {    return false;  }  const clientSet = new Set(clientIds);  for (const id of serverIds) {    if (!clientSet.has(id)) {      return false;    }  }  return true;} interface PersistentThreadTerminalDrawerProps {  threadRef: { environmentId: EnvironmentId; threadId: ThreadId };  threadId: ThreadId;  visible: boolean;  launchContext: PersistentTerminalLaunchContext | null;  focusRequestId: number;  splitShortcutLabel: string | undefined;  splitVerticalShortcutLabel: string | undefined;  newShortcutLabel: string | undefined;  closeShortcutLabel: string | undefined;  keybindings: ResolvedKeybindingsConfig;  onAddTerminalContext: (selection: TerminalContextSelection) => void;} const PersistentThreadTerminalDrawer = memo(function PersistentThreadTerminalDrawer({  threadRef,  threadId,  visible,  launchContext,  focusRequestId,  splitShortcutLabel,  splitVerticalShortcutLabel,  newShortcutLabel,  closeShortcutLabel,  keybindings,  onAddTerminalContext,}: PersistentThreadTerminalDrawerProps) {  const openTerminal = useAtomCommand(terminalEnvironment.open, "terminal open");  const writeTerminal = useAtomCommand(terminalEnvironment.write, "terminal write");  const closeTerminalMutation = useAtomCommand(terminalEnvironment.close, "terminal close");  const serverThread = useThread(threadRef);  const draftThread = useComposerDraftStore((store) => store.getDraftThreadByRef(threadRef));  const projectRef = serverThread    ? scopeProjectRef(serverThread.environmentId, serverThread.projectId)    : draftThread      ? scopeProjectRef(draftThread.environmentId, draftThread.projectId)      : null;  const project = useProject(projectRef);  const terminalUiState = useTerminalUiStateStore((state) =>    selectThreadTerminalUiState(state.terminalUiStateByThreadKey, threadRef),  );  const knownTerminalSessions = useKnownTerminalSessions({    environmentId: threadRef.environmentId,    threadId,  });  const panelSurfaces = useRightPanelStore(    (state) => selectThreadRightPanelState(state.byThreadKey, threadRef).surfaces,  );  const panelTerminalIds = useMemo(    () =>      new Set(        panelSurfaces.flatMap((surface) =>

Lines 1–500 of 4992.