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
115
116
117
118
119
120
121
122
|
// 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::path::PathBuf;
use walkdir::WalkDir;
// struct for CLIArgs
#[derive(Debug, Default, Clone)]
pub struct CLIArgs {
pub helparg: bool,
pub versionarg: bool,
pub pos_args: Vec<PathBuf>,
pub files: 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),
Value(pos_arg) => args.pos_args.push(pos_arg.into()),
_ => return Err(arg.unexpected()),
}
}
args.files = parse_files(args.pos_args.clone());
Ok(args)
}
}
fn parse_files(args: Vec<PathBuf>) -> Vec<PathBuf> {
// convert to PathBuf to use methods for testing the path
// let path = PathBuf::from(pos_arg);
let mut files: Vec<PathBuf> = Vec::new();
// If pos arg is file, just push it to path vec
for i in args {
if i.is_file() {
files.push(i);
// If pos arg is dir, walk dir and collect bibfiles
} else if i.is_dir() {
for file in WalkDir::new(i) {
let f = file.unwrap().into_path();
if f.is_file() && f.extension().unwrap_or_default() == "bib" {
files.push(f)
}
}
} else {
println!(
"{}\n{}",
"The positional argument is neither a valid file, nor a directory:"
.red()
.bold(),
i.as_os_str().to_string_lossy().bright_red().italic()
);
println!();
println!("{}", help_func());
std::process::exit(1)
}
}
files
}
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
}
|
