path: root/src/main.rs

// primstav - look up marked days on the norwegian primstav.
// Copyright (C) 2018  Harald Eilertsen <haraldei@anduin.net>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

use chrono;
use serde_yaml;
use std::io::BufReader;
use std::iter::Iterator;
use std::fs::File;

mod locale;

use primstav::{Primstav, Mark};

fn capitalize(s: &str) -> String {
    let mut chars = s.chars();
    match chars.next() {
        Some(ch) => ch.to_uppercase().chain(chars).collect::<String>(),
        None => String::new(),
    }
}

#[test]
fn test_capitalize() {
    assert_eq!("Flaccid", capitalize("flaccid"));
    assert_eq!("Ångström", capitalize("ångström"));
    assert_eq!("Å i lofoten", capitalize("å i lofoten"));
    assert_eq!("Ååååh", capitalize("ååååh"));
    assert_eq!("", capitalize(""));
}

fn print_entry(m: &Mark) {
    println!("{}", capitalize(&m.name));

    if !m.alternative_names.is_empty() {
        let names = m.alternative_names.iter()
            .map(|v| capitalize(&v))
            .collect::<Vec<_>>();
        println!("({})", names.join(", "));
    }

    if !m.symbols.is_empty() {
        println!("Symboler: {}", m.symbols.join(", "));
    }

    if !m.links.is_empty() {
        println!("Kilder:");
        for (source, url) in m.links.iter() {
            println!("  {}: {}", source, url);
        }
    }
}

fn main() {
    let f = BufReader::new(File::open("data/primstav.yml").unwrap());
    let p = Primstav::new(serde_yaml::from_reader(f).unwrap());

    let mut args = std::env::args().skip(1);
    let local_date = match args.next() {
        Some(s) => chrono::NaiveDate::parse_from_str(&s, "%d-%m-%Y").unwrap(),
        None => chrono::Local::today().naive_utc(),
    };

    let locale = locale::get();
    println!("{}", capitalize(&locale.format_date(&local_date)));

    match p.get(&local_date) {
        Some(d) => print_entry(&d),
        None => ()
    }
}