tzconvert/src/lib.rs

use std::str::FromStr;
use chrono::prelude::*;
use chrono_tz::Tz;
use colored::*;

lazy_static::lazy_static! {
    static ref HOUR_AMPM: regex::Regex = regex::Regex::new(r#"(?P<hr>\d{1,2})\s?(?P<ampm>(am|pm|AM|PM))"#).unwrap();
    static ref HOUR_MINUTE_AMPM: regex::Regex = regex::Regex::new(r#"(?P<hr>\d{1,2}):(?P<minute>\d{2})\s?(?P<ampm>(am|pm|AM|PM))"#).unwrap();
    static ref ZERO_PADDED_OFFSET: regex::Regex = regex::Regex::new(r#"(?P<sign>[+-])0(?P<hr>\d)h"#).unwrap();
}

fn parse_time(time_input: &str) -> NaiveTime {
    let time: String = time_input.split_whitespace().collect::<Vec<_>>().join("");
    match NaiveTime::parse_from_str(&time, "%H:%M") {
        Ok(t) => return t,
        Err(_e) => {}
    }

    if HOUR_MINUTE_AMPM.is_match(&time) {
        return NaiveTime::parse_from_str(&time, "%I:%M%p").unwrap();
    }

    let t2 = HOUR_AMPM.replace(&time, "$hr:00$ampm");

    match NaiveTime::parse_from_str(&t2, "%I:%M%p") {
        Ok(t) => return t,
        Err(_e) => {}
    }

    panic!("failed to parse time -- try HH:MM format (input = '{}')", time)
}

fn convert_common_tz_abbrs(tz: &str) -> &str {
    let tz = tz.trim();
    if tz.eq_ignore_ascii_case("EST") { return "America/New_York" }
    if tz.eq_ignore_ascii_case("EPT") { return "America/New_York" }
    if tz.eq_ignore_ascii_case("EDT") { return "America/New_York" }
    if tz.eq_ignore_ascii_case("US/Eastern") { return "America/New_York" }
    if tz.eq_ignore_ascii_case("CET") { return "Europe/Brussels" }
    if tz.eq_ignore_ascii_case("CEST") { return "Europe/Brussels" }
    if tz.eq_ignore_ascii_case("Brussels") { return "Europe/Brussels" }

    tz
}

pub fn convert(from: &str, to: &str, time: &str, day: Option<String>, date: Option<NaiveDate>) {
    let from = convert_common_tz_abbrs(from);
    let to = convert_common_tz_abbrs(to);

    let fromtz: Tz = match from.parse() {
        Ok(tz) => tz,
        Err(e) => panic!("failed to parse from tz: {} (input = '{}')", e, from),
    };

    let totz: Tz = match to.parse() {
        Ok(tz) => tz,
        Err(e) => panic!("failed to parse to tz: {} (input = '{}')", e, to),
    };

    let tm = parse_time(time);

    let dt: Date<Tz> = match date {
        Some(dt) => fromtz.from_local_date(&dt).unwrap(),
        None => Utc::today().with_timezone(&fromtz),
    };

    let dt = match day {
        Some(day_str) => {
            let target_day_of_week: Weekday = Weekday::from_str(&day_str)
                .map_err(|e| {
                    eprintln!("failed to parse day (input = '{}')", day_str);
                    e
                }).unwrap();
            let mut cur = dt;
            while cur.weekday() != target_day_of_week {
                cur = cur.succ();
            }
            cur
        }

        None => dt,
    };

    let src = dt.and_time(tm).unwrap();

    let dst = src.with_timezone(&totz);


    // let indent = "    ";
    // println!();
    // println!("{}{} ... {} ... {}",
    //     indent,
    //     pad_spaces(fromtz_str, n_spaces),
    //     src.format("%a, %b %e"),
    //     src.format("%l:%M%P"),
    // );

    // println!();

    // let dst_line = format!("{}{} ... {} ... {}",
    //     indent,
    //     pad_spaces(totz_str, n_spaces),
    //     dst.format("%a, %b %e"),
    //     dst.format("%l:%M%P").to_string(),
    // );

    // println!("{}", dst_line.bold());
    println!("tzconvert v{}", structopt::clap::crate_version!());
    println!();
    let line = get_output(src, dst);
    println!("    {}", line);
    println!();
}

fn get_output(src: DateTime<Tz>, dst: DateTime<Tz>) -> String {
    let mut out = String::with_capacity(128);

    let fromtz_str = src.timezone().to_string();
    let totz_str = dst.timezone().to_string();
    let n_spaces = std::cmp::max(fromtz_str.len(), totz_str.len());

    //let mut src_str = format!("{tz}  |  {offset}h  |  {dt}  |  {tm}",
    let mut src_str = format!("{tz}   --+--   {offset}h   --+--   {dt}   --+--   {tm}",
        tz = pad(fromtz_str, n_spaces, " "),
        offset = src.offset().fix(),
        dt = src.format("%a, %b %e"),
        tm = src.format("%l:%M%P"),
    );
    src_str = clean(src_str);
    out.push_str(&format!("\n    {}\n    {}\n    {}\n", pad("", src_str.len(), " "), src_str, pad("", src_str.len(), " ")));
    //out.push_str(" -> ");

    let abs_diff_in_hours = (src.offset().fix().local_minus_utc() - dst.offset().fix().local_minus_utc()).abs() / 60 / 60;
    out.push_str(&format!("    .\n    .\n    .    {diff}h\n    .\n    .\n    ", diff = abs_diff_in_hours));

    //let mut dst_str = format!("{tz} | {offset}h | {dt} | {tm}",
    let mut dst_str = format!("{tz}   --+--   {offset}h   --+--   {dt}   --+--   {tm}",
        //tz = pad_spaces(dst.timezone(), n_spaces),
        tz = pad(totz_str, n_spaces, " "),
        offset = dst.offset().fix(),
        dt = dst.format("%a, %b %e"),
        tm = dst.format("%l:%M%P"),
    );
    dst_str = clean(dst_str);
    out.push_str(&format!("    {}\n    {}\n    {}\n",
        pad("", dst_str.len(), " "),
        dst_str.bold(),
        pad("", dst_str.len(), " "))
    );
    //out.push_str(&format!("    {}\n    {}\n    {}\n\n",
    //    pad("", dst_str.len(), "-"), dst_str.bold(), pad("", dst_str.len(), "-"))
    //);

    //out = out.replace(
    //    " -> ",
    //    &format!("\n\n    .\n    . {diff}h\n    .\n\n    ", diff = abs_diff_in_hours),
    //);
    out
}

fn clean(mut out: String) -> String {
    out = out.replace(":00", "");
    //while out.contains("  ") {
    //    out = out.replace("  ", " ");
    //}

    out = ZERO_PADDED_OFFSET.replace_all(&out, "${sign}${hr}h").to_string();
    out
}

fn pad<S: ToString>(s: S, n: usize, with: &str) -> String {
    let s = s.to_string();
    let mut out = String::with_capacity(n);
    out.push_str(&s);
    while out.len() < n {
        out.push_str(with);
    }
    out
}

#[test]
fn parse_time_like_2pm() {
    dbg!(parse_time("2pm"));
    assert_eq!(parse_time("2pm"), NaiveTime::from_hms(14, 0, 0));
    assert_eq!(parse_time("7am"), NaiveTime::from_hms(7, 0, 0));
    dbg!(HOUR_AMPM.replace("8:30am", "$hr:00$ampm"));
    assert_eq!(parse_time("8:30am"), NaiveTime::from_hms(8, 30, 0));
}