Refactor the extension's rating mechanism from a webview message to a direct VS Code command

Author: robertpiosik

packages/vscode/package.json

@@ -116,6 +116,18 @@
         "title": "Apply Context from Clipboard",
         "category": "Code Web Chat"
       },
+      {
+        "command": "codeWebChat.openRepository",
+        "title": "Repository",
+        "category": "Code Web Chat",
+        "icon": "$(github-alt)"
+      },
+      {
+        "command": "codeWebChat.rate",
+        "title": "Rate",
+        "category": "Code Web Chat",
+        "icon": "$(star)"
+      },
       {
         "command": "codeWebChat.applyContext",
         "title": "Apply Context",
@@ -328,6 +340,10 @@
           "command": "codeWebChat.clearChecks",
           "when": "false"
         },
+        {
+          "command": "codeWebChat.rate",
+          "when": "false"
+        },
         {
           "command": "codeWebChat.clearChecksOpenEditors",
           "when": "false"
@@ -480,9 +496,14 @@
           "group": "navigation@1"
         },
         {
-          "command": "codeWebChat.settings",
+          "command": "codeWebChat.rate",
           "when": "view == codeWebChatView",
           "group": "navigation@2"
+        },
+        {
+          "command": "codeWebChat.settings",
+          "when": "view == codeWebChatView",
+          "group": "navigation@3"
         }
       ],
       "view/item/context": [

packages/vscode/src/commands/index.ts

@@ -14,6 +14,7 @@ export * from './open-file-from-workspace-command'
 export * from './open-url-command'
 export * from './open-settings-command'
 export * from './reference-in-chat-command'
+export * from './rate-command'
 export * from './rename-command'
 export * from './revert-command'
 export * from './save-all-command'

…ackend/message-handlers/handle-review.ts …ages/vscode/src/commands/rate-command.tspackages/vscode/src/view/backend/message-handlers/handle-review.ts renamed to packages/vscode/src/commands/rate-command.ts packages/vscode/src/view/backend/message-handlers/handle-review.ts renamed to packages/vscode/src/commands/rate-command.ts

@@ -1,6 +1,6 @@
 import * as vscode from 'vscode'
 
-export const handle_review = async (): Promise<void> => {
+const rate = async (): Promise<void> => {
   const options = [
     {
       label: 'Visual Studio Marketplace',
@@ -24,3 +24,6 @@ export const handle_review = async (): Promise<void> => {
     vscode.env.openExternal(vscode.Uri.parse(selected_option.url))
   }
 }
+
+export const rate_command = () =>
+  vscode.commands.registerCommand('codeWebChat.rate', rate)

packages/vscode/src/extension.ts

@@ -35,6 +35,7 @@ import {
   reference_in_chat_command,
   open_settings_command,
   open_url_command,
+  rate_command,
   edit_context_commands,
   apply_context_from_clipboard_command
 } from './commands'
@@ -158,6 +159,7 @@ export async function activate(context: vscode.ExtensionContext) {
       command: 'codeWebChat.openRepository',
       url: 'https://github.com/robertpiosik/CodeWebChat'
     }),
+    rate_command(),
     open_settings_command(context),
     apply_context_from_clipboard_command(workspace_provider)
   )

packages/vscode/src/utils/at-sign-quick-pick.ts

@@ -11,16 +11,15 @@ export async function at_sign_quick_pick(
   const items = [
     {
       label: '@Selection',
-      description: 'Inject text selection of the active editor'
+      description: 'Text selection of the active editor'
     },
     {
       label: '@Changes',
-      description:
-        'Inject changes between the current branch and the selected branch'
+      description: 'Changes between the current branch and the selected branch'
     },
     {
       label: '@SavedContext',
-      description: 'Inject files from a saved context'
+      description: 'Files from a saved context'
     }
   ]
 

packages/vscode/src/view/backend/message-handlers/index.ts

@@ -32,4 +32,3 @@ export * from './handle-get-mode-web'
 export * from './handle-get-mode-api'
 export * from './handle-save-mode-web'
 export * from './handle-save-mode-api'
-export * from './handle-review'

packages/vscode/src/view/backend/view-provider.ts

@@ -46,8 +46,7 @@ import {
   handle_get_mode_api,
   handle_save_mode_api,
   handle_save_home_view_type,
-  handle_get_home_view_type,
-  handle_review
+  handle_get_home_view_type
 } from './message-handlers'
 import {
   config_preset_to_ui_format,
@@ -398,8 +397,6 @@ export class ViewProvider implements vscode.WebviewViewProvider {
             handle_get_home_view_type(this)
           } else if (message.command == 'SHOW_AT_SIGN_QUICK_PICK') {
             await handle_at_sign_quick_pick(this, this.context)
-          } else if (message.command == 'REVIEW') {
-            await handle_review()
           }
         } catch (error: any) {
           console.error('Error handling message:', message, error)

packages/vscode/src/view/frontend/home/Home.tsx

@@ -390,12 +390,6 @@ export const Home: React.FC<Props> = (props) => {
     } as WebviewMessage)
   }
 
-  const handle_review_click = () => {
-    props.vscode.postMessage({
-      command: 'REVIEW'
-    } as WebviewMessage)
-  }
-
   const handle_quick_action_click = (command: string) => {
     props.vscode.postMessage({
       command: 'EXECUTE_COMMAND',
@@ -499,7 +493,6 @@ export const Home: React.FC<Props> = (props) => {
       on_code_completion_with_quick_pick_click={
         handle_code_completion_with_quick_pick_click
       }
-      on_review_click={handle_review_click}
     />
   )
 }

packages/vscode/src/view/frontend/home/HomeView/HomeView.tsx

@@ -53,7 +53,6 @@ type Props = {
   on_edit_context_with_quick_pick_click: () => void
   on_code_completion_click: () => void
   on_code_completion_with_quick_pick_click: () => void
-  on_review_click: () => void
 }
 
 const web_mode_labels: Record<WebMode, string> = {

packages/vscode/src/view/types/messages.ts

@@ -193,10 +193,6 @@ export interface SaveApiModeMessage extends BaseMessage {
   mode: ApiMode
 }
 
-export interface ReviewMessage extends BaseMessage {
-  command: 'REVIEW'
-}
-
 // Messages from extension to webview:
 export interface InstructionsMessage extends BaseMessage {
   command: 'INSTRUCTIONS'
@@ -375,7 +371,6 @@ export type WebviewMessage =
   | GetWebModeMessage
   | GetApiModeMessage
   | SaveApiModeMessage
-  | ReviewMessage
 
 export type ExtensionMessage =
   | InstructionsMessage