commit
94208e39fc
@ -0,0 +1 @@ |
|||||||
|
/target |
@ -0,0 +1,87 @@ |
|||||||
|
# This file is automatically @generated by Cargo. |
||||||
|
# It is not intended for manual editing. |
||||||
|
[[package]] |
||||||
|
name = "argh" |
||||||
|
version = "0.1.4" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "91792f088f87cdc7a2cfb1d617fa5ea18d7f1dc22ef0e1b5f82f3157cdc522be" |
||||||
|
dependencies = [ |
||||||
|
"argh_derive", |
||||||
|
"argh_shared", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "argh_derive" |
||||||
|
version = "0.1.4" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "c4eb0c0c120ad477412dc95a4ce31e38f2113e46bd13511253f79196ca68b067" |
||||||
|
dependencies = [ |
||||||
|
"argh_shared", |
||||||
|
"heck", |
||||||
|
"proc-macro2", |
||||||
|
"quote", |
||||||
|
"syn", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "argh_shared" |
||||||
|
version = "0.1.4" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "781f336cc9826dbaddb9754cb5db61e64cab4f69668bd19dcc4a0394a86f4cb1" |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "chop" |
||||||
|
version = "0.1.0" |
||||||
|
dependencies = [ |
||||||
|
"argh", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "heck" |
||||||
|
version = "0.3.2" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "87cbf45460356b7deeb5e3415b5563308c0a9b057c85e12b06ad551f98d0a6ac" |
||||||
|
dependencies = [ |
||||||
|
"unicode-segmentation", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "proc-macro2" |
||||||
|
version = "1.0.24" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "1e0704ee1a7e00d7bb417d0770ea303c1bccbabf0ef1667dae92b5967f5f8a71" |
||||||
|
dependencies = [ |
||||||
|
"unicode-xid", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "quote" |
||||||
|
version = "1.0.9" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "c3d0b9745dc2debf507c8422de05d7226cc1f0644216dfdfead988f9b1ab32a7" |
||||||
|
dependencies = [ |
||||||
|
"proc-macro2", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "syn" |
||||||
|
version = "1.0.64" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "3fd9d1e9976102a03c542daa2eff1b43f9d72306342f3f8b3ed5fb8908195d6f" |
||||||
|
dependencies = [ |
||||||
|
"proc-macro2", |
||||||
|
"quote", |
||||||
|
"unicode-xid", |
||||||
|
] |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "unicode-segmentation" |
||||||
|
version = "1.7.1" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "bb0d2e7be6ae3a5fa87eed5fb451aff96f2573d2694942e40543ae0bbe19c796" |
||||||
|
|
||||||
|
[[package]] |
||||||
|
name = "unicode-xid" |
||||||
|
version = "0.2.1" |
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||||
|
checksum = "f7fe0bb3479651439c9112f72b6c505038574c9fbb575ed1bf3b797fa39dd564" |
@ -0,0 +1,10 @@ |
|||||||
|
[package] |
||||||
|
name = "chop" |
||||||
|
version = "0.1.0" |
||||||
|
authors = ["Mike Cugini <mike@betamike.com>"] |
||||||
|
edition = "2018" |
||||||
|
|
||||||
|
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html |
||||||
|
|
||||||
|
[dependencies] |
||||||
|
argh = "0.1.4" |
@ -0,0 +1,141 @@ |
|||||||
|
use std::error::Error; |
||||||
|
use std::fs::File; |
||||||
|
use std::io::{self, BufReader}; |
||||||
|
use std::io::prelude::*; |
||||||
|
|
||||||
|
use argh::FromArgs; |
||||||
|
|
||||||
|
#[derive(FromArgs)] |
||||||
|
#[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, |
||||||
|
|
||||||
|
/// count of chunks to produce (default 1)
|
||||||
|
#[argh(option, short = 'c', default = "1")] |
||||||
|
count: usize, |
||||||
|
|
||||||
|
/// optional prefix to use for generated files (default "")
|
||||||
|
#[argh(option, short = 'p', default = "String::from(\"\")")] |
||||||
|
prefix: String, |
||||||
|
|
||||||
|
/// use numeric suffixes starting with 0, not alphabetic
|
||||||
|
#[argh(switch, short = 'd')] |
||||||
|
numeric: bool, |
||||||
|
|
||||||
|
/// use numeric suffixes starting with 0, not alphabetic
|
||||||
|
#[argh(option, long = "numeric-start", default = "0")] |
||||||
|
numeric_start: usize, |
||||||
|
|
||||||
|
/// filename to read from, or "-" for stdin (default "-")
|
||||||
|
#[argh(positional, default = "String::from(\"-\")")] |
||||||
|
filename: String, |
||||||
|
|
||||||
|
|
||||||
|
} |
||||||
|
|
||||||
|
struct AlphabeticSuffixGenerator { |
||||||
|
suffix: Vec<u8>, |
||||||
|
remaining: usize, |
||||||
|
} |
||||||
|
|
||||||
|
impl AlphabeticSuffixGenerator { |
||||||
|
fn new(count: usize) -> AlphabeticSuffixGenerator { |
||||||
|
// given 27 letters, minimum number of digits is floor( Log 27 (count) + 1 )
|
||||||
|
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; |
||||||
|
|
||||||
|
AlphabeticSuffixGenerator{suffix, remaining: count} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
impl Iterator for AlphabeticSuffixGenerator { |
||||||
|
type Item = String; |
||||||
|
|
||||||
|
fn next(&mut self) -> Option<String> { |
||||||
|
if self.remaining == 0 { |
||||||
|
return None |
||||||
|
} |
||||||
|
for idx in (0..self.suffix.len()).rev() { |
||||||
|
if self.suffix[idx] < b'z' { |
||||||
|
self.suffix[idx] += 1; |
||||||
|
break
|
||||||
|
} else { |
||||||
|
self.suffix[idx] = b'a'; |
||||||
|
} |
||||||
|
} |
||||||
|
self.remaining -= 1; |
||||||
|
Some(String::from_utf8(self.suffix.to_vec()).expect("invalid suffix generated")) |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
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) { |
||||||
|
Err(why) => panic!("failed to open {}: {}", path, why), |
||||||
|
Ok(file) => Box::new(file), |
||||||
|
} |
||||||
|
}, |
||||||
|
}; |
||||||
|
|
||||||
|
let prefix = match opts.prefix.as_str() { |
||||||
|
"" => String::from(""), |
||||||
|
_ => opts.prefix + "_", |
||||||
|
}; |
||||||
|
|
||||||
|
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))) |
||||||
|
} else { |
||||||
|
Box::new(AlphabeticSuffixGenerator::new(opts.count)) |
||||||
|
}; |
||||||
|
|
||||||
|
let mut eof = false; |
||||||
|
let mut reader = BufReader::new(input); |
||||||
|
for suffix in suffix_gen { |
||||||
|
let mut out_file = File::create(format!("{}{}", prefix, suffix))?; |
||||||
|
let mut line = String::new(); |
||||||
|
for _ in 0..opts.lines { |
||||||
|
match reader.read_line(&mut line)? { |
||||||
|
0 => { |
||||||
|
eof = true; |
||||||
|
break
|
||||||
|
}, |
||||||
|
_ => { |
||||||
|
out_file.write_all(line.as_bytes())?; |
||||||
|
line.clear(); |
||||||
|
} |
||||||
|
}; |
||||||
|
} |
||||||
|
} |
||||||
|
// see if there's anything left for the remainder file
|
||||||
|
if !eof { |
||||||
|
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())?; |
||||||
|
}; |
||||||
|
} |
||||||
|
|
||||||
|
Ok(()) |
||||||
|
} |
||||||
|
|
||||||
|
fn main() { |
||||||
|
// display nicer, non-debug representation of the error
|
||||||
|
if let Err(err) = try_main() { |
||||||
|
eprintln!("{}", err); |
||||||
|
std::process::exit(1); |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue