Added rev applet

This commit is contained in:
Nathan Fisher 2023-01-07 11:44:35 -05:00
parent e3071e5707
commit 3857265728
2 changed files with 106 additions and 3 deletions

View File

@ -22,6 +22,7 @@ pub mod mountpoint;
mod mv;
pub mod nologin;
mod pwd;
pub mod rev;
mod rm;
mod rmdir;
pub mod shitbox;
@ -32,7 +33,7 @@ pub mod r#true;
pub use {
self::hostname::Hostname, base32::Base32, base64::Base64, bootstrap::Bootstrap,
dirname::Dirname, echo::Echo, factor::Factor, head::Head, mountpoint::Mountpoint,
nologin::Nologin, r#false::False, r#true::True, shitbox::Shitbox, sleep::Sleep,
nologin::Nologin, r#false::False, r#true::True, rev::Rev, shitbox::Shitbox, sleep::Sleep,
};
pub trait Cmd: fmt::Debug + Sync {
@ -54,6 +55,7 @@ pub fn get(name: &str) -> Option<Box<dyn Cmd>> {
"head" => Some(Box::new(Head::default())),
"mountpoint" => Some(Box::new(Mountpoint::default())),
"nologin" => Some(Box::new(Nologin::default())),
"rev" => Some(Box::new(Rev::default())),
"shitbox" => Some(Box::new(Shitbox::default())),
"sleep" => Some(Box::new(Sleep::default())),
"true" => Some(Box::new(True::default())),
@ -61,7 +63,7 @@ pub fn get(name: &str) -> Option<Box<dyn Cmd>> {
}
}
pub static COMMANDS: [&'static str; 14] = [
pub static COMMANDS: [&'static str; 15] = [
"base32",
"base64",
"bootstrap",
@ -73,7 +75,8 @@ pub static COMMANDS: [&'static str; 14] = [
"hostname",
"mountpoint",
"nologin",
"true",
"rev",
"sleep",
"shitbox",
"true",
];

100
src/cmd/rev/mod.rs Normal file
View File

@ -0,0 +1,100 @@
use crate::Cmd;
use clap::{Arg, Command, ArgAction};
use std::{
fs::File,
io::{self, BufRead, BufReader, ErrorKind, Write},
};
use termcolor::{Color, ColorChoice, ColorSpec, StandardStream, WriteColor};
#[derive(Debug)]
pub struct Rev {
name: &'static str,
path: Option<crate::Path>,
}
impl Default for Rev {
fn default() -> Self {
Self {
name: "rev",
path: Some(crate::Path::UsrBin),
}
}
}
impl Cmd for Rev {
fn name(&self) -> &str {
self.name
}
fn cli(&self) -> clap::Command {
Command::new(self.name)
.about("reverse lines characterwise")
.author("Nathan Fisher")
.args([
Arg::new("verbose")
.short('v')
.long("verbose")
.help("print a header between each file")
.action(ArgAction::SetTrue),
Arg::new("color")
.short('c')
.long("color")
.value_parser(["always", "ansi", "auto", "never"]),
Arg::new("file")
.help("if file is '-' read from stdin")
.num_args(0..),
])
}
fn run(&self, matches: Option<&clap::ArgMatches>) -> Result<(), Box<dyn std::error::Error>> {
let Some(matches) = matches else {
return Err(Box::new(io::Error::new(ErrorKind::Other, "No input")));
};
let files: Vec<_> = match matches.get_many::<String>("file") {
Some(c) => c.cloned().collect(),
None => vec![String::from("-")],
};
let color = match matches.get_one::<String>("color").map(String::as_str) {
Some("always") => ColorChoice::Always,
Some("ansi") => ColorChoice::AlwaysAnsi,
Some("auto") => {
if atty::is(atty::Stream::Stdout) {
ColorChoice::Auto
} else {
ColorChoice::Never
}
}
_ => ColorChoice::Never,
};
for (index, file) in files.into_iter().enumerate() {
if matches.get_flag("verbose") {
let mut stdout = StandardStream::stdout(color);
stdout.set_color(ColorSpec::new().set_fg(Some(Color::Green)))?;
match index {
0 => writeln!(stdout, "===> {file} <==="),
_ => writeln!(stdout, "\n===> {file} <==="),
}?;
stdout.reset()?;
}
rev_file(&file)?;
}
Ok(())
}
fn path(&self) -> Option<crate::Path> {
self.path
}
}
fn rev_file(file: &str) -> Result<(), io::Error> {
let reader: Box<dyn BufRead> = if file == "-" {
Box::new(BufReader::new(io::stdin()))
} else {
let buf = File::open(file)?;
Box::new(BufReader::new(buf))
};
for line in reader.lines() {
println!("{}", line.unwrap().chars().rev().collect::<String>());
}
Ok(())
}