Auto-focus editor on typing and global Shift+Tab mode cycling

Author: charlesvien

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

@@ -20,6 +20,7 @@ import {
   isJsonRpcNotification,
   isJsonRpcResponse,
 } from "@shared/types/session-events";
+import { useAutoFocusOnTyping } from "@hooks/useAutoFocusOnTyping";
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { useHotkeys } from "react-hotkeys-hook";
 import { getSessionService } from "../service/service";
@@ -336,6 +337,8 @@ export function SessionView({
     editorRef.current?.focus();
   }, []);
 
+  useAutoFocusOnTyping(editorRef);
+
   return (
     <ContextMenu.Root>
       <ContextMenu.Trigger>

apps/twig/src/renderer/features/task-detail/components/TaskInput.tsx

@@ -10,12 +10,14 @@ import {
 } from "@features/sessions/stores/sessionStore";
 import type { AgentAdapter } from "@features/settings/stores/settingsStore";
 import { useSettingsStore } from "@features/settings/stores/settingsStore";
+import { useAutoFocusOnTyping } from "@hooks/useAutoFocusOnTyping";
 import { useRepositoryIntegration } from "@hooks/useIntegrations";
 import { Flex } from "@radix-ui/themes";
 import { useRegisteredFoldersStore } from "@renderer/stores/registeredFoldersStore";
 import { useNavigationStore } from "@stores/navigationStore";
 import { useTaskDirectoryStore } from "@stores/taskDirectoryStore";
 import { useCallback, useEffect, useRef, useState } from "react";
+import { useHotkeys } from "react-hotkeys-hook";
 import { usePreviewSession } from "../hooks/usePreviewSession";
 import { useTaskCreation } from "../hooks/useTaskCreation";
 import { TaskInputEditor } from "./TaskInputEditor";
@@ -107,6 +109,23 @@ export function TaskInput() {
     }
   }, [modeOption, allowBypassPermissions, previewTaskId]);
 
+  // Global shift+tab to cycle mode regardless of focus
+  useHotkeys(
+    "shift+tab",
+    (e) => {
+      e.preventDefault();
+      handleCycleMode();
+    },
+    {
+      enableOnFormTags: true,
+      enableOnContentEditable: true,
+      enabled: !!modeOption,
+    },
+    [handleCycleMode, modeOption],
+  );
+
+  useAutoFocusOnTyping(editorRef, isCreatingTask);
+
   const handleDragEnter = useCallback((e: React.DragEvent) => {
     e.preventDefault();
     e.stopPropagation();
@@ -253,7 +272,6 @@ export function TaskInput() {
             onEmptyChange={setEditorIsEmpty}
             adapter={adapter}
             previewTaskId={previewTaskId}
-            onCycleMode={handleCycleMode}
             onAdapterChange={setAdapter}
             isPreviewConnecting={isConnecting}
           />

apps/twig/src/renderer/features/task-detail/components/TaskInputEditor.tsx

@@ -25,7 +25,6 @@ interface TaskInputEditorProps {
   onEmptyChange?: (isEmpty: boolean) => void;
   adapter?: "claude" | "codex";
   previewTaskId?: string;
-  onCycleMode?: () => void;
   onAdapterChange?: (adapter: AgentAdapter) => void;
   isPreviewConnecting?: boolean;
 }
@@ -45,7 +44,6 @@ export const TaskInputEditor = forwardRef<
       onEmptyChange,
       adapter,
       previewTaskId,
-      onCycleMode,
       onAdapterChange,
       isPreviewConnecting,
     },
@@ -148,13 +146,6 @@ export const TaskInputEditor = forwardRef<
               focus();
             }
           }}
-          onKeyDown={(e) => {
-            if (e.key === "Tab" && e.shiftKey && onCycleMode) {
-              e.preventDefault();
-              e.stopPropagation();
-              onCycleMode();
-            }
-          }}
         >
           <AttachmentsBar
             attachments={attachments}

apps/twig/src/renderer/hooks/useAutoFocusOnTyping.ts

@@ -0,0 +1,29 @@
+import type { MessageEditorHandle } from "@features/message-editor/components/MessageEditor";
+import { type RefObject, useEffect } from "react";
+
+export function useAutoFocusOnTyping(
+  editorRef: RefObject<MessageEditorHandle | null>,
+  disabled = false,
+) {
+  useEffect(() => {
+    const handleKeyDown = (e: KeyboardEvent) => {
+      if (disabled) return;
+
+      const activeEl = document.activeElement;
+      const isInInput =
+        activeEl &&
+        (activeEl.tagName === "INPUT" ||
+          activeEl.tagName === "TEXTAREA" ||
+          activeEl.tagName === "SELECT" ||
+          activeEl.getAttribute("contenteditable") === "true");
+      if (isInInput) return;
+
+      if (e.key.length > 1 || e.metaKey || e.ctrlKey || e.altKey) return;
+
+      editorRef.current?.focus();
+    };
+
+    document.addEventListener("keydown", handleKeyDown);
+    return () => document.removeEventListener("keydown", handleKeyDown);
+  }, [editorRef, disabled]);
+}