jay-compositor 1.13.0

use {
    crate::{
        ifs::wl_surface::xdg_surface::xdg_toplevel::XdgToplevel,
        it::{
            test_error::{TestError, TestResult},
            test_object::TestObject,
            test_transport::TestTransport,
            test_utils::test_window::TestWindow,
            testrun::ParseFull,
        },
        tree::{ContainerNode, ContainingNode, FloatNode, ToplevelNodeBase},
        utils::buffd::MsgParser,
        wire::{XdgToplevelId, xdg_toplevel::*},
    },
    ahash::AHashSet,
    std::{
        cell::{Cell, RefCell},
        future::poll_fn,
        rc::Rc,
        task::{Poll, Waker},
    },
};

pub struct TestXdgToplevelCore {
    pub id: XdgToplevelId,
    pub tran: Rc<TestTransport>,
    pub destroyed: Cell<bool>,

    pub configured: Cell<bool>,
    pub configured_waiter: Cell<Option<Waker>>,

    pub width: Cell<i32>,
    pub height: Cell<i32>,
    pub states: RefCell<AHashSet<u32>>,

    pub close_requested: Cell<bool>,
}

pub struct TestXdgToplevel {
    pub core: Rc<TestXdgToplevelCore>,
    pub server: Rc<XdgToplevel>,
}

impl TestXdgToplevel {
    pub fn parent(&self) -> TestResult<Rc<dyn ContainingNode>> {
        match self.server.tl_data().parent.get() {
            Some(p) => Ok(p),
            _ => bail!("toplevel has no parent"),
        }
    }

    pub fn container_parent(&self) -> TestResult<Rc<ContainerNode>> {
        let parent = self.parent()?;
        match parent.node_into_container() {
            Some(p) => Ok(p),
            _ => bail!("toplevel parent is not a container"),
        }
    }

    pub fn float_parent(&self) -> TestResult<Rc<FloatNode>> {
        let parent = self.parent()?;
        match parent.node_into_float() {
            Some(p) => Ok(p),
            _ => bail!("toplevel parent is not a float"),
        }
    }
}

impl TestXdgToplevelCore {
    pub fn destroy(&self) -> Result<(), TestError> {
        if !self.destroyed.replace(true) {
            self.tran.send(Destroy { self_id: self.id })?;
        }
        Ok(())
    }

    pub fn set_title(&self, title: &str) -> Result<(), TestError> {
        self.tran.send(SetTitle {
            self_id: self.id,
            title,
        })?;
        Ok(())
    }

    pub fn set_parent(&self, parent: &TestWindow) -> Result<(), TestError> {
        self.tran.send(SetParent {
            self_id: self.id,
            parent: parent.tl.server.id,
        })?;
        Ok(())
    }

    pub async fn configured(&self) {
        poll_fn(|ctx| {
            if self.configured.get() {
                return Poll::Ready(());
            }
            self.configured_waiter.set(Some(ctx.waker().clone()));
            Poll::Pending
        })
        .await;
    }

    fn handle_configure(&self, parser: MsgParser<'_, '_>) -> Result<(), TestError> {
        let ev = Configure::parse_full(parser)?;
        self.width.set(ev.width);
        self.height.set(ev.height);
        *self.states.borrow_mut() = ev.states.iter().copied().collect();
        self.configured.set(true);
        if let Some(waker) = self.configured_waiter.take() {
            waker.wake();
        }
        Ok(())
    }

    fn handle_close(&self, parser: MsgParser<'_, '_>) -> Result<(), TestError> {
        let _ev = Close::parse_full(parser)?;
        self.close_requested.set(true);
        Ok(())
    }

    fn handle_configure_bounds(&self, parser: MsgParser<'_, '_>) -> Result<(), TestError> {
        let _ev = ConfigureBounds::parse_full(parser)?;
        Ok(())
    }

    fn handle_wm_capabilities(&self, parser: MsgParser<'_, '_>) -> Result<(), TestError> {
        let _ev = WmCapabilities::parse_full(parser)?;
        Ok(())
    }
}

impl Drop for TestXdgToplevelCore {
    fn drop(&mut self) {
        let _ = self.destroy();
    }
}

test_object! {
    TestXdgToplevelCore, XdgToplevel;

    CONFIGURE => handle_configure,
    CLOSE => handle_close,
    CONFIGURE_BOUNDS => handle_configure_bounds,
    WM_CAPABILITIES => handle_wm_capabilities,
}

impl TestObject for TestXdgToplevelCore {}