bofh 0.0.2

Command-line client for the Cerebrum IGA
Documentation 
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use rustyline::hint::{Hint, Hinter};
use rustyline::Context;
use rustyline_derive::{Completer, Helper, Highlighter, Validator};
use std::collections::BTreeMap;
#[derive(Completer, Helper, Validator, Highlighter)]
pub(crate) struct BofhHelper<'a> {
    pub(crate) commands: &'a BTreeMap<String, bofh::CommandGroup>,
}

#[derive(Hash, Debug, PartialEq, Eq)]
pub(crate) struct CommandHint {
    display: String,
}

impl Hint for CommandHint {
    fn display(&self) -> &str {
        &self.display
    }

    fn completion(&self) -> Option<&str> {
        Some(&self.display)
    }
}

impl CommandHint {
    fn suffix(&self, strip_chars: usize) -> CommandHint {
        CommandHint {
            display: self.display[strip_chars..].to_owned(),
        }
    }
}

impl Hinter for BofhHelper<'_> {
    type Hint = CommandHint;

    fn hint(&self, line: &str, pos: usize, _ctx: &Context<'_>) -> Option<CommandHint> {
        if line.is_empty() || pos < line.len() {
            return None;
        }

        let words: Vec<&str> = line.split_whitespace().collect();

        // Hint arguments
        if words.len() >= 2 {
            if let Some(command) = self.commands.get(words[0]) {
                if let Some(subcommand) = command.commands.get(words[1]) {
                    let args_to_hint = subcommand.args.len() - words.len() + 2;
                    if args_to_hint <= subcommand.args.len() {
                        return Some(CommandHint {
                            display: format!(
                                "{}{}",
                                if line.ends_with(char::is_whitespace) {
                                    ""
                                } else {
                                    " "
                                },
                                subcommand.args[subcommand.args.len() - args_to_hint..]
                                    .iter()
                                    .filter_map(|arg| arg.arg_type.clone())
                                    .collect::<Vec<String>>()
                                    .join(" ")
                            ),
                        });
                    }
                }
            }
        };

        // Hint commands
        let mut pos = pos;
        let candidates: Vec<&str> = if words.len() == 1 {
            // Complete command group
            self.commands
                .keys()
                .filter_map(|command| {
                    if command.starts_with(words[0]) && command != words[0] {
                        Some(command.as_str())
                    } else {
                        None
                    }
                })
                .collect()
        } else if words.len() == 2 {
            // Complete subcommand
            pos = pos - words[0].len() - 1;
            if let Some(command) = self.commands.get(words[0]) {
                command
                    .commands
                    .keys()
                    .filter_map(|command| {
                        if command.starts_with(words[1]) && command != words[1] {
                            Some(command.as_str())
                        } else {
                            None
                        }
                    })
                    .collect()
            } else {
                vec![]
            }
        } else {
            return None;
        };

        if candidates.len() == 1 {
            return Some(
                CommandHint {
                    display: String::from(candidates[0]),
                }
                .suffix(pos),
            );
        }
        None
    }
}