rustodo 2.26.0

A modern, powerful task manager built with Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147

//! Holiday data fetched from holidata.net and cached locally.
//!
//! Cache location: `~/.config/rustodo/holidays/<locale>/<year>.json`
//!
//! Usage:
//! ```no_run
//! use rustodo::services::holidays::HolidayCache;
//! use chrono::NaiveDate;
//! let holidays = HolidayCache::load("pt-BR", 2026)?;
//! let names = holidays.for_date(NaiveDate::from_ymd_opt(2026, 1, 1).unwrap());
//! # Ok::<(), anyhow::Error>(())
//! ```
use std::collections::HashMap;
use std::fs;
use std::path::PathBuf;

use anyhow::Result;
use chrono::NaiveDate;
use serde::{Deserialize, Serialize};

// ── Types ─────────────────────────────────────────────────────────────────────

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct HolidayEntry {
    pub date: String,
    pub description: String,
    #[serde(default)]
    pub region: String,
}

/// In-memory holiday map for a single locale+year.
pub struct HolidayCache {
    /// date → holiday name (national only, region == "")
    map: HashMap<NaiveDate, String>,
}

impl HolidayCache {
    /// Load holidays for `locale` and `year`.
    ///
    /// Strategy:
    /// 1. Check cache at `~/.config/rustodo/holidays/<locale>/<year>.json`
    /// 2. If missing, fetch from holidata.net and save to cache
    /// 3. Parse and return
    pub fn load(locale: &str, year: i32) -> Result<Self> {
        let path = cache_path(locale, year)?;

        let raw = if path.exists() {
            fs::read_to_string(&path)?
        } else {
            let data = fetch(locale, year)?;
            if let Some(parent) = path.parent() {
                fs::create_dir_all(parent)?;
            }
            fs::write(&path, &data)?;
            data
        };

        let map = parse_ndjson(&raw);
        Ok(Self { map })
    }

    /// Returns the holiday name for a date, if any.
    pub fn for_date(&self, date: NaiveDate) -> Option<&str> {
        self.map.get(&date).map(String::as_str)
    }

    /// Returns true if the date is a holiday.
    pub fn is_holiday(&self, date: NaiveDate) -> bool {
        self.map.contains_key(&date)
    }
}

/// Empty cache — used when holidays are disabled.
impl Default for HolidayCache {
    fn default() -> Self {
        Self {
            map: HashMap::new(),
        }
    }
}

// ── Fetch ─────────────────────────────────────────────────────────────────────

fn fetch(locale: &str, year: i32) -> Result<String> {
    let url = format!("https://holidata.net/{}/{}.json", locale, year);
    let mut response = ureq::get(&url).call()?;
    Ok(response.body_mut().read_to_string()?)
}

// ── Parse ─────────────────────────────────────────────────────────────────────

/// Parse NDJSON (one JSON object per line). Only national holidays (region == "").
fn parse_ndjson(raw: &str) -> HashMap<NaiveDate, String> {
    let mut map = HashMap::new();
    for line in raw.lines() {
        let line = line.trim();
        if line.is_empty() {
            continue;
        }
        if let Ok(entry) = serde_json::from_str::<HolidayEntry>(line) {
            // Only national holidays (no region)
            if !entry.region.is_empty() {
                continue;
            }
            if let Ok(date) = NaiveDate::parse_from_str(&entry.date, "%Y-%m-%d") {
                map.insert(date, entry.description);
            }
        }
    }
    map
}

// ── Cache path ────────────────────────────────────────────────────────────────

fn cache_path(locale: &str, year: i32) -> Result<PathBuf> {
    let config_dir = if let Ok(dir) = std::env::var("RUSTODO_CONFIG_DIR") {
        PathBuf::from(dir)
    } else {
        let dirs = directories::ProjectDirs::from("", "", "rustodo")
            .ok_or_else(|| anyhow::anyhow!("Could not determine config directory"))?;
        dirs.config_dir().to_path_buf()
    };
    Ok(config_dir
        .join("holidays")
        .join(locale)
        .join(format!("{}.json", year)))
}

// ── Refresh command ───────────────────────────────────────────────────────────

/// Force re-download of holiday data for `locale` and `year`.
/// Called by `todo holidays refresh`.
pub fn refresh(locale: &str, year: i32) -> Result<()> {
    let path = cache_path(locale, year)?;
    let data = fetch(locale, year)?;
    if let Some(parent) = path.parent() {
        fs::create_dir_all(parent)?;
    }
    fs::write(&path, &data)?;
    println!(
        "  Holidays updated: {} {} ({} entries)",
        locale,
        year,
        data.lines().filter(|l| !l.trim().is_empty()).count()
    );
    Ok(())
}