Extract webview message handling logic into dedicated handler files

Author: robertpiosik

packages/vscode/src/view/backend/message-handlers/handle-get-code-completion-suggestions.ts

@@ -0,0 +1,11 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_get_code_completion_suggestions = (
+  provider: ViewProvider
+): void => {
+  provider.send_message<ExtensionMessage>({
+    command: 'CODE_COMPLETION_SUGGESTIONS',
+    value: provider.code_completion_suggestions
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-get-code-completions-history.ts

@@ -0,0 +1,15 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_get_code_completions_history = (
+  provider: ViewProvider
+): void => {
+  const history = provider.context.workspaceState.get<string[]>(
+    'code-completions-history',
+    []
+  )
+  provider.send_message<ExtensionMessage>({
+    command: 'FIM_CHAT_HISTORY',
+    messages: history
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-get-code-completions-mode.ts

@@ -0,0 +1,22 @@
+import * as vscode from 'vscode'
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_get_code_completions_mode = (
+  provider: ViewProvider
+): void => {
+  const has_active_editor = !!vscode.window.activeTextEditor
+
+  if (provider.is_code_completions_mode && !has_active_editor) {
+    provider.is_code_completions_mode = false
+    provider.send_message<ExtensionMessage>({
+      command: 'CODE_COMPLETIONS_MODE',
+      enabled: false
+    })
+  } else {
+    provider.send_message<ExtensionMessage>({
+      command: 'CODE_COMPLETIONS_MODE',
+      enabled: provider.is_code_completions_mode
+    })
+  }
+}

packages/vscode/src/view/backend/message-handlers/handle-get-history.ts

@@ -0,0 +1,10 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_get_history = (provider: ViewProvider): void => {
+  const history = provider.context.workspaceState.get<string[]>('history', [])
+  provider.send_message<ExtensionMessage>({
+    command: 'CHAT_HISTORY',
+    messages: history
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-get-instructions.ts

@@ -0,0 +1,9 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_get_instructions = (provider: ViewProvider): void => {
+  provider.send_message<ExtensionMessage>({
+    command: 'INSTRUCTIONS',
+    value: provider.instructions
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-get-open-router-api-key.ts

@@ -0,0 +1,12 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_get_open_router_api_key = (
+  provider: ViewProvider
+): void => {
+  const api_key = provider.api_tools_settings_manager.get_open_router_api_key()
+  provider.send_message<ExtensionMessage>({
+    command: 'OPEN_ROUTER_API_KEY',
+    api_key
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-request-editor-selection-state.ts

@@ -0,0 +1,11 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_request_editor_selection_state = (
+  provider: ViewProvider
+): void => {
+  provider.send_message<ExtensionMessage>({
+    command: 'EDITOR_SELECTION_CHANGED',
+    has_selection: provider.has_active_selection
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-request-editor-state.ts

@@ -0,0 +1,9 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { ExtensionMessage } from '@/view/types/messages'
+
+export const handle_request_editor_state = (provider: ViewProvider): void => {
+  provider.send_message<ExtensionMessage>({
+    command: 'EDITOR_STATE_CHANGED',
+    has_active_editor: provider.has_active_editor
+  })
+}

packages/vscode/src/view/backend/message-handlers/handle-save-code-completion-suggestions.ts

@@ -0,0 +1,13 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { SaveCodeCompletionSuggestionsMessage } from '@/view/types/messages'
+
+export const handle_save_code_completion_suggestions = async (
+  provider: ViewProvider,
+  message: SaveCodeCompletionSuggestionsMessage
+): Promise<void> => {
+  provider.code_completion_suggestions = message.instruction
+  await provider.context.workspaceState.update(
+    'code-completion-suggestions',
+    message.instruction
+  )
+}

packages/vscode/src/view/backend/message-handlers/handle-save-code-completions-mode.ts

@@ -0,0 +1,14 @@
+import { ViewProvider } from '@/view/backend/view-provider'
+import { SaveCodeCompletionsModeMessage } from '@/view/types/messages'
+
+export const handle_save_code_completions_mode = async (
+  provider: ViewProvider,
+  message: SaveCodeCompletionsModeMessage
+): Promise<void> => {
+  provider.is_code_completions_mode = message.enabled
+  provider.calculate_token_count()
+  provider.send_message({
+    command: 'CODE_COMPLETIONS_MODE',
+    enabled: message.enabled
+  })
+}