use jiff::civil;
use showbits_common::{
    Node, Tree, WidgetExt,
    color::{BLACK, WHITE},
    widgets::{Block, Text},
};
use taffy::{
    AlignContent, AlignItems, Display, FlexDirection,
    style_helpers::{length, percent, repeat},
};

use crate::persistent_printer::PersistentPrinter;

use super::{Context, Drawing, FEED};

pub struct CalendarDrawing {
    pub year: i16,
    pub month: i8,
}

impl Drawing for CalendarDrawing {
    fn draw(&self, printer: &mut PersistentPrinter, ctx: &mut Context) -> anyhow::Result<()> {
        let mut date = civil::Date::new(self.year, self.month, 1)?;

        let mut tree = Tree::<Context>::new(WHITE);

        let mut grid = Node::empty()
            .with_display(Display::Grid)
            .with_grid_template_columns(vec![repeat(7, vec![length(50.0)])])
            .with_grid_auto_rows(vec![length(50.0)])
            .with_gap(length(2.0));

        for weekday in ["Mo", "Di", "Mi", "Do", "Fr", "Sa", "So"] {
            let text = Text::new()
                .with_metrics(Text::default_metrics().scale(2.0))
                .and_plain(weekday)
                .widget(&mut ctx.font_stuff)
                .node()
                .with_justify_self(Some(AlignItems::Center))
                .with_align_self(Some(AlignItems::Center))
                .register(&mut tree)?;

            grid = grid.and_child(text);
        }

        let placeholders = date.weekday().to_monday_zero_offset();
        for _ in 0..placeholders {
            let empty = Node::empty().register(&mut tree)?;
            grid = grid.and_child(empty);
        }

        loop {
            let day = Text::new()
                .and_plain(date.day().to_string())
                .widget(&mut ctx.font_stuff)
                .node()
                .register(&mut tree)?;

            let block = Block::new()
                .with_border(BLACK)
                .node()
                .with_border(length(2.0))
                .with_display(Display::Flex)
                .with_justify_content(Some(AlignContent::Center))
                .with_align_items(Some(AlignItems::Center))
                .and_child(day)
                .register(&mut tree)?;

            grid = grid.and_child(block);

            let next_day = date.tomorrow()?;
            if date.month() != next_day.month() {
                break;
            }
            date = next_day;
        }

        let title = Text::new()
            .and_plain(format!(
                "Ankreuzkalender {:04}-{:02}",
                self.year, self.month
            ))
            .widget(&mut ctx.font_stuff)
            .node()
            .register(&mut tree)?;

        let root = Node::empty()
            .with_size_width(percent(1.0))
            .with_padding_bottom(length(FEED))
            .with_display(Display::Flex)
            .with_flex_direction(FlexDirection::Column)
            .with_align_items(Some(AlignItems::Center))
            .and_child(title)
            .and_child(grid.register(&mut tree)?)
            .register(&mut tree)?;

        printer.print_tree(&mut tree, ctx, root)?;
        Ok(())
    }
}