mirror of
https://github.com/Llewellynvdm/exa.git
synced 2024-09-28 21:29:02 +00:00
183 lines
5.5 KiB
Rust
183 lines
5.5 KiB
Rust
use ansi_term::Style;
|
|
use locale::Numeric as NumericLocale;
|
|
use number_prefix::Prefix;
|
|
|
|
use crate::fs::fields as f;
|
|
use crate::output::cell::{TextCell, DisplayWidth};
|
|
use crate::output::table::SizeFormat;
|
|
|
|
|
|
impl f::Size {
|
|
pub fn render<C: Colours>(self, colours: &C, size_format: SizeFormat, numerics: &NumericLocale) -> TextCell {
|
|
use number_prefix::NumberPrefix;
|
|
|
|
let size = match self {
|
|
Self::Some(s) => s,
|
|
Self::None => return TextCell::blank(colours.no_size()),
|
|
Self::DeviceIDs(ref ids) => return ids.render(colours),
|
|
};
|
|
|
|
let result = match size_format {
|
|
SizeFormat::DecimalBytes => NumberPrefix::decimal(size as f64),
|
|
SizeFormat::BinaryBytes => NumberPrefix::binary(size as f64),
|
|
SizeFormat::JustBytes => {
|
|
|
|
// Use the binary prefix to select a style.
|
|
let prefix = match NumberPrefix::binary(size as f64) {
|
|
NumberPrefix::Standalone(_) => None,
|
|
NumberPrefix::Prefixed(p, _) => Some(p),
|
|
};
|
|
|
|
// But format the number directly using the locale.
|
|
let string = numerics.format_int(size);
|
|
|
|
return TextCell::paint(colours.size(prefix), string);
|
|
}
|
|
};
|
|
|
|
let (prefix, n) = match result {
|
|
NumberPrefix::Standalone(b) => return TextCell::paint(colours.size(None), b.to_string()),
|
|
NumberPrefix::Prefixed(p, n) => (p, n),
|
|
};
|
|
|
|
let symbol = prefix.symbol();
|
|
let decimal_to_diplay = if n < 10_f64 { 1 } else { 0 };
|
|
let number = numerics.format_float(n, decimal_to_diplay);
|
|
|
|
// The numbers and symbols are guaranteed to be written in ASCII, so
|
|
// we can skip the display width calculation.
|
|
let width = DisplayWidth::from(number.len() + symbol.len());
|
|
|
|
TextCell {
|
|
width,
|
|
contents: vec![
|
|
colours.size(Some(prefix)).paint(number),
|
|
colours.unit(Some(prefix)).paint(symbol),
|
|
].into(),
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
impl f::DeviceIDs {
|
|
fn render<C: Colours>(self, colours: &C) -> TextCell {
|
|
let major = self.major.to_string();
|
|
let minor = self.minor.to_string();
|
|
|
|
TextCell {
|
|
width: DisplayWidth::from(major.len() + 1 + minor.len()),
|
|
contents: vec![
|
|
colours.major().paint(major),
|
|
colours.comma().paint(","),
|
|
colours.minor().paint(minor),
|
|
].into(),
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
pub trait Colours {
|
|
fn size(&self, prefix: Option<Prefix>) -> Style;
|
|
fn unit(&self, prefix: Option<Prefix>) -> Style;
|
|
fn no_size(&self) -> Style;
|
|
|
|
fn major(&self) -> Style;
|
|
fn comma(&self) -> Style;
|
|
fn minor(&self) -> Style;
|
|
}
|
|
|
|
|
|
#[cfg(test)]
|
|
pub mod test {
|
|
use super::Colours;
|
|
use crate::output::cell::{TextCell, DisplayWidth};
|
|
use crate::output::table::SizeFormat;
|
|
use crate::fs::fields as f;
|
|
|
|
use locale::Numeric as NumericLocale;
|
|
use ansi_term::Colour::*;
|
|
use ansi_term::Style;
|
|
use number_prefix::Prefix;
|
|
|
|
|
|
struct TestColours;
|
|
|
|
impl Colours for TestColours {
|
|
fn size(&self, _prefix: Option<Prefix>) -> Style { Fixed(66).normal() }
|
|
fn unit(&self, _prefix: Option<Prefix>) -> Style { Fixed(77).bold() }
|
|
fn no_size(&self) -> Style { Black.italic() }
|
|
|
|
fn major(&self) -> Style { Blue.on(Red) }
|
|
fn comma(&self) -> Style { Green.italic() }
|
|
fn minor(&self) -> Style { Cyan.on(Yellow) }
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn directory() {
|
|
let directory = f::Size::None;
|
|
let expected = TextCell::blank(Black.italic());
|
|
assert_eq!(expected, directory.render(&TestColours, SizeFormat::JustBytes, &NumericLocale::english()))
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn file_decimal() {
|
|
let directory = f::Size::Some(2_100_000);
|
|
let expected = TextCell {
|
|
width: DisplayWidth::from(4),
|
|
contents: vec![
|
|
Fixed(66).paint("2.1"),
|
|
Fixed(77).bold().paint("M"),
|
|
].into(),
|
|
};
|
|
|
|
assert_eq!(expected, directory.render(&TestColours, SizeFormat::DecimalBytes, &NumericLocale::english()))
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn file_binary() {
|
|
let directory = f::Size::Some(1_048_576);
|
|
let expected = TextCell {
|
|
width: DisplayWidth::from(5),
|
|
contents: vec![
|
|
Fixed(66).paint("1.0"),
|
|
Fixed(77).bold().paint("Mi"),
|
|
].into(),
|
|
};
|
|
|
|
assert_eq!(expected, directory.render(&TestColours, SizeFormat::BinaryBytes, &NumericLocale::english()))
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn file_bytes() {
|
|
let directory = f::Size::Some(1048576);
|
|
let expected = TextCell {
|
|
width: DisplayWidth::from(9),
|
|
contents: vec![
|
|
Fixed(66).paint("1,048,576"),
|
|
].into(),
|
|
};
|
|
|
|
assert_eq!(expected, directory.render(&TestColours, SizeFormat::JustBytes, &NumericLocale::english()))
|
|
}
|
|
|
|
|
|
#[test]
|
|
fn device_ids() {
|
|
let directory = f::Size::DeviceIDs(f::DeviceIDs { major: 10, minor: 80 });
|
|
let expected = TextCell {
|
|
width: DisplayWidth::from(5),
|
|
contents: vec![
|
|
Blue.on(Red).paint("10"),
|
|
Green.italic().paint(","),
|
|
Cyan.on(Yellow).paint("80"),
|
|
].into(),
|
|
};
|
|
|
|
assert_eq!(expected, directory.render(&TestColours, SizeFormat::JustBytes, &NumericLocale::english()))
|
|
}
|
|
}
|