use super::*;
use crate::layout::{
    Dir, UiPos, UiRegion, Vec2, WidgetArrLike, WidgetFnRet, WidgetIdFnRet, WidgetLike,
};

pub trait CoreWidget<W, Tag> {
    fn pad(self, padding: impl Into<Padding>) -> WidgetFnRet!(Regioned);
    fn center(self, size: impl Into<Vec2>) -> WidgetFnRet!(Regioned);
    fn region(self, region: UiRegion) -> WidgetFnRet!(Regioned);
    fn label(self, label: impl Into<String>) -> WidgetIdFnRet!(W);
}

impl<W: WidgetLike<Tag>, Tag> CoreWidget<W::Widget, Tag> for W {
    fn pad(self, padding: impl Into<Padding>) -> WidgetFnRet!(Regioned) {
        |ui| Regioned {
            region: padding.into().region(),
            inner: self.add(ui).erase_type(),
        }
    }

    fn center(self, size: impl Into<Vec2>) -> WidgetFnRet!(Regioned) {
        |ui| Regioned {
            region: UiPos::center().expand(size.into()),
            inner: self.add(ui).erase_type(),
        }
    }

    fn region(self, region: UiRegion) -> WidgetFnRet!(Regioned) {
        move |ui| Regioned {
            region,
            inner: self.add(ui).erase_type(),
        }
    }

    fn label(self, label: impl Into<String>) -> WidgetIdFnRet!(W::Widget) {
        |ui| {
            let id = self.add(ui);
            ui.set_label(&id, label.into());
            id
        }
    }
}

pub trait CoreWidgetArr<const LEN: usize, Tag> {
    fn span(self, dir: Dir, lengths: [impl Into<SpanLen>; LEN]) -> WidgetFnRet!(Span);
    fn stack(self) -> WidgetFnRet!(Stack);
}

impl<const LEN: usize, Wa: WidgetArrLike<LEN, Tag>, Tag> CoreWidgetArr<LEN, Tag> for Wa {
    fn span(self, dir: Dir, lengths: [impl Into<SpanLen>; LEN]) -> WidgetFnRet!(Span) {
        let lengths = lengths.map(Into::into);
        move |ui| Span {
            dir,
            children: self.ui(ui).arr.into_iter().zip(lengths).collect(),
        }
    }
    fn stack(self) -> WidgetFnRet!(Stack) {
        move |ui| Stack {
            children: self.ui(ui).arr.to_vec(),
        }
    }
}