yz-diary-date 0.1.2

// Copyright (c) 2021-2024 Alain Emilia Anna Zscheile
// SPDX-License-Identifier: MIT OR Apache-2.0

use chrono::NaiveDate;
use std::path::Path;

#[cfg(test)]
mod tests;

fn fti_digits(s: &str) -> bool {
    s.len() >= 2 && s.chars().take(2).all(|i| i.is_ascii_digit())
}

/// usually, `parse_from_path` should be used instead of this low-level function
pub fn parse(par: &str, fin: &str) -> Option<NaiveDate> {
    if !fti_digits(fin) || par.len() < 4 {
        return None;
    }

    let y = par.parse::<i32>().ok()?;
    let m = fin[..2].parse::<u32>().unwrap();

    // verify the day info
    let mut dayinf = &fin[2..];
    if dayinf.starts_with(|i| matches!(i, '-' | '_')) {
        dayinf = &dayinf[1..];
    }
    if !fti_digits(dayinf) {
        return None;
    }
    let d = dayinf[..2].parse::<u32>().unwrap();

    chrono::NaiveDate::from_ymd_opt(y, m, d)
}

/// tries to parse a diary entry path to extract the reference date
/// should be usually given a path containing at least 2 components
pub fn parse_from_path(x: &Path) -> Option<NaiveDate> {
    let mut fin = x;
    let mut fins = fin.file_name()?.to_str()?;

    // we allow 1 additional path component after the date part
    if !fti_digits(fins) {
        fin = x.parent()?;
        fins = fin.file_name()?.to_str()?;
        if !fti_digits(fins) {
            return None;
        }
    }

    let par = fin.parent()?.file_name()?.to_str()?;
    parse(par, fins)
}

/// tries to parse a diary entry path to extract the reference date
/// should be usually given a path containing at least 2 components
/// but additionally requires that the whole path must be UTF-8
#[cfg(feature = "camino")]
pub fn parse_from_utf8path(x: &camino::Utf8Path) -> Option<NaiveDate> {
    let mut fin = x;
    let mut fins = fin.file_name()?;

    // we allow 1 additional path component after the date part
    if !fti_digits(fins) {
        fin = x.parent()?;
        fins = fin.file_name()?;
        if !fti_digits(fins) {
            return None;
        }
    }

    let par = fin.parent()?.file_name()?;
    parse(par, fins)
}

pub fn fmt(date: &NaiveDate, daysep: Option<char>) -> String {
    let tmp;
    date.format(if let Some(daysep) = daysep {
        assert!(matches!(daysep, '-' | '_'));
        tmp = format!("%Y/%m{}%d", daysep);
        &tmp
    } else {
        "%Y/%m%d"
    })
    .to_string()
}