Make mode preference sticky!

Author: jahooma

cli/src/state/chat-store.ts

@@ -3,6 +3,7 @@ import { create } from 'zustand'
 import { immer } from 'zustand/middleware/immer'
 
 import { clamp } from '../utils/math'
+import { loadModePreference, saveModePreference } from '../utils/settings'
 
 import type { ChatMessage } from '../types/chat'
 import type { AgentMode } from '../utils/constants'
@@ -87,7 +88,7 @@ const initialState: ChatStoreState = {
   isChainInProgress: false,
   slashSelectedIndex: 0,
   agentSelectedIndex: 0,
-  agentMode: 'DEFAULT',
+  agentMode: loadModePreference(),
   hasReceivedPlanResponse: false,
   lastMessageMode: null,
   sessionCreditsUsed: 0,
@@ -171,6 +172,7 @@ export const useChatStore = create<ChatStore>()(
     setAgentMode: (mode) =>
       set((state) => {
         state.agentMode = mode
+        saveModePreference(mode)
       }),
 
     toggleAgentMode: () =>
@@ -182,6 +184,7 @@ export const useChatStore = create<ChatStore>()(
         } else {
           state.agentMode = 'DEFAULT'
         }
+        saveModePreference(state.agentMode)
       }),
 
     setHasReceivedPlanResponse: (value) =>

cli/src/utils/settings.ts

@@ -0,0 +1,117 @@
+import fs from 'fs'
+import path from 'path'
+
+import { getConfigDir } from './auth'
+import { logger } from './logger'
+
+import type { AgentMode } from './constants'
+
+/**
+ * Settings schema - add new settings here as the product evolves
+ */
+export interface Settings {
+  mode?: AgentMode
+  // Add new settings here over time
+}
+
+const VALID_MODES: AgentMode[] = ['DEFAULT', 'MAX', 'PLAN']
+
+/**
+ * Get the settings file path
+ */
+export const getSettingsPath = (): string => {
+  return path.join(getConfigDir(), 'settings.json')
+}
+
+/**
+ * Load all settings from file system
+ * @returns The saved settings object, with defaults for missing values
+ */
+export const loadSettings = (): Settings => {
+  const settingsPath = getSettingsPath()
+
+  if (!fs.existsSync(settingsPath)) {
+    return {}
+  }
+
+  try {
+    const settingsFile = fs.readFileSync(settingsPath, 'utf8')
+    const parsed = JSON.parse(settingsFile)
+    return validateSettings(parsed)
+  } catch (error) {
+    logger.debug(
+      {
+        error: error instanceof Error ? error.message : String(error),
+      },
+      'Error reading settings',
+    )
+    return {}
+  }
+}
+
+/**
+ * Validate and sanitize settings from file
+ */
+const validateSettings = (parsed: unknown): Settings => {
+  if (typeof parsed !== 'object' || parsed === null) {
+    return {}
+  }
+
+  const settings: Settings = {}
+  const obj = parsed as Record<string, unknown>
+
+  // Validate mode
+  if (
+    typeof obj.mode === 'string' &&
+    VALID_MODES.includes(obj.mode as AgentMode)
+  ) {
+    settings.mode = obj.mode as AgentMode
+  }
+
+  // Add validation for new settings here
+
+  return settings
+}
+
+/**
+ * Save settings to file system (merges with existing settings)
+ */
+export const saveSettings = (newSettings: Partial<Settings>): void => {
+  const configDir = getConfigDir()
+  const settingsPath = getSettingsPath()
+
+  try {
+    if (!fs.existsSync(configDir)) {
+      fs.mkdirSync(configDir, { recursive: true })
+    }
+
+    // Load existing settings and merge
+    const existingSettings = loadSettings()
+    const mergedSettings = { ...existingSettings, ...newSettings }
+
+    fs.writeFileSync(settingsPath, JSON.stringify(mergedSettings, null, 2))
+  } catch (error) {
+    logger.debug(
+      {
+        error: error instanceof Error ? error.message : String(error),
+      },
+      'Error saving settings',
+    )
+  }
+}
+
+/**
+ * Load the saved agent mode preference
+ * @returns The saved mode, or 'DEFAULT' if not found or invalid
+ */
+export const loadModePreference = (): AgentMode => {
+  const settings = loadSettings()
+  return settings.mode ?? 'DEFAULT'
+}
+
+/**
+ * Save the agent mode preference
+ */
+export const saveModePreference = (mode: AgentMode): void => {
+  saveSettings({ mode })
+}