mod ext_workspace_unstable;
mod river;

use wayrs_client::connection::Connection;
use wayrs_client::global::*;

use crate::pointer_btn::PointerBtn;
use crate::protocol::*;
use crate::state::State;
use ext_workspace_unstable::*;
use river::*;

pub enum WmInfoProvider {
    None,
    River(RiverInfoProvider),
    EWU(ExtWorkspaceUnstable),
}

pub type WmInfoCallback = fn(&mut Connection<State>, &mut State, WlOutput, WmInfo);

impl WmInfoProvider {
    pub fn bind(
        conn: &mut Connection<State>,
        globals: &Globals,
        callback: WmInfoCallback,
    ) -> WmInfoProvider {
        if let Some(river) = RiverInfoProvider::bind(conn, globals, callback) {
            Self::River(river)
        } else if let Some(ext_wp_u) = ExtWorkspaceUnstable::bind(conn, globals, callback) {
            Self::EWU(ext_wp_u)
        } else {
            Self::None
        }
    }

    pub fn new_ouput(&mut self, conn: &mut Connection<State>, output: WlOutput) {
        match self {
            Self::None => (),
            Self::River(x) => x.new_output(conn, output),
            Self::EWU(x) => x.new_ouput(conn, output),
        }
    }

    pub fn output_removed(&mut self, conn: &mut Connection<State>, output: WlOutput) {
        match self {
            Self::None => (),
            Self::River(x) => x.output_removed(conn, output),
            Self::EWU(x) => x.output_removed(conn, output),
        }
    }

    pub fn click_on_tag(
        &mut self,
        conn: &mut Connection<State>,
        output: WlOutput,
        seat: WlSeat,
        tag: &str,
        btn: PointerBtn,
    ) {
        match self {
            Self::None => (),
            Self::River(x) => x.click_on_tag(conn, output, seat, tag, btn),
            Self::EWU(x) => x.click_on_tag(conn, output, seat, tag, btn),
        }
    }
}

#[derive(Default, Debug)]
pub struct WmInfo {
    pub layout_name: Option<String>,
    pub tags: Vec<Tag>,
}

#[derive(Debug)]
pub struct Tag {
    pub name: String,
    pub is_focused: bool,
    pub is_active: bool,
    pub is_urgent: bool,
}