obsidian-logging 1.3.2

use crate::config::{Config, ListType, TimeFormat};
use chrono::{NaiveDate, NaiveTime, Timelike};
use lazy_static::lazy_static;
use regex::Regex;
use std::path::PathBuf;

lazy_static! {
    static ref TIME_PATTERN: Regex =
        Regex::new(r"^(?:[-*]\s*)?(\d{2}:\d{2}(?::\d{2})?(?:\s*[AaPp][Mm])?)\s*(.+)$").unwrap();
}

/// Format time according to the specified format (12 or 24 hour)
pub fn format_time(time: NaiveTime, format: &TimeFormat) -> String {
    match format {
        TimeFormat::Hour24 => time.format("%H:%M:%S").to_string(),
        TimeFormat::Hour12 => {
            let hour = time.hour();
            let minute = time.minute();
            let second = time.second();
            let period = if hour < 12 { "AM" } else { "PM" };
            let hour12 = match hour {
                0 => 12,
                13..=23 => hour - 12,
                _ => hour,
            };
            format!("{:02}:{:02}:{:02} {}", hour12, minute, second, period)
        }
    }
}

/// Parse time string in either 12 or 24 hour format
/// Supports both HH:MM and HH:MM:SS formats. If seconds are not provided, defaults to 00.
pub fn parse_time(time_str: &str) -> Option<NaiveTime> {
    // Try 24-hour format with seconds first
    // Validate seconds are in range 0-59 before parsing
    if time_str.matches(':').count() >= 2 {
        // Has seconds, validate format
        let parts: Vec<&str> = time_str.split(':').collect();
        if parts.len() >= 3 {
            if let Ok(seconds) = parts[2]
                .split_whitespace()
                .next()
                .unwrap_or("")
                .parse::<u32>()
            {
                if seconds >= 60 {
                    return None;
                }
            }
        }
    }

    if let Ok(time) = NaiveTime::parse_from_str(time_str, "%H:%M:%S") {
        return Some(time);
    }

    // Try 24-hour format without seconds (default to 00 seconds)
    if let Ok(time) = NaiveTime::parse_from_str(time_str, "%H:%M") {
        return Some(NaiveTime::from_hms_opt(time.hour(), time.minute(), 0).unwrap());
    }

    // Try various 12-hour formats with seconds
    let formats_with_seconds = vec![
        "%I:%M:%S %p", // "02:30:45 PM"
        "%I:%M:%S%p",  // "02:30:45PM"
        "%l:%M:%S %p", // "2:30:45 PM"
        "%l:%M:%S%p",  // "2:30:45PM"
    ];

    for format in formats_with_seconds {
        if let Ok(time) = NaiveTime::parse_from_str(&time_str.to_uppercase(), format) {
            // Validate that seconds are in valid range (0-59)
            if time.second() >= 60 {
                continue;
            }
            return Some(time);
        }
    }

    // Try various 12-hour formats without seconds (default to 00 seconds)
    let formats = vec![
        "%I:%M %p", // "02:30 PM"
        "%I:%M%p",  // "02:30PM"
        "%l:%M %p", // "2:30 PM"
        "%l:%M%p",  // "2:30PM"
    ];

    for format in formats {
        if let Ok(time) = NaiveTime::parse_from_str(&time_str.to_uppercase(), format) {
            return Some(NaiveTime::from_hms_opt(time.hour(), time.minute(), 0).unwrap());
        }
    }

    None
}

/// Build file path for given date and format string from configuration yaml
/// Supported tokens: {year}, {month}, {date}
pub fn get_log_path_for_date(date: NaiveDate, config: &Config) -> PathBuf {
    let mut path = PathBuf::from(&config.vault);

    let year = date.format("%Y").to_string();
    let month = date.format("%m").to_string();
    let date_str = date.format("%Y-%m-%d").to_string();

    let file_path = config
        .file_path_format
        .replace("{year}", &year)
        .replace("{month}", &month)
        .replace("{date}", &date_str);

    path.push(file_path);
    path
}

/// Format a table row with given widths for timestamp and entry columns
fn format_table_row(timestamp: &str, entry: &str, time_width: usize, entry_width: usize) -> String {
    format!(
        "| {:<width_t$} | {:<width_e$} |",
        timestamp,
        entry,
        width_t = time_width,
        width_e = entry_width
    )
}

/// Format a table separator line with given column widths
fn format_table_separator(time_width: usize, entry_width: usize) -> String {
    format!(
        "|{}|{}|",
        "-".repeat(time_width + 2),
        "-".repeat(entry_width + 2)
    )
}

/// Parse an entry to extract timestamp and content
fn parse_entry(entry: &str) -> (String, String) {
    if entry.starts_with('|') {
        // Parse table format
        let parts: Vec<&str> = entry.split('|').collect();
        if parts.len() >= 4 {
            return (parts[1].trim().to_string(), parts[2].trim().to_string());
        }
    } else if entry.starts_with(['*', '-']) {
        // Parse bullet format - handle both 24-hour and 12-hour time formats
        let content = entry.trim_start_matches(['-', '*', ' ']);

        // Try to find a valid time pattern at the beginning
        let time_patterns = [
            // 24-hour format: HH:MM:SS
            r"^(\d{1,2}:\d{2}:\d{2})\s+(.+)$",
            // 24-hour format: HH:MM (backward compatibility)
            r"^(\d{1,2}:\d{2})\s+(.+)$",
            // 12-hour format: HH:MM:SS AM/PM
            r"^(\d{1,2}:\d{2}:\d{2}\s+[AaPp][Mm])\s+(.+)$",
            // 12-hour format: HH:MM AM/PM (backward compatibility)
            r"^(\d{1,2}:\d{2}\s+[AaPp][Mm])\s+(.+)$",
        ];

        for pattern in &time_patterns {
            if let Ok(regex) = Regex::new(pattern) {
                if let Some(captures) = regex.captures(content) {
                    let time = captures.get(1).unwrap().as_str().trim();
                    let entry_text = captures.get(2).unwrap().as_str().trim();
                    return (time.to_string(), entry_text.to_string());
                }
            }
        }

        // Fallback to original behavior for backward compatibility
        if let Some(space_pos) = content.find(' ') {
            if let Some(second_space) = content[space_pos + 1..].find(' ') {
                return (
                    content[..space_pos + 1 + second_space].trim().to_string(),
                    content[space_pos + 1 + second_space + 1..]
                        .trim()
                        .to_string(),
                );
            }
        }
    }
    (String::new(), String::new())
}

/// Extract log entries from the log section
/// Returns ( content before log section, content after log section, list of log entries, and detected list type)
/// Section heading retrieved from yaml config
pub fn extract_log_entries(
    content: &str,
    section_header: &str,
    list_type: &ListType,
    config: &Config,
    include_header: bool,
) -> (String, String, Vec<String>, ListType) {
    let mut before = String::new();
    let mut after = String::new();
    let mut entries = Vec::new();
    let mut found_type = list_type.clone();
    let mut in_section = false;
    let mut found_section = false;

    let lines = content.lines().peekable();
    for line in lines {
        if line.starts_with(section_header) {
            found_section = true;
            in_section = true;
            before = before.trim_end().to_string() + "\n\n";
            continue;
        }

        if in_section {
            if line.starts_with("##") {
                in_section = false;
                after = line.to_string();
                continue;
            }

            let trimmed = line.trim();
            if !trimmed.is_empty() {
                if trimmed.starts_with('|') {
                    found_type = ListType::Table;
                } else if trimmed.starts_with(['*', '-']) {
                    found_type = ListType::Bullet;
                }

                // Skip table separator and header rows
                if !trimmed.contains("---")
                    && trimmed != format!("| {} | {} |", config.time_label, config.event_label)
                {
                    entries.push(line.to_string());
                }
            }
        } else if !found_section {
            before.push_str(line);
            before.push('\n');
        } else if !line.is_empty() {
            after.push('\n');
            after.push_str(line);
        }
    }

    // Convert entries if needed
    if found_type != *list_type {
        let mut converted_entries = Vec::new();

        if *list_type == ListType::Table {
            // Convert from bullet to table
            let mut max_time_width = config.time_label.len();
            let mut max_entry_width = config.event_label.len();

            // First pass: calculate widths
            for entry in &entries {
                let (time, text) = parse_entry(entry);
                // Parse and reformat time according to config
                let formatted_time = if let Some(parsed_time) = parse_time(&time) {
                    format_time(parsed_time, &config.time_format)
                } else {
                    time
                };
                max_time_width = max_time_width.max(formatted_time.len());
                max_entry_width = max_entry_width.max(text.len());
            }

            // Add header only if include_header is true
            if include_header {
                converted_entries.push(format_table_row(
                    &config.time_label,
                    &config.event_label,
                    max_time_width,
                    max_entry_width,
                ));
                converted_entries.push(format_table_separator(max_time_width, max_entry_width));
            }

            // Second pass: format entries
            for entry in entries {
                let (time, text) = parse_entry(&entry);
                // Parse and reformat time according to config
                let formatted_time = if let Some(parsed_time) = parse_time(&time) {
                    format_time(parsed_time, &config.time_format)
                } else {
                    time
                };
                converted_entries.push(format_table_row(
                    &formatted_time,
                    &text,
                    max_time_width,
                    max_entry_width,
                ));
            }
        } else {
            // Convert from table to bullet
            // Add table header as a comment only if include_header is true
            if include_header {
                converted_entries.push(format!(
                    "<!-- {} | {} -->",
                    config.time_label, config.event_label
                ));
            }

            for entry in entries {
                let (time, text) = parse_entry(&entry);
                if !time.is_empty() && !text.is_empty() {
                    // Parse and reformat time according to config
                    let formatted_time = if let Some(parsed_time) = parse_time(&time) {
                        format_time(parsed_time, &config.time_format)
                    } else {
                        time
                    };
                    converted_entries.push(format!("- {} {}", formatted_time, text));
                }
            }
        }

        entries = converted_entries;
    } else if *list_type == ListType::Table && found_type == ListType::Table && include_header {
        // Format hasn't changed, but ensure table format has proper header
        // Rebuild table with proper header and separator
        let mut max_time_width = config.time_label.len();
        let mut max_entry_width = config.event_label.len();

        // First pass: calculate widths from existing entries
        for entry in &entries {
            let (time, text) = parse_entry(entry);
            max_time_width = max_time_width.max(time.len());
            max_entry_width = max_entry_width.max(text.len());
        }

        // Rebuild table with header
        let mut rebuilt_entries = Vec::new();
        rebuilt_entries.push(format_table_row(
            &config.time_label,
            &config.event_label,
            max_time_width,
            max_entry_width,
        ));
        rebuilt_entries.push(format_table_separator(max_time_width, max_entry_width));

        // Add data rows
        for entry in entries {
            let (time, text) = parse_entry(&entry);
            if !time.is_empty() && !text.is_empty() {
                rebuilt_entries.push(format_table_row(
                    &time,
                    &text,
                    max_time_width,
                    max_entry_width,
                ));
            }
        }

        entries = rebuilt_entries;
        // If include_header is false, keep original entries as-is
        // For bullet format, entries are already in the correct format
    }

    (before, after, entries, found_type)
}