aopt 1.4.5

A flexible and typed getopt tools
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 std::borrow::Cow;

use crate::str::CowStrUtils;
use crate::{error, Error};

pub trait OptValidator {
    type Error: Into<Error>;

    /// Check the option string.
    fn check(&mut self, name: &str) -> Result<bool, Self::Error>;

    /// Split the option string into prefix and name.
    fn split<'a>(&self, name: &Cow<'a, str>) -> Result<(Cow<'a, str>, Cow<'a, str>), Self::Error>;
}

pub trait PrefixedValidator {
    type Error: Into<Error>;

    /// Register the prefix to current validator.
    fn reg_prefix(&mut self, val: &str) -> Result<(), Self::Error>;

    /// Unregister the prefix to current validator.
    fn unreg_prefix(&mut self, val: &str) -> Result<(), Self::Error>;
}

/// A prefixed validator used in [`Policy`](crate::parser::Policy) and [`InvokeGuess`](crate::guess::InvokeGuess).
///
/// The default prefixes are `--/`, `--`, `-/`, `-` and `/`(only for windows).
#[derive(Debug, Clone)]
pub struct PrefixOptValidator(Vec<String>);

#[cfg(target_os = "windows")]
impl Default for PrefixOptValidator {
    fn default() -> Self {
        Self::new(
            ["--/", "--", "-/", "-", "/"]
                .map(|v| v.to_string())
                .to_vec(),
        )
    }
}

#[cfg(not(target_os = "windows"))]
impl Default for PrefixOptValidator {
    fn default() -> Self {
        Self::new(["--/", "--", "-/", "-"].map(|v| v.to_string()).to_vec())
    }
}

impl PrefixOptValidator {
    pub fn new(prefix: Vec<String>) -> Self {
        // sort the prefix by length
        let mut _self = Self(prefix);

        _self.sort_prefix();
        _self
    }

    fn sort_prefix(&mut self) {
        self.0.sort_by_key(|b| std::cmp::Reverse(b.len()));
    }

    pub fn add_prefix(&mut self, prefix: &str) -> &mut Self {
        self.0.push(prefix.to_string());
        self.sort_prefix();
        self
    }
}

impl OptValidator for PrefixOptValidator {
    type Error = Error;

    fn check(&mut self, name: &str) -> Result<bool, Self::Error> {
        for prefix in self.0.iter() {
            if name.starts_with(prefix) {
                return Ok(true);
            }
        }
        Ok(false)
    }

    fn split<'a>(&self, name: &Cow<'a, str>) -> Result<(Cow<'a, str>, Cow<'a, str>), Self::Error> {
        for prefix in self.0.iter() {
            if name.starts_with(prefix) {
                return Ok(name.split_at(prefix.len()));
            }
        }
        Err(error!(
            "can not split the {}: invalid option name string",
            name
        ))
    }
}

impl PrefixedValidator for PrefixOptValidator {
    type Error = Error;

    fn reg_prefix(&mut self, val: &str) -> Result<(), Self::Error> {
        if self.0.iter().any(|v| v == val) {
            Err(error!("the prefix already exist"))
        } else {
            self.add_prefix(val);
            Ok(())
        }
    }

    fn unreg_prefix(&mut self, val: &str) -> Result<(), Self::Error> {
        if let Some(index) = self.0.iter().position(|v| v == val) {
            self.0.remove(index);
            Ok(())
        } else {
            Err(error!("the prefix not exist"))
        }
    }
}