Shell completion command

Cargo.toml

@@ -16,6 +16,8 @@ path = "./src/huak/lib.rs"
 
 [dependencies]
 clap = {version = "4.0.15", features = ["cargo", "derive"]}
+clap_complete = "4.0"
+
 thiserror = "1.0.36"
 fs_extra = "1.2.0"
 glob = "0.3.0"

README.md

@@ -68,25 +68,26 @@ A Python package manager written in Rust inspired by Cargo.
 Usage: huak <COMMAND>
 
 Commands:
-  activate  Activate the project's virtual environment
-  add       Add a dependency to the existing project
-  audit     Check for vulnerable dependencies and license compatibility*
-  build     Build tarball and wheel for the project
-  clean     Remove tarball and wheel from the built project
-  doc       Generates documenation for the project*
-  fix       Auto-fix fixable lint conflicts
-  fmt       Format the project's Python code
-  init      Initialize the existing project
-  install   Install the dependencies of an existing project
-  lint      Lint the project's Python code
-  new       Create a new project at <path>
-  publish   Builds and uploads current project to a registry*
-  remove    Remove a dependency from the project
-  run       Run a command within the project's environment context
-  test      Test the project's Python code
-  update    Update dependencies added to the project*
-  version   Display the version of the project
-  help      Print this message or the help of the given subcommand(s)
+  activate    Activate the project's virtual environment
+  add         Add a dependency to the existing project
+  audit       Check for vulnerable dependencies and license compatibility*
+  build       Build tarball and wheel for the project
+  completion  Generates a shell completion script for supported shells. See the help menu for more information on supported shells
+  clean       Remove tarball and wheel from the built project
+  doc         Generates documentation for the project*
+  fix         Auto-fix fixable lint conflicts
+  fmt         Format the project's Python code
+  init        Initialize the existing project
+  install     Install the dependencies of an existing project
+  lint        Lint the project's Python code
+  new         Create a new project at <path>
+  publish     Builds and uploads current project to a registry*
+  remove      Remove a dependency from the project
+  run         Run a command within the project's environment context
+  test        Test the project's Python code
+  update      Update dependencies added to the project*
+  version     Display the version of the project
+  help        Print this message or the help of the given subcommand(s)
 ```
 _"*" indicates first-pass of implementation is incomplete._
 

src/bin/huak/commands/config/completion.rs

@@ -0,0 +1,63 @@
+use crate::commands::Cli;
+use crate::errors::CliResult;
+
+use huak::ops::config;
+
+use clap::{Args, Command, CommandFactory, Subcommand};
+use clap_complete::{generate, Shell};
+
+/// Prints the script to stdout and a way to add the script to the shell init file to stderr. This
+/// way if the user runs completion <shell> > completion.sh only the stdout will be redirected into
+/// completion.sh.
+pub fn run(config_command: Config) -> CliResult<()> {
+    match config_command.command {
+        ConfigCommand::Completion { shell } => {
+            generate_shell_completion_script(shell)
+        }
+        ConfigCommand::Install { shell } => {
+            let mut cmd: Command = Cli::command();
+            let _result = match shell {
+                Shell::Bash => config::_add_completion_bash(),
+                Shell::Elvish => config::_add_completion_elvish(),
+                Shell::Fish => config::_add_completion_fish(&mut cmd),
+                Shell::PowerShell => config::_add_completion_powershell(),
+                Shell::Zsh => config::_add_completion_zsh(&mut cmd),
+                _ => Ok(()),
+            };
+        }
+        ConfigCommand::Uninstall { shell } => {
+            let _result = match shell {
+                Shell::Bash => config::_remove_completion_bash(),
+                Shell::Elvish => config::_remove_completion_elvish(),
+                Shell::Fish => config::_remove_completion_fish(),
+                Shell::PowerShell => config::_remove_completion_powershell(),
+                Shell::Zsh => config::_remove_completion_zsh(),
+                _ => Ok(()),
+            };
+        }
+    }
+    Ok(())
+}
+
+fn generate_shell_completion_script(shell: Shell) {
+    let mut cmd = Cli::command();
+
+    generate(shell, &mut cmd, "huak", &mut std::io::stdout())
+}
+
+#[derive(Args)]
+pub struct Config {
+    #[command(subcommand)]
+    command: ConfigCommand,
+}
+
+#[derive(Debug, Subcommand)]
+pub enum ConfigCommand {
+    /// Generates a shell completion script for supported shells.
+    /// See the help menu for more information on supported shells.
+    Completion { shell: Shell },
+    /// Installs the completion script in your shell init file.
+    Install { shell: Shell },
+    /// Uninstalls the completion script from your shell init file.
+    Uninstall { shell: Shell },
+}

src/bin/huak/commands/config/mod.rs

@@ -0,0 +1,2 @@
+mod completion;
+pub use completion::*;

src/bin/huak/commands/mod.rs

@@ -1,11 +1,13 @@
 use crate::errors::CliResult;
+
 use clap::{Parser, Subcommand};
 
 pub(crate) mod activate;
 pub(crate) mod add;
 pub(crate) mod audit;
 pub(crate) mod build;
 pub(crate) mod clean;
+pub(crate) mod config;
 pub(crate) mod doc;
 pub(crate) mod fix;
 pub(crate) mod fmt;
@@ -46,6 +48,8 @@ pub enum Commands {
     Audit,
     /// Build tarball and wheel for the project.
     Build,
+    /// Interact with the configuration of huak.
+    Config(config::Config),
     /// Remove tarball and wheel from the built project.
     Clean {
         #[arg(long, required = false)]
@@ -119,6 +123,7 @@ pub enum Commands {
 impl Cli {
     pub fn run(self) -> CliResult<()> {
         match self.command {
+            Commands::Config(config_command) => config::run(config_command),
             Commands::Activate => activate::run(),
             Commands::Add { dependency, dev } => add::run(dependency, dev),
             Commands::Audit => audit::run(),

src/huak/lib.rs

@@ -28,6 +28,61 @@
 //!
 //!Around 0.1.0 you'll be able to install `huak` using `brew` or `pip`. Distribution plans will be finalized closer to 0.1.0.
 //!
+//!## Shell completion
+//!
+//!##### Bash
+//!First, ensure that you install `bash-completion` using your package manager.
+//!After, add this to your `~/.bash_profile`:
+//!
+//!```bash
+//! eval "$(huak config completion bash)"
+//!```
+//!
+//!##### Elvish
+//!This shell is supported, but the suggestion as to how this should be added to your shell init file is missing.
+//!
+//!If you are able to test this please head over to [github](https://github.com/cnpryer/huak/issues) and file an issue
+
+//!##### Fish
+//!Generate a `huak.fish` completion script:
+//!
+//!```bash
+//! huak config completion fish > ~/.config/fish/completions/huak.fish
+//!```
+//!
+//!##### Powershell
+//!Open your profile script with:
+//!
+//!```bash
+//! mkdir -Path (Split-Path -Parent $profile) -ErrorAction SilentlyContinue
+//!```
+//!
+//!```bash
+//! notepad $profile
+//!```
+//!
+//!Add the line and save the file:
+//!```bash
+//! Invoke-Expression -Command $(huak config completion powershell| Out-String)
+//!```
+//!
+//!##### Zsh
+//!Generate a `_huak` completion script and put it somewhere in your `$fpath`:
+//!```bash
+//! huak config completion zsh > /usr/local/share/zsh/site-functions/_huak
+//!```
+//!
+//!Ensure that the following is present in your `~/.zshrc`:
+//!
+//!```bash
+//! autoload -U compinit
+//!```
+//!
+//!```bash
+//! compinit -i
+//!```
+//!
+//!
 //!## Goals
 //!
 //!Besides some of my own experience with the Python ecosystem, there are a few additional guiding principles steering the development of Huak: