Introduce testing/resetStroage endpoint and add a related action to JB (#8144)

There are some functionalities that use the local storage. We want to be
able to test them. One of them is auto-edit enrolment that uses storage
to mark a user as enrolled. Thanks to this change, a user will be able
to reset the storage from the client and retry the features.

## Test plan
1. cmd + shift + A
2. Search for: "Cody: Reset Cody Storage"
3. Execute



<!-- Required. See
https://docs-legacy.sourcegraph.com/dev/background-information/testing_principles.
-->

Author: mkondratek

agent/bindings/kotlin/lib/src/main/kotlin/com/sourcegraph/cody/agent/protocol_generated/CodyAgentServer.kt

@@ -186,4 +186,6 @@ interface CodyAgentServer {
   fun secrets_didChange(params: Secrets_DidChangeParams)
   @JsonNotification("window/didChangeFocus")
   fun window_didChangeFocus(params: Window_DidChangeFocusParams)
+  @JsonNotification("testing/resetStorage")
+  fun testing_resetStorage(params: Null?)
 }

agent/src/agent.ts

@@ -1417,6 +1417,10 @@ export class Agent extends MessageHandler implements ExtensionClient {
             this.secretsDidChange.fire({ key })
         })
 
+        this.registerNotification('testing/resetStorage', () => {
+            localStorage.resetStorage()
+        })
+
         this.registerAuthenticatedRequest('featureFlags/getFeatureFlag', async ({ flagName }) => {
             return featureFlagProvider.evaluateFeatureFlagEphemerally(
                 FeatureFlag[flagName as keyof typeof FeatureFlag]

jetbrains/src/main/kotlin/com/sourcegraph/cody/ResetStorage.kt

@@ -0,0 +1,11 @@
+package com.sourcegraph.cody
+
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.sourcegraph.cody.agent.CodyAgentService
+import com.sourcegraph.common.ui.DumbAwareEDTAction
+
+class ResetStorage : DumbAwareEDTAction() {
+  override fun actionPerformed(e: AnActionEvent) {
+    CodyAgentService.withAgent(e.project!!) { agent -> agent.server.testing_resetStorage(null) }
+  }
+}

jetbrains/src/main/resources/META-INF/plugin.xml

@@ -200,6 +200,13 @@
                     <override-text place="GoToAction" text="Cody: Export All Chats As JSON"/>
                 </action>
 
+                <action id="cody.resetStorage"
+                        class="com.sourcegraph.cody.ResetStorage"
+                        text="Reset Cody Storage"
+                        description="Reset Cody Storage">
+                    <override-text place="GoToAction" text="Cody: Reset Cody Storage"/>
+                </action>
+
                 <action id="cody.openChat"
                         class="com.sourcegraph.cody.chat.OpenChatAction"
                         text="Open Chat"

vscode/src/jsonrpc/agent-protocol.ts

@@ -378,6 +378,8 @@ export type ClientNotifications = {
     'secrets/didChange': [{ key: string }]
 
     'window/didChangeFocus': [{ focused: boolean }]
+
+    'testing/resetStorage': [null]
 }
 
 // ================

vscode/src/services/LocalStorageProvider.ts

@@ -241,6 +241,11 @@ class LocalStorage implements LocalStorageForModelPreferences {
         await this.set(this.AUTO_EDITS_BETA_ENROLLED, true)
     }
 
+    public async resetStorage(): Promise<void> {
+        for (const key of this.storage.keys()) {
+            await this.storage.update(key, undefined)
+        }
+    }
     public async setGitHubRepoAccessibility(data: GitHubDotComRepoMetaData[]): Promise<void> {
         await this.set(this.GIT_REPO_ACCESSIBILITY_KEY, data)
     }