This commit is contained in:
Mike Cugini 2021-03-16 00:25:29 -04:00
parent 84c926b8fa
commit 107e692081

View File

@ -15,19 +15,17 @@
use std::error::Error;
use std::fs::File;
use std::io::{self, BufReader};
use std::io::prelude::*;
use std::io::{self, BufReader};
use argh::FromArgs;
#[derive(FromArgs)]
#[argh(
description = "\
#[argh(description = "\
chop off bits of file\n\n\
chop will take produce --count files of --lines lines from the beginning of a\n\
file or stdin. Any remaining lines will be written to a final catchall file.")]
struct Options {
/// number of lines in each chunk
#[argh(option, short = 'n')]
lines: usize,
@ -51,8 +49,6 @@ struct Options {
/// filename to read from, or "-" for stdin (default "-")
#[argh(positional, default = "String::from(\"-\")")]
filename: String,
}
struct AlphabeticSuffixGenerator {
@ -66,9 +62,12 @@ impl AlphabeticSuffixGenerator {
let suffix_length = ((count as f64).log(27f64) + 1.0).floor() as usize;
let mut suffix = vec![b'a'; suffix_length];
suffix[suffix_length-1] -= 1;
suffix[suffix_length - 1] -= 1;
AlphabeticSuffixGenerator{suffix, remaining: count}
AlphabeticSuffixGenerator {
suffix,
remaining: count,
}
}
}
@ -77,12 +76,12 @@ impl Iterator for AlphabeticSuffixGenerator {
fn next(&mut self) -> Option<String> {
if self.remaining == 0 {
return None
return None;
}
for idx in (0..self.suffix.len()).rev() {
if self.suffix[idx] < b'z' {
self.suffix[idx] += 1;
break
break;
} else {
self.suffix[idx] = b'a';
}
@ -92,17 +91,14 @@ impl Iterator for AlphabeticSuffixGenerator {
}
}
fn try_main() -> Result<(), Box<dyn Error>> {
let opts: Options = argh::from_env();
let input: Box<dyn Read> = match opts.filename.as_str() {
"-" => Box::new(io::stdin()),
path => {
match File::open(path) {
path => match File::open(path) {
Err(why) => panic!("failed to open {}: {}", path, why),
Ok(file) => Box::new(file),
}
},
};
@ -112,7 +108,9 @@ fn try_main() -> Result<(), Box<dyn Error>> {
};
let suffix_gen: Box<dyn Iterator<Item = String>> = if opts.numeric {
Box::new((opts.numeric_start..(opts.numeric_start + opts.count)).map(|num| format!("{}", num)))
Box::new(
(opts.numeric_start..(opts.numeric_start + opts.count)).map(|num| format!("{}", num)),
)
} else {
Box::new(AlphabeticSuffixGenerator::new(opts.count))
};
@ -126,8 +124,8 @@ fn try_main() -> Result<(), Box<dyn Error>> {
match reader.read_line(&mut line)? {
0 => {
eof = true;
break
},
break;
}
_ => {
out_file.write_all(line.as_bytes())?;
line.clear();
@ -140,8 +138,8 @@ fn try_main() -> Result<(), Box<dyn Error>> {
let mut out_file = File::create(format!("{}rest", prefix))?;
for result in reader.lines() {
// lines() strips newline characters, so add one
out_file.write_all((result?+"\n").as_bytes())?;
};
out_file.write_all((result? + "\n").as_bytes())?;
}
}
Ok(())