Handle non-git drafts in local mode

Author: SAKETH11111

apps/web/src/components/BranchToolbar.logic.test.ts

@@ -138,6 +138,17 @@ describe("resolveEffectiveEnvMode", () => {
       }),
     ).toBe("worktree");
   });
+
+  it("falls back to local mode for non-git projects even if the draft prefers worktree mode", () => {
+    expect(
+      resolveEffectiveEnvMode({
+        activeWorktreePath: null,
+        hasServerThread: false,
+        draftThreadEnvMode: "worktree",
+        isGitRepo: false,
+      }),
+    ).toBe("local");
+  });
 });
 
 describe("resolveEnvModeLabel", () => {

apps/web/src/components/BranchToolbar.logic.ts

@@ -54,8 +54,12 @@ export function resolveEffectiveEnvMode(input: {
   activeWorktreePath: string | null;
   hasServerThread: boolean;
   draftThreadEnvMode: EnvMode | undefined;
+  isGitRepo?: boolean;
 }): EnvMode {
-  const { activeWorktreePath, hasServerThread, draftThreadEnvMode } = input;
+  const { activeWorktreePath, hasServerThread, draftThreadEnvMode, isGitRepo = true } = input;
+  if (!isGitRepo) {
+    return "local";
+  }
   if (!hasServerThread) {
     if (activeWorktreePath) {
       return "local";

apps/web/src/components/ChatView.browser.tsx

@@ -49,8 +49,19 @@ import { BrowserWsRpcHarness, type NormalizedWsRpcRequestBody } from "../../test
 import { estimateTimelineMessageHeight } from "./timelineHeight";
 import { DEFAULT_CLIENT_SETTINGS } from "@t3tools/contracts/settings";
 
+const { gitStatusStateRef } = vi.hoisted(() => ({
+  gitStatusStateRef: {
+    current: { data: null, error: null, cause: null, isPending: false } as {
+      data: unknown;
+      error: unknown;
+      cause: unknown;
+      isPending: boolean;
+    },
+  },
+}));
+
 vi.mock("../lib/gitStatusState", () => ({
-  useGitStatus: () => ({ data: null, error: null, cause: null, isPending: false }),
+  useGitStatus: () => gitStatusStateRef.current,
   useGitStatuses: () => new Map(),
   refreshGitStatus: () => Promise.resolve(null),
   resetGitStatusStateForTests: () => undefined,
@@ -1458,6 +1469,7 @@ describe("ChatView timeline estimator parity (full app)", () => {
     document.body.innerHTML = "";
     wsRequests.length = 0;
     customWsRpcResolver = null;
+    gitStatusStateRef.current = { data: null, error: null, cause: null, isPending: false };
     useComposerDraftStore.setState({
       draftsByThreadKey: {},
       draftThreadsByThreadKey: {},
@@ -2448,6 +2460,100 @@ describe("ChatView timeline estimator parity (full app)", () => {
     }
   });
 
+  it("falls back to local send behavior for non-git drafts even when worktree mode is persisted", async () => {
+    gitStatusStateRef.current = {
+      data: {
+        isRepo: false,
+        hasOriginRemote: false,
+        isDefaultBranch: false,
+        branch: null,
+        hasWorkingTreeChanges: false,
+        workingTree: { files: [], insertions: 0, deletions: 0 },
+        hasUpstream: false,
+        aheadCount: 0,
+        behindCount: 0,
+        pr: null,
+      },
+      error: null,
+      cause: null,
+      isPending: false,
+    };
+    useComposerDraftStore.setState({
+      draftThreadsByThreadKey: {
+        [THREAD_KEY]: {
+          threadId: THREAD_ID,
+          environmentId: LOCAL_ENVIRONMENT_ID,
+          projectId: PROJECT_ID,
+          logicalProjectKey: PROJECT_DRAFT_KEY,
+          createdAt: NOW_ISO,
+          runtimeMode: "full-access",
+          interactionMode: "default",
+          branch: null,
+          worktreePath: null,
+          envMode: "worktree",
+        },
+      },
+      logicalProjectDraftThreadKeyByLogicalProjectKey: {
+        [PROJECT_DRAFT_KEY]: THREAD_KEY,
+      },
+    });
+
+    const mounted = await mountChatView({
+      viewport: DEFAULT_VIEWPORT,
+      snapshot: createDraftOnlySnapshot(),
+      resolveRpc: (body) => {
+        if (body._tag === ORCHESTRATION_WS_METHODS.dispatchCommand) {
+          return {
+            sequence: fixture.snapshot.snapshotSequence + 1,
+          };
+        }
+        return undefined;
+      },
+    });
+
+    try {
+      useComposerDraftStore.getState().setPrompt(THREAD_REF, "Ship it");
+      await waitForLayout();
+
+      const sendButton = await waitForSendButton();
+      expect(sendButton.disabled).toBe(false);
+      sendButton.click();
+
+      await vi.waitFor(
+        () => {
+          const dispatchRequest = wsRequests.find(
+            (request) => request._tag === ORCHESTRATION_WS_METHODS.dispatchCommand,
+          ) as
+            | {
+                _tag: string;
+                bootstrap?: {
+                  createThread?: { projectId?: string };
+                  prepareWorktree?: unknown;
+                };
+              }
+            | undefined;
+          expect(dispatchRequest).toMatchObject({
+            _tag: ORCHESTRATION_WS_METHODS.dispatchCommand,
+            bootstrap: {
+              createThread: {
+                projectId: PROJECT_ID,
+              },
+            },
+          });
+          expect(dispatchRequest?.bootstrap?.prepareWorktree).toBeUndefined();
+          expect(document.querySelector('button[aria-label="Preparing worktree"]')).toBeNull();
+        },
+        { timeout: 8_000, interval: 16 },
+      );
+
+      expect(useComposerDraftStore.getState().getDraftThread(THREAD_REF)?.envMode ?? null).toBe(
+        "local",
+      );
+    } finally {
+      await mounted.cleanup();
+    }
+  });
+
   it("toggles plan mode with Shift+Tab only while the composer is focused", async () => {
     const mounted = await mountChatView({
       viewport: DEFAULT_VIEWPORT,

apps/web/src/components/ChatView.tsx

@@ -2196,7 +2196,38 @@ export default function ChatView(props: ChatViewProps) {
     activeWorktreePath,
     hasServerThread: isServerThread,
     draftThreadEnvMode: isLocalDraftThread ? draftThread?.envMode : undefined,
+    isGitRepo,
   });
+  const draftThreadEnvMode = draftThread?.envMode;
+  const draftThreadBranch = draftThread?.branch;
+
+  useEffect(() => {
+    if (!isLocalDraftThread) {
+      return;
+    }
+    if (!draftThread) {
+      return;
+    }
+    if (gitStatusQuery.data?.isRepo !== false) {
+      return;
+    }
+    if (draftThreadEnvMode !== "worktree" && draftThreadBranch === null) {
+      return;
+    }
+    setDraftThreadContext(composerDraftTarget, {
+      envMode: "local",
+      branch: null,
+      worktreePath: null,
+    });
+  }, [
+    composerDraftTarget,
+    draftThread,
+    draftThreadBranch,
+    draftThreadEnvMode,
+    gitStatusQuery.data?.isRepo,
+    isLocalDraftThread,
+    setDraftThreadContext,
+  ]);
 
   useEffect(() => {
     if (!activeThreadId) {