use std::sync::Arc;

use crate::{widgets::*, *};

use super::*;

/// A wrapper around other containers for things you often want in a window
pub struct Window {
    pub title_label: Label,
    pub floating: Floating,
    pub frame: Frame,
    pub resize: Resize,
    pub scroll: ScrollArea,
}

impl Window {
    // TODO: Into<Label>
    pub fn new(title: impl Into<String>) -> Self {
        let title = title.into();
        let floating = Floating::new(&title);
        let title_label = Label::new(title)
            .text_style(TextStyle::Heading)
            .multiline(false);
        Self {
            title_label,
            floating,
            frame: Frame::default(),
            resize: Resize::default()
                .handle_offset(Vec2::splat(4.0))
                .auto_shrink_width(true)
                .auto_expand_width(true)
                .auto_shrink_height(false)
                .auto_expand_height(false),
            scroll: ScrollArea::default()
                .always_show_scroll(false)
                .max_height(f32::INFINITY), // As large as we can be
        }
    }

    /// This is quite a crap idea
    /// Usage: `Winmdow::new(...).mutate(|w| w.resize = w.resize.auto_expand_width(true))`
    pub fn mutate(mut self, mutate: impl Fn(&mut Self)) -> Self {
        mutate(&mut self);
        self
    }

    /// This is quite a crap idea
    /// Usage: `Winmdow::new(...).resize(|r| r.auto_expand_width(true))`
    pub fn resize(mut self, mutate: impl Fn(Resize) -> Resize) -> Self {
        self.resize = mutate(self.resize);
        self
    }

    pub fn default_pos(mut self, default_pos: Pos2) -> Self {
        self.floating = self.floating.default_pos(default_pos);
        self
    }

    pub fn default_size(mut self, default_size: Vec2) -> Self {
        self.resize = self.resize.default_size(default_size);
        self
    }

    pub fn min_size(mut self, min_size: Vec2) -> Self {
        self.resize = self.resize.min_size(min_size);
        self
    }

    pub fn max_size(mut self, max_size: Vec2) -> Self {
        self.resize = self.resize.max_size(max_size);
        self
    }

    pub fn fixed_size(mut self, size: Vec2) -> Self {
        self.resize = self.resize.fixed_size(size);
        self
    }

    /// Can you resize it with the mouse?
    /// Note that a window can still auto-resize
    pub fn resizable(mut self, resizable: bool) -> Self {
        self.resize = self.resize.resizable(resizable);
        self
    }
}

impl Window {
    pub fn show(self, ctx: &Arc<Context>, add_contents: impl FnOnce(&mut Ui)) {
        let Window {
            title_label,
            floating,
            mut frame,
            resize,
            scroll,
        } = self;
        frame.margin = Some(frame.margin.unwrap_or_else(|| ctx.style().window_padding));

        // TODO: easier way to compose these
        floating.show(ctx, |ui| {
            frame.show(ui, |ui| {
                resize.show(ui, |ui| {
                    ui.add(title_label);
                    ui.add(Separator::new().line_width(1.0)); // TODO: nicer way to split window title from contents
                    scroll.show(ui, |ui| add_contents(ui))
                })
            })
        })
    }
}