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

use crate::script_type::ScriptType;
use serde::Serialize;
use structopt::clap::AppSettings::AllowLeadingHyphen;
use structopt::StructOpt;

#[derive(StructOpt, Debug, Serialize)]
pub struct Types {
    #[structopt(subcommand)]
    pub subcmd: Option<TypesSubs>,
}

#[derive(StructOpt, Debug, Serialize)]
#[structopt(settings = &[AllowLeadingHyphen])]
pub enum TypesSubs {
    #[structopt(external_subcommand)]
    Other(Vec<String>),
    LS,
    Template {
        #[structopt(long, short)]
        edit: bool,
        ty: ScriptType,
    },
}

impl Types {
    pub fn sanitize(&mut self) {
        match self.subcmd.as_ref() {
            None => self.subcmd = Some(TypesSubs::LS),
            Some(TypesSubs::Other(args)) => {
                let args = std::array::IntoIter::new(["types", "template"])
                    .chain(args.iter().map(|s| s.as_str()));
                self.subcmd = Some(TypesSubs::from_iter(args));
            }
            _ => (),
        }
    }
}