exa/exa.rs

83 lines
2.1 KiB
Rust
Raw Normal View History

#![feature(phase)]
extern crate regex;
#[phase(syntax)] extern crate regex_macros;
2014-05-03 10:30:37 +00:00
use std::os;
use std::io::fs;
use file::File;
2014-05-25 16:14:50 +00:00
use options::Options;
2014-05-03 10:30:37 +00:00
pub mod colours;
pub mod column;
pub mod format;
pub mod file;
pub mod unix;
pub mod options;
2014-05-03 10:30:37 +00:00
fn main() {
2014-05-25 16:14:50 +00:00
let args = os::args().iter()
2014-05-21 23:16:05 +00:00
.map(|x| x.to_strbuf())
.collect();
2014-05-25 16:14:50 +00:00
match Options::getopts(args) {
Err(err) => println!("Invalid options:\n{}", err.to_err_msg()),
Ok(opts) => {
let strs = if opts.dirs.is_empty() {
vec!("./".to_strbuf())
}
else {
opts.dirs.clone()
};
for dir in strs.move_iter() {
exa(&opts, Path::new(dir))
}
}
};
2014-05-03 10:30:37 +00:00
}
2014-05-25 16:14:50 +00:00
fn exa(options: &Options, path: Path) {
let paths = match fs::readdir(&path) {
Ok(paths) => paths,
Err(e) => fail!("readdir: {}", e),
};
let mut files = paths.iter().map(|path| File::from_path(path)).collect();
options.sort(&mut files);
2014-05-24 20:45:24 +00:00
if options.reverse {
files.reverse();
}
2014-05-25 18:42:31 +00:00
let columns = options.columns();
2014-05-26 10:08:33 +00:00
let table: Vec<Vec<String>> = files.iter()
.filter(|&f| options.show(f))
.map(|f| columns.iter().map(|c| f.display(c)).collect())
.collect();
2014-05-03 13:26:49 +00:00
let lengths: Vec<Vec<uint>> = table.iter()
2014-05-25 18:42:31 +00:00
.map(|row| row.iter().map(|col| colours::strip_formatting(col).len() ).collect())
.collect();
2014-05-25 16:14:50 +00:00
let maxes: Vec<uint> = range(0, columns.len())
.map(|n| lengths.iter().map(|row| *row.get(n)).max().unwrap())
.collect();
2014-05-03 10:30:37 +00:00
for (field_lengths, row) in lengths.iter().zip(table.iter()) {
2014-05-03 11:15:35 +00:00
let mut first = true;
for ((column_length, cell), field_length) in maxes.iter().zip(row.iter()).zip(field_lengths.iter()) {
2014-05-03 11:15:35 +00:00
if first {
first = false;
} else {
print!(" ");
}
print!("{}", cell.as_slice());
for _ in range(*field_length, *column_length) {
print!(" ");
}
2014-05-03 11:15:35 +00:00
}
print!("\n");
2014-05-03 10:30:37 +00:00
}
}