feat: v0.4.1, enchance config outputs
This commit is contained in:
@@ -2,7 +2,7 @@ use std::cmp::Ordering;
|
||||
use std::collections::HashMap;
|
||||
|
||||
use clap::Args;
|
||||
use rust_util::XResult;
|
||||
use rust_util::{iff, information, warning, XResult};
|
||||
use tabled::{Table, Tabled};
|
||||
use tabled::settings::Style;
|
||||
|
||||
@@ -33,16 +33,68 @@ impl PartialOrd for ConfigProfile {
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Tabled)]
|
||||
pub struct ConfigEnvelop {
|
||||
pub r#type: String,
|
||||
pub sid: String,
|
||||
pub kid: String,
|
||||
pub desc: String,
|
||||
pub args: String,
|
||||
}
|
||||
|
||||
#[derive(Debug, Args)]
|
||||
pub struct CmdConfig {
|
||||
/// Show KID
|
||||
#[arg(long)]
|
||||
pub show_kid: bool,
|
||||
/// Encryption profile (use default when --key-filter is assigned)
|
||||
#[arg(long, short = 'p')]
|
||||
pub profile: Option<String>,
|
||||
/// Encryption key filter (key_id or type:TYPE(e.g. ecdh, pgp, ecdh-p384, pgp-ed25519), multiple joined by ',', ALL for all)
|
||||
#[arg(long, short = 'k')]
|
||||
pub key_filter: Option<String>,
|
||||
}
|
||||
|
||||
pub fn config(cmd_version: CmdConfig) -> XResult<()> {
|
||||
let config = TinyEncryptConfig::load(TINY_ENC_CONFIG_FILE)?;
|
||||
|
||||
if cmd_version.profile.is_some() || cmd_version.key_filter.is_some() {
|
||||
return config_key_filter(&cmd_version, &config);
|
||||
}
|
||||
config_profiles(&cmd_version, &config)
|
||||
}
|
||||
|
||||
fn config_key_filter(cmd_version: &CmdConfig, config: &TinyEncryptConfig) -> XResult<()> {
|
||||
let envelops = config.find_envelops(&cmd_version.profile, &cmd_version.key_filter)?;
|
||||
if envelops.is_empty() { warning!("Found no envelops"); }
|
||||
information!("Found {} envelops", envelops.len());
|
||||
let mut config_envelops = vec![];
|
||||
for envelop in envelops {
|
||||
config_envelops.push(ConfigEnvelop {
|
||||
r#type: envelop.r#type.get_name().to_string(),
|
||||
sid: envelop.sid.as_ref().map(ToString::to_string).unwrap_or_else(|| "-".to_string()),
|
||||
kid: process_kid(&envelop.kid),
|
||||
desc: envelop.desc.as_ref().map(ToString::to_string).unwrap_or_else(|| "-".to_string()),
|
||||
args: envelop.args.as_ref().map(|a| format!("[{}]", a.join(", "))).unwrap_or_else(|| "-".to_string()),
|
||||
});
|
||||
}
|
||||
let mut table = Table::new(config_envelops);
|
||||
table.with(Style::sharp());
|
||||
println!("{}", table);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn process_kid(kid: &str) -> String {
|
||||
if kid.len() < 10 {
|
||||
kid.to_string()
|
||||
} else {
|
||||
kid.chars().enumerate()
|
||||
.filter(|(i, _c)| *i <= 50)
|
||||
.map(|(i, c)| iff!(i >= 48, '.', c)).collect()
|
||||
}
|
||||
}
|
||||
|
||||
fn config_profiles(cmd_version: &CmdConfig, config: &TinyEncryptConfig) -> XResult<()> {
|
||||
let mut reverse_map = HashMap::new();
|
||||
for (p, v) in &config.profiles {
|
||||
let p = p;
|
||||
|
||||
@@ -2,7 +2,7 @@ use std::cmp::Ordering;
|
||||
use std::collections::HashMap;
|
||||
use std::fs;
|
||||
|
||||
use rust_util::{debugging, opt_result, simple_error, XResult};
|
||||
use rust_util::{debugging, iff, opt_result, simple_error, XResult};
|
||||
use rust_util::util_file::resolve_file_path;
|
||||
use serde::{Deserialize, Serialize};
|
||||
|
||||
@@ -20,6 +20,7 @@ use crate::spec::TinyEncryptEnvelopType;
|
||||
/// },
|
||||
/// {
|
||||
/// "type": "ecdh",
|
||||
/// "sid": "SHORT-ID-1",
|
||||
/// "kid": "KID-2",
|
||||
/// "desc": "this is key 002",
|
||||
/// "publicPart": "04..."
|
||||
@@ -83,11 +84,7 @@ impl TinyEncryptConfig {
|
||||
|
||||
pub fn find_by_kid(&self, kid: &str) -> Option<&TinyEncryptConfigEnvelop> {
|
||||
let config_envelops = self.find_by_kid_or_filter(kid, |_| false);
|
||||
if config_envelops.is_empty() {
|
||||
None
|
||||
} else {
|
||||
Some(config_envelops[0])
|
||||
}
|
||||
iff!(config_envelops.is_empty(), None, Some(config_envelops[0]))
|
||||
}
|
||||
|
||||
pub fn find_by_kid_or_type(&self, k_filter: &str) -> Vec<&TinyEncryptConfigEnvelop> {
|
||||
@@ -99,11 +96,9 @@ impl TinyEncryptConfig {
|
||||
pub fn find_by_kid_or_filter<F>(&self, kid: &str, f: F) -> Vec<&TinyEncryptConfigEnvelop>
|
||||
where F: Fn(&TinyEncryptConfigEnvelop) -> bool {
|
||||
self.envelops.iter().filter(|e| {
|
||||
if e.kid == kid {
|
||||
return true;
|
||||
}
|
||||
if e.kid == kid { return true; }
|
||||
if let Some(sid) = &e.sid {
|
||||
return sid == kid;
|
||||
if sid == kid { return true; }
|
||||
}
|
||||
f(e)
|
||||
}).collect()
|
||||
|
||||
Reference in New Issue
Block a user