use crate::prelude::*;
use iris_macro::widget_trait;

// TODO: naming in here is a bit weird like eventable
#[macro_export]
macro_rules! event_ctx {
    ($ty: ty) => {
        mod local_event_trait {
            use super::*;
            use std::marker::Sized;
            #[allow(unused_imports)]
            use $crate::prelude::*;

            #[allow(unused)]
            pub trait EventableCtx<W: ?Sized, Tag, Ctx: 'static> {
                fn on<E: Event>(
                    self,
                    event: E,
                    f: impl WidgetEventFn<Ctx, E::Data, W>,
                ) -> impl WidgetIdFn<W> + EventableCtx<W, IdFnTag, Ctx>;
            }

            impl<WL: WidgetLike<Tag>, Tag> EventableCtx<WL::Widget, Tag, $ty> for WL {
                fn on<E: Event>(
                    self,
                    event: E,
                    f: impl WidgetEventFn<$ty, E::Data, WL::Widget>,
                ) -> impl WidgetIdFn<WL::Widget> + EventableCtx<WL::Widget, IdFnTag, $ty> {
                    eventable::Eventable::on(self, event, f)
                }
            }

            #[allow(unused)]
            pub trait EventableCtxUi<W: ?Sized, Tag, Ctx: 'static>
            where
                WidgetRef<W>: EventableCtx<W, Tag, Ctx>,
            {
                fn on<E: Event>(
                    &mut self,
                    widget: &WidgetRef<W>,
                    event: E,
                    f: impl WidgetEventFn<Ctx, E::Data, W>,
                );
            }

            impl<W: ?Sized + 'static, Tag> EventableCtxUi<W, Tag, $ty> for Ui
            where
                WidgetRef<W>: EventableCtx<W, Tag, $ty>,
            {
                fn on<E: Event>(
                    &mut self,
                    widget: &WidgetRef<W>,
                    event: E,
                    f: impl WidgetEventFn<$ty, E::Data, W>,
                ) {
                    self.register_widget_event(&widget, event, f);
                }
            }
        }
        use local_event_trait::*;
    };
}
pub use event_ctx;

pub mod eventable {
    use super::*;

    widget_trait!(
        pub Eventable;
        fn on<E: Event, Ctx: 'static>(
            self,
            event: E,
            f: impl WidgetEventFn<Ctx, E::Data, WL::Widget>,
        ) -> impl WidgetIdFn<WL::Widget> {
            move |ui| {
                let id = self.add(ui);
                ui.register_widget_event(&id, event, f);
                id
            }
        }
    );
}