pmsh 0.0.1

A custom shell written in Rust
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

use std::collections::HashMap;
use std::sync::RwLock;

#[derive(Debug, Clone)]
pub struct CompSpec {
    pub wordlist: Option<String>,
}

pub struct CompletionRegistry {
    specs: Option<HashMap<String, CompSpec>>,
}

impl CompletionRegistry {
    pub const fn new() -> Self {
        Self { specs: None }
    }

    fn ensure_init(&mut self) {
        if self.specs.is_none() {
            self.specs = Some(HashMap::new());
        }
    }

    pub fn register(&mut self, command: String, spec: CompSpec) {
        self.ensure_init();
        self.specs.as_mut().unwrap().insert(command, spec);
    }

    pub fn remove(&mut self, command: &str) {
        if let Some(specs) = &mut self.specs {
            specs.remove(command);
        }
    }

    pub fn get(&self, command: &str) -> Option<CompSpec> {
        self.specs
            .as_ref()
            .and_then(|specs| specs.get(command).cloned())
    }
}

pub static COMP_REGISTRY: RwLock<CompletionRegistry> = RwLock::new(CompletionRegistry::new());