feat: add command to remove key

Author: Barbapapazes

package.json

@@ -151,6 +151,11 @@
         "title": "Add Key",
         "category": "Utils AI"
       },
+      {
+        "command": "barbapapazes.utils-ai.removeKey",
+        "title": "Remove Key",
+        "category": "Utils AI"
+      },
       {
         "command": "barbapapazes.utils-ai.runAction",
         "title": "Run action",

src/ai/base_ai.ts

@@ -1,6 +1,8 @@
 import type { Awaitable } from '../types/index.js'
 
 export class BaseAI {
+  static name: string
+
   protected key: string
   protected options: Record<string, unknown>
 
@@ -9,7 +11,7 @@ export class BaseAI {
     this.options = options
   }
 
-  ask(_: string, __: string): Awaitable<string> {
+  ask(_prompt: string, _content: string): Awaitable<string> {
     throw new Error('Method not implemented.')
   }
 }

src/ai/openai.ts

@@ -1,6 +1,8 @@
 import { BaseAI } from './base_ai.js'
 
 export class OpenAI extends BaseAI {
+  static name = 'OpenAI'
+
   constructor(key: string, options: Record<string, unknown> = {}) {
     super(key, options)
   }

src/commands/index.ts

@@ -1,10 +1,12 @@
 import { AddKeyCommand } from './add_key_command.js'
 import { CheckConfigCommand } from './check_config_command.js'
+import { RemoveKeyCommand } from './remove_key_command.js'
 import { RunActionCommand } from './run_action_command.js'
 import { RunQuickActionCommand } from './run_quick_action_command.js'
 
 export const commands = {
   addKey: AddKeyCommand,
+  removeKey: RemoveKeyCommand,
   runAction: RunActionCommand,
   runQuickAction: RunQuickActionCommand,
   checkConfig: CheckConfigCommand,

src/commands/remove_key_command.ts

@@ -0,0 +1,27 @@
+import { window } from 'vscode'
+import { BaseCommand } from './base_command.js'
+
+export class RemoveKeyCommand extends BaseCommand {
+  readonly keysKey = 'keys'
+
+  protected async run(): Promise<void> {
+    const keyName = await window.showQuickPick(this.getKeyNames())
+
+    this.assert(keyName, 'Key name is required.')
+
+    await this.remove(keyName)
+
+    this.logger.log('Key removed successfully.', {
+      notification: true,
+    })
+  }
+
+  protected async remove(name: string): Promise<void> {
+    await this.context.secrets.delete(name)
+    await this.context.globalState.update(this.keysKey, this.getKeyNames().filter(keyName => keyName !== name))
+  }
+
+  protected getKeyNames(): string[] {
+    return this.context.globalState.get<[] | undefined>(this.keysKey) || []
+  }
+}