From ea2245d7b1dd6116bf85b4b903626b4846ad984c Mon Sep 17 00:00:00 2001
From: Gabriel <68819302+obsidianical@users.noreply.github.com>
Date: Thu, 21 Jul 2022 10:34:48 +0200
Subject: [PATCH] did stuff

---
 src/bin/easymacrorec.rs | 267 ++++++++++++++++++++++++----------------
 src/lib.rs              |  85 ++++++++++---
 src/x11_safe_wrapper.rs |   5 +-
 3 files changed, 229 insertions(+), 128 deletions(-)

diff --git a/src/bin/easymacrorec.rs b/src/bin/easymacrorec.rs
index f7e993f..1b74121 100644
--- a/src/bin/easymacrorec.rs
+++ b/src/bin/easymacrorec.rs
@@ -4,17 +4,25 @@ use std::os::raw::{c_char, c_uchar, c_uint};
 use std::process::{exit, ExitCode};
 use std::ptr::{addr_of, slice_from_raw_parts};
 use std::{io, slice, thread};
+use std::borrow::BorrowMut;
+use std::cmp::max;
+use std::env::args;
 use std::io::Write;
 use std::ffi::c_void;
-use std::fmt::{format};
+use std::fmt::format;
 use std::fs::{File, OpenOptions};
+use std::mem::size_of;
+use std::time::{Duration, SystemTime, UNIX_EPOCH};
 use clap::Parser;
 use x11::keysym::XK_Escape;
 use x11::xinput2::XIGrabModeSync;
-use x11::xlib::{ButtonPress, ButtonRelease, CurrentTime, GrabModeAsync, GrabModeSync, GrabSuccess, KeyCode, KeyPress, KeyPressMask, KeyRelease, MotionNotify, SyncPointer, XEvent, XFree, XKeyEvent, XKeyPressedEvent, XPointer};
+use x11::xlib::{ButtonPress, ButtonRelease, CurrentTime, GrabModeAsync, GrabModeSync, GrabSuccess, KeyCode, KeyPress, KeyPressMask, KeyRelease, MotionNotify, SyncPointer, Time, XEvent, XFree, XKeyEvent, XKeyPressedEvent, XPointer};
 use x11::xrecord::{XRecordAllocRange, XRecordContext, XRecordCreateContext, XRecordDisableContext, XRecordEndOfData, XRecordFreeData, XRecordInterceptData, XRecordStartOfData};
+use easymacros::{Instructions, Position};
 use easymacros::x11_safe_wrapper::{Keycode, XDisplay};
 
+const DELAY_DEFAULT_MS: u16 = 10;
+
 /// Macro recording module for easymacros. Outputs are partially compatible with xmacro.
 #[derive(Parser, Debug)]
 #[clap(author, version, about, long_about = None)]
@@ -23,7 +31,7 @@ struct Args {
 	#[clap(value_parser, value_name = "output_file", value_hint = clap::ValueHint::FilePath)]
 	output_file: Option<std::path::PathBuf>,
 	/// Display to run the macro on. This uses the $DISPLAY environment variable by default.
-	#[clap(short='D', long)]
+	#[clap(short = 'D', long)]
 	display: Option<String>,
 	/// Max Delay in milliseconds for macro delays
 	#[clap(short, long)]
@@ -32,25 +40,22 @@ struct Args {
 	#[clap(short, long)]
 	ignore_delay_capturing: bool,
 }
+
 fn main() {
 	let args = Args::parse();
 
 	let display = XDisplay::open(args.display.clone());
 	let recorded_display = XDisplay::open(args.display.clone());
-
 	let stop_key = get_stop_key(display);
+	let writer = MacroWriter::new(args.output_file, args.ignore_delay_capturing);
 
-	let screen = display.get_default_screen();
-	dbg!(stop_key);
-
-	// check if the output file argument is specified, if not use stdout
-	let mut outfile: Box<dyn Write> = if let Some(outfile) = args.output_file {
-		Box::new(File::create(outfile).expect("Failed to create output file"))
-	} else {
-		Box::new(io::stdout())
-	};
-
-	event_loop(display, recorded_display, screen, stop_key, outfile, args.ignore_delay_capturing, args.max_delay);
+	event_loop(
+		display,
+		recorded_display,
+		stop_key,
+		writer,
+		args.max_delay
+	);
 
 	display.close();
 }
@@ -69,18 +74,11 @@ fn get_stop_key(display: XDisplay) -> Keycode {
 	println!("Press the key you want to use to stop recording the macro.");
 
 	let mut stop_key: Keycode = XK_Escape;
-
 	loop {
 		display.allow_events(SyncPointer, CurrentTime);
-		let ev = display.window_event(root, KeyPressMask);
-
-		unsafe {
-			//if it's a key event, save the keycode and break out of the loop
-			if let XEvent { key } = ev {
-				stop_key = key.keycode;
-				break;
-			}
-		}
+		let ev = XKeyEvent::from(display.window_event(root, KeyPressMask));
+		stop_key = ev.keycode;
+		break;
 	}
 
 	display.ungrab_keyboard(CurrentTime);
@@ -89,65 +87,99 @@ fn get_stop_key(display: XDisplay) -> Keycode {
 	stop_key
 }
 
-fn event_loop(display: XDisplay, recordeddpy: XDisplay, screen: i32, stop_key: Keycode, mut outfile: Box<dyn Write>, ignore_delay_capturing: bool, max_delay: Option<u64>) {
+fn event_loop(
+	xdpy: XDisplay,
+	recdpy: XDisplay,
+	stop_key: Keycode,
+	mut writer: MacroWriter,
+	max_delay: Option<Time>
+) {
 	// let root = display.get_root_window(screen);
 	let protocol_ranges = unsafe { XRecordAllocRange() };
 
-	let pointer_pos = display.query_pointer_pos();
+	let pointer_pos: Position<i16> = Position::from(xdpy.query_pointer_pos());
 
-	if pointer_pos != (-1, -1) {
-		writeln!(&mut outfile, "MotionNotify {} {}", pointer_pos.0, pointer_pos.1)
-			.expect("Failed to write to outfile");
+	if pointer_pos != Position (-1, -1) {
+		writer.write(Instructions::MotionNotify(pointer_pos))
 	}
 
-	let ctx = recordeddpy.create_record_context(protocol_ranges);
-	let ev_cb_data = EvCallbackData {
-		outfile,
-		xdpy: display,
-		recdpy: recordeddpy,
-		ctx,
-		stop_key,
-		ev_nr: 0,
-		working: true,
-		x: pointer_pos.0 as i16,
-		y: pointer_pos.1 as i16,
-		ignore_delay_capturing,
-		max_delay,
-		no_keypress_yet: true,
-		last_event: 0,
-		moving: false,
-	};
+	let ctx = recdpy.create_record_context(protocol_ranges);
+	let ev_cb_data = EvCallbackData::new(writer, xdpy, recdpy, ctx, stop_key, pointer_pos, max_delay);
 
-	if !recordeddpy.enable_context_async(ctx, Some(ev_callback), addr_of!(ev_cb_data) as *mut c_char) {
+	if !recdpy.enable_context_async(ctx, Some(ev_callback), addr_of!(ev_cb_data) as *mut c_char) {
 		panic!("Failed to enable record context")
 	}
 	while ev_cb_data.working {
-		recordeddpy.process_replies();
+		recdpy.process_replies();
 	}
 
-	display.disable_context(ctx);
-	display.free_context(ctx);
+	xdpy.disable_context(ctx);
+	xdpy.free_context(ctx);
 	unsafe { XFree(protocol_ranges as *mut c_void) };
 }
 
 #[repr(C)]
 pub struct EvCallbackData {
-	outfile: Box<dyn Write>,
-	xdpy: XDisplay,
-	recdpy: XDisplay,
+	pub writer: MacroWriter,
+	pub xdpy: XDisplay,
+	pub recdpy: XDisplay,
+	pub ctx: XRecordContext,
 	stop_key: Keycode,
 	ev_nr: u32,
 	working: bool,
-	ctx: XRecordContext,
-	x: i16,
-	y: i16,
-	ignore_delay_capturing: bool,
-	max_delay: Option<u64>,
+	pos: Position<i16>,
+	max_delay: Option<Time>,
 	no_keypress_yet: bool,
-	last_event: u64,
+	last_event: Time,
 	moving: bool,
 }
 
+impl EvCallbackData {
+	pub fn new(
+		writer: MacroWriter,
+		xdpy: XDisplay,
+		recdpy: XDisplay,
+		ctx: XRecordContext,
+		stop_key: Keycode,
+		pos: Position<i16>,
+		max_delay: Option<Time>,
+	) -> Self {
+		EvCallbackData {
+			writer,
+			xdpy,
+			recdpy,
+			ctx,
+			stop_key,
+			ev_nr: 0,
+			working: true,
+			pos,
+			max_delay,
+			no_keypress_yet: true,
+			last_event: SystemTime::now().duration_since(UNIX_EPOCH).unwrap().as_millis() as Time,
+			moving: false,
+		}
+	}
+
+	pub fn ptr_is_moving(&self) -> bool { self.moving }
+
+	pub unsafe fn update_pos(&mut self, intercept_data: &mut XRecordInterceptData) -> Position<i16> {
+		self.pos.0 = *((intercept_data.data as usize + size_of::<i16>() * 10) as *const i16);
+		self.pos.1 = *((intercept_data.data as usize + size_of::<i16>() * 11) as *const i16);
+		self.pos
+	}
+
+	pub fn write_pos(&mut self) {
+		self.writer.write(Instructions::MotionNotify(self.pos));
+		self.moving = false;
+	}
+
+	pub fn maybe_write_delay(&mut self, server_time: Time) {
+		if server_time - self.last_event > 1 {
+			self.writer.write(Instructions::Delay(calculate_delay(server_time, self.last_event, self.max_delay)));
+			self.last_event = server_time;
+		}
+	}
+}
 
 unsafe extern "C" fn ev_callback(closure: *mut c_char, intercept_data: *mut XRecordInterceptData) {
 	let data = &mut *(closure as *mut EvCallbackData);
@@ -155,6 +187,7 @@ unsafe extern "C" fn ev_callback(closure: *mut c_char, intercept_data: *mut XRec
 
 	if intercept_data.category == XRecordStartOfData {
 		println!("Got start of data!");
+		data.last_event = intercept_data.server_time;
 		XRecordFreeData(intercept_data);
 		return;
 	} else if intercept_data.category == XRecordEndOfData {
@@ -164,14 +197,14 @@ unsafe extern "C" fn ev_callback(closure: *mut c_char, intercept_data: *mut XRec
 	}
 
 	let ev_type = *(intercept_data.data as *const u8);
-	if data.x == -1 || data.y == -1 {
+
+	if data.pos.0 == 0 || data.pos.1 == -1 {
 		if ev_type == MOTIONNOTIFY_U8 {
-			data.x = *((intercept_data.data as usize + std::mem::size_of::<i16>() * 10) as *const i16);
-			data.y = *((intercept_data.data as usize + std::mem::size_of::<i16>() * 11) as *const i16);
-			println!("MotionNotify {} {}", data.x, data.y);
+			data.update_pos(intercept_data);
+			data.write_pos();
 		} else {
-			println!("Move your cursor so the macro can start with a fixed cursor position!");
-			println!("Skipping event...");
+			println!("Move your cursor so the macro can start with a fixed cursor position!
+			Skipping event...");
 		}
 	} else if data.no_keypress_yet && ev_type == KEYRELEASE_U8 {
 		println!("Skipping KeyRelease without recorded KeyPress...");
@@ -179,10 +212,7 @@ unsafe extern "C" fn ev_callback(closure: *mut c_char, intercept_data: *mut XRec
 		// if
 		match ev_type {
 			MOTIONNOTIFY_U8 => {
-				let root_x = *((intercept_data.data as usize + std::mem::size_of::<i16>() * 10) as *const i16);
-				let root_y = *((intercept_data.data as usize + std::mem::size_of::<i16>() * 11) as *const i16);
-				data.x = root_x;
-				data.y = root_y;
+				data.update_pos(intercept_data);
 
 				if !data.moving {
 					data.moving = true;
@@ -199,62 +229,81 @@ unsafe extern "C" fn ev_callback(closure: *mut c_char, intercept_data: *mut XRec
 						data.no_keypress_yet = false;
 					}
 
-					if !data.ignore_delay_capturing && (intercept_data.server_time - data.last_event) != 0 {
-						writeln!(&mut data.outfile,"Delay {}", if let Some(max_delay) = data.max_delay {
-							let delay_res = intercept_data.server_time - data.last_event;
-							if delay_res > max_delay { max_delay } else { delay_res }
-						} else { intercept_data.server_time - data.last_event })
-							.expect("Failed to write to outfile.");
-						data.last_event = intercept_data.server_time;
-					}
+					data.maybe_write_delay(intercept_data.server_time);
 
-					if data.moving {
-						writeln!(&mut data.outfile, "MotionNotify {} {}", data.x, data.y)
-							.expect("Failed to write to outfile.");
-						data.moving = false;
-					}
+					if data.ptr_is_moving() { data.write_pos(); }
 
-					writeln!(
-						&mut data.outfile,
-						"{} {}",
-						if ev_type == KEYPRESS_U8 { "KeyStrPress" } else { "KeyRelease" },
-						keyname.to_str().unwrap()
-					).expect("Failed to write to outfile.");
+					data.writer.write(if ev_type == KEYPRESS_U8 {
+							Instructions::KeyStrPress(keyname)
+						} else {
+							Instructions::KeyStrRelease(keyname)
+						});
 				}
 			}
 			BUTTONPRESS_U8 | BUTTONRELEASE_U8 => {
 				let bc: u8 = *((intercept_data.data as usize + 1) as *const u8);
 
-				if !data.ignore_delay_capturing && (intercept_data.server_time - data.last_event) != 0 {
-					writeln!(&mut data.outfile,"Delay {}", if let Some(max_delay) = data.max_delay {
-						let delay_res = intercept_data.server_time - data.last_event;
-						if delay_res > max_delay { max_delay } else { delay_res }
-					} else { intercept_data.server_time - data.last_event })
-						.expect("Failed to write to outfile.");
-					data.last_event = intercept_data.server_time;
-				}
+				data.maybe_write_delay(intercept_data.server_time);
 
-				if data.moving {
-					writeln!(&mut data.outfile, "MotionNotify {} {}", data.x, data.y)
-						.expect("Failed to write to outfile.");
-					data.moving = false;
-				}
+				if data.ptr_is_moving() { data.write_pos(); }
 
-				writeln!(
-					&mut data.outfile,
-					"{} {}",
-					if ev_type == BUTTONPRESS_U8 { "ButtonPress" } else { "ButtonRelease" },
-					bc
-				).expect("Failed to write to outfile.");
+				if ev_type == BUTTONPRESS_U8 {
+					data.writer.write(Instructions::ButtonPress(bc));
+				} else {
+					data.writer.write(Instructions::ButtonRelease(bc));
+				}
 			}
 			_ => eprintln!("Unknown event type: {:?}", ev_type)
 		}
 	}
 
-	data.ev_nr += 1;
+	data.ev_nr += 2;
 	XRecordFreeData(intercept_data)
 }
 
+fn calculate_delay(server_time: Time, last_event: Time, max_delay: Option<Time>) -> Time {
+	if let Some(max) = max_delay {
+		let max = max as u64;
+		let delay = server_time - last_event;
+
+		if delay > max {
+			max
+		} else {
+			delay
+		}
+	} else {
+		server_time - last_event
+	}
+}
+
+pub struct MacroWriter {
+	outfile: Box<dyn Write>,
+	ignore_delay_capturing: bool,
+}
+
+impl MacroWriter {
+	pub fn new(outfile: Option<std::path::PathBuf>, ignore_delay_capturing: bool) -> Self {
+		Self {
+			outfile: if let Some(outfile) = outfile {
+				Box::new(File::create(outfile).expect("Failed to create output file"))
+			} else {
+				Box::new(io::stdout())
+			},
+			ignore_delay_capturing,
+		}
+	}
+
+	pub fn write(&mut self, instruction: Instructions) {
+		if self.ignore_delay_capturing {
+			if let Instructions::Delay(_) = instruction {
+				return;
+			}
+		}
+
+		writeln!(&mut self.outfile, "{}", instruction).expect("Failed to write instruction to outfile");
+	}
+}
+
 const KEYPRESS_U8: u8 = KeyPress as u8;
 const KEYRELEASE_U8: u8 = KeyRelease as u8;
 const BUTTONPRESS_U8: u8 = ButtonPress as u8;
diff --git a/src/lib.rs b/src/lib.rs
index dc7937a..be16e94 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,23 +1,74 @@
 extern crate core;
 
+use std::ffi::CString;
+use std::fmt::{Display, Formatter};
+use std::time::Duration;
+use x11::xlib::Time;
+use crate::x11_safe_wrapper::{Keycode, Keysym};
+
 pub mod x11_safe_wrapper;
 pub mod chartbl;
 
-pub enum XMacroInstructions {
-    Delay,
-    ButtonPress,
-    ButtonRelease,
-    MotionNotify,
-    KeyCodePress,
-    KeyCodeRelease,
-    // Screenshot,
-    KeySymPress,
-    KeySymRelease,
-    KeySym,
-    KeyStrPress,
-    KeyStrRelease,
-    KeyStr,
-    String,
-    // ExecBlock,
-    // ExecNoBlock,
+#[derive(Copy, Clone, PartialEq, Eq)]
+pub struct Position<T> (pub T, pub T);
+
+impl From<Position<i32>> for Position<i16> {
+    fn from(pos: Position<i32>) -> Self {
+        Self (pos.0 as i16, pos.1 as i16)
+    }
+}
+
+impl From<(i32, i32)> for Position<i32> {
+    fn from(v: (i32, i32)) -> Self {
+        Position(v.0, v.1)
+    }
+}
+
+impl From<(i32, i32)> for Position<i16> {
+    fn from(v: (i32, i32)) -> Position<i16> {
+        let p1: Position<i32> = Position::from(v);
+        Position::from(p1)
+    }
+}
+
+pub enum Instructions<'a> {
+    Delay(Time),
+    ButtonPress(u8),
+    ButtonRelease(u8),
+    MotionNotify(Position<i16>),
+    KeyCodePress(Keycode),
+    KeyCodeRelease(Keycode),
+    KeySymPress(Keysym),
+    KeySymRelease(Keysym),
+    KeySym(Keysym),
+    KeyStrPress(CString),
+    KeyStrRelease(CString),
+    KeyStr(CString),
+    String(&'a str),
+    ExecBlock(&'a str),
+    ExecNoBlock(&'a str),
+}
+
+impl Display for Instructions<'_> {
+    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{}",
+            match self {
+                Instructions::Delay(d) => format!("Delay {}", d),
+                Instructions::ButtonPress(b) => format!("ButtonPress {}", b),
+                Instructions::ButtonRelease(b) => format!("ButtonRelease {}", b),
+                Instructions::MotionNotify(pos) => format!("MotionNotify {} {}", pos.0, pos.1),
+                Instructions::KeyCodePress(kc) => format!("KeyCodePress {}", kc),
+                Instructions::KeyCodeRelease(kc) => format!("KeyCodeRelease {}", kc),
+                Instructions::KeySymPress(ks) => format!("KeySymPress {}", ks),
+                Instructions::KeySymRelease(ks) => format!("KeySymRelease {}", ks),
+                Instructions::KeySym(ks) => format!("KeySym {}", ks),
+                Instructions::KeyStrPress(kstr) => format!("KeyStrPress {}", kstr.to_str().unwrap()),
+                Instructions::KeyStrRelease(kstr) => format!("KeyStrRelease {}", kstr.to_str().unwrap()),
+                Instructions::KeyStr(kstr) => format!("KeyStr {}", kstr.to_str().unwrap()),
+                Instructions::String(str) => format!("String {}", str),
+                Instructions::ExecBlock(cmd) => format!("ExecBlock {}", cmd),
+                Instructions::ExecNoBlock(cmd) => format!("ExecNoBlock {}", cmd),
+            },
+        )
+    }
 }
diff --git a/src/x11_safe_wrapper.rs b/src/x11_safe_wrapper.rs
index fcd89e5..ed13b16 100644
--- a/src/x11_safe_wrapper.rs
+++ b/src/x11_safe_wrapper.rs
@@ -7,7 +7,7 @@ use x11::xtest::{
 	XTestFakeButtonEvent, XTestFakeKeyEvent, XTestFakeMotionEvent, XTestGrabControl,
 	XTestQueryExtension,
 };
-use crate::XMacroInstructions::KeySym;
+use crate::Instructions::KeySym;
 
 #[derive(Debug, Clone, Copy)]
 pub struct XDisplay {
@@ -277,9 +277,10 @@ impl XDisplay {
 pub fn string_to_keysym(string: &[u8]) -> Keysym {
 	unsafe { XStringToKeysym(string.as_ptr() as *const c_char) }
 }
+
 pub fn keysym_to_string(keysym: Keysym) -> CString {
 	unsafe {
 		let cstr = CStr::from_ptr(XKeysymToString(keysym));
 		CString::from(cstr)
 	}
-}
+}
\ No newline at end of file