// 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;
use std::str::FromStr;
use std::string::ToString;

mod locale;

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(d: &serde_yaml::Value) {
    let name = d.get("name").unwrap();
    println!("{}", capitalize(name.as_str().unwrap()));

    if let Some(alt_names) = d.get("alternative_names") {
        let names = alt_names.as_sequence().unwrap().iter()
            .map(|v| capitalize(v.as_str().unwrap()))
            .collect::<Vec<_>>();
        println!("({})", names.join(", "));
    }

    if let Some(syms) = d.get("symbols") {
        let s = syms.as_sequence().unwrap().iter()
            .map(|v| v.as_str().unwrap())
            .collect::<Vec<_>>();
        println!("Symboler: {}", s.join(", "));
    }
}

fn main() {
    let f = BufReader::new(File::open("data/primstav.yml").unwrap());
    let data: serde_yaml::Value = 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)));

    let key = local_date.format("%d%m").to_string();
    match data.get(serde_yaml::Value::Number(u32::from_str(&key).unwrap().into())) {
        Some(d) => print_entry(&d),
        None => ()
    }
}