kraven 0.3.0

Manage named environment variable profiles
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

//! Shell completion support using clap's dynamic completion system.

use std::ffi::OsStr;

use clap::CommandFactory;
use clap_complete::engine::{ArgValueCompleter, CompletionCandidate};
use clap_complete::CompleteEnv;

use crate::config;
use crate::Cli;

/// Initialize dynamic shell completions.
///
/// This must be called before argument parsing. If the `COMPLETE` environment
/// variable is set, it handles the completion request and exits.
pub fn init() {
    CompleteEnv::with_factory(build_cli).complete();
}

/// Build the CLI command with profile completers attached.
fn build_cli() -> clap::Command {
    Cli::command()
        .mut_subcommand("activate", add_profile_completer)
        .mut_subcommand("edit", add_profile_completer)
        .mut_subcommand("show", add_profile_completer)
        .mut_subcommand("remove", add_profile_completer)
        .mut_subcommand("encrypt", add_profile_completer)
        .mut_subcommand("decrypt", add_profile_completer)
}

/// Add profile completer to a subcommand's "profile" argument.
fn add_profile_completer(cmd: clap::Command) -> clap::Command {
    cmd.mut_arg("profile", |arg| {
        arg.add(ArgValueCompleter::new(complete_profiles))
    })
}

/// Complete profile names from the profile directory.
fn complete_profiles(current: &OsStr) -> Vec<CompletionCandidate> {
    let current_str = current.to_string_lossy();

    let Ok(profile_dir) = config::get_profile_dir() else {
        return vec![];
    };

    let Ok(entries) = std::fs::read_dir(&profile_dir) else {
        return vec![];
    };

    let mut seen = std::collections::HashSet::new();

    entries
        .flatten()
        .filter_map(|entry| {
            let path = entry.path();
            if !path.is_file() {
                return None;
            }
            let file_name = path.file_name()?.to_str()?;
            if file_name.starts_with('.') {
                return None;
            }
            let name = file_name.strip_suffix(".gpg").unwrap_or(file_name);
            if name.starts_with(&*current_str) && seen.insert(name.to_string()) {
                Some(CompletionCandidate::new(name))
            } else {
                None
            }
        })
        .collect()
}