Allow making a toolset the default for plan mode (#238)

* Allow making a toolset the default for plan mode

Closes #187

* Reorder the indicators

* Format

* Changelog

Author: bl-ue

CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+- Allow making a toolset the default for plan mode (#238) - @bl-ue
+
 ## [v3.1.3](https://github.com/Piebald-AI/tweakcc/releases/tag/v3.1.3) - 2025-11-26
 
 - Add paths for mise npm backend (#234) - @coryzibell

src/components/ToolsetEditView.tsx

@@ -47,8 +47,19 @@ export function ToolsetEditView({
   useEffect(() => {
     if (toolset) {
       updateSettings(settings => {
+        const oldName = settings.toolsets[toolsetIndex].name;
         settings.toolsets[toolsetIndex].name = name;
         settings.toolsets[toolsetIndex].allowedTools = allowedTools;
+
+        // Update references if name changed
+        if (oldName !== name) {
+          if (settings.defaultToolset === oldName) {
+            settings.defaultToolset = name;
+          }
+          if (settings.planModeToolset === oldName) {
+            settings.planModeToolset = name;
+          }
+        }
       });
     }
   }, [name, allowedTools]);

src/components/ToolsetsView.tsx

@@ -1,20 +1,31 @@
 import React, { useState, useContext } from 'react';
 import { Box, Text, useInput } from 'ink';
-import { Toolset } from '../utils/types.js';
+import { Toolset, DEFAULT_SETTINGS } from '../utils/types.js';
 import { SettingsContext } from '../App.js';
 import { ToolsetEditView } from './ToolsetEditView.js';
 import Header from './Header.js';
+import { getCurrentClaudeCodeTheme } from '../utils/misc.js';
 
 interface ToolsetsViewProps {
   onBack: () => void;
 }
 
 export function ToolsetsView({ onBack }: ToolsetsViewProps) {
   const {
-    settings: { toolsets, defaultToolset },
+    settings: { toolsets, defaultToolset, planModeToolset, themes },
     updateSettings,
   } = useContext(SettingsContext);
 
+  // Get current theme colors
+  const currentThemeId = getCurrentClaudeCodeTheme();
+  const currentTheme = themes.find(t => t.id === currentThemeId) || themes[0];
+
+  const defaultTheme = DEFAULT_SETTINGS.themes[0]; // Dark mode theme
+  const planModeColor =
+    currentTheme?.colors.planMode || defaultTheme.colors.planMode;
+  const autoAcceptColor =
+    currentTheme?.colors.autoAccept || defaultTheme.colors.autoAccept;
+
   const [selectedIndex, setSelectedIndex] = useState(0);
   const [editingToolsetIndex, setEditingToolsetIndex] = useState<number | null>(
     null
@@ -43,6 +54,10 @@ export function ToolsetsView({ onBack }: ToolsetsViewProps) {
       if (settings.defaultToolset === toolsetToDelete.name) {
         settings.defaultToolset = null;
       }
+      // Clear plan mode if we're deleting the plan mode toolset
+      if (settings.planModeToolset === toolsetToDelete.name) {
+        settings.planModeToolset = null;
+      }
     });
 
     if (selectedIndex >= toolsets.length - 1) {
@@ -57,6 +72,13 @@ export function ToolsetsView({ onBack }: ToolsetsViewProps) {
     });
   };
 
+  const handleSetPlanModeToolset = (index: number) => {
+    const toolset = toolsets[index];
+    updateSettings(settings => {
+      settings.planModeToolset = toolset.name;
+    });
+  };
+
   useInput(
     (input, key) => {
       if (key.escape) {
@@ -70,10 +92,12 @@ export function ToolsetsView({ onBack }: ToolsetsViewProps) {
         setInputActive(false);
       } else if (input === 'n') {
         handleCreateToolset();
-      } else if (input === 'd' && toolsets.length > 0) {
+      } else if (input === 'x' && toolsets.length > 0) {
         handleDeleteToolset(selectedIndex);
-      } else if (input === 's' && toolsets.length > 0) {
+      } else if (input === 'd' && toolsets.length > 0) {
         handleSetDefaultToolset(selectedIndex);
+      } else if (input === 'p' && toolsets.length > 0) {
+        handleSetPlanModeToolset(selectedIndex);
       }
     },
     { isActive: inputActive }
@@ -108,9 +132,12 @@ export function ToolsetsView({ onBack }: ToolsetsViewProps) {
       <Box marginBottom={1} flexDirection="column">
         <Text dimColor>n to create a new toolset</Text>
         {toolsets.length > 0 && (
-          <Text dimColor>s to set as default toolset</Text>
+          <Text dimColor>d to set as default toolset</Text>
+        )}
+        {toolsets.length > 0 && (
+          <Text dimColor>p to set as plan mode toolset</Text>
         )}
-        {toolsets.length > 0 && <Text dimColor>d to delete a toolset</Text>}
+        {toolsets.length > 0 && <Text dimColor>x to delete a toolset</Text>}
         {toolsets.length > 0 && <Text dimColor>enter to edit toolset</Text>}
         <Text dimColor>esc to go back</Text>
       </Box>
@@ -121,17 +148,31 @@ export function ToolsetsView({ onBack }: ToolsetsViewProps) {
         <Box flexDirection="column">
           {toolsets.map((toolset, index) => {
             const isDefault = toolset.name === defaultToolset;
+            const isPlanMode = toolset.name === planModeToolset;
             const isSelected = selectedIndex === index;
+
+            // Determine the color for the entire line
+            let lineColor: string | undefined = undefined;
+            if (isSelected) {
+              lineColor = 'yellow';
+            }
+
             return (
-              <Text
-                key={index}
-                color={isSelected ? 'yellow' : undefined}
-                bold={isDefault}
-              >
-                {isSelected ? '❯ ' : '  '}
-                {toolset.name}
-                {isDefault && ' [DEFAULT]'} ({getToolsetDescription(toolset)})
-              </Text>
+              <Box key={index} flexDirection="row">
+                <Text color={lineColor}>
+                  {isSelected ? '❯ ' : '  '}
+                  {toolset.name}{' '}
+                </Text>
+
+                <Text color={lineColor}>
+                  ({getToolsetDescription(toolset)})
+                </Text>
+
+                {isDefault && (
+                  <Text color={autoAcceptColor}> ⏵⏵ accept edits</Text>
+                )}
+                {isPlanMode && <Text color={planModeColor}> ⏸ plan mode</Text>}
+              </Box>
             );
           })}
         </Box>

src/utils/config.ts

@@ -104,6 +104,9 @@ export const readConfigFile = async (): Promise<TweakccConfig> => {
     if (!Object.hasOwn(readConfig.settings, 'defaultToolset')) {
       readConfig.settings.defaultToolset = DEFAULT_SETTINGS.defaultToolset;
     }
+    if (!Object.hasOwn(readConfig.settings, 'planModeToolset')) {
+      readConfig.settings.planModeToolset = DEFAULT_SETTINGS.planModeToolset;
+    }
 
     // Add any colors that the user doesn't have to any built-in themes.
     for (const defaultTheme of DEFAULT_SETTINGS.themes) {

src/utils/patches/index.ts

@@ -50,7 +50,11 @@ import { writeThinkingVisibility } from './thinkingVisibility.js';
 import { writePatchesAppliedIndication } from './patchesAppliedIndication.js';
 import { applySystemPrompts } from './systemPrompts.js';
 import { writeFixLspSupport } from './fixLspSupport.js';
-import { writeToolsets } from './toolsets.js';
+import {
+  writeToolsets,
+  writeModeChangeUpdateToolset,
+  addSetStateFnAccessAtToolChangeComponentScope,
+} from './toolsets.js';
 import { writeConversationTitle } from './conversationTitle.js';
 
 export interface LocationResult {
@@ -612,6 +616,23 @@ export const applyCustomization = async (
       content = result;
   }
 
+  // Apply mode-change toolset switching (if both toolsets are configured)
+  if (config.settings.planModeToolset && config.settings.defaultToolset) {
+    // First, add setState access at the tool change component scope
+    if ((result = addSetStateFnAccessAtToolChangeComponentScope(content)))
+      content = result;
+
+    // Then, inject the mode change toolset switching code
+    if (
+      (result = writeModeChangeUpdateToolset(
+        content,
+        config.settings.planModeToolset,
+        config.settings.defaultToolset
+      ))
+    )
+      content = result;
+  }
+
   // Apply conversation title management (always enabled)
   if ((result = writeConversationTitle(content))) content = result;
 

src/utils/patches/toolsets.ts

@@ -739,6 +739,114 @@ export const writeSlashCommandDefinition = (oldFile: string): string | null => {
   return writeSlashCommandDefinitionToArray(oldFile, commandDef);
 };
 
+// ============================================================================
+// MODE CHANGE TOOLSET FUNCTIONS
+// ============================================================================
+
+/**
+ * Find the tool change component scope
+ * Pattern: X(Y,function(Z){W("tengu_ext_at_mentioned",{});
+ * Returns the start index
+ */
+export const findToolChangeComponentScope = (
+  fileContents: string
+): number | null => {
+  const pattern =
+    /[\w$]+\([\w$]+,function\([\w$]+\)\{[\w$]+\("tengu_ext_at_mentioned",\{\}\);/;
+  const match = fileContents.match(pattern);
+
+  if (!match || match.index === undefined) {
+    console.error(
+      'patch: findToolChangeComponentScope: failed to find tool change component scope'
+    );
+    return null;
+  }
+
+  return match.index;
+};
+
+/**
+ * Add setState function access at the tool change component scope
+ * Injects: const [state, setState] = appStateGetterFn();
+ */
+export const addSetStateFnAccessAtToolChangeComponentScope = (
+  oldFile: string
+): string | null => {
+  const scopeIndex = findToolChangeComponentScope(oldFile);
+  if (scopeIndex === null) {
+    return null;
+  }
+
+  const stateInfo = getAppStateVarAndGetterFunction(oldFile);
+  if (!stateInfo) {
+    console.error(
+      'patch: addSetStateFnAccessAtToolChangeComponentScope: failed to get app state getter function'
+    );
+    return null;
+  }
+
+  const { appStateGetterFunction } = stateInfo;
+
+  // Inject the setState access right at the start of the component scope
+  const injectionCode = `const [state, setState] = ${appStateGetterFunction}();`;
+
+  const newFile =
+    oldFile.slice(0, scopeIndex) + injectionCode + oldFile.slice(scopeIndex);
+
+  return newFile;
+};
+
+/**
+ * Find the mode change location in the code
+ * Pattern: let X=Y(Z,{type:"setMode",mode:W,destination:"session"});
+ * Returns the start index and the mode variable (W)
+ */
+export const findModeChange = (
+  fileContents: string
+): { index: number; modeVar: string } | null => {
+  const pattern =
+    /let [\w$]+=[\w$]+\([\w$]+,\{type:"setMode",mode:([\w$]+),destination:"session"\}\);/;
+  const match = fileContents.match(pattern);
+
+  if (!match || match.index === undefined) {
+    console.error('patch: findModeChange: failed to find mode change location');
+    return null;
+  }
+
+  return {
+    index: match.index,
+    modeVar: match[1],
+  };
+};
+
+/**
+ * Write the mode change toolset update code
+ * This injects code before the mode change to automatically switch toolsets
+ */
+export const writeModeChangeUpdateToolset = (
+  oldFile: string,
+  planModeToolset: string,
+  defaultToolset: string
+): string | null => {
+  const modeChangeResult = findModeChange(oldFile);
+  if (!modeChangeResult) {
+    return null;
+  }
+
+  const { index: modeChangeIndex, modeVar } = modeChangeResult;
+
+  // Build the injection code using setState directly
+  const injectionCode = `if(${modeVar}==="plan"){setState((prev)=>({...prev,toolset:${JSON.stringify(planModeToolset)}}));}else{setState((prev)=>({...prev,toolset:${JSON.stringify(defaultToolset)}}));}`;
+
+  // Inject right before the mode change
+  const newFile =
+    oldFile.slice(0, modeChangeIndex) +
+    injectionCode +
+    oldFile.slice(modeChangeIndex);
+
+  return newFile;
+};
+
 // ============================================================================
 // MAIN ORCHESTRATOR
 // ============================================================================

src/utils/types.ts

@@ -118,6 +118,7 @@ export interface Settings {
   misc: MiscConfig;
   toolsets: Toolset[];
   defaultToolset: string | null;
+  planModeToolset: string | null;
 }
 
 export interface TweakccConfig {
@@ -910,6 +911,7 @@ export const DEFAULT_SETTINGS: Settings = {
   },
   toolsets: [],
   defaultToolset: null,
+  planModeToolset: null,
 };
 
 // Support XDG Base Directory Specification with backward compatibility