use super::*;
use crate::prelude::*;

pub trait CoreWidget<W, Tag> {
    fn pad(self, padding: impl Into<Padding>) -> impl WidgetFn<Pad>;
    fn align(self, align: Align) -> impl WidgetFn<Aligned>;
    fn center(self) -> impl WidgetFn<Aligned>;
    fn label(self, label: impl Into<String>) -> impl WidgetIdFn<W>;
    fn sized(self, size: impl Into<Size>) -> impl WidgetFn<Sized>;
    fn width(self, len: impl Into<Len>) -> impl WidgetFn<Sized>;
    fn height(self, len: impl Into<Len>) -> impl WidgetFn<Sized>;
    fn offset(self, amt: impl Into<UiVec2>) -> impl WidgetFn<Offset>;
    fn scroll(self) -> impl WidgetIdFn<Offset>;
    fn masked(self) -> impl WidgetFn<Masked>;
    fn background<T>(self, w: impl WidgetLike<T>) -> impl WidgetFn<Stack>;
    fn z_offset(self, offset: usize) -> impl WidgetFn<Stack>;
}

impl<W: WidgetLike<Tag>, Tag> CoreWidget<W::Widget, Tag> for W {
    fn pad(self, padding: impl Into<Padding>) -> impl WidgetFn<Pad> {
        |ui| Pad {
            padding: padding.into(),
            inner: self.add(ui).any(),
        }
    }

    fn align(self, align: Align) -> impl WidgetFn<Aligned> {
        move |ui| Aligned {
            inner: self.add(ui).any(),
            align,
        }
    }

    fn center(self) -> impl WidgetFn<Aligned> {
        self.align(Align::Center)
    }

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

    fn sized(self, size: impl Into<Size>) -> impl WidgetFn<Sized> {
        let size = size.into();
        move |ui| Sized {
            inner: self.add(ui).any(),
            x: Some(size.x),
            y: Some(size.y),
        }
    }

    fn width(self, len: impl Into<Len>) -> impl WidgetFn<Sized> {
        let len = len.into();
        move |ui| Sized {
            inner: self.add(ui).any(),
            x: Some(len),
            y: None,
        }
    }

    fn height(self, len: impl Into<Len>) -> impl WidgetFn<Sized> {
        let len = len.into();
        move |ui| Sized {
            inner: self.add(ui).any(),
            x: None,
            y: Some(len),
        }
    }

    fn offset(self, amt: impl Into<UiVec2>) -> impl WidgetFn<Offset> {
        move |ui| Offset {
            inner: self.add(ui).any(),
            amt: amt.into(),
        }
    }

    fn scroll(self) -> impl WidgetIdFn<Offset> {
        self.offset(UiVec2::ZERO)
            .edit_on(CursorSense::Scroll, |w, data| {
                w.amt += UiVec2::abs(data.scroll_delta * 50.0);
            })
    }

    fn masked(self) -> impl WidgetFn<Masked> {
        move |ui| Masked {
            inner: self.add(ui).any(),
        }
    }

    fn background<T>(self, w: impl WidgetLike<T>) -> impl WidgetFn<Stack> {
        move |ui| Stack {
            children: vec![w.add(ui).any(), self.add(ui).any()],
            size: StackSize::Child(1),
            offset: 0,
        }
    }

    fn z_offset(self, offset: usize) -> impl WidgetFn<Stack> {
        move |ui| Stack {
            children: vec![self.add(ui).any()],
            size: StackSize::Child(0),
            offset,
        }
    }
}

pub trait CoreWidgetArr<const LEN: usize, Wa: WidgetArrLike<LEN, Tag>, Tag> {
    fn span(self, dir: Dir) -> SpanBuilder<LEN, Wa, Tag>;
    fn stack(self) -> StackBuilder<LEN, Wa, Tag>;
}

impl<const LEN: usize, Wa: WidgetArrLike<LEN, Tag>, Tag> CoreWidgetArr<LEN, Wa, Tag> for Wa {
    fn span(self, dir: Dir) -> SpanBuilder<LEN, Wa, Tag> {
        SpanBuilder::new(self, dir)
    }
    fn stack(self) -> StackBuilder<LEN, Wa, Tag> {
        StackBuilder::new(self)
    }
}