use gregorian::{Date, DateResultExt};
use regex::Regex;

use std::borrow::Cow;

use ropey::RopeSlice;

use crate::{Range, Tendril};

#[derive(Copy, Clone, Debug, PartialEq, Eq)]
struct Format {
    regex: &'static str,
    separator: char,
}

// Only support formats that aren't region specific.
static FORMATS: &[Format] = &[
    Format {
        regex: r"(\d{4})-(\d{2})-(\d{2})",
        separator: '-',
    },
    Format {
        regex: r"(\d{4})/(\d{2})/(\d{2})",
        separator: '/',
    },
];

const DATE_LENGTH: usize = 10;

#[derive(Copy, Clone, Debug, PartialEq, Eq)]
enum DateField {
    Year,
    Month,
    Day,
}

#[derive(Debug, PartialEq, Eq)]
pub struct DateIncrementor {
    pub date: Date,
    pub range: Range,

    field: DateField,
    format: Format,
}

impl DateIncrementor {
    pub fn from_range(text: RopeSlice, range: Range) -> Option<DateIncrementor> {
        let from = range.from().saturating_sub(DATE_LENGTH);
        let to = (range.from() + DATE_LENGTH).min(text.len_chars());
        let (from_in_text, to_in_text) = (range.from() - from, range.to() - from);
        let text: Cow<str> = text.slice(from..to).into();

        FORMATS.iter().find_map(|&format| {
            let re = Regex::new(format.regex).ok()?;
            let captures = re.captures(&text)?;

            let date = captures.get(0)?;
            let offset = range.from() - from_in_text;
            let range = Range::new(date.start() + offset, date.end() + offset);

            let year = captures.get(1)?;
            let month = captures.get(2)?;
            let day = captures.get(3)?;

            let year_range = year.range();
            let month_range = month.range();
            let day_range = day.range();

            let to_inclusive = if to_in_text > from_in_text {
                to_in_text - 1
            } else {
                to_in_text
            };
            let field = if year_range.contains(&from_in_text) && year_range.contains(&to_inclusive)
            {
                DateField::Year
            } else if month_range.contains(&from_in_text) && month_range.contains(&to_inclusive) {
                DateField::Month
            } else if day_range.contains(&from_in_text) && day_range.contains(&to_inclusive) {
                DateField::Day
            } else {
                return None;
            };

            let year: i16 = year.as_str().parse().ok()?;
            let month: u8 = month.as_str().parse().ok()?;
            let day: u8 = day.as_str().parse().ok()?;

            let date = Date::new(year, month, day).ok()?;

            Some(DateIncrementor {
                date,
                field,
                range,
                format,
            })
        })
    }

    pub fn incremented_text(&self, amount: i64) -> Tendril {
        let date = match self.field {
            DateField::Year => self
                .date
                .add_years(amount.try_into().unwrap_or(0))
                .or_next_valid(),
            DateField::Month => self
                .date
                .add_months(amount.try_into().unwrap_or(0))
                .or_prev_valid(),
            DateField::Day => self.date.add_days(amount.try_into().unwrap_or(0)),
        };

        format!(
            "{:04}{}{:02}{}{:02}",
            date.year(),
            self.format.separator,
            date.month().to_number(),
            self.format.separator,
            date.day()
        )
        .into()
    }
}

#[cfg(test)]
mod test {
    use super::*;
    use crate::Rope;

    #[test]
    fn test_create_incrementor_for_year_with_dashes() {
        let rope = Rope::from_str("2021-11-15");

        for head in 0..=3 {
            let range = Range::point(head);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range),
                Some(DateIncrementor {
                    date: Date::new(2021, 11, 15).unwrap(),
                    range: Range::new(0, 10),
                    field: DateField::Year,
                    format: FORMATS[0],
                })
            );
        }
    }

    #[test]
    fn test_create_incrementor_for_month_with_dashes() {
        let rope = Rope::from_str("2021-11-15");

        for head in 5..=6 {
            let range = Range::point(head);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range),
                Some(DateIncrementor {
                    date: Date::new(2021, 11, 15).unwrap(),
                    range: Range::new(0, 10),
                    field: DateField::Month,
                    format: FORMATS[0],
                })
            );
        }
    }

    #[test]
    fn test_create_incrementor_for_day_with_dashes() {
        let rope = Rope::from_str("2021-11-15");

        for head in 8..=9 {
            let range = Range::point(head);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range),
                Some(DateIncrementor {
                    date: Date::new(2021, 11, 15).unwrap(),
                    range: Range::new(0, 10),
                    field: DateField::Day,
                    format: FORMATS[0],
                })
            );
        }
    }

    #[test]
    fn test_try_create_incrementor_on_dashes() {
        let rope = Rope::from_str("2021-11-15");

        for head in &[4, 7] {
            let range = Range::point(*head);
            assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None,);
        }
    }

    #[test]
    fn test_create_incrementor_for_year_with_slashes() {
        let rope = Rope::from_str("2021/11/15");

        for head in 0..=3 {
            let range = Range::point(head);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range),
                Some(DateIncrementor {
                    date: Date::new(2021, 11, 15).unwrap(),
                    range: Range::new(0, 10),
                    field: DateField::Year,
                    format: FORMATS[1],
                })
            );
        }
    }

    #[test]
    fn test_create_incrementor_for_month_with_slashes() {
        let rope = Rope::from_str("2021/11/15");

        for head in 5..=6 {
            let range = Range::point(head);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range),
                Some(DateIncrementor {
                    date: Date::new(2021, 11, 15).unwrap(),
                    range: Range::new(0, 10),
                    field: DateField::Month,
                    format: FORMATS[1],
                })
            );
        }
    }

    #[test]
    fn test_create_incrementor_for_day_with_slashes() {
        let rope = Rope::from_str("2021/11/15");

        for head in 8..=9 {
            let range = Range::point(head);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range),
                Some(DateIncrementor {
                    date: Date::new(2021, 11, 15).unwrap(),
                    range: Range::new(0, 10),
                    field: DateField::Day,
                    format: FORMATS[1],
                })
            );
        }
    }

    #[test]
    fn test_try_create_incrementor_on_slashes() {
        let rope = Rope::from_str("2021/11/15");

        for head in &[4, 7] {
            let range = Range::point(*head);
            assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None,);
        }
    }

    #[test]
    fn test_date_surrounded_by_spaces() {
        let rope = Rope::from_str("   2021-11-15  ");
        let range = Range::point(3);
        assert_eq!(
            DateIncrementor::from_range(rope.slice(..), range),
            Some(DateIncrementor {
                date: Date::new(2021, 11, 15).unwrap(),
                range: Range::new(3, 13),
                field: DateField::Year,
                format: FORMATS[0],
            })
        );
    }

    #[test]
    fn test_date_in_single_quotes() {
        let rope = Rope::from_str("date = '2021-11-15'");
        let range = Range::point(10);
        assert_eq!(
            DateIncrementor::from_range(rope.slice(..), range),
            Some(DateIncrementor {
                date: Date::new(2021, 11, 15).unwrap(),
                range: Range::new(8, 18),
                field: DateField::Year,
                format: FORMATS[0],
            })
        );
    }

    #[test]
    fn test_date_in_double_quotes() {
        let rope = Rope::from_str("let date = \"2021-11-15\";");
        let range = Range::point(12);
        assert_eq!(
            DateIncrementor::from_range(rope.slice(..), range),
            Some(DateIncrementor {
                date: Date::new(2021, 11, 15).unwrap(),
                range: Range::new(12, 22),
                field: DateField::Year,
                format: FORMATS[0],
            })
        );
    }

    #[test]
    fn test_date_cursor_one_right_of_date() {
        let rope = Rope::from_str("2021-11-15 ");
        let range = Range::point(10);
        assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
    }

    #[test]
    fn test_date_cursor_one_left_of_number() {
        let rope = Rope::from_str(" 2021-11-15");
        let range = Range::point(0);
        assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
    }

    #[test]
    fn test_invalid_dates() {
        let tests = [
            "0000-00-00",
            "1980-2-21",
            "1980-12-1",
            "12345",
            "2020-02-30",
            "1999-12-32",
            "19-12-32",
            "1-2-3",
            "0000/00/00",
            "1980/2/21",
            "1980/12/1",
            "12345",
            "2020/02/30",
            "1999/12/32",
            "19/12/32",
            "1/2/3",
        ];

        for invalid in tests {
            let rope = Rope::from_str(invalid);
            let range = Range::point(0);

            assert_eq!(DateIncrementor::from_range(rope.slice(..), range), None);
        }
    }

    #[test]
    fn test_increment_dates() {
        let tests = [
            // (original, cursor, amount, expected)
            ("2020-02-28", 0, 1, "2021-02-28"),
            ("2020-02-29", 0, 1, "2021-03-01"),
            ("2020-01-31", 5, 1, "2020-02-29"),
            ("2020-01-20", 5, 1, "2020-02-20"),
            ("2020-02-28", 8, 1, "2020-02-29"),
            ("2021-02-28", 8, 1, "2021-03-01"),
            ("2021-02-28", 0, -1, "2020-02-28"),
            ("2021-03-01", 0, -1, "2020-03-01"),
            ("2020-02-29", 5, -1, "2020-01-29"),
            ("2020-02-20", 5, -1, "2020-01-20"),
            ("2020-02-29", 8, -1, "2020-02-28"),
            ("2021-03-01", 8, -1, "2021-02-28"),
            ("1980/12/21", 8, 100, "1981/03/31"),
            ("1980/12/21", 8, -100, "1980/09/12"),
            ("1980/12/21", 8, 1000, "1983/09/17"),
            ("1980/12/21", 8, -1000, "1978/03/27"),
        ];

        for (original, cursor, amount, expected) in tests {
            let rope = Rope::from_str(original);
            let range = Range::point(cursor);
            assert_eq!(
                DateIncrementor::from_range(rope.slice(..), range)
                    .unwrap()
                    .incremented_text(amount),
                expected.into()
            );
        }
    }
}