use crate::mini_salsa::{run_ui, setup_logging, MiniSalsaState};
use crate::substratum1::{Substratum, SubstratumState};
use crate::substratum2::{Substratum2, Substratum2State};
use rat_event::{ConsumedEvent, HandleEvent, Outcome, Regular};
use rat_focus::{Focus, FocusBuilder};
use ratatui::layout::{Constraint, Layout, Rect};
use ratatui::widgets::Block;
use ratatui::Frame;

mod adapter;
mod mini_salsa;

fn main() -> Result<(), anyhow::Error> {
    setup_logging()?;

    let mut data = Data {};

    let mut state = State {
        sub1: Substratum2State::named("sub1"),
        sub3: SubstratumState::named("sub3"),
        sub4: SubstratumState::named("sub4"),
    };
    focus_input(&mut state).next();

    run_ui(
        "focus_recursive2",
        handle_input,
        repaint_input,
        &mut data,
        &mut state,
    )
}

struct Data {}

struct State {
    pub(crate) sub1: Substratum2State,
    pub(crate) sub3: SubstratumState,
    pub(crate) sub4: SubstratumState,
}

fn repaint_input(
    frame: &mut Frame<'_>,
    area: Rect,
    _data: &mut Data,
    _istate: &mut MiniSalsaState,
    state: &mut State,
) -> Result<(), anyhow::Error> {
    let l0 = Layout::horizontal([
        Constraint::Length(25),
        Constraint::Length(25),
        Constraint::Fill(1),
    ])
    .split(area);

    let w1 = Substratum2::new().block(Block::bordered().title("First"));
    frame.render_stateful_widget(w1, l0[0], &mut state.sub1);

    let l11 = Layout::vertical([Constraint::Length(8), Constraint::Length(8)]).split(l0[1]);

    let w3 = Substratum::new().block(Block::bordered().title("Third"));
    frame.render_stateful_widget(w3, l11[0], &mut state.sub3);
    let w4 = Substratum::new().block(Block::bordered().title("Forth"));
    frame.render_stateful_widget(w4, l11[1], &mut state.sub4);

    let cursor = state.sub1.screen_cursor().or_else(|| {
        state
            .sub3
            .screen_cursor()
            .or_else(|| state.sub4.screen_cursor())
    });
    if let Some((x, y)) = cursor {
        frame.set_cursor_position((x, y));
    }
    Ok(())
}

fn focus_input(state: &mut State) -> Focus {
    let mut fb = FocusBuilder::default();
    fb.container(&state.sub1)
        .container(&state.sub3)
        .container(&state.sub4);
    fb.build()
}

fn handle_input(
    event: &crossterm::event::Event,
    _data: &mut Data,
    _istate: &mut MiniSalsaState,
    state: &mut State,
) -> Result<Outcome, anyhow::Error> {
    let f = focus_input(state).handle(event, Regular);
    let r = f.and(|| {
        (state.sub1.handle(event, Regular))
            .or_else(|| state.sub3.handle(event, Regular))
            .or_else(|| state.sub4.handle(event, Regular))
    });
    Ok(r)
}

pub mod substratum2 {
    use crate::mini_salsa::theme::THEME;
    use crate::substratum1::{Substratum, SubstratumState};
    use rat_event::{ConsumedEvent, HandleEvent, Outcome, Regular};
    use rat_focus::{ContainerFlag, FocusBuilder, FocusContainer};
    use ratatui::buffer::Buffer;
    use ratatui::layout::{Constraint, Layout, Rect};
    use ratatui::prelude::{BlockExt, Style, Widget};
    use ratatui::widgets::{Block, StatefulWidget};

    #[derive(Debug, Default)]
    pub struct Substratum2<'a> {
        block: Option<Block<'a>>,
    }

    impl<'a> Substratum2<'a> {
        pub fn new() -> Self {
            Self::default()
        }

        pub fn block(mut self, block: Block<'a>) -> Self {
            self.block = Some(block);
            self
        }
    }

    #[derive(Debug, Default)]
    pub struct Substratum2State {
        pub container_focus: ContainerFlag,
        pub area: Rect,
        pub stratum1: SubstratumState,
        pub stratum2: SubstratumState,
    }

    impl Substratum2State {
        pub fn named(name: &str) -> Self {
            Self {
                container_focus: ContainerFlag::named(name),
                area: Default::default(),
                stratum1: SubstratumState::named(format!("{}.1", name).as_str()),
                stratum2: SubstratumState::named(format!("{}.2", name).as_str()),
            }
        }
    }

    impl<'a> StatefulWidget for Substratum2<'a> {
        type State = Substratum2State;

        fn render(mut self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
            state.area = area;

            let inner = self.block.inner_if_some(area);

            self.block = if state.container_focus.get() {
                if let Some(block) = self.block {
                    Some(block.border_style(Style::default().fg(THEME.secondary[2])))
                } else {
                    self.block
                }
            } else {
                self.block
            };

            self.block.render(area, buf);

            let ll = Layout::vertical([Constraint::Length(8), Constraint::Length(8)]).split(inner);

            let ss1 = Substratum::new().block(Block::bordered().title("Primus"));
            ss1.render(ll[0], buf, &mut state.stratum1);

            let ss2 = Substratum::new().block(Block::bordered().title("Secundus"));
            ss2.render(ll[1], buf, &mut state.stratum2);
        }
    }

    impl Substratum2State {
        pub fn screen_cursor(&self) -> Option<(u16, u16)> {
            if self.stratum1.is_container_focused() {
                self.stratum1.screen_cursor()
            } else if self.stratum2.is_container_focused() {
                self.stratum2.screen_cursor()
            } else {
                None
            }
        }
    }

    impl FocusContainer for Substratum2State {
        fn build(&self, builder: &mut FocusBuilder) {
            builder.container(&self.stratum1);
            builder.container(&self.stratum2);
        }

        fn container(&self) -> Option<ContainerFlag> {
            Some(self.container_focus.clone())
        }

        fn area(&self) -> Rect {
            self.area
        }
    }

    impl HandleEvent<crossterm::event::Event, Regular, Outcome> for Substratum2State {
        fn handle(&mut self, event: &crossterm::event::Event, _keymap: Regular) -> Outcome {
            self.stratum1
                .handle(event, Regular) //
                .or_else(|| self.stratum2.handle(event, Regular))
        }
    }
}

pub mod substratum1 {
    use crate::adapter::textinputf::{TextInputF, TextInputFState};
    use crate::mini_salsa::layout_grid;
    use crate::mini_salsa::theme::THEME;
    use rat_event::{ConsumedEvent, HandleEvent, Outcome, Regular};
    use rat_focus::{ContainerFlag, FocusBuilder, FocusContainer, HasFocus};
    use ratatui::buffer::Buffer;
    use ratatui::layout::{Constraint, Layout, Rect};
    use ratatui::prelude::{BlockExt, Span, StatefulWidget, Style};
    use ratatui::widgets::{Block, Widget};

    #[derive(Debug, Default)]
    pub struct Substratum<'a> {
        block: Option<Block<'a>>,
    }

    impl<'a> Substratum<'a> {
        pub fn new() -> Self {
            Self {
                block: Default::default(),
            }
        }

        pub fn block(mut self, block: Block<'a>) -> Self {
            self.block = Some(block);
            self
        }
    }

    #[derive(Debug, Default)]
    pub struct SubstratumState {
        pub container_focus: ContainerFlag,
        pub area: Rect,
        pub input1: TextInputFState,
        pub input2: TextInputFState,
        pub input3: TextInputFState,
        pub input4: TextInputFState,
    }

    impl SubstratumState {
        pub fn named(name: &str) -> Self {
            Self {
                container_focus: ContainerFlag::named(name),
                area: Default::default(),
                input1: Default::default(),
                input2: Default::default(),
                input3: Default::default(),
                input4: Default::default(),
            }
        }
    }

    impl<'a> StatefulWidget for Substratum<'a> {
        type State = SubstratumState;

        fn render(mut self, area: Rect, buf: &mut Buffer, state: &mut Self::State) {
            state.area = area;

            let inner = self.block.inner_if_some(area);

            self.block = if state.container_focus.get() {
                if let Some(block) = self.block {
                    Some(block.border_style(Style::default().fg(THEME.secondary[3])))
                } else {
                    self.block
                }
            } else {
                self.block
            };

            self.block.render(area, buf);

            let l_grid = layout_grid::<2, 4>(
                inner,
                Layout::horizontal([Constraint::Length(10), Constraint::Length(20)]),
                Layout::vertical([
                    Constraint::Length(1),
                    Constraint::Length(1),
                    Constraint::Length(1),
                    Constraint::Length(1),
                ]),
            );

            Span::from("Text 1").render(l_grid[0][0], buf);
            let input1 = TextInputF::default()
                .style(THEME.text_input())
                .focus_style(THEME.text_input_focus());
            input1.render(l_grid[1][0], buf, &mut state.input1);

            Span::from("Text 2").render(l_grid[0][1], buf);
            let input2 = TextInputF::default()
                .style(THEME.text_input())
                .focus_style(THEME.text_input_focus());
            input2.render(l_grid[1][1], buf, &mut state.input2);

            Span::from("Text 3").render(l_grid[0][2], buf);
            let input3 = TextInputF::default()
                .style(THEME.text_input())
                .focus_style(THEME.text_input_focus());
            input3.render(l_grid[1][2], buf, &mut state.input3);

            Span::from("Text 4").render(l_grid[0][3], buf);
            let input4 = TextInputF::default()
                .style(THEME.text_input())
                .focus_style(THEME.text_input_focus());
            input4.render(l_grid[1][3], buf, &mut state.input4);
        }
    }

    impl SubstratumState {
        pub fn screen_cursor(&self) -> Option<(u16, u16)> {
            if self.input1.is_focused() {
                self.input1.screen_cursor()
            } else if self.input2.is_focused() {
                self.input2.screen_cursor()
            } else if self.input3.is_focused() {
                self.input3.screen_cursor()
            } else if self.input4.is_focused() {
                self.input4.screen_cursor()
            } else {
                None
            }
        }
    }

    impl FocusContainer for SubstratumState {
        fn build(&self, builder: &mut FocusBuilder) {
            builder
                .widget(&self.input1)
                .widget(&self.input2)
                .widget(&self.input3)
                .widget(&self.input4);
        }

        fn container(&self) -> Option<ContainerFlag> {
            Some(self.container_focus.clone())
        }

        fn area(&self) -> Rect {
            self.area
        }
    }

    impl HandleEvent<crossterm::event::Event, Regular, Outcome> for SubstratumState {
        fn handle(&mut self, event: &crossterm::event::Event, _keymap: Regular) -> Outcome {
            self.input1
                .handle(event, Regular) //
                .or_else(|| self.input2.handle(event, Regular))
                .or_else(|| self.input3.handle(event, Regular))
                .or_else(|| self.input4.handle(event, Regular))
        }
    }
}