Refactor panel message handlers to dedicated files for managing configurations, retrieving collapsed states, and saving component collapsed states

Author: robertpiosik

packages/vscode/src/views/panel/backend/message-handlers/handle-get-collapsed-states.ts

@@ -0,0 +1,39 @@
+import {
+  get_configurations_collapsed_state_key,
+  get_presets_collapsed_state_key
+} from '@/constants/state-keys'
+import { PanelProvider } from '@/views/panel/backend/panel-provider'
+import { ApiMode, WebMode } from '@shared/types/modes'
+
+export const handle_get_collapsed_states = (
+  panel_provider: PanelProvider
+): void => {
+  const WEB_MODES: WebMode[] = [
+    'ask',
+    'edit-context',
+    'code-completions',
+    'no-context'
+  ]
+  const API_MODES: ApiMode[] = ['edit-context', 'code-completions']
+  panel_provider.send_message({
+    command: 'COLLAPSED_STATES',
+    presets_collapsed_by_web_mode: Object.fromEntries(
+      WEB_MODES.map((mode) => [
+        mode,
+        panel_provider.context.globalState.get<boolean>(
+          get_presets_collapsed_state_key(mode),
+          false
+        )
+      ])
+    ),
+    configurations_collapsed_by_api_mode: Object.fromEntries(
+      API_MODES.map((mode) => [
+        mode,
+        panel_provider.context.globalState.get<boolean>(
+          get_configurations_collapsed_state_key(mode),
+          false
+        )
+      ])
+    )
+  })
+}

packages/vscode/src/views/panel/backend/message-handlers/handle-manage-configurations.ts

@@ -0,0 +1,10 @@
+import * as vscode from 'vscode'
+import { ManageConfigurationsMessage } from '@/views/panel/types/messages'
+
+export const handle_manage_configurations = async (
+  message: ManageConfigurationsMessage
+): Promise<void> => {
+  const section =
+    message.api_mode == 'edit-context' ? 'edit-context' : 'code-completions'
+  await vscode.commands.executeCommand('codeWebChat.settings', section)
+}

packages/vscode/src/views/panel/backend/message-handlers/handle-save-component-collapsed-state.ts

@@ -0,0 +1,24 @@
+import {
+  get_configurations_collapsed_state_key,
+  get_presets_collapsed_state_key
+} from '@/constants/state-keys'
+import { PanelProvider } from '@/views/panel/backend/panel-provider'
+import { SaveComponentCollapsedStateMessage } from '@/views/panel/types/messages'
+import { ApiMode, WebMode } from '@shared/types/modes'
+
+export const handle_save_component_collapsed_state = async (
+  panel_provider: PanelProvider,
+  message: SaveComponentCollapsedStateMessage
+): Promise<void> => {
+  if (message.component == 'presets') {
+    await panel_provider.context.globalState.update(
+      get_presets_collapsed_state_key(message.mode as WebMode),
+      message.is_collapsed
+    )
+  } else if (message.component == 'configurations') {
+    await panel_provider.context.globalState.update(
+      get_configurations_collapsed_state_key(message.mode as ApiMode),
+      message.is_collapsed
+    )
+  }
+}

packages/vscode/src/views/panel/backend/message-handlers/index.ts

@@ -43,3 +43,6 @@ export * from './handle-intelligent-update-file-in-preview'
 export * from './handle-commit-changes'
 export * from './handle-accept-commit-message'
 export * from './handle-cancel-commit-message'
+export * from './handle-get-collapsed-states'
+export * from './handle-manage-configurations'
+export * from './handle-save-component-collapsed-state'

packages/vscode/src/views/panel/backend/panel-provider.ts

@@ -53,7 +53,10 @@ import {
   handle_accept_commit_message,
   handle_cancel_commit_message,
   handle_hash_sign_quick_pick,
-  handle_proceed_with_commit
+  handle_proceed_with_commit,
+  handle_get_collapsed_states,
+  handle_manage_configurations,
+  handle_save_component_collapsed_state
 } from './message-handlers'
 import {
   API_EDIT_FORMAT_STATE_KEY,
@@ -549,55 +552,11 @@ export class PanelProvider implements vscode.WebviewViewProvider {
           } else if (message.command == 'CANCEL_COMMIT_MESSAGE') {
             await handle_cancel_commit_message(this)
           } else if (message.command == 'MANAGE_CONFIGURATIONS') {
-            const section =
-              message.api_mode == 'edit-context'
-                ? 'edit-context'
-                : 'code-completions'
-            await vscode.commands.executeCommand(
-              'codeWebChat.settings',
-              section
-            )
+            await handle_manage_configurations(message)
           } else if (message.command == 'GET_COLLAPSED_STATES') {
-            const WEB_MODES: WebMode[] = [
-              'ask',
-              'edit-context',
-              'code-completions',
-              'no-context'
-            ]
-            const API_MODES: ApiMode[] = ['edit-context', 'code-completions']
-            this.send_message({
-              command: 'COLLAPSED_STATES',
-              presets_collapsed_by_web_mode: Object.fromEntries(
-                WEB_MODES.map((mode) => [
-                  mode,
-                  this.context.globalState.get<boolean>(
-                    get_presets_collapsed_state_key(mode),
-                    false
-                  )
-                ])
-              ),
-              configurations_collapsed_by_api_mode: Object.fromEntries(
-                API_MODES.map((mode) => [
-                  mode,
-                  this.context.globalState.get<boolean>(
-                    get_configurations_collapsed_state_key(mode),
-                    false
-                  )
-                ])
-              )
-            })
+            handle_get_collapsed_states(this)
           } else if (message.command == 'SAVE_COMPONENT_COLLAPSED_STATE') {
-            if (message.component == 'presets') {
-              await this.context.globalState.update(
-                get_presets_collapsed_state_key(message.mode as WebMode),
-                message.is_collapsed
-              )
-            } else if (message.component == 'configurations') {
-              await this.context.globalState.update(
-                get_configurations_collapsed_state_key(message.mode as ApiMode),
-                message.is_collapsed
-              )
-            }
+            await handle_save_component_collapsed_state(this, message)
           }
         } catch (error: any) {
           Logger.error({