shitbox/corebox/commands/fold/mod.rs

117 lines
3.7 KiB
Rust
Raw Permalink Normal View History

2023-01-08 11:40:08 -05:00
use super::Cmd;
use clap::{value_parser, Arg, ArgAction, ArgMatches, Command};
2023-01-10 18:28:33 -05:00
use std::{
fs::File,
io::{self, BufRead, BufReader},
};
2023-01-08 11:40:08 -05:00
use textwrap::{fill, wrap_algorithms::WrapAlgorithm};
#[derive(Debug)]
pub struct Fold;
2023-01-08 11:40:08 -05:00
impl Cmd for Fold {
fn cli(&self) -> Command {
Command::new("fold")
2023-01-08 11:40:08 -05:00
.about("Wrap each input line to fit in specified width")
.author("Nathan Fisher")
.after_long_help("With no FILE, or when FILE is -, read standard input")
.args([
Arg::new("FILE").help("The input file to use").num_args(0..),
Arg::new("BYTES")
.help("Count bytes rather than columns")
.short('b')
.long("bytes")
.action(ArgAction::SetTrue),
Arg::new("WORDS")
.help("Break at spaces")
.short('s')
.long("spaces")
.action(ArgAction::SetTrue),
Arg::new("OPTIMAL")
.help("Optimal fit")
.long_help(
"Uses a look ahad algorithm to avoid unnecessarily short or long lines",
)
.short('o')
.long("optimal")
.action(ArgAction::SetTrue)
.conflicts_with_all(["BYTES", "WORDS"]),
Arg::new("WIDTH")
.help("Use width columns")
.short('w')
.long("width")
.default_value("80")
.num_args(1)
.value_parser(value_parser!(usize)),
])
}
fn run(&self, matches: &ArgMatches) -> Result<(), Box<dyn std::error::Error>> {
2023-01-08 11:40:08 -05:00
let files = match matches.get_many::<String>("FILE") {
Some(c) => c.map(String::to_string).collect(),
None => vec!["-".to_string()],
};
for file in files {
if file.as_str() == "-" {
wrap_stdin(matches)?;
} else {
wrap_file(&file, matches)?;
}
}
if matches.get_flag("BYTES") {
println!();
}
Ok(())
}
fn path(&self) -> Option<shitbox::Path> {
Some(shitbox::Path::UsrBin)
2023-01-08 11:40:08 -05:00
}
}
fn wrap_line(line: &str, args: &ArgMatches) {
let width = args.get_one("WIDTH").map_or(80, |x| *x);
2023-01-08 11:40:08 -05:00
if args.get_flag("OPTIMAL") {
let line = line.replace('\t', " ");
let opts = textwrap::Options::new(width).wrap_algorithm(WrapAlgorithm::new_optimal_fit());
println!("{}", fill(line.trim_end(), opts));
} else if args.get_flag("WORDS") {
let line = line.replace('\t', " ");
let opts = textwrap::Options::new(width).wrap_algorithm(WrapAlgorithm::FirstFit);
println!("{}", fill(line.trim_end(), opts));
} else if args.get_flag("BYTES") {
for (index, b) in line.as_bytes().iter().enumerate() {
if index % width == 0 {
println!();
}
print!("{}", *b as char);
}
} else {
let line = line
.chars()
.collect::<Vec<char>>()
.chunks(width)
.map(|c| c.iter().collect::<String>())
.collect::<Vec<String>>();
for line in &line {
2023-01-14 02:34:27 -05:00
println!("{line}");
2023-01-08 11:40:08 -05:00
}
}
}
fn wrap_stdin(args: &ArgMatches) -> Result<(), io::Error> {
for line in io::stdin().lock().lines() {
wrap_line(&line?, args);
}
Ok(())
}
fn wrap_file(file: &str, args: &ArgMatches) -> Result<(), io::Error> {
let fd = File::open(file)?;
let buf = BufReader::new(fd);
for line in buf.lines() {
wrap_line(&line?, args);
}
Ok(())
}