amberwindow 0.3.14

use macroquad::prelude::*;
use super::*;

/// # Manages window update and creation.
/// 
/// Handles all windows and creates new windows.
/// 
/// ## Examples
/// 
/// ### Creating and updating windows
/// 
/// ```
/// let windows = WindowManager::new(); // Create new window manager
/// 
/// loop {
///     windows.begin("window"); // returns `Option<Window>`
///     windows.update_windows();       // Update + Render all windows.
/// }
/// ```
/// 
/// ### Getting window index from id
/// 
/// ```
/// // Get window with id = "window"
/// let idx = windows.get_window_index("window");
/// 
/// windows[idx].queue_free(); // Kill the window
/// ```
pub struct WindowManager {
    pub windows: Vec<Window>,
    pub frame_pushed: Vec<String>,
    pub freed: Vec<String>,
    pub font: Option<Font>,
}
impl WindowManager {
    /// Create a new WindowManager
    pub fn new() -> Self {
        Self {
            windows: vec![],
            frame_pushed: vec![],
            freed: vec![],
            font: None
        }
    }

    pub async fn set_font(&mut self, font_path: &str) -> &mut Self {
        self.font = Some(
            load_ttf_font(font_path).await.unwrap()
        );
        self
    }

    // Select a window (passing in a window)
    pub fn select(&mut self, idx: usize) {
        for i in self.windows.iter_mut() {
            i.selected = false;
        }
        self.windows[idx].selected = true;
    }

    /// Create a new immediate window with a ***unique*** `id`.
    /// 
    /// Returns an Option<&mut Window> which can have methods called to change itself.
    /// # Example
    /// 
    /// ```
    /// if let Some(win) = windows.begin("my_window") {
    ///     widget.Text(win, "hello");
    /// }
    /// ```
    pub fn begin(&mut self, id: &str) -> Option<&mut Window> {
        self.frame_pushed.push(id.to_string());

        if let Some(idx) = self.get_window_index(id) {
            if !self.check_freed(id) {
                self.windows[idx].frame_pushed.clear();
        
                return Some(&mut self.windows[idx]);
            }
        } else if !self.check_freed(id) {
            let uuid = RandomRange::gen_range(0, 999999999);
            let mut win = Window::new(
                "Window",
                Rect::new(0., 0., 200., 200.),
                self.font.clone(),
                None,
                id.to_string(),
                Some(uuid.to_string())
            );
            self.windows.push(win);
            let last = self.windows.len();
            return Some(&mut self.windows[last-1]);
        }
        None
    }

    /// Check if the a window with `id` has been freed.
    pub fn check_freed(&mut self, id: &str) -> bool {
        for i in self.windows.iter() {
            if i.id != id.to_owned() { continue }
            if i.queue_free {
                self.freed.push(i.uuid.clone());
                return true;
            } else if self.freed.contains(&i.uuid) {
                return true;
            }
        }
        false
    }

    /// Updates (ONLY) all windows.
    /// *(After calling begin and changing them)*
    pub fn update_windows(&mut self) {
        let mouse_position = vec2(mouse_position().0, mouse_position().1);

        let mut win_idx: usize = 0;
        let mut selected: Option<usize> = None;

        let mut windows = &mut self.windows;

        windows.retain(|x| self.frame_pushed.contains(&x.id));

        for win in windows.iter_mut() {
            if self.freed.contains(&win.uuid) { continue }

            let mut idx = 0;
            win.widgets.retain(|_| {
                if idx > win.frame_pushed.len()-1 {
                    return false;
                }
                idx += 1;
                true
            });
            win.update(selected, &mouse_position);

            if win.selected {
                selected = Some(win_idx);
            }

            win_idx += 1;
        }

        if let Some(idx) = selected {
            if windows.len() > 0 {
                let clone = windows[idx].clone();
                windows.remove(idx);
                windows.insert(0, clone);
            }
        }

        self.frame_pushed.clear();
    }

    /// Renders (ONLY) all windows.
    /// *(After calling `update_windows()``)*
    pub fn render_windows(&mut self) {
        let mut reversed = self.windows.clone();
        reversed.reverse();
        for win in reversed.iter_mut() {
            if self.freed.contains(&win.uuid) { continue }

            win.render();
        }

        self.frame_pushed.clear();
    }

    /// Updates and renders all windows.
    /// ## MUST BE CALLED AT THE END
    /// *(After calling begin and changing them)*
    pub fn end_windows(&mut self) {
        let mouse_position = vec2(mouse_position().0, mouse_position().1);

        let mut win_idx: usize = 0;
        let mut selected: Option<usize> = None;

        let mut windows = &mut self.windows;

        windows.retain(|x| self.frame_pushed.contains(&x.id));

        for win in windows.iter_mut() {
            if self.freed.contains(&win.uuid) { continue }

            let mut idx = 0;
            win.widgets.retain(|_| {
                if idx > win.frame_pushed.len()-1 {
                    return false;
                }
                idx += 1;
                true
            });
            win.update(selected, &mouse_position);

            if win.selected {
                selected = Some(win_idx);
            }

            win_idx += 1;
        }

        if let Some(idx) = selected {
            if windows.len() > 0 {
                let clone = windows[idx].clone();
                windows.remove(idx);
                windows.insert(0, clone);
            }
        }

        let mut reversed = windows.clone();
        reversed.reverse();
        for win in reversed.iter_mut() {
            if self.freed.contains(&win.uuid) { continue }

            win.render();
        }

        self.frame_pushed.clear();
    }

    /// Get the index of a window using its `id`.
    pub fn get_window_index(&mut self, id: &str) -> Option<usize> {
        let mut idx = 0;
        for i in self.windows.iter() {
            if i.id == id.to_owned() {
                return Some(idx);
            }
            idx += 1;
        }
        None
    }
}