use ratatui_core::layout;
use ratatui_core::style::Style;
use ratatui_core::terminal::Frame;

use ratatui_widgets::block;
use ratatui_widgets::borders;

use super::element::{Element, ElementWithChildren};
use crate::error;

#[derive(Default)]
pub struct Container<'a> {
    children: Vec<Box<dyn Element>>,
    constraint: layout::Constraint,
    layout: layout::Layout,
    widgets: ContainerWidgets<'a>,
}

impl ElementWithChildren for Container<'_> {
    #[inline(always)]
    fn add_child(&mut self, child: impl Into<Box<dyn Element>>) -> Result<(), error::LintuiniError>{
        self.children.push(child.into());
        

        Ok(())
    }
    #[inline(always)]
    fn children(&self) -> &[Box<dyn Element>] {
        self.children.as_slice()
    }
    #[inline(always)]
    fn children_mut(&mut self) -> &mut [Box<dyn Element>] {
        self.children.as_mut_slice()
    }
}

impl Element for Container<'_> {
    #[inline(always)]
    fn constraint(&self) -> layout::Constraint {
        self.constraint
    }

    fn render(&self, frm: &mut Frame, rect: layout::Rect) -> Result<(), error::LintuiniError> {
        frm.render_widget(&self.widgets.block0, rect);

        for (el, r) in self.children.iter().zip(&*self.layout.split(rect)) {
            el.render(frm, *r);
        }

        Ok(())
    }
}

impl Container<'_> {
    #[inline(always)]
    pub fn new() -> Self {
        Container::default()
    }

    #[inline(always)]
    pub fn constraint(mut self, constraint: layout::Constraint) -> Self {
        self.constraint = constraint;
        self
    }

    #[inline(always)]
    pub fn direction(mut self, direction: layout::Direction) -> Self {
        self.layout = self.layout.direction(direction);
        self.regenerate_layout()
    }

    fn regenerate_layout(mut self) -> Self {
        if self.children.len() == 0 { return self; }
        self.layout = self.layout.constraints(
            self.children
                .iter()
                .map(|el| el.constraint())
        );
        self
    }
}
#[derive(Clone, Default)]
struct ContainerWidgets<'a> {
    block0: block::Block<'a>,
}