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
|
// bibiman - a TUI for managing BibLaTeX databases
// Copyright (C) 2024 lukeflo
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
/////
use color_eyre::eyre::Result;
use color_eyre::owo_colors::OwoColorize;
use lexopt::prelude::*;
use std::env;
use std::ffi::OsString;
use std::path::PathBuf;
use walkdir::WalkDir;
// struct for CLIArgs
#[derive(Debug, Default)]
pub struct CLIArgs {
pub helparg: bool,
pub versionarg: bool,
pub pos_args: Vec<PathBuf>,
}
impl CLIArgs {
pub fn parse_args() -> Result<CLIArgs, lexopt::Error> {
let mut args = CLIArgs::default();
let mut parser = lexopt::Parser::from_env();
while let Some(arg) = parser.next()? {
match arg {
Short('h') | Long("help") => args.helparg = true,
Short('v') | Long("version") => args.versionarg = true,
Value(pos_arg) => parse_files(&mut args, pos_arg),
_ => return Err(arg.unexpected()),
}
}
Ok(args)
}
}
pub fn parse_files(args: &mut CLIArgs, pos_arg: OsString) {
// convert to PathBuf to use methods for testing the path
let path = PathBuf::from(pos_arg);
// If pos arg is file, just push it to path vec
if path.is_file() {
args.pos_args.push(path);
// If pos arg is dir, walk dir and collect bibfiles
} else if path.is_dir() {
for file in WalkDir::new(path) {
let f = file.unwrap().into_path();
if f.is_file() && f.extension().unwrap() == "bib" {
args.pos_args.push(f)
}
}
} else {
println!(
"{} {}",
"The positional argument is neither a valid file, nor a path:"
.red()
.bold(),
path.as_os_str().to_string_lossy().bright_red().italic()
);
println!();
println!("{}", help_func());
std::process::exit(1)
}
}
pub fn help_func() -> String {
let help = format!(
"\
{} {}
USAGE:
bibiman [FLAGS] [file]
POSITIONAL ARGS:
<file> Path to .bib file
FLAGS:
-h, --help Show this help and exit
-v, --version Show the version and exit",
env!("CARGO_PKG_NAME"),
env!("CARGO_PKG_VERSION"),
);
help
}
pub fn version_func() -> String {
let version = format!(
"\
{} {}
{}
{}
Target Triple: {}",
env!("CARGO_PKG_NAME"),
env!("CARGO_PKG_VERSION"),
env!("CARGO_PKG_AUTHORS"),
env!("CARGO_PKG_LICENSE"),
env!("TARGET")
);
version
}
|
