fpr_cli_util/
lib.rs

1use core::fmt;
2pub use fpr_cli::*;
3use std::{
4    env::{args, var},
5    ffi::OsStr,
6    fs::{self, File},
7    io::{BufReader, Read},
8    panic::Location,
9    path::Path,
10};
11
12#[macro_export]
13macro_rules! version {
14    ($c:ty) => {
15        mod build {
16            include!(concat!(env!("OUT_DIR"), "/built.rs"));
17        }
18        #[derive(Args)]
19        #[args(f = version, desc = "Print version.")]
20        pub struct Version {}
21        fn version(_: &$c, _: Version) -> Result<(), String> {
22            use build::*;
23            println!(
24                "{} v{PKG_VERSION}: {TARGET}, {PROFILE}",
25                env!("CARGO_PKG_NAME"),
26            );
27            if !PKG_DESCRIPTION.is_empty() {
28                println!("{PKG_DESCRIPTION}");
29            }
30            if !PKG_HOMEPAGE.is_empty() {
31                println!("Homeage: {PKG_HOMEPAGE}");
32            }
33            if !PKG_LICENSE.is_empty() {
34                println!("License: {PKG_LICENSE}");
35            }
36            if !PKG_AUTHORS.is_empty() {
37                println!("Authors: {PKG_AUTHORS}");
38            }
39            if !PKG_REPOSITORY.is_empty() {
40                println!(
41                    "Repository: {PKG_REPOSITORY}, commit = {}, dirty = {}",
42                    GIT_COMMIT_HASH_SHORT.unwrap_or("none"),
43                    GIT_DIRTY
44                        .map(|e| e.to_string())
45                        .unwrap_or("none".to_string()),
46                );
47            }
48            Ok(())
49        }
50    };
51}
52
53pub trait Ctx
54where
55    Self: Sized,
56{
57    fn new(s: &str) -> Result<Self, String>;
58}
59
60fn irun<C: Ctx, M: Acts<C>, K: AsRef<OsStr>, P: AsRef<str>>(
61    config_path_override_env: K,
62    config_path_default_home_rel: P,
63) -> Result<(), String> {
64    let c = C::new(&var(config_path_override_env).unwrap_or_else(|_| {
65        let home = var("HOME").expect("Failed to get variable '$HOME'");
66        format!("{home}/{}", config_path_default_home_rel.as_ref())
67    }))?;
68    M::run(&c).map_err(|e| format!("{e}"))?;
69    Ok(())
70}
71
72pub fn run<C: Ctx, M: Acts<C>, K: AsRef<OsStr>, P: AsRef<str>>(
73    config_path_override_env: K,
74    config_path_default_home_rel: P,
75) {
76    if let Err(e) = irun::<C, M, K, P>(config_path_override_env, config_path_default_home_rel) {
77        eprintln!("{e}");
78    }
79}
80
81fn irund<C: Ctx, M: Args<C>, V: Args<C>, K: AsRef<OsStr>, P: AsRef<str>>(
82    config_path_override_env: K,
83    config_path_default_home_rel: P,
84) -> Result<(), String> {
85    let c = C::new(&var(config_path_override_env).unwrap_or_else(|_| {
86        let home = var("HOME").expect("Failed to get variable '$HOME'");
87        format!("{home}/{}", config_path_default_home_rel.as_ref())
88    }))?;
89    let args: Vec<_> = args().collect();
90    let args: Vec<_> = args.iter().skip(1).map(|e| e.as_str()).collect();
91    if args.iter().any(|e| *e == "--version") {
92        V::next_impl(&c, &args).map_err(|e| format!("{e}"))?;
93    } else {
94        M::next_impl(&c, &args).map_err(|e| format!("{e}"))?;
95    }
96    Ok(())
97}
98pub fn rund<M: Args<C>, V: Args<C>, C: Ctx, K: AsRef<OsStr>, P: AsRef<str>>(
99    config_path_override_env: K,
100    config_path_default_home_rel: P,
101) {
102    if let Err(e) = irund::<C, M, V, K, P>(config_path_override_env, config_path_default_home_rel) {
103        eprintln!("{e}");
104    }
105}
106
107pub fn reader<P: AsRef<Path>>(p: P) -> Result<BufReader<Box<dyn Read>>, String> {
108    let f: Box<dyn Read> = Box::new(File::open(p.as_ref()).map_err(|e| {
109        format!(
110            "Failed to open file '{}' because '{e}'.",
111            p.as_ref().to_string_lossy()
112        )
113    })?);
114    Ok(BufReader::new(f))
115}
116pub fn from_toml<P: AsRef<Path>, O: serde::de::DeserializeOwned>(p: P) -> Result<O, String> {
117    let mut buf = String::new();
118    reader(p.as_ref())?.read_to_string(&mut buf).map_err(|e| {
119        format!(
120            "Error while reading '{}' because '{e}'.",
121            p.as_ref().to_string_lossy()
122        )
123    })?;
124    Ok(toml::from_str(&buf).map_err(|e| {
125        format!(
126            "Failed to parse '{}' because '{e}'.",
127            p.as_ref().to_string_lossy()
128        )
129    })?)
130}
131#[track_caller]
132pub fn write_toml<P: AsRef<Path>, O: serde::Serialize + fmt::Debug>(
133    o: &O,
134    p: P,
135) -> Result<(), String> {
136    let l = Location::caller();
137    let s = toml::to_string_pretty(o)
138        .map_err(|e| format!("Failed to serialize '{o:#?}' because '{e}' at {l}"))?;
139    fs::write(&p, s).map_err(|e| {
140        format!(
141            "Failed to write to '{}' because '{e}'",
142            p.as_ref().to_string_lossy()
143        )
144    })?;
145    Ok(())
146}
147#[track_caller]
148pub fn write_ctoml<P: AsRef<Path>, O: serde::Serialize + fmt::Debug>(
149    o: &O,
150    p: P,
151) -> Result<(), String> {
152    let l = Location::caller();
153    let s = toml::to_string(o)
154        .map_err(|e| format!("Failed to serialize '{o:#?}' because '{e}' at {l}"))?;
155    fs::write(&p, s).map_err(|e| {
156        format!(
157            "Failed to write to '{}' because '{e}'",
158            p.as_ref().to_string_lossy()
159        )
160    })?;
161    Ok(())
162}
fpr_cli_util/lib.rs

fpr_cli_util/
lib.rs
