add shell completion for flamegraph

README.md

@@ -33,6 +33,16 @@ This will make the `flamegraph` and
 binary directory. On most systems this is
 usually something like `~/.cargo/bin`.
 
+## Shell auto-completion
+
+At the moment, only `flamegraph` supports auto-completion. Supported shells are `bash`, `fish`, `zsh`, `powershell` and `elvish`.
+`cargo-flamegraph` does not support auto-completion because it is not as straight-forward to implement for custom cargo subcommands. See #153 for details.
+
+How you enable auto-completion depends on your shell, e.g.
+```bash
+flamegraph --completions bash > $XDG_CONFIG_HOME/bash_completion # or /etc/bash_completion.d/
+```
+
 ## Examples
 
 ```

src/bin/flamegraph.rs

@@ -12,6 +12,10 @@ struct Opt {
     #[structopt(short = "p", long = "pid")]
     pid: Option<u32>,
 
+    /// Generate shell completions for the given shell.
+    #[structopt(long = "completions")]
+    completions: Option<structopt::clap::Shell>,
+
     #[structopt(flatten)]
     graph: flamegraph::Options,
 
@@ -20,6 +24,12 @@ struct Opt {
 
 fn main() -> anyhow::Result<()> {
     let opt = Opt::from_args();
+
+    if let Some(shell) = opt.completions {
+        Opt::clap().gen_completions_to("flamegraph", shell, &mut std::io::stdout().lock());
+        return Ok(());
+    }
+
     let workload = match (opt.pid, opt.trailing_arguments.is_empty()) {
         (Some(p), true) => Workload::Pid(p),
         (None, false) => Workload::Command(opt.trailing_arguments.clone()),