Add token count display and improve unstaged file selection with a quick pick UI

Author: robertpiosik

apps/editor/src/commands/generate-commit-message-command/generate-commit-message-command.ts

@@ -3,6 +3,7 @@ import {
   get_git_repository,
   prepare_staged_changes
 } from '../../utils/git-repository-utils'
+import { display_token_count } from '../../utils/display-token-count'
 import { get_commit_message_config } from './utils/get-commit-message-config'
 import { build_commit_message_prompt } from './utils/build-commit-message-prompt'
 import { generate_commit_message_with_api } from './utils/generate-commit-message-with-api'
@@ -122,8 +123,11 @@ export const generate_commit_message_command = (
       const { was_empty_stage, message_prompt } = data
 
       await vscode.env.clipboard.writeText(message_prompt)
+      const token_count = Math.floor(message_prompt.length / 4)
       vscode.window.showInformationMessage(
-        'Commit message prompt copied to clipboard'
+        `Commit message prompt of about ${display_token_count(
+          token_count
+        )} tokens copied to clipboard.`
       )
 
       if (was_empty_stage) {

apps/editor/src/utils/git-repository-utils.ts

@@ -96,9 +96,32 @@ export const prepare_staged_changes = async (
     staged_changes.length == 0 &&
     repository.state.workingTreeChanges.length > 0
   ) {
-    const files_to_stage = repository.state.workingTreeChanges.map(
-      (change: any) => change.uri.fsPath
-    )
+    const items = repository.state.workingTreeChanges.map((change: any) => {
+      const relative_path = path.relative(
+        repository.rootUri.fsPath,
+        change.uri.fsPath
+      )
+      const dir_name = path.dirname(relative_path)
+
+      return {
+        label: path.basename(relative_path),
+        description: dir_name == '.' ? '' : dir_name,
+        picked: true,
+        fsPath: change.uri.fsPath
+      }
+    })
+
+    const selected = await vscode.window.showQuickPick(items, {
+      canPickMany: true,
+      title: 'Unstaged Files',
+      placeHolder: 'Select files to commit'
+    })
+
+    if (!selected || selected.length == 0) {
+      return null
+    }
+
+    const files_to_stage = selected.map((item) => item.fsPath)
     const file_args = files_to_stage
       .map((file: string) => `"${file.replace(/"/g, '\\"')}"`)
       .join(' ')