Add user settings management with load and save functionality-Architect Claude

Author: PrashamTrivedi

src/commands/settings.ts

@@ -0,0 +1,34 @@
+import { parseArgs } from "jsr:@std/cli/parse-args";
+import { loadUserSettings, saveUserSettings } from "../config/settings.ts";
+
+export async function handleSettings(args: string[]): Promise<void> {
+    const flags = parseArgs(args, {
+        string: ["set-name", "add-command", "remove-command"],
+        boolean: ["list"],
+    });
+
+    const settings = await loadUserSettings();
+
+    if (flags["set-name"]) {
+        settings.userName = flags["set-name"];
+    }
+    else if (flags["add-command"]) {
+        const [name, description, helpCmd, ...helpFlags] = flags["add-command"].split(",");
+        settings.customCommands.push({
+            name,
+            description,
+            helpCommand: helpCmd,
+            helpFlags: helpFlags
+        });
+    }
+    else if (flags["remove-command"]) {
+        const index = settings.customCommands.findIndex(cmd => cmd.name === flags["remove-command"]);
+        if (index >= 0) settings.customCommands.splice(index, 1);
+    }
+    else if (flags.list) {
+        console.log(JSON.stringify(settings, null, 2));
+        return;
+    }
+
+    await saveUserSettings(settings);
+}

src/config/constants.ts

@@ -1,6 +1,7 @@
 import {join} from "jsr:@std/path"
 import Anthropic from "npm:@anthropic-ai/sdk"
 import {homedir} from "node:os"
+import { loadUserSettings } from "./settings.ts";
 
 export const EDITOR_DIR = join(homedir(), ".ComputerUseAgent", "editor_dir")
 export const SESSIONS_DIR = join(homedir(), ".ComputerUseAgent", "sessions")
@@ -147,4 +148,20 @@ export const CLIPBOARD_TOOLS: Anthropic.Beta.BetaTool[] = [
         input_schema: {type: "object", properties: {}, required: []}
 
     }
-]
+]
+
+export async function getSystemContext(basePrompt: string): Promise<string> {
+    const settings = await loadUserSettings();
+    const customCommandsContext = settings.customCommands.length > 0 
+        ? "\nCustom Commands:\n" + settings.customCommands
+            .map(cmd => `- ${cmd.name}: ${cmd.description}${
+                cmd.helpCommand ? `\n  Help: ${cmd.helpCommand}` : ''
+            }${
+                cmd.helpFlags ? `\n  Flags: ${cmd.helpFlags.join(', ')}` : ''
+            }`).join('\n')
+        : '';
+
+    return `${basePrompt}
+User Context:
+- Name: ${settings.userName}${customCommandsContext}`;
+}

src/config/settings.ts

@@ -0,0 +1,23 @@
+import { join } from "jsr:@std/path";
+import { homedir } from "node:os";
+import { UserSettings } from "../types/interfaces.ts";
+
+const DEFAULT_SETTINGS: UserSettings = {
+    userName: "User",
+    customCommands: []
+};
+
+const SETTINGS_PATH = join(homedir(), ".ComputerUseAgent", "settings.json");
+
+export async function loadUserSettings(): Promise<UserSettings> {
+    try {
+        const content = await Deno.readTextFile(SETTINGS_PATH);
+        return { ...DEFAULT_SETTINGS, ...JSON.parse(content) };
+    } catch {
+        return DEFAULT_SETTINGS;
+    }
+}
+
+export async function saveUserSettings(settings: UserSettings): Promise<void> {
+    await Deno.writeTextFile(SETTINGS_PATH, JSON.stringify(settings, null, 2));
+}

src/main.ts

@@ -12,6 +12,7 @@ import {BashSession} from "./modules/bash/bash_session.ts"
 import {HybridSession} from "./modules/hybrid/hybrid_session.ts"
 import {determineIntent} from "./utils/intent.ts"
 import {handleHistory} from "./commands/history.ts"
+import { handleSettings } from "./commands/settings.ts"
 
 async function main() {
   await setupLogging()
@@ -35,6 +36,11 @@ async function main() {
     return
   }
 
+  if (flags._[0] === "settings") {
+    await handleSettings(flags._.slice(1));
+    return;
+  }
+
   if (mode === "editor") {
     const session = new EditorSession(sessionId)
     await session.processEdit(prompt)

src/types/interfaces.ts

@@ -38,3 +38,13 @@ export interface Memory {
 export interface MemoryFile {
   memories: Memory[];
 }
+
+export interface UserSettings {
+  userName: string;
+  customCommands: {
+    name: string;
+    description: string;
+    helpCommand?: string;
+    helpFlags?: string[];
+  }[];
+}

src/utils/session.ts

@@ -3,6 +3,7 @@ import {format} from "jsr:@std/datetime"
 import {crypto} from "jsr:@std/crypto"
 import {log} from "../config/logging.ts"
 import {PromptDatabase} from "../modules/db/database.ts"
+import { getSystemContext } from "../config/constants.ts";
 
 export class SessionLogger {
   private sessionId: string
@@ -91,4 +92,8 @@ export class BaseSession {
       cost: totalCost
     })
   }
+
+  protected async getSystemPrompt(basePrompt: string): Promise<string> {
+    return await getSystemContext(basePrompt);
+  }
 }