use std::collections::BTreeMap;

use clap::{App, Arg, ArgMatches, SubCommand};
use rust_util::util_clap::{Command, CommandError};
use yubikey::YubiKey;

pub struct CommandImpl;

impl Command for CommandImpl {
    fn name(&self) -> &str { "list" }

    fn subcommand<'a>(&self) -> App<'a, 'a> {
        SubCommand::with_name(self.name()).about("YubiKey list")
            .arg(Arg::with_name("json").long("json").help("JSON output"))
    }

    fn run(&self, _arg_matches: &ArgMatches, sub_arg_matches: &ArgMatches) -> CommandError {
        let json_output = sub_arg_matches.is_present("json");
        if json_output { rust_util::util_msg::set_logger_std_out(false); }

        let mut yk = opt_result!(YubiKey::open(), "YubiKey not found: {}");

        if json_output {
            let mut json = BTreeMap::<&'_ str, String>::new();
            json.insert("name", yk.name().to_string());
            json.insert("version", yk.version().to_string());
            json.insert("serial", yk.serial().0.to_string());
            if let Ok(pin_retries) = yk.get_pin_retries() {
                json.insert("pin_retries", pin_retries.to_string());
            }

            println!("{}", serde_json::to_string_pretty(&json).expect("Convert to JSON failed!"));
        } else {
            success!("Name: {}", yk.name());
            success!("Version: {}", yk.version());
            success!("Serial: {}", yk.serial().0);
            // success!("{:?}", yk.config());
            if let Ok(pin_retries) = yk.get_pin_retries() {
                success!("Pin retries: {}", pin_retries);
            }
        }

        Ok(None)
    }
}