413 lines
12 KiB
Rust
413 lines
12 KiB
Rust
|
use std::{
|
||
|
|
env,
|
||
|
|
process::exit,
|
||
|
|
};
|
||
|
|
|
||
|
|
use wayland_client::{
|
||
|
|
protocol::{
|
||
|
|
wl_buffer, wl_compositor, wl_keyboard, wl_registry, wl_seat, wl_shm, wl_shm_pool,
|
||
|
|
wl_surface,
|
||
|
|
},
|
||
|
|
Connection, Dispatch, QueueHandle, WEnum,
|
||
|
|
};
|
||
|
|
|
||
|
|
use wayland_protocols::xdg::{
|
||
|
|
activation::v1::client::{xdg_activation_token_v1, xdg_activation_v1},
|
||
|
|
shell::client::{xdg_surface, xdg_toplevel, xdg_wm_base},
|
||
|
|
};
|
||
|
|
|
||
|
|
use crate::{
|
||
|
|
buffer::BufferManager,
|
||
|
|
fonts::MaybeFontMap,
|
||
|
|
keyboard::Keyboard,
|
||
|
|
widgets::{
|
||
|
|
Geometry, HorizontalLayout, IndexedLayout, Layout, Margin, VerticalLayout, Widget,
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
pub struct State {
|
||
|
|
pub running: bool,
|
||
|
|
pub dirty: bool,
|
||
|
|
pub activated: bool,
|
||
|
|
pub base_surface: Option<wl_surface::WlSurface>,
|
||
|
|
pub wm_base: Option<xdg_wm_base::XdgWmBase>,
|
||
|
|
pub xdg_surface: Option<(xdg_surface::XdgSurface, xdg_toplevel::XdgToplevel)>,
|
||
|
|
pub wl_shm: Option<wl_shm::WlShm>,
|
||
|
|
pub xdg_activation: Option<xdg_activation_v1::XdgActivationV1>,
|
||
|
|
pub configured: bool,
|
||
|
|
pub dimensions: (i32, i32),
|
||
|
|
pub bufmgr: BufferManager,
|
||
|
|
pub widgets: Vec<Box<dyn Widget>>,
|
||
|
|
pub keyboard: Keyboard,
|
||
|
|
pub fonts: MaybeFontMap,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl State {
|
||
|
|
pub fn add_buffer(&mut self, qh: &QueueHandle<Self>) {
|
||
|
|
self.bufmgr.add_buffer(
|
||
|
|
self.wl_shm.as_ref().expect("missing wl_shm"),
|
||
|
|
self.dimensions,
|
||
|
|
qh,
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn activate(&mut self, qh: &QueueHandle<Self>) {
|
||
|
|
let key = match env::var("XDG_ACTIVATION_TOKEN") {
|
||
|
|
Ok(token) => token,
|
||
|
|
Err(_) => return,
|
||
|
|
};
|
||
|
|
match (&self.base_surface, &self.xdg_activation) {
|
||
|
|
(Some(surface), Some(activation)) => {
|
||
|
|
activation.activate(key, surface);
|
||
|
|
let activation_token = activation.get_activation_token(qh, ());
|
||
|
|
activation_token.commit();
|
||
|
|
env::remove_var("XDG_ACTIVATION_TOKEN");
|
||
|
|
}
|
||
|
|
_ => (),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_registry::WlRegistry, ()> for State {
|
||
|
|
fn event(
|
||
|
|
state: &mut Self,
|
||
|
|
registry: &wl_registry::WlRegistry,
|
||
|
|
event: wl_registry::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
qh: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
if let wl_registry::Event::Global {
|
||
|
|
name, interface, ..
|
||
|
|
} = event
|
||
|
|
{
|
||
|
|
match &interface[..] {
|
||
|
|
"wl_compositor" => {
|
||
|
|
let compositor =
|
||
|
|
registry.bind::<wl_compositor::WlCompositor, _, _>(name, 4, qh, ());
|
||
|
|
let surface = compositor.create_surface(qh, ());
|
||
|
|
state.base_surface = Some(surface);
|
||
|
|
state.activate(qh);
|
||
|
|
|
||
|
|
if state.wm_base.is_some() && state.xdg_surface.is_none() {
|
||
|
|
state.init_xdg_surface(qh);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
"wl_shm" => {
|
||
|
|
state.wl_shm = Some(registry.bind::<wl_shm::WlShm, _, _>(name, 1, qh, ()));
|
||
|
|
}
|
||
|
|
"wl_seat" => {
|
||
|
|
registry.bind::<wl_seat::WlSeat, _, _>(name, 1, qh, ());
|
||
|
|
}
|
||
|
|
"xdg_wm_base" => {
|
||
|
|
let wm_base = registry.bind::<xdg_wm_base::XdgWmBase, _, _>(name, 1, qh, ());
|
||
|
|
state.wm_base = Some(wm_base);
|
||
|
|
|
||
|
|
if state.base_surface.is_some() && state.xdg_surface.is_none() {
|
||
|
|
state.init_xdg_surface(qh);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
"xdg_activation_v1" => {
|
||
|
|
state.xdg_activation = Some(
|
||
|
|
registry.bind::<xdg_activation_v1::XdgActivationV1, _, _>(name, 1, qh, ()),
|
||
|
|
);
|
||
|
|
state.activate(qh);
|
||
|
|
}
|
||
|
|
_ => {}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_compositor::WlCompositor, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
_: &wl_compositor::WlCompositor,
|
||
|
|
_: wl_compositor::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_surface::WlSurface, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
_: &wl_surface::WlSurface,
|
||
|
|
_: wl_surface::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_shm::WlShm, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
_: &wl_shm::WlShm,
|
||
|
|
_: wl_shm::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_shm_pool::WlShmPool, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
_: &wl_shm_pool::WlShmPool,
|
||
|
|
_: wl_shm_pool::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_buffer::WlBuffer, ()> for State {
|
||
|
|
fn event(
|
||
|
|
state: &mut Self,
|
||
|
|
buffer: &wl_buffer::WlBuffer,
|
||
|
|
_: wl_buffer::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
for buf in state.bufmgr.buffers.iter_mut() {
|
||
|
|
if &buf.buffer == buffer {
|
||
|
|
buf.release();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl State {
|
||
|
|
fn init_xdg_surface(&mut self, qh: &QueueHandle<State>) {
|
||
|
|
let wm_base = self.wm_base.as_ref().unwrap();
|
||
|
|
let base_surface = self.base_surface.as_ref().unwrap();
|
||
|
|
|
||
|
|
let xdg_surface = wm_base.get_xdg_surface(base_surface, qh, ());
|
||
|
|
let toplevel = xdg_surface.get_toplevel(qh, ());
|
||
|
|
toplevel.set_title("wldash".into());
|
||
|
|
|
||
|
|
base_surface.commit();
|
||
|
|
|
||
|
|
self.xdg_surface = Some((xdg_surface, toplevel));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<xdg_wm_base::XdgWmBase, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
wm_base: &xdg_wm_base::XdgWmBase,
|
||
|
|
event: xdg_wm_base::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
if let xdg_wm_base::Event::Ping { serial } = event {
|
||
|
|
wm_base.pong(serial);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<xdg_surface::XdgSurface, ()> for State {
|
||
|
|
fn event(
|
||
|
|
state: &mut Self,
|
||
|
|
xdg_surface: &xdg_surface::XdgSurface,
|
||
|
|
event: xdg_surface::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
if let xdg_surface::Event::Configure { serial, .. } = event {
|
||
|
|
xdg_surface.ack_configure(serial);
|
||
|
|
state.configured = true;
|
||
|
|
state.dirty = true;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<xdg_toplevel::XdgToplevel, ()> for State {
|
||
|
|
fn event(
|
||
|
|
state: &mut Self,
|
||
|
|
_: &xdg_toplevel::XdgToplevel,
|
||
|
|
event: xdg_toplevel::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
match event {
|
||
|
|
xdg_toplevel::Event::Configure {
|
||
|
|
width,
|
||
|
|
height,
|
||
|
|
states,
|
||
|
|
..
|
||
|
|
} => {
|
||
|
|
if (width, height) == (0, 0) {
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
let activated = states
|
||
|
|
.iter()
|
||
|
|
.find(|&st| *st as u32 == xdg_toplevel::State::Activated as u32)
|
||
|
|
.is_some();
|
||
|
|
|
||
|
|
if activated {
|
||
|
|
state.activated = true;
|
||
|
|
} else if state.activated && !activated {
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
if state.dimensions != (width, height) {
|
||
|
|
state.bufmgr.clear_buffers();
|
||
|
|
state.dimensions = (width, height);
|
||
|
|
let mut layout = VerticalLayout {
|
||
|
|
widgets: vec![
|
||
|
|
Box::new(HorizontalLayout {
|
||
|
|
widgets: vec![
|
||
|
|
Box::new(IndexedLayout { widget_idx: 0 }),
|
||
|
|
Box::new(Margin {
|
||
|
|
widget: Box::new(IndexedLayout { widget_idx: 1 }),
|
||
|
|
margin: (16, 8, 0, 0),
|
||
|
|
}),
|
||
|
|
Box::new(VerticalLayout {
|
||
|
|
widgets: vec![
|
||
|
|
Box::new(Margin {
|
||
|
|
widget: Box::new(IndexedLayout { widget_idx: 2 }),
|
||
|
|
margin: (16, 8, 8, 0),
|
||
|
|
}),
|
||
|
|
Box::new(Margin {
|
||
|
|
widget: Box::new(IndexedLayout { widget_idx: 3 }),
|
||
|
|
margin: (16, 8, 8, 0),
|
||
|
|
}),
|
||
|
|
],
|
||
|
|
}),
|
||
|
|
],
|
||
|
|
}),
|
||
|
|
Box::new(IndexedLayout { widget_idx: 4 }),
|
||
|
|
Box::new(IndexedLayout { widget_idx: 5 }),
|
||
|
|
],
|
||
|
|
};
|
||
|
|
|
||
|
|
state.fonts.resolve();
|
||
|
|
let fonts = state.fonts.unwrap();
|
||
|
|
layout.geometry_update(
|
||
|
|
&mut fonts.borrow_mut(),
|
||
|
|
&Geometry {
|
||
|
|
x: 0,
|
||
|
|
y: 0,
|
||
|
|
width: width as u32,
|
||
|
|
height: height as u32,
|
||
|
|
},
|
||
|
|
state,
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
xdg_toplevel::Event::Close => {
|
||
|
|
state.running = false;
|
||
|
|
}
|
||
|
|
_ => (),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_seat::WlSeat, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
seat: &wl_seat::WlSeat,
|
||
|
|
event: wl_seat::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
qh: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
if let wl_seat::Event::Capabilities {
|
||
|
|
capabilities: WEnum::Value(capabilities),
|
||
|
|
} = event
|
||
|
|
{
|
||
|
|
if capabilities.contains(wl_seat::Capability::Keyboard) {
|
||
|
|
seat.get_keyboard(qh, ());
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<wl_keyboard::WlKeyboard, ()> for State {
|
||
|
|
fn event(
|
||
|
|
state: &mut Self,
|
||
|
|
_: &wl_keyboard::WlKeyboard,
|
||
|
|
event: wl_keyboard::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
match event {
|
||
|
|
wl_keyboard::Event::Key {
|
||
|
|
serial,
|
||
|
|
time,
|
||
|
|
key,
|
||
|
|
state: kbstate,
|
||
|
|
} => {
|
||
|
|
if key == 1 {
|
||
|
|
state.running = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
let k = state.keyboard.key(serial, time, key, kbstate);
|
||
|
|
for widget in state.widgets.iter_mut() {
|
||
|
|
widget.keyboard_input(&k);
|
||
|
|
}
|
||
|
|
state.dirty = true;
|
||
|
|
}
|
||
|
|
wl_keyboard::Event::Modifiers {
|
||
|
|
mods_depressed,
|
||
|
|
mods_latched,
|
||
|
|
mods_locked,
|
||
|
|
group,
|
||
|
|
..
|
||
|
|
} => {
|
||
|
|
state
|
||
|
|
.keyboard
|
||
|
|
.modifiers(mods_depressed, mods_latched, mods_locked, group);
|
||
|
|
}
|
||
|
|
wl_keyboard::Event::Keymap { format, fd, size } => {
|
||
|
|
state.keyboard.keymap(format, fd, size);
|
||
|
|
}
|
||
|
|
_ => (),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<xdg_activation_v1::XdgActivationV1, ()> for State {
|
||
|
|
fn event(
|
||
|
|
_: &mut Self,
|
||
|
|
_: &xdg_activation_v1::XdgActivationV1,
|
||
|
|
_: xdg_activation_v1::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Dispatch<xdg_activation_token_v1::XdgActivationTokenV1, ()> for State {
|
||
|
|
fn event(
|
||
|
|
state: &mut Self,
|
||
|
|
_: &xdg_activation_token_v1::XdgActivationTokenV1,
|
||
|
|
event: xdg_activation_token_v1::Event,
|
||
|
|
_: &(),
|
||
|
|
_: &Connection,
|
||
|
|
_: &QueueHandle<Self>,
|
||
|
|
) {
|
||
|
|
match event {
|
||
|
|
xdg_activation_token_v1::Event::Done { token } => {
|
||
|
|
for widget in state.widgets.iter_mut() {
|
||
|
|
widget.token_update(&token);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
_ => (),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|