use ntime::Format;
use std::io;
use crate::attributes::Map;
use crate::attributes::value::Value;
use crate::console::Color;
use crate::constant::{ATTRIBUTE_KEY_ERROR, DEFAULT_LOG_DELIMITER_STRING};
use crate::format::compact;
use crate::format::{FormatterConfig, OutputFormat};
use crate::level::Level;
use crate::sink::LogUpdate;
pub fn default_format_config() -> FormatterConfig {
FormatterConfig {
format: OutputFormat::ColorCompact,
time_format: ntime::Format::LocalMillisDateTime,
delimiter: DEFAULT_LOG_DELIMITER_STRING.into(),
}
}
pub fn write_value<T: io::Write>(out: &mut T, val: &Value) -> io::Result<()> {
compact::write_value(out, val)
}
pub fn write<T: io::Write>(out: &mut T, time_format: &Format, update: &LogUpdate, attrs: &Map) -> io::Result<()> {
let msg_color = if Level::Debug.includes(&update.level) { Color::Default } else { Color::BrightWhite };
let level_color = update.level.color();
update.when.write(out, time_format)?;
write!(
out,
" {level_open}{level}{level_close} {msg_open}{msg}{msg_close}",
level_open = level_color.to_escape_str(),
level = update.level.as_short_str(),
level_close = Color::Default.to_escape_str(),
msg_open = msg_color.to_escape_str(),
msg = update.msg,
msg_close = Color::Default.to_escape_str(),
)?;
for (key, val) in attrs.into_iter() {
write!(
out,
" {key_open}{key}{key_close}={val_open}",
key_open = Color::Cyan.to_escape_str(),
key_close = Color::Default.to_escape_str(),
val_open = if key == ATTRIBUTE_KEY_ERROR { Color::BrightRed.to_escape_str() } else { "" }
)?;
write_value(out, val)?;
write!(out, "{val_close}", val_close = Color::Default.to_escape_str())?;
}
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn serialize_write() {
for tc in [
(Value::Bool(true), "true"),
(Value::String("".into()), "\"\""),
(Value::String("abcd 1234".into()), "\"abcd 1234\""),
(Value::Int(-123), "-123"),
(Value::LongInt(-12345678901234567), "-0x2bdc545d6b4b87"),
(Value::Size(89801234567890123), "0x13f09bf3ecf84cb"),
(Value::Uint(123456), "123456"),
(Value::LongUint(12345678901234567), "0x2bdc545d6b4b87"),
(Value::Usize(89801234567890123), "0x13f09bf3ecf84cb"),
(Value::Float(-1.2345), "-1.2345"),
] {
let (v, want): (Value, &str) = tc;
let mut out = Vec::new();
assert!(write_value(&mut out, &v).is_ok());
assert_eq!(String::from_utf8(out).unwrap(), String::from(want));
}
}
}