Allow for /commit and /review commands to be run in sidekick

Resolves #63

In `cli.py`, the commit process has been refined to provide better user feedback and control. The user is now asked for confirmation before committing changes, and has the option to abort the commit if they choose. This makes the commit process more interactive and user-friendly. 🎉

In `coder.py`, the autocomplete feature has been enhanced. Now, it supports autocomplete for the `/review` and `/commit` commands, and the file listing for the `/add` command is now cached for performance. This should make the CLI more responsive and easier to use. 🚀

Additionally, the `/commit` and `/review` commands have been added to the sidekick command in `cli.py`, providing more functionality to the user. 🛠️

These changes should improve the overall user experience of the CLI. 😊

Author: TechNickAI

aicodebot/cli.py

@@ -106,13 +106,16 @@ def commit(verbose, response_token_size, yes, skip_pre_commit, files):  # noqa:
     if not staged_files:
         # If no files are staged, they probably want to commit all changed files, confirm.
         if not files and not yes:
-            click.confirm(
+            confirm = click.confirm(
                 "Since there are no git staged files, all of the modified files will be committed:\n\t"
                 + "\n\t".join(unstaged_files)
                 + "\nDoes that look correct?",
                 default=True,
-                abort=True,
             )
+            if not confirm:
+                console.print("Aborting commit.")
+                return
+
         files = unstaged_files
     else:
         # The list of files to be committed is the same as the list of staged files
@@ -189,10 +192,14 @@ def commit(verbose, response_token_size, yes, skip_pre_commit, files):  # noqa:
         subprocess.call([editor, temp_file_name])  # noqa: S603
 
     # Ask the user if they want to commit the changes
-    if yes or click.confirm("Are you ready to commit the changes?", default=True, abort=True):
-        # Commit the changes using the temporary file for the commit message
-        exec_and_get_output(["git", "commit", "-F", temp_file_name])
-        console.print(f"✅ {len(files)} file(s) committed.")
+    if not yes:
+        confirm = click.confirm("Are you ready to commit the changes?", default=True)
+        if confirm:
+            # Commit the changes using the temporary file for the commit message
+            exec_and_get_output(["git", "commit", "-F", temp_file_name])
+            console.print(f"✅ {len(files)} file(s) committed.")
+        else:
+            console.print("Aborting commit.")
 
     # Delete the temporary file
     Path(temp_file_name).unlink()
@@ -562,12 +569,23 @@ def calc_response_token_size(files):
                 show_file_context(files)
                 continue
 
+            elif cmd == "/commit":
+                # Call the commit function with the parsed arguments
+                args = human_input.split()[1:]
+                ctx = click.get_current_context()
+                ctx.invoke(commit, *args)
+                continue
             elif cmd == "/edit":
                 human_input = edited_input = click.edit()
             elif cmd == "/files":
                 show_file_context(files)
                 continue
-
+            elif cmd == "/review":
+                # Call the review function with the parsed arguments
+                args = human_input.split()[1:]
+                ctx = click.get_current_context()
+                ctx.invoke(review, *args)
+                continue
             elif cmd == "/quit":
                 break
 

aicodebot/coder.py

@@ -445,12 +445,13 @@ class SidekickCompleter(Completer):
     """
 
     files = []  # List of files that we have loaded in the current context
+    project_files = Coder.filtered_file_list(".", use_gitignore=True, ignore_patterns=[".git"])
 
     def get_completions(self, document, complete_event):
         # Get the text before the cursor
         text = document.text_before_cursor
 
-        supported_commands = ["/add", "/drop", "/edit", "/files", "/quit"]
+        supported_commands = ["/add", "/commit", "/drop", "/edit", "/files", "/review", "/quit"]
 
         # If the text starts with a slash, it's a command
         if text.startswith("/"):
@@ -459,15 +460,21 @@ def get_completions(self, document, complete_event):
                     yield Completion(command, start_position=-len(text))
 
         if text.startswith("/add "):
-            # If the text starts with /add or /drop, it's a file, so autocomplete the file name
+            # For /add autocomplete the file name from the project file listing
             # Get the list of files in the current directory, filtered by the .gitignore file
-            project_files = Coder.filtered_file_list(".", use_gitignore=True, ignore_patterns=[".git"])
-            for file in project_files:
+            for file in self.project_files:
                 if str(file).startswith(text.split()[-1]):
                     yield Completion(str(file), start_position=-len(text.split()[-1]))
 
         elif text.startswith("/drop "):
-            # If the text starts with /drop, use the current context files for autocomplete
+            # For /drop, use the current context files for autocomplete
             for file in self.files:
                 if file.startswith(text.split()[-1]):
                     yield Completion(file, start_position=-len(text.split()[-1]))
+
+        elif text.startswith(("/review ", "/commit ")):
+            # For /review and /commit, use the staged/unstaged files for autocomplete
+            changed_files = Coder.git_staged_files() + Coder.git_unstaged_files()
+            for file in changed_files:
+                if file.startswith(text.split()[-1]):
+                    yield Completion(file, start_position=-len(text.split()[-1]))