Implement the Generate Commit Message command and integrate it into the Source Control view title menu

Author: robertpiosik

packages/vscode/package.json

@@ -277,6 +277,12 @@
         "command": "codeWebChat.settings",
         "title": "Settings",
         "category": "Code Web Chat"
+      },
+      {
+        "command": "codeWebChat.generateCommitMessage",
+        "title": "Generate Commit Message with CWC",
+        "category": "Code Web Chat",
+        "icon": "$(sparkle)"
       }
     ],
     "menus": {
@@ -459,6 +465,13 @@
           "group": "navigation@1"
         }
       ],
+      "scm/title": [
+        {
+          "command": "codeWebChat.generateCommitMessage",
+          "group": "navigation",
+          "when": "scmProvider == git"
+        }
+      ],
       "view/item/context": [
         {
           "command": "codeWebChat.closeEditor",

packages/vscode/src/extension.ts

@@ -1,5 +1,10 @@
 import * as vscode from 'vscode'
 import { context_initialization } from './context/context-initialization'
+import {
+  get_git_repository,
+  prepare_staged_changes
+} from './utils/git-repository-utils'
+import { generate_commit_message_from_diff } from './views/panel/backend/message-handlers/handle-commit-changes/utils'
 import { PanelProvider } from './views/panel/backend/panel-provider'
 import { WebSocketManager } from './services/websocket-manager'
 import {
@@ -107,4 +112,27 @@ export async function activate(context: vscode.ExtensionContext) {
       }
     )
   )
+
+  context.subscriptions.push(
+    vscode.commands.registerCommand(
+      'codeWebChat.generateCommitMessage',
+      async (source_control?: vscode.SourceControl) => {
+        const repository = get_git_repository(source_control)
+        if (!repository) return
+
+        const diff = await prepare_staged_changes(repository)
+
+        if (!diff) return
+
+        const commit_message = await generate_commit_message_from_diff({
+          context,
+          repository,
+          diff
+        })
+        if (commit_message) {
+          repository.inputBox.value = commit_message
+        }
+      }
+    )
+  )
 }

packages/vscode/src/utils/git-repository-utils.ts

@@ -63,11 +63,19 @@ export function get_git_repository(
 export async function prepare_staged_changes(
   repository: GitRepository
 ): Promise<string | null> {
+  await repository.status()
   const staged_changes = repository.state.indexChanges || []
 
-  if (staged_changes.length === 0) {
-    await repository.add([])
+  if (
+    staged_changes.length === 0 &&
+    repository.state.workingTreeChanges.length > 0
+  ) {
+    const uris_to_stage = repository.state.workingTreeChanges.map(
+      (change: any) => change.uri.fsPath
+    )
+    await repository.add(uris_to_stage)
     await new Promise((resolve) => setTimeout(resolve, 500))
+    await repository.status()
   }
 
   const diff = execSync('git diff --staged', {