summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/vtcol.rs184
1 files changed, 87 insertions, 97 deletions
diff --git a/src/vtcol.rs b/src/vtcol.rs
index 174b62f..665b600 100644
--- a/src/vtcol.rs
+++ b/src/vtcol.rs
@@ -1,5 +1,6 @@
-use lazy_static::lazy_static;
use std::sync::Mutex;
+use std::path::{Path, PathBuf};
+use lazy_static::lazy_static;
type Fd = libc::c_int;
@@ -95,7 +96,7 @@ enum Scheme
Default,
SolarizedDark,
SolarizedLight,
- Custom(Option<String>),
+ Custom(Option<PathBuf>),
}
impl<'a> std::fmt::Display for Scheme
@@ -106,13 +107,8 @@ impl<'a> std::fmt::Display for Scheme
Scheme::Default => "default",
Scheme::SolarizedDark => "solarized_dark",
Scheme::SolarizedLight => "solarized_light",
- Scheme::Custom(ref kind) => {
- let tmp: &str = match *kind {
- Some(ref fname) => fname.as_str(),
- None => "<read stdin>",
- };
- tmp
- },
+ Scheme::Custom(None) => "<read stdin>",
+ Scheme::Custom(Some(fname)) => fname.display(),
};
write!(f, "{}", id)
}
@@ -127,104 +123,106 @@ extern "C" {
#[derive(Debug)]
struct Job
{
- this: String, /* argv[0] */
scheme: Scheme, /* The color scheme to switch to. */
}
impl<'a> Job
{
- pub fn new() -> Job
+ pub fn from_argv() -> Job
{
- let argv: Vec<String> = std::env::args().collect();
- let this = argv[0].clone();
- let mut opts = getopts::Options::new();
-
- opts.optopt("s", "scheme", "predefined color scheme", "NAME");
- opts.optopt("d", "dump", "dump predefined scheme", "NAME");
- opts.optopt("f", "file", "apply scheme from file", "PATH");
- opts.optflag("v", "verbose", "enable verbose messages");
- opts.optflag("l", "list", "list available color schemes");
- opts.optflag("h", "help", "print this message");
-
- let matches = match opts.parse(&argv[1..]) {
- Ok(m) => m,
- Err(f) => panic!(f.to_string()),
- };
-
- if matches.opt_present("h") {
- Job::usage(&this, opts);
- unsafe { exit(0) };
- };
-
- if matches.opt_present("v") {
+ use clap::{App, Arg};
+
+ let matches = App::new(clap::crate_name!())
+ .version(clap::crate_version!())
+ .author(clap::crate_authors!())
+ .about(clap::crate_description!())
+ .arg(
+ Arg::with_name("scheme")
+ .short("s")
+ .long("scheme")
+ .value_name("NAME")
+ .help("predefined color scheme")
+ .takes_value(true),
+ )
+ .arg(
+ Arg::with_name("dump")
+ .short("d")
+ .long("dump")
+ .value_name("NAME")
+ .help("dump predefined scheme")
+ .takes_value(true),
+ )
+ .arg(
+ Arg::with_name("file")
+ .short("f")
+ .long("file")
+ .value_name("PATH")
+ .help("apply scheme from file")
+ .takes_value(true),
+ )
+ .arg(
+ Arg::with_name("verbose")
+ .short("v")
+ .long("verbose")
+ .help("enable extra diagnostics")
+ .takes_value(false),
+ )
+ .arg(
+ Arg::with_name("list")
+ .short("l")
+ .long("list")
+ .help("list available color schemes")
+ .takes_value(false),
+ )
+ .arg(
+ Arg::with_name("help")
+ .short("h")
+ .long("help")
+ .help("print this message")
+ .takes_value(false),
+ )
+ .get_matches();
+
+ if matches.is_present("v") {
VERBOSITY.lock().unwrap() = true;
}
- if matches.opt_present("l") {
+ if matches.is_present("l") {
Job::schemes();
- unsafe { exit(0) };
+ std::process::exit(0);
};
- if matches.opt_present("d") {
- match matches.opt_str("d") {
- None => {
- Job::usage(&this, opts);
- panic!("no color scheme given, aborting")
- },
- Some(name) => {
- let scm = Job::pick_scheme(&name);
- Job::dump(scm);
- unsafe { exit(0) };
- },
- };
+ if let Some(name) = matches.value_of("dump") {
+ let scm = Job::pick_scheme(name);
+ Job::dump(scm);
+ std::process::exit(0);
}
- let scheme = if matches.opt_present("f") {
- match matches.opt_str("f") {
- None => {
- Job::usage(&this, opts);
- panic!("no file name specified, aborting")
+ let scheme = match matches.value_of("file") {
+ Some("-") => Job::scheme_from_stdin(),
+ Some(fname) => Scheme::Custom(Some(PathBuf::from(fname))),
+ None =>
+ match matches.value_of("scheme") {
+ Some("-") | None => Job::scheme_from_stdin(),
+ Some(name) => Job::pick_scheme(name),
},
- Some(fname) =>
- if fname == "-" {
- Job::scheme_from_stdin()
- } else {
- Scheme::Custom(Some(fname.clone()))
- },
- }
- } else {
- match matches.opt_str("s") {
- None => Job::scheme_from_stdin(),
- Some(name) =>
- if name == "-" {
- Job::scheme_from_stdin()
- } else {
- Job::pick_scheme(&name)
- },
- }
- }; /* [let scheme] */
+ };
- Job { this, scheme }
+ Job { scheme }
}
- fn pick_scheme<'b>(name: &String) -> Scheme
+ fn pick_scheme<'b>(name: &str) -> Scheme
{
- match name.as_str() {
+ match name {
"solarized" | "solarized_dark" | "sd" => Scheme::SolarizedDark,
"solarized_light" | "sl" => Scheme::SolarizedLight,
"default" | "normal" => Scheme::Default,
- _any => Scheme::Custom(Some(name.clone())),
+ _any => Scheme::Custom(Some(PathBuf::from(name))),
}
}
fn scheme_from_stdin() -> Scheme { Scheme::Custom(None) }
- fn usage(this: &String, opts: getopts::Options)
- {
- let brief = format!("usage: {} [options]", this);
- print!("{}", opts.usage(&brief));
- }
-
fn schemes()
{
println!("Available color schemes:");
@@ -240,12 +238,9 @@ impl<'a> Job
Scheme::Default => Job::dump_scheme(&DEFAULT_COLORS),
Scheme::SolarizedDark => Job::dump_scheme(&SOLARIZED_COLORS_DARK),
Scheme::SolarizedLight => Job::dump_scheme(&SOLARIZED_COLORS_LIGHT),
- Scheme::Custom(kind) =>
- match kind {
- Some(fname) =>
+ Scheme::Custom(None) => Job::dump_palette(Palette::from_stdin()),
+ Scheme::Custom(Some(fname)) =>
Job::dump_palette(Palette::from_file(&fname)),
- None => Job::dump_palette(Palette::from_stdin()),
- },
}
}
@@ -431,12 +426,11 @@ impl Palette
/* [Palette::from_buffered_reader] */
- pub fn from_file(fname: &String) -> Palette
+ pub fn from_file(fname: &Path) -> Palette
{
/* Check if file exists
*/
- let path = std::path::Path::new(fname);
- let file = match std::fs::File::open(&path) {
+ let file = match std::fs::File::open(&fname) {
Err(e) => panic!("failed to open {} as file ({})", fname, e),
Ok(f) => f,
};
@@ -583,20 +577,16 @@ fn clear_term(fd: Fd)
fn main()
{
- let job = Job::new();
+ let job = Job::from_argv();
vrb!("job parms: {:?}", job);
- let mut pal: Palette = {
+ let mut pal: Palette =
match job.scheme {
Scheme::Default => Palette::new(&DEFAULT_COLORS),
Scheme::SolarizedDark => Palette::new(&SOLARIZED_COLORS_DARK),
Scheme::SolarizedLight => Palette::new(&SOLARIZED_COLORS_LIGHT),
- Scheme::Custom(ref kind) =>
- match *kind {
- Some(ref fname) => Palette::from_file(fname),
- None => Palette::from_stdin(),
- },
- }
- };
+ Scheme::Custom(None) => Palette::from_stdin(),
+ Scheme::Custom(Some(fname)) => Palette::from_file(&fname),
+ };
vrb!("Using palette:");
vrb!("{}", pal);
let fd = get_console_fd(None).unwrap();
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-22 05:10:23 +0000