moved code around, essentially refactored everything
This commit is contained in:
parent
90f5d9b9b8
commit
f0a32fbf3c
540
src/main.rs
540
src/main.rs
|
@ -1,534 +1,34 @@
|
||||||
use x11::xlib;
|
use std::io::Result;
|
||||||
|
|
||||||
use std::ffi::CString;
|
mod wm;
|
||||||
use std::io::{Error, ErrorKind, Result};
|
|
||||||
use std::ptr::{null, null_mut};
|
|
||||||
use std::sync::atomic::{AtomicPtr, Ordering};
|
|
||||||
use std::sync::Arc;
|
|
||||||
|
|
||||||
use x11::xlib::{
|
#[allow(dead_code)]
|
||||||
ButtonPressMask, ButtonReleaseMask, ControlMask, GrabModeAsync, LockMask, Mod1Mask, Mod2Mask,
|
unsafe extern "C" fn xlib_error_handler(
|
||||||
Mod3Mask, Mod4Mask, Mod5Mask, PointerMotionMask, ShiftMask, XEvent,
|
_dpy: *mut x11::xlib::Display,
|
||||||
};
|
ee: *mut x11::xlib::XErrorEvent,
|
||||||
|
) -> std::os::raw::c_int {
|
||||||
|
let err = ee.as_ref().unwrap();
|
||||||
|
|
||||||
use nix::unistd::{close, execvp, fork, setsid, ForkResult};
|
if err.error_code == x11::xlib::BadWindow
|
||||||
|
|| err.error_code == x11::xlib::BadDrawable
|
||||||
#[derive(Clone)]
|
|| err.error_code == x11::xlib::BadAccess
|
||||||
struct Display(Arc<AtomicPtr<xlib::Display>>);
|
|| err.error_code == x11::xlib::BadMatch
|
||||||
|
|
||||||
impl Display {
|
|
||||||
fn get(&self) -> *mut xlib::Display {
|
|
||||||
self.0.load(Ordering::SeqCst)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct WMAtoms {
|
|
||||||
protocols: Option<xlib::Atom>,
|
|
||||||
delete: Option<xlib::Atom>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for WMAtoms {
|
|
||||||
fn default() -> Self {
|
|
||||||
Self {
|
|
||||||
protocols: None,
|
|
||||||
delete: None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
struct XlibState {
|
|
||||||
display: Display,
|
|
||||||
keys: Vec<(i32, u32, Box<dyn Fn(&Self, &XEvent)>)>,
|
|
||||||
//move_window:
|
|
||||||
// u64 : window to move
|
|
||||||
// (i32, i32) : initial cursor position
|
|
||||||
// (i32, i32) : initial window position
|
|
||||||
move_window: Option<(u64, (i32, i32), (i32, i32))>,
|
|
||||||
//resize_window:
|
|
||||||
// u64 : window to move
|
|
||||||
// (i32, i32) : initial window position
|
|
||||||
resize_window: Option<(u64, (i32, i32))>,
|
|
||||||
atoms: WMAtoms,
|
|
||||||
event_handlers: Vec<Arc<dyn Fn(&mut Self, &XEvent)>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Default for XlibState {
|
|
||||||
fn default() -> Self {
|
|
||||||
let display = unsafe { xlib::XOpenDisplay(null()) };
|
|
||||||
assert_ne!(display, null_mut());
|
|
||||||
|
|
||||||
let display = Display {
|
|
||||||
0: Arc::new(AtomicPtr::new(display)),
|
|
||||||
};
|
|
||||||
|
|
||||||
Self {
|
|
||||||
display,
|
|
||||||
keys: vec![],
|
|
||||||
move_window: None,
|
|
||||||
resize_window: None,
|
|
||||||
atoms: WMAtoms::default(),
|
|
||||||
event_handlers: vec![],
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl XlibState {
|
|
||||||
fn new() -> Result<Self> {
|
|
||||||
let display = unsafe { xlib::XOpenDisplay(null()) };
|
|
||||||
assert_ne!(display, null_mut());
|
|
||||||
|
|
||||||
let display = Display {
|
|
||||||
0: Arc::new(AtomicPtr::new(display)),
|
|
||||||
};
|
|
||||||
|
|
||||||
Ok(Self {
|
|
||||||
display: display.clone(),
|
|
||||||
keys: vec![],
|
|
||||||
move_window: None,
|
|
||||||
resize_window: None,
|
|
||||||
atoms: WMAtoms {
|
|
||||||
protocols: {
|
|
||||||
Some(unsafe {
|
|
||||||
let wm_protocols_str = CString::new("WM_PROTOCOLS").unwrap();
|
|
||||||
xlib::XInternAtom(display.get(), wm_protocols_str.as_c_str().as_ptr(), 0)
|
|
||||||
})
|
|
||||||
.filter(|&atom| atom != 0)
|
|
||||||
},
|
|
||||||
delete: {
|
|
||||||
Some(unsafe {
|
|
||||||
let wm_delete_str = CString::new("WM_DELETE_WINDOW").unwrap();
|
|
||||||
xlib::XInternAtom(display.get(), wm_delete_str.as_c_str().as_ptr(), 0)
|
|
||||||
})
|
|
||||||
.filter(|&atom| atom != 0)
|
|
||||||
},
|
|
||||||
},
|
|
||||||
event_handlers: vec![],
|
|
||||||
}
|
|
||||||
// handle_move_window
|
|
||||||
.grab_button(
|
|
||||||
1,
|
|
||||||
Mod1Mask,
|
|
||||||
ButtonPressMask | ButtonReleaseMask | PointerMotionMask,
|
|
||||||
)
|
|
||||||
.add_event_handler(|state, event| {
|
|
||||||
let clean_mask = state.clean_mask();
|
|
||||||
|
|
||||||
if unsafe {
|
|
||||||
state.move_window.is_none()
|
|
||||||
&& event.get_type() == xlib::ButtonPress
|
|
||||||
&& event.button.button == 1
|
|
||||||
&& event.button.state & clean_mask == Mod1Mask & clean_mask
|
|
||||||
&& event.button.subwindow != 0
|
|
||||||
} {
|
|
||||||
let win_pos = unsafe {
|
|
||||||
let mut attr: xlib::XWindowAttributes =
|
|
||||||
std::mem::MaybeUninit::uninit().assume_init();
|
|
||||||
xlib::XGetWindowAttributes(state.dpy(), event.button.subwindow, &mut attr);
|
|
||||||
|
|
||||||
(attr.x, attr.y)
|
|
||||||
};
|
|
||||||
|
|
||||||
state.move_window = unsafe {
|
|
||||||
Some((
|
|
||||||
event.button.subwindow,
|
|
||||||
(event.button.x, event.button.y),
|
|
||||||
win_pos,
|
|
||||||
))
|
|
||||||
};
|
|
||||||
} else if unsafe {
|
|
||||||
state.move_window.is_some()
|
|
||||||
&& event.get_type() == xlib::ButtonRelease
|
|
||||||
&& event.button.button == 1
|
|
||||||
} {
|
|
||||||
state.move_window = None;
|
|
||||||
} else if state.move_window.is_some() && event.get_type() == xlib::MotionNotify {
|
|
||||||
let move_window = state.move_window.unwrap();
|
|
||||||
|
|
||||||
let attr = unsafe {
|
|
||||||
let mut attr: xlib::XWindowAttributes =
|
|
||||||
std::mem::MaybeUninit::uninit().assume_init();
|
|
||||||
xlib::XGetWindowAttributes(state.dpy(), move_window.0, &mut attr);
|
|
||||||
|
|
||||||
attr
|
|
||||||
};
|
|
||||||
|
|
||||||
let (x, y) = unsafe {
|
|
||||||
(
|
|
||||||
event.motion.x - move_window.1 .0 + move_window.2 .0,
|
|
||||||
event.motion.y - move_window.1 .1 + move_window.2 .1,
|
|
||||||
)
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut wc = xlib::XWindowChanges {
|
|
||||||
x,
|
|
||||||
y,
|
|
||||||
width: attr.width,
|
|
||||||
height: attr.height,
|
|
||||||
border_width: 0,
|
|
||||||
sibling: 0,
|
|
||||||
stack_mode: 0,
|
|
||||||
};
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
xlib::XConfigureWindow(
|
|
||||||
state.dpy(),
|
|
||||||
state.move_window.unwrap().0,
|
|
||||||
(xlib::CWX | xlib::CWY) as u32,
|
|
||||||
&mut wc,
|
|
||||||
);
|
|
||||||
|
|
||||||
xlib::XSync(state.dpy(), 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
// resize window handler
|
|
||||||
.grab_button(
|
|
||||||
3,
|
|
||||||
Mod1Mask,
|
|
||||||
ButtonPressMask | ButtonReleaseMask | PointerMotionMask,
|
|
||||||
)
|
|
||||||
.add_event_handler(|state, event| {
|
|
||||||
let clean_mask = state.clean_mask();
|
|
||||||
|
|
||||||
if unsafe {
|
|
||||||
state.resize_window.is_none()
|
|
||||||
&& event.get_type() == xlib::ButtonPress
|
|
||||||
&& event.button.button == 3
|
|
||||||
&& event.button.state & clean_mask == Mod1Mask & clean_mask
|
|
||||||
&& event.button.subwindow != 0
|
|
||||||
} {
|
|
||||||
unsafe {
|
|
||||||
let mut attr: xlib::XWindowAttributes =
|
|
||||||
std::mem::MaybeUninit::uninit().assume_init();
|
|
||||||
|
|
||||||
xlib::XGetWindowAttributes(state.dpy(), event.button.subwindow, &mut attr);
|
|
||||||
|
|
||||||
state.resize_window = Some((event.button.subwindow, (attr.x, attr.y)));
|
|
||||||
|
|
||||||
xlib::XWarpPointer(
|
|
||||||
state.dpy(),
|
|
||||||
0,
|
|
||||||
event.button.subwindow,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
attr.width + attr.border_width - 1,
|
|
||||||
attr.height + attr.border_width - 1,
|
|
||||||
);
|
|
||||||
};
|
|
||||||
} else if unsafe {
|
|
||||||
state.resize_window.is_some()
|
|
||||||
&& event.get_type() == xlib::ButtonRelease
|
|
||||||
&& event.button.button == 3
|
|
||||||
} {
|
|
||||||
state.resize_window = None;
|
|
||||||
} else if state.resize_window.is_some() && event.get_type() == xlib::MotionNotify {
|
|
||||||
let resize_window = state.resize_window.unwrap();
|
|
||||||
|
|
||||||
let attr = unsafe {
|
|
||||||
let mut attr: xlib::XWindowAttributes =
|
|
||||||
std::mem::MaybeUninit::uninit().assume_init();
|
|
||||||
xlib::XGetWindowAttributes(state.dpy(), resize_window.0, &mut attr);
|
|
||||||
|
|
||||||
attr
|
|
||||||
};
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
let (width, height) = {
|
|
||||||
(
|
|
||||||
std::cmp::max(
|
|
||||||
1,
|
|
||||||
event.motion.x - resize_window.1 .0 + 2 * attr.border_width + 1,
|
|
||||||
),
|
|
||||||
std::cmp::max(
|
|
||||||
1,
|
|
||||||
event.motion.y - resize_window.1 .1 + 2 * attr.border_width + 1,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
};
|
|
||||||
|
|
||||||
let mut wc = xlib::XWindowChanges {
|
|
||||||
x: attr.x,
|
|
||||||
y: attr.y,
|
|
||||||
width,
|
|
||||||
height,
|
|
||||||
border_width: attr.border_width,
|
|
||||||
sibling: 0,
|
|
||||||
stack_mode: 0,
|
|
||||||
};
|
|
||||||
|
|
||||||
xlib::XConfigureWindow(
|
|
||||||
state.dpy(),
|
|
||||||
resize_window.0,
|
|
||||||
(xlib::CWWidth | xlib::CWHeight) as u32,
|
|
||||||
&mut wc,
|
|
||||||
);
|
|
||||||
|
|
||||||
xlib::XSync(state.dpy(), 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
})
|
|
||||||
.add_key_with_handler(
|
|
||||||
"T",
|
|
||||||
Mod1Mask,
|
|
||||||
|state, _| {
|
|
||||||
let _ = state.spawn("xterm", &[]);
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.add_key_with_handler(
|
|
||||||
"Q",
|
|
||||||
Mod1Mask,
|
|
||||||
|state, event| unsafe {
|
|
||||||
if event.key.subwindow != 0 {
|
|
||||||
if state.atoms.delete.is_none()
|
|
||||||
|| !state.send_event(event.key.subwindow, state.atoms.delete.unwrap())
|
|
||||||
{
|
|
||||||
println!("delete atmom: {:?}", state.atoms.delete);
|
|
||||||
xlib::XKillClient(state.dpy(), event.key.subwindow);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
)
|
|
||||||
.add_key_with_handler(
|
|
||||||
"Q",
|
|
||||||
Mod1Mask | ShiftMask,
|
|
||||||
|state, _event| {
|
|
||||||
unsafe {
|
|
||||||
xlib::XCloseDisplay(state.dpy());
|
|
||||||
}
|
|
||||||
|
|
||||||
std::process::exit(0);
|
|
||||||
},
|
|
||||||
))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn dpy(&self) -> *mut xlib::Display {
|
|
||||||
self.display.get()
|
|
||||||
}
|
|
||||||
|
|
||||||
fn root(&self) -> u64 {
|
|
||||||
unsafe { xlib::XDefaultRootWindow(self.dpy()) }
|
|
||||||
}
|
|
||||||
|
|
||||||
fn main_loop(mut self) -> Result<Self> {
|
|
||||||
loop {
|
|
||||||
let event = unsafe {
|
|
||||||
let mut event: xlib::XEvent = std::mem::MaybeUninit::uninit().assume_init();
|
|
||||||
xlib::XNextEvent(self.dpy(), &mut event);
|
|
||||||
|
|
||||||
event
|
|
||||||
};
|
|
||||||
|
|
||||||
// run keypress handlers
|
|
||||||
if event.get_type() == xlib::KeyPress {
|
|
||||||
// cache clean mask, that way numlock_mask doesnt get called for every cmp
|
|
||||||
let clean_mask = self.clean_mask();
|
|
||||||
|
|
||||||
for (key, mask, handler) in self.keys.iter().clone() {
|
|
||||||
// check if key and mask with any numlock state fit
|
|
||||||
if unsafe {
|
|
||||||
event.key.keycode == *key as u32
|
|
||||||
&& event.key.state & clean_mask == *mask & clean_mask
|
|
||||||
} {
|
|
||||||
handler(&self, &event);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
self.event_handlers
|
|
||||||
.clone()
|
|
||||||
.iter()
|
|
||||||
.for_each(|handler| handler(&mut self, &event));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn add_event_handler<F>(mut self, handler: F) -> Self
|
|
||||||
where
|
|
||||||
F: Fn(&mut Self, &XEvent) + 'static,
|
|
||||||
{
|
{
|
||||||
self.event_handlers.push(Arc::new(handler));
|
|
||||||
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
fn add_key_with_handler<S, F>(
|
|
||||||
mut self,
|
|
||||||
key: S,
|
|
||||||
mask: u32,
|
|
||||||
handler: F,
|
|
||||||
) -> Self where S: Into<String>, F: Fn(&Self, &XEvent) + 'static {
|
|
||||||
let keycode = self.keycode(key);
|
|
||||||
|
|
||||||
self.keys.push((keycode, mask, Box::new(handler)));
|
|
||||||
self.grab_key(keycode, mask)
|
|
||||||
}
|
|
||||||
|
|
||||||
fn grab_key(self, keycode: i32, mask: u32) -> Self {
|
|
||||||
let numlock_mask = self.numlock_mask();
|
|
||||||
let modifiers = vec![0, LockMask, numlock_mask, LockMask | numlock_mask];
|
|
||||||
for &modifier in modifiers.iter() {
|
|
||||||
unsafe {
|
|
||||||
xlib::XGrabKey(
|
|
||||||
self.dpy(),
|
|
||||||
keycode,
|
|
||||||
mask | modifier,
|
|
||||||
self.root(),
|
|
||||||
1, /* true */
|
|
||||||
GrabModeAsync,
|
|
||||||
GrabModeAsync,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
fn grab_button(self, button: u32, mod_mask: u32, button_mask: i64) -> Self {
|
|
||||||
let numlock_mask = self.numlock_mask();
|
|
||||||
let modifiers = vec![0, LockMask, numlock_mask, LockMask | numlock_mask];
|
|
||||||
|
|
||||||
modifiers.iter().for_each(|&modifier| {
|
|
||||||
unsafe {
|
|
||||||
xlib::XGrabButton(
|
|
||||||
self.dpy(),
|
|
||||||
button,
|
|
||||||
mod_mask | modifier,
|
|
||||||
self.root(),
|
|
||||||
1, /*true */
|
|
||||||
button_mask as u32,
|
|
||||||
GrabModeAsync,
|
|
||||||
GrabModeAsync,
|
|
||||||
0,
|
|
||||||
0,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
self
|
|
||||||
}
|
|
||||||
|
|
||||||
// spawn a new process / calls execvp
|
|
||||||
pub fn spawn<T: ToString>(&self, command: T, args: &[T]) -> Result<()> {
|
|
||||||
let fd = unsafe { xlib::XConnectionNumber(self.dpy()) };
|
|
||||||
|
|
||||||
match unsafe { fork() } {
|
|
||||||
Ok(ForkResult::Parent { .. }) => Ok(()),
|
|
||||||
Ok(ForkResult::Child) => {
|
|
||||||
// i dont think i want to exit this block without closing the program,
|
|
||||||
// so unwrap everything
|
|
||||||
|
|
||||||
close(fd)
|
|
||||||
.or_else(|_| Err("failed to close x connection"))
|
|
||||||
.unwrap();
|
|
||||||
setsid().ok().ok_or("failed to setsid").unwrap();
|
|
||||||
|
|
||||||
let c_cmd = CString::new(command.to_string()).unwrap();
|
|
||||||
|
|
||||||
let c_args: Vec<_> = args
|
|
||||||
.iter()
|
|
||||||
.map(|s| CString::new(s.to_string()).unwrap())
|
|
||||||
.collect();
|
|
||||||
|
|
||||||
execvp(
|
|
||||||
&c_cmd,
|
|
||||||
&c_args.iter().map(|s| s.as_c_str()).collect::<Vec<_>>(),
|
|
||||||
)
|
|
||||||
.or(Err("failed to execvp()"))
|
|
||||||
.unwrap();
|
|
||||||
|
|
||||||
eprintln!("execvp({}) failed.", c_cmd.to_str().unwrap());
|
|
||||||
std::process::exit(0);
|
|
||||||
}
|
|
||||||
Err(_) => Err(Error::new(ErrorKind::Other, "failed to fork.")),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn keycode<S>(&self, string: S) -> i32 where S: Into<String> {
|
|
||||||
let c_string = CString::new(string.into()).unwrap();
|
|
||||||
unsafe {
|
|
||||||
let keysym = xlib::XStringToKeysym(c_string.as_ptr());
|
|
||||||
xlib::XKeysymToKeycode(self.dpy(), keysym) as i32
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn check_for_protocol(&self, window: xlib::Window, proto: xlib::Atom) -> bool {
|
|
||||||
let mut protos: *mut xlib::Atom = null_mut();
|
|
||||||
let mut num_protos: i32 = 0;
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
if xlib::XGetWMProtocols(self.dpy(), window, &mut protos, &mut num_protos) != 0 {
|
|
||||||
for i in 0..num_protos {
|
|
||||||
if *protos.offset(i as isize) == proto {
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn send_event(&self, window: xlib::Window, proto: xlib::Atom) -> bool {
|
|
||||||
if self.check_for_protocol(window, proto) && self.atoms.protocols.is_some() {
|
|
||||||
let mut data = xlib::ClientMessageData::default();
|
|
||||||
data.set_long(0, proto as i64);
|
|
||||||
let mut event = XEvent {
|
|
||||||
client_message: xlib::XClientMessageEvent {
|
|
||||||
type_: xlib::ClientMessage,
|
|
||||||
serial: 0,
|
|
||||||
display: self.dpy(),
|
|
||||||
send_event: 0,
|
|
||||||
window,
|
|
||||||
format: 32,
|
|
||||||
message_type: self.atoms.protocols.unwrap(),
|
|
||||||
data,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
|
|
||||||
unsafe {
|
|
||||||
xlib::XSendEvent(self.dpy(), window, 0, xlib::NoEventMask, &mut event);
|
|
||||||
}
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
fn numlock_mask(&self) -> u32 {
|
|
||||||
unsafe {
|
|
||||||
let modmap = xlib::XGetModifierMapping(self.dpy());
|
|
||||||
let max_keypermod = (*modmap).max_keypermod;
|
|
||||||
|
|
||||||
for i in 0..8 {
|
|
||||||
for j in 0..max_keypermod {
|
|
||||||
if *(*modmap)
|
|
||||||
.modifiermap
|
|
||||||
.offset((i * max_keypermod + j) as isize)
|
|
||||||
== xlib::XKeysymToKeycode(self.dpy(), x11::keysym::XK_Num_Lock as u64)
|
|
||||||
{
|
|
||||||
return 1 << i;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
0
|
0
|
||||||
}
|
} else {
|
||||||
|
eprintln!(
|
||||||
fn clean_mask(&self) -> u32 {
|
"wm: fatal error:\nrequest_code: {}\nerror_code: {}",
|
||||||
!(self.numlock_mask() | LockMask)
|
err.request_code, err.error_code
|
||||||
& (ShiftMask | ControlMask | Mod1Mask | Mod2Mask | Mod3Mask | Mod4Mask | Mod5Mask)
|
);
|
||||||
|
std::process::exit(1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
fn main() -> Result<()> {
|
fn main() -> Result<()> {
|
||||||
println!("Hello, world!");
|
println!("Hello, world!");
|
||||||
|
|
||||||
let state = XlibState::new()?;
|
wm::WMState::init().run();
|
||||||
|
|
||||||
state.main_loop()?;
|
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue