dora-cli 0.5.0-rc0

`dora` goal is to be a low latency, composable, and distributed data flow.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use clap::{CommandFactory, ValueEnum};
use clap_complete::Shell;

use crate::command::Executable;

#[derive(Debug, clap::Args)]
#[command(after_help = r#"
USAGE:
  eval $(dora completion)              # Auto-detect shell
  eval $(dora completion <SHELL>)      # Specify shell explicitly

PERSIST COMPLETION:

  Bash:
    echo 'eval "$(dora completion bash)"' >> ~/.bashrc
    # Then restart shell

  Zsh:
    echo 'eval "$(dora completion zsh)"' >> ~/.zshrc
    # Then restart shell

    # If you get 'command not found: compdef', add this before the eval line:
    autoload -Uz compinit
    compinit

  Fish:
    # Add to ~/.config/fish/config.fish
    if status is-interactive
        eval "$(dora completion fish)"
    end
    # Then restart shell

"#)]
pub struct Completion {
    /// The shell to generate the script for
    #[arg(value_enum)]
    shell: Option<Shell>,
}
impl Executable for Completion {
    async fn execute(self) -> eyre::Result<()> {
        let shell = if let Some(sh) = self.shell {
            sh
        } else {
            get_shell().map_err(|e| {
                eyre::eyre!(
                    "Please specify the shell via parameter. Unable to get the current shell. ({})",
                    e
                )
            })?
        };
        clap_complete::generate(
            shell,
            &mut crate::Args::command(),
            "dora",
            &mut std::io::stdout(),
        );
        Ok(())
    }
}

fn get_shell() -> eyre::Result<Shell> {
    let pid =
        sysinfo::get_current_pid().map_err(|_| eyre::eyre!("Unable to get the current PID"))?;
    let system = sysinfo::System::new_all();
    let process = system
        .process(pid)
        .ok_or(eyre::eyre!("Unable to get the current process"))?;
    let parent_pid = process
        .parent()
        .ok_or(eyre::eyre!("Unable to get the parent process PID"))?;
    let parent_process = system
        .process(parent_pid)
        .ok_or(eyre::eyre!("Unable to get the parent process"))?;
    let current_shell = Shell::from_str(
        parent_process
            .name()
            .to_str()
            .ok_or(eyre::eyre!("Unable to get the shell name"))?,
        true,
    )
    .map_err(|_| eyre::eyre!("Unable to get the shell type"))?;
    Ok(current_shell)
}