feat(code): warn on local task branch mismatch (#1596)

## Problem

we want task isolation, but we can only do so much with local tasks, since they are simply not in an isolated environment

downstack #1594 + #1595 add automatic branch linking to tasks, so when an agent touches a file, or an agent/human creates a branch from posthog code, we associate that branch to that task

now, we need a way to keep the user in the loop on what's happening, and steer them in the right direction

<!-- Who is this for and what problem does it solve? -->

<!-- Closes #ISSUE_ID -->

## Changes

adds a UI warning if you send a prompt in a task that has a linked branch, but that branch is not currently checked out

details:

- hooks into tiptap with a new `onBeforeSend`so prompt is never lost if you cancel
- the warning is "blocking" in the sense that you have to make a decision. you can:
    - "continue anyway" -> ignore the warning, agent works in your current branch
        - automatic branch linking will still occur and maybe overwrite the current link, per the standard rules (see #1595 )
    - "switch branch" - checks out the task's linked branch
        - warns if there are uncommitted changes on current branch. this does not block, and surfaces errors from the switch (if any) in the dialog

<!-- What did you change and why? -->

<!-- If there are frontend changes, include screenshots. -->

## How did you test this?

manually

<!-- Describe what you tested -- manual steps, automated tests, or both. -->

<!-- If you're an agent, only list tests you actually ran. -->

Author: adboio

apps/code/src/main/services/workspace/service.ts

@@ -369,6 +369,8 @@ export class WorkspaceService extends TypedEventEmitter<WorkspaceServiceEvents>
       branchName,
     });
     trackAppEvent("branch_linked", {
+      task_id: taskId,
+      branch_name: branchName,
       source: source ?? "unknown",
     });
     log.info("Linked branch to task", { taskId, branchName, source });
@@ -381,6 +383,7 @@ export class WorkspaceService extends TypedEventEmitter<WorkspaceServiceEvents>
       branchName: null,
     });
     trackAppEvent("branch_unlinked", {
+      task_id: taskId,
       source: source ?? "unknown",
     });
     log.info("Unlinked branch from task", { taskId, source });

apps/code/src/renderer/features/message-editor/components/PromptInput.tsx

@@ -45,6 +45,7 @@ export interface PromptInputProps {
   // prompt history provider
   getPromptHistory?: () => string[];
   // callbacks
+  onBeforeSubmit?: (text: string, clearEditor: () => void) => boolean;
   onSubmit?: (text: string) => void;
   onBashCommand?: (command: string) => void;
   onBashModeChange?: (isBashMode: boolean) => void;
@@ -80,6 +81,7 @@ export const PromptInput = forwardRef<EditorHandle, PromptInputProps>(
       reasoningSelector,
       tourHighlightSubmit = false,
       getPromptHistory,
+      onBeforeSubmit,
       onSubmit,
       onBashCommand,
       onBashModeChange,
@@ -127,6 +129,7 @@ export const PromptInput = forwardRef<EditorHandle, PromptInputProps>(
         commands: enableCommands,
       },
       getPromptHistory,
+      onBeforeSubmit,
       onSubmit,
       onBashCommand,
       onBashModeChange,

apps/code/src/renderer/features/message-editor/tiptap/useTiptapEditor.ts

@@ -30,6 +30,7 @@ export interface UseTiptapEditorOptions {
   };
   clearOnSubmit?: boolean;
   getPromptHistory?: () => string[];
+  onBeforeSubmit?: (text: string, clearEditor: () => void) => boolean;
   onSubmit?: (text: string) => void;
   onBashCommand?: (command: string) => void;
   onBashModeChange?: (isBashMode: boolean) => void;
@@ -85,6 +86,7 @@ export function useTiptapEditor(options: UseTiptapEditorOptions) {
     capabilities = {},
     clearOnSubmit = true,
     getPromptHistory,
+    onBeforeSubmit,
     onSubmit,
     onBashCommand,
     onBashModeChange,
@@ -100,6 +102,7 @@ export function useTiptapEditor(options: UseTiptapEditorOptions) {
   } = capabilities;
 
   const callbackRefs = useRef({
+    onBeforeSubmit,
     onSubmit,
     onBashCommand,
     onBashModeChange,
@@ -108,6 +111,7 @@ export function useTiptapEditor(options: UseTiptapEditorOptions) {
     onBlur,
   });
   callbackRefs.current = {
+    onBeforeSubmit,
     onSubmit,
     onBashCommand,
     onBashModeChange,
@@ -450,26 +454,39 @@ export function useTiptapEditor(options: UseTiptapEditorOptions) {
 
     const text = editor.getText().trim();
 
+    const doClear = () => {
+      if (!clearOnSubmit) return;
+      editor.commands.clearContent();
+      prevBashModeRef.current = false;
+      pasteCountRef.current = 0;
+      setAttachments([]);
+      draft.clearDraft();
+    };
+
     if (enableBashMode && text.startsWith("!")) {
-      // Bash mode requires immediate execution, can't be queued
+      // Bash mode requires immediate execution, can't be queued.
+      // Intentionally bypasses onBeforeSubmit — bash commands run inline and
+      // cannot be deferred the way normal prompts can.
       if (isLoading) {
         toast.error("Cannot run shell commands while agent is generating");
         return;
       }
       const command = text.slice(1).trim();
       if (command) callbackRefs.current.onBashCommand?.(command);
     } else {
+      const serialized = contentToXml(content);
+
+      if (callbackRefs.current.onBeforeSubmit) {
+        if (!callbackRefs.current.onBeforeSubmit(serialized, doClear)) {
+          return;
+        }
+      }
+
       // Normal prompts can be queued when loading
-      callbackRefs.current.onSubmit?.(contentToXml(content));
+      callbackRefs.current.onSubmit?.(serialized);
     }
 
-    if (clearOnSubmit) {
-      editor.commands.clearContent();
-      prevBashModeRef.current = false;
-      pasteCountRef.current = 0;
-      setAttachments([]);
-      draft.clearDraft();
-    }
+    doClear();
   }, [
     editor,
     disabled,

apps/code/src/renderer/features/sessions/components/SessionView.tsx

@@ -39,6 +39,7 @@ interface SessionViewProps {
   isRunning: boolean;
   isPromptPending?: boolean | null;
   promptStartedAt?: number | null;
+  onBeforeSubmit?: (text: string, clearEditor: () => void) => boolean;
   onSendPrompt: (text: string) => void;
   onBashCommand?: (command: string) => void;
   onCancelPrompt: () => void;
@@ -69,6 +70,7 @@ export function SessionView({
   isRunning,
   isPromptPending = false,
   promptStartedAt,
+  onBeforeSubmit,
   onSendPrompt,
   onBashCommand,
   onCancelPrompt,
@@ -531,6 +533,7 @@ export function SessionView({
                             disabled={!isRunning}
                           />
                         }
+                        onBeforeSubmit={onBeforeSubmit}
                         onSubmit={handleSubmit}
                         onBashCommand={onBashCommand}
                         onCancel={onCancelPrompt}

apps/code/src/renderer/features/task-detail/components/BranchMismatchDialog.tsx

@@ -0,0 +1,148 @@
+import { GitBranch, Warning } from "@phosphor-icons/react";
+import {
+  AlertDialog,
+  Button,
+  Callout,
+  Code,
+  Flex,
+  Text,
+} from "@radix-ui/themes";
+
+interface BranchMismatchDialogProps {
+  open: boolean;
+  linkedBranch: string;
+  currentBranch: string;
+  hasUncommittedChanges: boolean;
+  switchError: string | null;
+  onSwitch: () => void;
+  onContinue: () => void;
+  onCancel: () => void;
+  isSwitching?: boolean;
+}
+
+function BranchLabel({ name }: { name: string }) {
+  return (
+    <Code
+      size="2"
+      variant="ghost"
+      truncate
+      style={{
+        maxWidth: "100%",
+        display: "inline-flex",
+        alignItems: "center",
+        gap: "4px",
+      }}
+    >
+      <GitBranch size={12} style={{ flexShrink: 0 }} />
+      <span
+        style={{
+          overflow: "hidden",
+          textOverflow: "ellipsis",
+          whiteSpace: "nowrap",
+        }}
+      >
+        {name}
+      </span>
+    </Code>
+  );
+}
+
+export function BranchMismatchDialog({
+  open,
+  linkedBranch,
+  currentBranch,
+  hasUncommittedChanges,
+  switchError,
+  onSwitch,
+  onContinue,
+  onCancel,
+  isSwitching,
+}: BranchMismatchDialogProps) {
+  return (
+    <AlertDialog.Root
+      open={open}
+      onOpenChange={(isOpen) => {
+        if (!isOpen) onCancel();
+      }}
+    >
+      <AlertDialog.Content maxWidth="420px" size="2">
+        <AlertDialog.Title size="3">
+          <Flex align="center" gap="2">
+            <Warning size={18} weight="fill" color="var(--orange-9)" />
+            Wrong branch
+          </Flex>
+        </AlertDialog.Title>
+        <AlertDialog.Description size="2">
+          This task is linked to a different branch than the one you're
+          currently on. The agent will make changes on the current branch.
+        </AlertDialog.Description>
+        <Flex direction="column" gap="1" mt="3" style={{ minWidth: 0 }}>
+          <Flex align="center" gap="2" style={{ minWidth: 0 }}>
+            <Text
+              size="1"
+              color="gray"
+              style={{ flexShrink: 0, width: "64px" }}
+            >
+              Linked
+            </Text>
+            <BranchLabel name={linkedBranch} />
+          </Flex>
+          <Flex align="center" gap="2" style={{ minWidth: 0 }}>
+            <Text
+              size="1"
+              color="gray"
+              style={{ flexShrink: 0, width: "64px" }}
+            >
+              Current
+            </Text>
+            <BranchLabel name={currentBranch} />
+          </Flex>
+        </Flex>
+
+        {hasUncommittedChanges && !switchError && (
+          <Callout.Root size="1" color="gray" mt="3">
+            <Callout.Text size="1">
+              You have uncommitted changes on your current branch. If needed,
+              commit or stash them first.
+            </Callout.Text>
+          </Callout.Root>
+        )}
+
+        {switchError && (
+          <Callout.Root size="1" color="red" mt="3">
+            <Callout.Text size="1">{switchError}</Callout.Text>
+          </Callout.Root>
+        )}
+
+        <Flex justify="end" gap="2" mt="4">
+          <AlertDialog.Cancel>
+            <Button variant="soft" color="gray" size="1" disabled={isSwitching}>
+              Cancel
+            </Button>
+          </AlertDialog.Cancel>
+
+          <Button
+            variant="soft"
+            color="orange"
+            size="1"
+            onClick={onContinue}
+            disabled={isSwitching}
+          >
+            Continue anyway
+          </Button>
+
+          <AlertDialog.Action>
+            <Button
+              variant="solid"
+              size="1"
+              onClick={onSwitch}
+              loading={isSwitching}
+            >
+              Switch branch
+            </Button>
+          </AlertDialog.Action>
+        </Flex>
+      </AlertDialog.Content>
+    </AlertDialog.Root>
+  );
+}

apps/code/src/renderer/features/task-detail/components/TaskLogsPanel.tsx

@@ -10,7 +10,9 @@ import { useSessionConnection } from "@features/sessions/hooks/useSessionConnect
 import { useSessionViewState } from "@features/sessions/hooks/useSessionViewState";
 import { useRestoreTask } from "@features/suspension/hooks/useRestoreTask";
 import { useSuspendedTaskIds } from "@features/suspension/hooks/useSuspendedTaskIds";
+import { BranchMismatchDialog } from "@features/task-detail/components/BranchMismatchDialog";
 import { WorkspaceSetupPrompt } from "@features/task-detail/components/WorkspaceSetupPrompt";
+import { useBranchMismatchDialog } from "@features/workspace/hooks/useBranchMismatchDialog";
 import {
   useCreateWorkspace,
   useWorkspaceLoaded,
@@ -76,6 +78,12 @@ export function TaskLogsPanel({ taskId, task, hideInput }: TaskLogsPanelProps) {
     handleBashCommand,
   } = useSessionCallbacks({ taskId, task, session, repoPath });
 
+  const { handleBeforeSubmit, dialogProps } = useBranchMismatchDialog({
+    taskId,
+    repoPath,
+    onSendPrompt: handleSendPrompt,
+  });
+
   const slackThreadUrl =
     typeof task.latest_run?.state?.slack_thread_url === "string"
       ? task.latest_run.state.slack_thread_url
@@ -126,6 +134,7 @@ export function TaskLogsPanel({ taskId, task, hideInput }: TaskLogsPanelProps) {
               isRestoring={isRestoring}
               isPromptPending={isPromptPending}
               promptStartedAt={promptStartedAt}
+              onBeforeSubmit={handleBeforeSubmit}
               onSendPrompt={handleSendPrompt}
               onBashCommand={isCloud ? undefined : handleBashCommand}
               onCancelPrompt={handleCancelPrompt}
@@ -143,6 +152,8 @@ export function TaskLogsPanel({ taskId, task, hideInput }: TaskLogsPanelProps) {
           </ErrorBoundary>
         </Box>
       </Flex>
+
+      {dialogProps && <BranchMismatchDialog {...dialogProps} />}
     </BackgroundWrapper>
   );
 }