mirror of
https://github.com/uutils/coreutils
synced 2024-12-13 14:52:41 +00:00
parent
e3ca4fdcf8
commit
4388c72bf3
4 changed files with 140 additions and 1 deletions
2
Makefile
2
Makefile
|
@ -21,6 +21,7 @@ PROGS := \
|
|||
rm \
|
||||
rmdir \
|
||||
sleep \
|
||||
seq \
|
||||
tee \
|
||||
true \
|
||||
users \
|
||||
|
@ -39,6 +40,7 @@ EXES := \
|
|||
TEST_PROGS := \
|
||||
cat \
|
||||
mkdir \
|
||||
seq \
|
||||
|
||||
TEST ?= $(TEST_PROGS)
|
||||
|
||||
|
|
|
@ -138,7 +138,6 @@ To do
|
|||
- relpath
|
||||
- remove
|
||||
- runcon
|
||||
- seq
|
||||
- setuidgid
|
||||
- shred
|
||||
- shuf
|
||||
|
|
109
seq/seq.rs
Normal file
109
seq/seq.rs
Normal file
|
@ -0,0 +1,109 @@
|
|||
#[crate_id(name="seq", vers="1.0.0", author="Daniel MacDougall")];
|
||||
|
||||
// TODO: Make -w flag work with decimals
|
||||
// TODO: Support -f flag
|
||||
|
||||
extern mod extra;
|
||||
|
||||
use std::os;
|
||||
use std::cmp::max;
|
||||
use extra::getopts::groups;
|
||||
|
||||
fn print_usage(opts: ~[groups::OptGroup]) {
|
||||
println!("seq 1.0.0\n");
|
||||
println!("Usage:\n seq [-w] [-s string] [-t string] [first [step]] last\n");
|
||||
println!("{:s}", groups::usage("Print sequences of numbers", opts));
|
||||
}
|
||||
|
||||
fn parse_float(s: &str) -> Result<f32, ~str>{
|
||||
match from_str(s) {
|
||||
Some(n) => Ok(n),
|
||||
None => Err(format!("seq: invalid floating point argument: {:s}", s))
|
||||
}
|
||||
}
|
||||
|
||||
fn escape_sequences(s: &str) -> ~str {
|
||||
s.replace("\\n", "\n").
|
||||
replace("\\t", "\t")
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let args = os::args();
|
||||
let opts = ~[
|
||||
groups::optopt("s", "separator", "Separator character (defaults to \\n)", ""),
|
||||
groups::optopt("t", "terminator", "Terminator character (defaults to separator)", ""),
|
||||
groups::optflag("w", "widths", "Equalize widths of all numbers by padding with zeros"),
|
||||
groups::optflag("h", "help", "print this help text and exit"),
|
||||
groups::optflag("V", "version", "print version and exit"),
|
||||
];
|
||||
let matches = match groups::getopts(args.tail(), opts) {
|
||||
Ok(m) => { m }
|
||||
Err(f) => {
|
||||
println!("{:s}", f.to_err_msg());
|
||||
print_usage(opts);
|
||||
return;
|
||||
}
|
||||
};
|
||||
if matches.opt_present("help") {
|
||||
print_usage(opts);
|
||||
return;
|
||||
}
|
||||
if matches.opt_present("version") {
|
||||
println!("seq 1.0.0");
|
||||
return;
|
||||
}
|
||||
if matches.free.len() < 1 || matches.free.len() > 3 {
|
||||
print_usage(opts);
|
||||
return;
|
||||
}
|
||||
let first = if matches.free.len() > 1 {
|
||||
match parse_float(matches.free[0]) {
|
||||
Ok(n) => n,
|
||||
Err(s) => { println!("{:s}", s); return; }
|
||||
}
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
let step = if matches.free.len() > 2 {
|
||||
match parse_float(matches.free[1]) {
|
||||
Ok(n) => n,
|
||||
Err(s) => { println!("{:s}", s); return; }
|
||||
}
|
||||
} else {
|
||||
1.0
|
||||
};
|
||||
let last = match parse_float(matches.free[matches.free.len()-1]) {
|
||||
Ok(n) => n,
|
||||
Err(s) => { println!("{:s}", s); return; }
|
||||
};
|
||||
let separator = escape_sequences(matches.opt_str("s").unwrap_or(~"\n"));
|
||||
let terminator = escape_sequences(matches.opt_str("t").unwrap_or(separator.clone()));
|
||||
print_seq(first, step, last, separator, terminator, matches.opt_present("w"));
|
||||
}
|
||||
|
||||
fn done_printing(next: f32, step: f32, last: f32) -> bool {
|
||||
if step > 0f32 {
|
||||
next > last
|
||||
} else {
|
||||
next < last
|
||||
}
|
||||
}
|
||||
|
||||
fn print_seq(first: f32, step: f32, last: f32, separator: ~str, terminator: ~str, pad: bool) {
|
||||
let mut i = first;
|
||||
let maxlen = max(first, last).to_str().len();
|
||||
while !done_printing(i, step, last) {
|
||||
let ilen = i.to_str().len();
|
||||
if pad && ilen < maxlen {
|
||||
for _ in range(0, maxlen - ilen) {
|
||||
print!("0");
|
||||
}
|
||||
}
|
||||
print!("{:f}", i);
|
||||
i += step;
|
||||
if !done_printing(i, step, last) {
|
||||
print!("{:s}", separator);
|
||||
}
|
||||
}
|
||||
print!("{:s}", terminator);
|
||||
}
|
29
seq/test.rs
Normal file
29
seq/test.rs
Normal file
|
@ -0,0 +1,29 @@
|
|||
use std::{run,str};
|
||||
|
||||
#[test]
|
||||
fn test_count_up() {
|
||||
let p = run::process_output("build/seq", [~"10"]).unwrap();
|
||||
let out = str::from_utf8(p.output).unwrap().into_owned();
|
||||
assert_eq!(out, ~"1\n2\n3\n4\n5\n6\n7\n8\n9\n10\n");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_count_down() {
|
||||
let p = run::process_output("build/seq", [~"--", ~"5", ~"-1", ~"1"]).unwrap();
|
||||
let out = str::from_utf8(p.output).unwrap().into_owned();
|
||||
assert_eq!(out, ~"5\n4\n3\n2\n1\n");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_separator_and_terminator() {
|
||||
let p = run::process_output("build/seq", [~"-s", ~",", ~"-t", ~"!", ~"2", ~"6"]).unwrap();
|
||||
let out = str::from_utf8(p.output).unwrap().into_owned();
|
||||
assert_eq!(out, ~"2,3,4,5,6!");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_equalize_widths() {
|
||||
let p = run::process_output("build/seq", [~"-w", ~"5", ~"10"]).unwrap();
|
||||
let out = str::from_utf8(p.output).unwrap().into_owned();
|
||||
assert_eq!(out, ~"05\n06\n07\n08\n09\n10\n");
|
||||
}
|
Loading…
Reference in a new issue