wry/src/config/handler.rs

use crate::backend;
use crate::backend::InputDeviceId;
use crate::ifs::wl_seat::{SeatId, WlSeatGlobal};
use crate::state::{DeviceHandlerData, State};
use crate::tree::walker::NodeVisitorBase;
use crate::tree::{ContainerNode, ContainerSplit, FloatNode, Node};
use crate::utils::copyhashmap::CopyHashMap;
use crate::utils::debug_fn::debug_fn;
use crate::utils::errorfmt::ErrorFmt;
use crate::utils::numcell::NumCell;
use crate::utils::stack::Stack;
use crate::xkbcommon::XkbKeymap;
use bincode::error::DecodeError;
use jay_config::_private::bincode_ops;
use jay_config::_private::ipc::{ClientMessage, Response, ServerMessage};
use jay_config::keyboard::keymap::Keymap;
use jay_config::keyboard::mods::Modifiers;
use jay_config::keyboard::syms::KeySym;
use jay_config::{Axis, Direction, InputDevice, LogLevel, Seat};
use libloading::Library;
use log::Level;
use std::cell::Cell;
use std::rc::Rc;
use thiserror::Error;

pub(super) struct ConfigProxyHandler {
    pub client_data: Cell<*const u8>,
    pub dropped: Cell<bool>,
    pub _lib: Option<Library>,
    pub _version: u32,
    pub unref: unsafe extern "C" fn(data: *const u8),
    pub handle_msg: unsafe extern "C" fn(data: *const u8, msg: *const u8, size: usize),
    pub state: Rc<State>,
    pub next_id: NumCell<u64>,
    pub keymaps: CopyHashMap<Keymap, Rc<XkbKeymap>>,
    pub bufs: Stack<Vec<u8>>,
}

impl ConfigProxyHandler {
    pub fn send(&self, msg: &ServerMessage) {
        let mut buf = self.bufs.pop().unwrap_or_default();
        buf.clear();
        bincode::encode_into_std_write(msg, &mut buf, bincode_ops()).unwrap();
        unsafe {
            (self.handle_msg)(self.client_data.get(), buf.as_ptr(), buf.len());
        }
        self.bufs.push(buf);
    }

    fn id(&self) -> u64 {
        self.next_id.fetch_add(1)
    }

    fn handle_log_request(
        &self,
        level: LogLevel,
        msg: &str,
        file: Option<&str>,
        line: Option<u32>,
    ) {
        let level = match level {
            LogLevel::Error => Level::Error,
            LogLevel::Warn => Level::Warn,
            LogLevel::Info => Level::Info,
            LogLevel::Debug => Level::Debug,
            LogLevel::Trace => Level::Trace,
        };
        let debug = debug_fn(|fmt| {
            if let Some(file) = file {
                write!(fmt, "{}", file)?;
                if let Some(line) = line {
                    write!(fmt, ":{}", line)?;
                }
                write!(fmt, ": ")?;
            }
            write!(fmt, "{}", msg)?;
            Ok(())
        });
        log::log!(level, "{:?}", debug);
    }

    fn handle_create_seat(&self, name: &str) {
        let global_name = self.state.globals.name();
        let seat = WlSeatGlobal::new(global_name, name, &self.state);
        self.state.globals.add_global(&self.state, &seat);
        self.send(&ServerMessage::Response {
            response: Response::CreateSeat {
                seat: Seat(seat.id().raw() as _),
            },
        });
    }

    fn handle_parse_keymap(&self, keymap: &str) -> Result<(), ParseKeymapError> {
        let (keymap, res) = match self.state.xkb_ctx.keymap_from_str(keymap) {
            Ok(keymap) => {
                let id = Keymap(self.id());
                self.keymaps.set(id, keymap);
                (id, Ok(()))
            }
            _ => (Keymap::INVALID, Err(ParseKeymapError::ParsingFailed)),
        };
        self.send(&ServerMessage::Response {
            response: Response::ParseKeymap { keymap },
        });
        res
    }

    fn handle_set_keymap(&self, seat: Seat, keymap: Keymap) -> Result<(), SeatSetKeymapError> {
        let seat = self.get_seat(seat)?;
        let keymap = if keymap.is_invalid() {
            self.state.default_keymap.clone()
        } else {
            self.get_keymap(keymap)?
        };
        seat.set_keymap(&keymap);
        Ok(())
    }

    fn handle_focus(&self, seat: Seat, direction: Direction) -> Result<(), FocusError> {
        let seat = self.get_seat(seat)?;
        seat.move_focus(direction);
        Ok(())
    }

    fn handle_move(&self, seat: Seat, direction: Direction) -> Result<(), MoveError> {
        let seat = self.get_seat(seat)?;
        seat.move_focused(direction);
        Ok(())
    }

    fn handle_get_repeat_rate(&self, seat: Seat) -> Result<(), SeatGetRepeatRateError> {
        let seat = self.get_seat(seat)?;
        let (rate, delay) = seat.get_rate();
        self.send(&ServerMessage::Response {
            response: Response::GetRepeatRate { rate, delay },
        });
        Ok(())
    }

    fn handle_set_repeat_rate(
        &self,
        seat: Seat,
        rate: i32,
        delay: i32,
    ) -> Result<(), SeatSetRepeatRateError> {
        let seat = self.get_seat(seat)?;
        if rate < 0 {
            return Err(SeatSetRepeatRateError::NegativeRate);
        }
        if delay < 0 {
            return Err(SeatSetRepeatRateError::NegativeDelay);
        }
        seat.set_rate(rate, delay);
        Ok(())
    }

    fn get_device_handler_data(
        &self,
        device: InputDevice,
    ) -> Result<Rc<DeviceHandlerData>, CphError> {
        let data = self
            .state
            .input_device_handlers
            .borrow_mut()
            .get(&InputDeviceId::from_raw(device.0 as _))
            .map(|d| d.data.clone());
        match data {
            Some(d) => Ok(d),
            _ => Err(CphError::DeviceDoesNotExist(device)),
        }
    }

    fn get_seat(&self, seat: Seat) -> Result<Rc<WlSeatGlobal>, CphError> {
        let seats = self.state.globals.seats.lock();
        for seat_global in seats.values() {
            if seat_global.id().raw() == seat.0 as _ {
                return Ok(seat_global.clone());
            }
        }
        Err(CphError::SeatDoesNotExist(seat))
    }

    fn get_kb(&self, kb: InputDevice) -> Result<Rc<dyn backend::InputDevice>, CphError> {
        let kbs = self.state.input_device_handlers.borrow_mut();
        match kbs.get(&(InputDeviceId::from_raw(kb.0 as _))) {
            None => Err(CphError::KeyboardDoesNotExist(kb)),
            Some(kb) => Ok(kb.device.clone()),
        }
    }

    fn get_keymap(&self, keymap: Keymap) -> Result<Rc<XkbKeymap>, CphError> {
        match self.keymaps.get(&keymap) {
            Some(k) => Ok(k),
            None => Err(CphError::KeymapDoesNotExist(keymap)),
        }
    }

    fn handle_set_seat(&self, device: InputDevice, seat: Seat) -> Result<(), SetSeatError> {
        let seat = if seat.is_invalid() {
            None
        } else {
            Some(self.get_seat(seat)?)
        };
        let dev = self.get_device_handler_data(device)?;
        dev.seat.set(seat);
        Ok(())
    }

    fn handle_get_mono(&self, seat: Seat) -> Result<(), GetMonoError> {
        let seat = self.get_seat(seat)?;
        self.send(&ServerMessage::Response {
            response: Response::GetMono {
                mono: seat.get_mono().unwrap_or(false).into(),
            },
        });
        Ok(())
    }

    fn handle_set_mono(&self, seat: Seat, mono: bool) -> Result<(), SetMonoError> {
        let seat = self.get_seat(seat)?;
        seat.set_mono(mono);
        Ok(())
    }

    fn handle_get_split(&self, seat: Seat) -> Result<(), GetSplitError> {
        let seat = self.get_seat(seat)?;
        self.send(&ServerMessage::Response {
            response: Response::GetSplit {
                axis: seat
                    .get_split()
                    .unwrap_or(ContainerSplit::Horizontal)
                    .into(),
            },
        });
        Ok(())
    }

    fn handle_set_split(&self, seat: Seat, axis: Axis) -> Result<(), SetSplitError> {
        let seat = self.get_seat(seat)?;
        seat.set_split(axis.into());
        Ok(())
    }

    fn handle_add_shortcut(
        &self,
        seat: Seat,
        mods: Modifiers,
        sym: KeySym,
    ) -> Result<(), AddShortcutError> {
        let seat = self.get_seat(seat)?;
        seat.add_shortcut(mods, sym);
        Ok(())
    }

    fn handle_remove_shortcut(
        &self,
        seat: Seat,
        mods: Modifiers,
        sym: KeySym,
    ) -> Result<(), AddShortcutError> {
        let seat = self.get_seat(seat)?;
        seat.remove_shortcut(mods, sym);
        Ok(())
    }

    fn handle_get_input_devices(&self, seat: Option<Seat>) {
        let id = seat.map(|s| SeatId::from_raw(s.0 as _));
        let matches = |dhd: &DeviceHandlerData| {
            let id = match id {
                Some(id) => id,
                _ => return true,
            };
            if let Some(seat) = dhd.seat.get() {
                return seat.id() == id;
            }
            false
        };
        let mut res = vec![];
        {
            let devs = self.state.input_device_handlers.borrow_mut();
            for dev in devs.values() {
                if matches(&dev.data) {
                    res.push(InputDevice(dev.id.raw() as _));
                }
            }
        }
        self.send(&ServerMessage::Response {
            response: Response::GetInputDevices { devices: res },
        });
    }

    fn handle_get_seats(&self) {
        let seats = {
            let seats = self.state.globals.seats.lock();
            seats
                .values()
                .map(|seat| Seat::from_raw(seat.id().raw() as _))
                .collect()
        };
        self.send(&ServerMessage::Response {
            response: Response::GetSeats { seats },
        });
    }

    fn handle_run(
        &self,
        prog: &str,
        args: Vec<String>,
        env: Vec<(String, String)>,
    ) -> Result<(), RunError> {
        let forker = match self.state.forker.get() {
            Some(f) => f,
            _ => return Err(RunError::NoForker),
        };
        forker.spawn(prog.to_string(), args, env, None);
        Ok(())
    }

    fn handle_grab(&self, kb: InputDevice, grab: bool) -> Result<(), GrabError> {
        let kb = self.get_kb(kb)?;
        kb.grab(grab);
        Ok(())
    }

    fn handle_get_title_height(&self) {
        self.send(&ServerMessage::Response {
            response: Response::GetTitleHeight {
                height: self.state.theme.title_height.get(),
            },
        });
    }

    fn handle_get_border_width(&self) {
        self.send(&ServerMessage::Response {
            response: Response::GetBorderWidth {
                width: self.state.theme.border_width.get(),
            },
        });
    }

    fn handle_create_split(&self, seat: Seat, axis: Axis) -> Result<(), CreateSplitError> {
        let seat = self.get_seat(seat)?;
        seat.create_split(axis.into());
        Ok(())
    }

    fn handle_focus_parent(&self, seat: Seat) -> Result<(), FocusParentError> {
        let seat = self.get_seat(seat)?;
        seat.focus_parent();
        Ok(())
    }

    fn handle_quit(&self) {
        log::info!("Quitting");
        self.state.el.stop();
    }

    fn handle_switch_to(&self, vtnr: u32) {
        match self.state.backend.get() {
            Some(b) => b.switch_to(vtnr),
            _ => log::warn!("Cannot switch to VT {}: Backend has not yet started", vtnr),
        }
    }

    fn handle_toggle_floating(&self, seat: Seat) -> Result<(), FocusParentError> {
        let seat = self.get_seat(seat)?;
        seat.toggle_floating();
        Ok(())
    }

    fn spaces_change(&self) {
        struct V;
        impl NodeVisitorBase for V {
            fn visit_container(&mut self, node: &Rc<ContainerNode>) {
                node.on_spaces_changed();
                node.visit_children(self);
            }
            fn visit_float(&mut self, node: &Rc<FloatNode>) {
                node.on_spaces_changed();
                node.visit_children(self);
            }
        }
        self.state.root.clone().visit(&mut V);
    }

    fn colors_change(&self) {
        struct V;
        impl NodeVisitorBase for V {
            fn visit_container(&mut self, node: &Rc<ContainerNode>) {
                node.on_colors_changed();
                node.visit_children(self);
            }
            fn visit_float(&mut self, node: &Rc<FloatNode>) {
                node.on_colors_changed();
                node.visit_children(self);
            }
        }
        self.state.root.clone().visit(&mut V);
    }

    fn handle_set_title_height(&self, height: i32) -> Result<(), SetTitleHeightError> {
        if height < 0 {
            return Err(SetTitleHeightError::Negative(height));
        }
        if height > 1000 {
            return Err(SetTitleHeightError::Excessive(height));
        }
        self.state.theme.title_height.set(height);
        self.spaces_change();
        Ok(())
    }

    fn handle_set_border_width(&self, width: i32) -> Result<(), SetBorderWidthError> {
        if width < 0 {
            return Err(SetBorderWidthError::Negative(width));
        }
        if width > 1000 {
            return Err(SetBorderWidthError::Excessive(width));
        }
        self.state.theme.border_width.set(width);
        self.spaces_change();
        Ok(())
    }

    fn handle_set_title_color(&self, color: jay_config::theme::Color) {
        self.state.theme.title_color.set(color.into());
        self.colors_change();
    }

    fn handle_set_border_color(&self, color: jay_config::theme::Color) {
        self.state.theme.border_color.set(color.into());
    }

    fn handle_set_background_color(&self, color: jay_config::theme::Color) {
        self.state.theme.background_color.set(color.into());
    }

    fn handle_set_title_underline_color(&self, color: jay_config::theme::Color) {
        self.state.theme.underline_color.set(color.into());
    }

    pub fn handle_request(&self, msg: &[u8]) {
        if let Err(e) = self.handle_request_(msg) {
            log::error!("Could not handle client request: {}", ErrorFmt(e));
        }
    }

    fn handle_request_(&self, msg: &[u8]) -> Result<(), CphError> {
        let (request, _) = match bincode::decode_from_slice::<ClientMessage, _>(msg, bincode_ops())
        {
            Ok(msg) => msg,
            Err(e) => return Err(CphError::ParsingFailed(e)),
        };
        match request {
            ClientMessage::Log {
                level,
                msg,
                file,
                line,
            } => self.handle_log_request(level, msg, file, line),
            ClientMessage::CreateSeat { name } => self.handle_create_seat(name),
            ClientMessage::ParseKeymap { keymap } => self.handle_parse_keymap(keymap)?,
            ClientMessage::SeatSetKeymap { seat, keymap } => {
                self.handle_set_keymap(seat, keymap)?
            }
            ClientMessage::SeatGetRepeatRate { seat } => self.handle_get_repeat_rate(seat)?,
            ClientMessage::SeatSetRepeatRate { seat, rate, delay } => {
                self.handle_set_repeat_rate(seat, rate, delay)?
            }
            ClientMessage::SetSeat { device, seat } => self.handle_set_seat(device, seat)?,
            ClientMessage::GetMono { seat } => self.handle_get_mono(seat)?,
            ClientMessage::SetMono { seat, mono } => self.handle_set_mono(seat, mono)?,
            ClientMessage::GetSplit { seat } => self.handle_get_split(seat)?,
            ClientMessage::SetSplit { seat, axis } => self.handle_set_split(seat, axis)?,
            ClientMessage::AddShortcut { seat, mods, sym } => {
                self.handle_add_shortcut(seat, mods, sym)?
            }
            ClientMessage::RemoveShortcut { seat, mods, sym } => {
                self.handle_remove_shortcut(seat, mods, sym)?
            }
            ClientMessage::Focus { seat, direction } => self.handle_focus(seat, direction)?,
            ClientMessage::Move { seat, direction } => self.handle_move(seat, direction)?,
            ClientMessage::GetInputDevices { seat } => self.handle_get_input_devices(seat),
            ClientMessage::GetSeats => self.handle_get_seats(),
            ClientMessage::RemoveSeat { .. } => {}
            ClientMessage::Run { prog, args, env } => self.handle_run(prog, args, env)?,
            ClientMessage::GrabKb { kb, grab } => self.handle_grab(kb, grab)?,
            ClientMessage::SetTitleHeight { height } => self.handle_set_title_height(height)?,
            ClientMessage::SetBorderWidth { width } => self.handle_set_border_width(width)?,
            ClientMessage::SetTitleColor { color } => self.handle_set_title_color(color),
            ClientMessage::SetTitleUnderlineColor { color } => {
                self.handle_set_title_underline_color(color)
            }
            ClientMessage::SetBorderColor { color } => self.handle_set_border_color(color),
            ClientMessage::SetBackgroundColor { color } => self.handle_set_background_color(color),
            ClientMessage::GetTitleHeight => self.handle_get_title_height(),
            ClientMessage::GetBorderWidth => self.handle_get_border_width(),
            ClientMessage::CreateSplit { seat, axis } => self.handle_create_split(seat, axis)?,
            ClientMessage::FocusParent { seat } => self.handle_focus_parent(seat)?,
            ClientMessage::ToggleFloating { seat } => self.handle_toggle_floating(seat)?,
            ClientMessage::Quit => self.handle_quit(),
            ClientMessage::SwitchTo { vtnr } => self.handle_switch_to(vtnr),
        }
        Ok(())
    }
}

#[derive(Debug, Error)]
enum CphError {
    #[error("Could not process a `parse_keymap` request")]
    ParseKeymapError(#[from] ParseKeymapError),
    #[error("Could not process a `set_seat` request")]
    SetSeatError(#[from] SetSeatError),
    #[error("Could not process a `add_shortcut` request")]
    AddShortcutError(#[from] AddShortcutError),
    #[error("Could not process a `remove_shortcut` request")]
    RemoveShortcutError(#[from] RemoveShortcutError),
    #[error("Could not process a `set_keymap` request")]
    SeatSetKeymapError(#[from] SeatSetKeymapError),
    #[error("Could not process a `get_repeat_rate` request")]
    SeatGetRepeatRateError(#[from] SeatGetRepeatRateError),
    #[error("Could not process a `set_repeat_rate` request")]
    SeatSetRepeatRateError(#[from] SeatSetRepeatRateError),
    #[error("Could not process a `focus` request")]
    FocusError(#[from] FocusError),
    #[error("Could not process a `move` request")]
    MoveError(#[from] MoveError),
    #[error("Could not process a `set_mono` request")]
    SetMonoError(#[from] SetMonoError),
    #[error("Could not process a `get_mono` request")]
    GetMonoError(#[from] GetMonoError),
    #[error("Could not process a `set_split` request")]
    SetSplitError(#[from] SetSplitError),
    #[error("Could not process a `get_split` request")]
    GetSplitError(#[from] GetSplitError),
    #[error("Could not process a `run` request")]
    RunError(#[from] RunError),
    #[error("Could not process a `grab` request")]
    GrabError(#[from] GrabError),
    #[error("Could not process a `create_split` request")]
    CreateSplitError(#[from] CreateSplitError),
    #[error("Could not process a `focus_parent` request")]
    FocusParentError(#[from] FocusParentError),
    #[error("Could not process a `toggle_floating` request")]
    ToggleFloatingError(#[from] ToggleFloatingError),
    #[error("Could not process a `set_title_height` request")]
    SetTitleHeightError(#[from] SetTitleHeightError),
    #[error("Could not process a `set_border_width` request")]
    SetBorderWidthError(#[from] SetBorderWidthError),
    #[error("Device {0:?} does not exist")]
    DeviceDoesNotExist(InputDevice),
    #[error("Device {0:?} does not exist")]
    KeymapDoesNotExist(Keymap),
    #[error("Seat {0:?} does not exist")]
    SeatDoesNotExist(Seat),
    #[error("Keyboard {0:?} does not exist")]
    KeyboardDoesNotExist(InputDevice),
    #[error("Could not parse the message")]
    ParsingFailed(#[source] DecodeError),
}

#[derive(Debug, Error)]
enum ParseKeymapError {
    #[error("Parsing failed")]
    ParsingFailed,
}

#[derive(Debug, Error)]
enum SetSeatError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(SetSeatError, CphError);

#[derive(Debug, Error)]
enum AddShortcutError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(AddShortcutError, CphError);

#[derive(Debug, Error)]
enum RemoveShortcutError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(RemoveShortcutError, CphError);

#[derive(Debug, Error)]
enum SeatSetKeymapError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(SeatSetKeymapError, CphError);

#[derive(Debug, Error)]
enum SeatSetRepeatRateError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
    #[error("Rate is negative")]
    NegativeRate,
    #[error("Delay is negative")]
    NegativeDelay,
}
efrom!(SeatSetRepeatRateError, CphError);

#[derive(Debug, Error)]
enum SeatGetRepeatRateError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(SeatGetRepeatRateError, CphError);

#[derive(Debug, Error)]
enum FocusError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(FocusError, CphError);

#[derive(Debug, Error)]
enum MoveError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(MoveError, CphError);

#[derive(Debug, Error)]
enum SetMonoError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(SetMonoError, CphError);

#[derive(Debug, Error)]
enum GetMonoError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(GetMonoError, CphError);

#[derive(Debug, Error)]
enum SetSplitError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(SetSplitError, CphError);

#[derive(Debug, Error)]
enum GetSplitError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(GetSplitError, CphError);

#[derive(Debug, Error)]
enum RunError {
    #[error("The ol' forker is not available")]
    NoForker,
}

#[derive(Debug, Error)]
enum GrabError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(GrabError, CphError);

#[derive(Debug, Error)]
enum SetTitleHeightError {
    #[error("The height {0} is negative")]
    Negative(i32),
    #[error("The height {0} is larger than the maximum 1000")]
    Excessive(i32),
}

#[derive(Debug, Error)]
enum SetBorderWidthError {
    #[error("The width {0} is negative")]
    Negative(i32),
    #[error("The width {0} is larger than the maximum 1000")]
    Excessive(i32),
}

#[derive(Debug, Error)]
enum CreateSplitError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(CreateSplitError, CphError);

#[derive(Debug, Error)]
enum FocusParentError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(FocusParentError, CphError);

#[derive(Debug, Error)]
enum ToggleFloatingError {
    #[error(transparent)]
    CphError(#[from] Box<CphError>),
}
efrom!(ToggleFloatingError, CphError);