use crate::date::utils::{parse_date_from_string, unsupported_input_error}; use chrono::{DateTime, Datelike, FixedOffset, Local, Timelike}; use nu_protocol::ast::Call; use nu_protocol::engine::{Command, EngineState, Stack}; use nu_protocol::{Category, Example, PipelineData, Signature, Span, Value}; #[derive(Clone)] pub struct SubCommand; impl Command for SubCommand { fn name(&self) -> &str { "date to-table" } fn signature(&self) -> Signature { Signature::build("date to-table").category(Category::Date) } fn usage(&self) -> &str { "Print the date in a structured table." } fn run( &self, engine_state: &EngineState, _stack: &mut Stack, call: &Call, input: PipelineData, ) -> Result { let head = call.head; input.map(move |value| helper(value, head), engine_state.ctrlc.clone()) } fn examples(&self) -> Vec { vec![ Example { description: "Print the date in a structured table.", example: "date to-table", result: None, }, Example { description: "Print the date in a structured table.", example: "date now | date to-table", result: None, }, Example { description: "Print the date in a structured table.", example: " '2020-04-12 22:10:57 +0200' | date to-table", result: { let span = Span::unknown(); let cols = vec![ "year".into(), "month".into(), "day".into(), "hour".into(), "minute".into(), "second".into(), "timezone".into(), ]; let vals = vec![ Value::Int { val: 2020, span }, Value::Int { val: 4, span }, Value::Int { val: 12, span }, Value::Int { val: 22, span }, Value::Int { val: 10, span }, Value::Int { val: 57, span }, Value::String { val: "+02:00".to_string(), span, }, ]; Some(Value::List { vals: vec![Value::Record { cols, vals, span }], span, }) }, }, ] } } fn parse_date_into_table(date: Result, Value>, head: Span) -> Value { let cols = vec![ "year".into(), "month".into(), "day".into(), "hour".into(), "minute".into(), "second".into(), "timezone".into(), ]; match date { Ok(x) => { let vals = vec![ Value::Int { val: x.year() as i64, span: head, }, Value::Int { val: x.month() as i64, span: head, }, Value::Int { val: x.day() as i64, span: head, }, Value::Int { val: x.hour() as i64, span: head, }, Value::Int { val: x.minute() as i64, span: head, }, Value::Int { val: x.second() as i64, span: head, }, Value::String { val: x.offset().to_string(), span: head, }, ]; Value::List { vals: vec![Value::Record { cols, vals, span: head, }], span: head, } } Err(e) => e, } } fn helper(val: Value, head: Span) -> Value { match val { Value::String { val, span: _ } => { let date = parse_date_from_string(val); parse_date_into_table(date, head) } Value::Nothing { span: _ } => { let now = Local::now(); let n = now.with_timezone(now.offset()); parse_date_into_table(Ok(n), head) } Value::Date { val, span: _ } => parse_date_into_table(Ok(val), head), _ => unsupported_input_error(), } } #[cfg(test)] mod test { use super::*; #[test] fn test_examples() { use crate::test_examples; test_examples(SubCommand {}) } }