use xcb::xinerama::ScreenInfo;

use super::{Client, Tag};

/// Info stored for each monitor
#[derive(Debug)]
pub struct MonitorInfo {
    /// Clients attached to that monitor
    pub clients: Vec<Client>,

    /// How clients should be filtered by tag
    pub focused_tag: TagFocus,

    /// The monitor's geometry
    pub screen_info: MonitorGeometry,
}

/// How clients are currently being filtered by tag
#[derive(Debug, Clone, Copy)]
pub enum TagFocus {
    /// Only show clients with the given tag
    Tag(Tag),

    /// Show all clients
    All,
}
impl TagFocus {
    /// Check if a client with the given tag should be displayed when using this filter
    pub const fn matches(&self, tag: Tag) -> bool {
        match self {
            Self::Tag(x) => *x == tag,
            Self::All => true,
        }
    }

    /// Get the tag that new clients should be assigned when using this filter
    pub const fn create_tag(&self) -> Tag {
        match self {
            Self::Tag(x) => *x,
            Self::All => 1,
        }
    }
}

impl MonitorInfo {
    /// Iterate over all tiled clients, returning a mutable reference to each.
    pub fn clients_tiled_mut(&mut self) -> impl Iterator<Item = &mut Client> {
        self.clients
            .iter_mut()
            .filter(|c| c.tiled())
            .filter(|c| self.focused_tag.matches(c.tag))
    }
}

impl Default for MonitorInfo {
    fn default() -> Self {
        Self {
            clients: vec![],
            screen_info: MonitorGeometry {
                x_org: 0,
                y_org: 0,
                width: 0,
                height: 0,
            },
            focused_tag: TagFocus::Tag(1),
        }
    }
}

/// Info on the monitor's geometry.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct MonitorGeometry {
    pub x_org: i16,
    pub y_org: i16,
    pub width: u16,
    pub height: u16,
}

impl From<ScreenInfo> for MonitorGeometry {
    fn from(value: ScreenInfo) -> Self {
        Self {
            x_org: value.x_org,
            y_org: value.y_org,
            width: value.width,
            height: value.height,
        }
    }
}