Initial commit
This commit is contained in:
cyber-sushi
GitHub
parent
edf987c204
commit
be6338e87a
1 changed files with 302 additions and 0 deletions
302
src/main.rs
Normal file
302
src/main.rs
Normal file
|
|
@ -0,0 +1,302 @@
|
|||
use std::{collections::HashMap, fmt::Debug, ffi::OsStr, path::Path, sync::Arc};
|
||||
use evdev::{Device, EventStream, Key, AbsoluteAxisType, EventType, InputEvent};
|
||||
use evdev::uinput::{VirtualDevice, VirtualDeviceBuilder};
|
||||
use tokio::{sync::Mutex, task::JoinHandle};
|
||||
use tokio_stream::StreamExt;
|
||||
use tokio_udev;
|
||||
use tokio;
|
||||
use serde;
|
||||
use home;
|
||||
|
||||
struct EventReader {
|
||||
config: Config,
|
||||
stream: Arc<Mutex<EventStream>>,
|
||||
virt_dev: Arc<Mutex<VirtualDevices>>,
|
||||
analog_position: Arc<Mutex<Vec<i32>>>,
|
||||
device_is_connected: Arc<Mutex<bool>>,
|
||||
}
|
||||
|
||||
impl EventReader {
|
||||
fn new(config: Config, stream: Arc<Mutex<EventStream>>, virt_dev: Arc<Mutex<VirtualDevices>>) -> Self {
|
||||
let mut position_vector: Vec<i32> = Vec::new();
|
||||
for i in [0, 0] {position_vector.push(i)};
|
||||
let position_vector_mutex = Arc::new(Mutex::new(position_vector));
|
||||
let device_is_connected: Arc<Mutex<bool>> = Arc::new(Mutex::new(true));
|
||||
Self {
|
||||
config: config,
|
||||
stream: stream,
|
||||
virt_dev: virt_dev,
|
||||
analog_position: position_vector_mutex,
|
||||
device_is_connected: device_is_connected,
|
||||
}
|
||||
}
|
||||
|
||||
async fn start(&self) {
|
||||
let mut stream = self.stream.lock().await;
|
||||
let mut analog_mode: &str = "left";
|
||||
if let Some(stick) = self.config.settings.get("POINTER_STICK") {
|
||||
analog_mode = stick.as_str();
|
||||
};
|
||||
let mut is_xbox: &str = "false";
|
||||
if let Some(xbox) = self.config.settings.get("XBOX_CONTROLLER") {
|
||||
is_xbox = xbox.as_str();
|
||||
};
|
||||
while let Some(Ok(event)) = stream.next().await {
|
||||
match (event.event_type(), AbsoluteAxisType(event.code()), analog_mode) {
|
||||
(EventType::KEY, _, _) => {
|
||||
if let Some(event_list) = self.config.keys.get(&Key(event.code())) {
|
||||
self.emit_event(event_list, event.value()).await
|
||||
} else {
|
||||
println!("Button has no binding in the config file.");
|
||||
}
|
||||
},
|
||||
(_, AbsoluteAxisType::ABS_HAT0X, _) => {
|
||||
let event_list_option: Option<&Vec<Key>> = match event.value() {
|
||||
-1 => self.config.keys.get(&Key::BTN_DPAD_LEFT),
|
||||
0 => self.config.abs.get(&"NONE_X".to_string()),
|
||||
1 => self.config.keys.get(&Key::BTN_DPAD_RIGHT),
|
||||
_ => self.config.abs.get(&"NONE_X".to_string()),
|
||||
};
|
||||
if let Some(event_list) = event_list_option {
|
||||
self.emit_event(event_list, event.value()).await;
|
||||
} else {
|
||||
println!("Button has no binding in the config file.");
|
||||
}
|
||||
},
|
||||
(_, AbsoluteAxisType::ABS_HAT0Y, _) => {
|
||||
let event_list_option: Option<&Vec<Key>> = match event.value() {
|
||||
-1 => self.config.keys.get(&Key::BTN_DPAD_UP),
|
||||
0 => self.config.abs.get(&"NONE_Y".to_string()),
|
||||
1 => self.config.keys.get(&Key::BTN_DPAD_DOWN),
|
||||
_ => self.config.abs.get(&"NONE_Y".to_string()),
|
||||
};
|
||||
if let Some(event_list) = event_list_option {
|
||||
self.emit_event(event_list, event.value()).await;
|
||||
} else {
|
||||
println!("Button has no binding in the config file.");
|
||||
}
|
||||
},
|
||||
(EventType::ABSOLUTE, AbsoluteAxisType::ABS_X | AbsoluteAxisType::ABS_Y, "left") => {
|
||||
let rel_value = self.get_rel_value(&is_xbox, &event).await;
|
||||
let mut analog_position = self.analog_position.lock().await;
|
||||
analog_position[event.code() as usize] = rel_value;
|
||||
},
|
||||
(EventType::ABSOLUTE, AbsoluteAxisType::ABS_RX | AbsoluteAxisType::ABS_RY, "right") => {
|
||||
let rel_value = self.get_rel_value(&is_xbox, &event).await;
|
||||
let mut analog_position = self.analog_position.lock().await;
|
||||
analog_position[(event.code() as usize) -3] = rel_value;
|
||||
},
|
||||
(EventType::ABSOLUTE, AbsoluteAxisType::ABS_Z, _) => {
|
||||
if let Some(event_list) = self.config.keys.get(&Key::BTN_TL2) {
|
||||
if event.value() == 0 {
|
||||
self.emit_event(event_list, event.value()).await
|
||||
} else {
|
||||
self.emit_event(event_list, 1).await
|
||||
};
|
||||
} else {
|
||||
println!("Button has no binding in the config file.");
|
||||
};
|
||||
},
|
||||
(EventType::ABSOLUTE, AbsoluteAxisType::ABS_RZ, _) => {
|
||||
if let Some(event_list) = self.config.keys.get(&Key::BTN_TR2) {
|
||||
if event.value() == 0 {
|
||||
self.emit_event(event_list, event.value()).await
|
||||
} else {
|
||||
self.emit_event(event_list, 1).await
|
||||
};
|
||||
} else {
|
||||
println!("Button has no binding in the config file.");
|
||||
};
|
||||
},
|
||||
_ => {}
|
||||
}
|
||||
}
|
||||
let mut device_is_connected = self.device_is_connected.lock().await;
|
||||
*device_is_connected = false;
|
||||
}
|
||||
|
||||
async fn emit_event(&self, event_list: &Vec<Key>, value: i32) {
|
||||
for key in event_list {
|
||||
let virtual_event: InputEvent = InputEvent::new_now(EventType::KEY, key.code(), value);
|
||||
let mut virt_dev = self.virt_dev.lock().await;
|
||||
virt_dev.keys.emit(&[virtual_event]).unwrap();
|
||||
}
|
||||
}
|
||||
|
||||
async fn get_rel_value(&self, is_xbox: &str, event: &InputEvent) -> i32 {
|
||||
let rel_value: i32 = match &is_xbox {
|
||||
&"false" => {
|
||||
let distance_from_center: i32 = event.value() as i32 - 128;
|
||||
distance_from_center / 10
|
||||
}
|
||||
_ => {
|
||||
event.value() as i32 / 2000
|
||||
}
|
||||
};
|
||||
return rel_value
|
||||
}
|
||||
|
||||
async fn cursor_loop(&self) {
|
||||
let polling_rate: u64 = self.config.settings.get("ANALOG_SENSITIVITY")
|
||||
.expect("No analog sensitivity found in config file.")
|
||||
.parse::<u64>()
|
||||
.expect("Invalid analog sensitivity.");
|
||||
while *self.device_is_connected.lock().await {
|
||||
{
|
||||
let analog_position = self.analog_position.lock().await;
|
||||
if analog_position[0] != 0 || analog_position[1] != 0 {
|
||||
let virtual_event_x: InputEvent = InputEvent::new_now(EventType::RELATIVE, 0, analog_position[0]);
|
||||
let virtual_event_y: InputEvent = InputEvent::new_now(EventType::RELATIVE, 1, analog_position[1]);
|
||||
let mut virt_dev = self.virt_dev.lock().await;
|
||||
virt_dev.relative_axes.emit(&[virtual_event_x]).unwrap();
|
||||
virt_dev.relative_axes.emit(&[virtual_event_y]).unwrap();
|
||||
}
|
||||
}
|
||||
tokio::time::sleep(std::time::Duration::from_millis(polling_rate)).await;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(serde::Deserialize, Debug, Clone)]
|
||||
struct Config {
|
||||
keys: HashMap<Key, Vec<Key>>,
|
||||
settings: HashMap<String, String>,
|
||||
#[serde(skip)]
|
||||
abs: HashMap<String, Vec<Key>>,
|
||||
}
|
||||
|
||||
impl Config {
|
||||
fn new_from_file(file: &str) -> Self {
|
||||
let config_str: String = std::fs::read_to_string(file).expect("No config file found at '~/.config/makima/config.toml");
|
||||
let config: Config = toml::from_str(&config_str).expect("Couldn't parse config file.");
|
||||
let keys: HashMap<Key, Vec<Key>> = config.keys;
|
||||
let mut abs: HashMap<String, Vec<Key>> = HashMap::new();
|
||||
let mut pad_horizontal: Vec<Key> = keys.get(&Key::BTN_DPAD_LEFT).unwrap_or(&Vec::new()).clone();
|
||||
pad_horizontal.extend(keys.get(&Key::BTN_DPAD_RIGHT).unwrap_or(&Vec::new()));
|
||||
let mut pad_vertical: Vec<Key> = keys.get(&Key::BTN_DPAD_UP).unwrap_or(&Vec::new()).clone();
|
||||
pad_vertical.extend(keys.get(&Key::BTN_DPAD_DOWN).unwrap_or(&Vec::new()));
|
||||
abs.insert("NONE_X".to_string(), pad_horizontal);
|
||||
abs.insert("NONE_Y".to_string(), pad_vertical);
|
||||
let settings: HashMap<String, String> = config.settings;
|
||||
Self {
|
||||
keys: keys,
|
||||
settings: settings,
|
||||
abs: abs,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
struct VirtualDevices {
|
||||
keys: VirtualDevice,
|
||||
relative_axes: VirtualDevice,
|
||||
}
|
||||
|
||||
impl VirtualDevices {
|
||||
fn new(keys: VirtualDevice, relative_axes: VirtualDevice) -> Self {
|
||||
Self {
|
||||
keys: keys,
|
||||
relative_axes: relative_axes,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async fn create_new_reader(device: String, config: Config) {
|
||||
let stream: Arc<Mutex<EventStream>> = Arc::new(Mutex::new(get_event_stream(Path::new(&device), config.clone())));
|
||||
let virt_dev: Arc<Mutex<VirtualDevices>> = Arc::new(Mutex::new(new_virtual_devices()));
|
||||
let reader = EventReader::new(config.clone(), stream, virt_dev);
|
||||
println!("Device detected at {}, reading events.", device);
|
||||
tokio::join!(
|
||||
reader.start(),
|
||||
reader.cursor_loop(),
|
||||
);
|
||||
println!("Disconnected device at {}.", device);
|
||||
}
|
||||
|
||||
async fn start_monitoring_udev(config: Config, mut tasks: Vec<JoinHandle<()>>) {
|
||||
let mut monitor = tokio_udev::AsyncMonitorSocket::new(
|
||||
tokio_udev::MonitorBuilder::new().unwrap()
|
||||
.match_subsystem(OsStr::new("input")).unwrap()
|
||||
.listen().unwrap()
|
||||
).unwrap();
|
||||
while let Some(Ok(event)) = monitor.next().await {
|
||||
if is_controller(&event.device()) {
|
||||
println!("Reinitializing...");
|
||||
let devices = scan_connected_devices();
|
||||
for task in &tasks {
|
||||
task.abort();
|
||||
}
|
||||
tasks.clear();
|
||||
for device in devices {
|
||||
tasks.push(tokio::spawn(create_new_reader(device, config.clone())));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
fn get_event_stream(path: &Path, config: Config) -> EventStream {
|
||||
let mut device: Device = Device::open(path).expect("Couldn't open device path.");
|
||||
if config.settings.get("GRAB_DEVICE").unwrap() == &"true".to_string() {
|
||||
device.grab().unwrap();
|
||||
};
|
||||
let stream: EventStream = device.into_event_stream().unwrap();
|
||||
return stream
|
||||
}
|
||||
|
||||
fn new_virtual_devices() -> VirtualDevices {
|
||||
let mut key_capabilities = evdev::AttributeSet::new();
|
||||
for i in 1..334 {key_capabilities.insert(Key(i));};
|
||||
let mut rel_capabilities = evdev::AttributeSet::new();
|
||||
for i in 0..13 {rel_capabilities.insert(evdev::RelativeAxisType(i));};
|
||||
let keys_builder = VirtualDeviceBuilder::new().unwrap()
|
||||
.name("Makima Virtual Keyboard/Mouse")
|
||||
.with_keys(&key_capabilities).unwrap();
|
||||
let rel_builder = VirtualDeviceBuilder::new().unwrap()
|
||||
.name("Makima Virtual Pointer")
|
||||
.with_relative_axes(&rel_capabilities).unwrap();
|
||||
let virtual_device_keys = keys_builder.build().unwrap();
|
||||
let virtual_device_rel = rel_builder.build().unwrap();
|
||||
let virtual_devices = VirtualDevices::new(virtual_device_keys, virtual_device_rel);
|
||||
return virtual_devices;
|
||||
}
|
||||
|
||||
fn scan_connected_devices() -> Vec<String> {
|
||||
let mut connected_devices: Vec<String> = Vec::new();
|
||||
let mut enumerator = tokio_udev::Enumerator::new().unwrap();
|
||||
enumerator.match_subsystem(
|
||||
OsStr::new("input"),
|
||||
).unwrap();
|
||||
let device_list = enumerator.scan_devices().unwrap();
|
||||
for device in device_list {
|
||||
if is_controller(&device) {
|
||||
connected_devices.push(device.devnode().unwrap().to_str().unwrap().to_string());
|
||||
}
|
||||
}
|
||||
return connected_devices
|
||||
}
|
||||
|
||||
fn is_controller(device: &tokio_udev::Device) -> bool {
|
||||
match (device.devnode(), device.property_value(OsStr::new("ID_INPUT_JOYSTICK"))) {
|
||||
(Some(_), Some(_)) => {
|
||||
if device.devnode().unwrap().to_str().unwrap().to_string().contains("event")
|
||||
&& !device.devpath().to_str().unwrap().contains(&"virtual") {
|
||||
true
|
||||
} else {
|
||||
false
|
||||
}
|
||||
}
|
||||
_ => false
|
||||
}
|
||||
}
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
let config_path = std::path::PathBuf::from(format!("{}/.config/makima/config.toml", home::home_dir().unwrap().display()));
|
||||
let config: Config = Config::new_from_file(config_path.to_str().unwrap());
|
||||
let devices: Vec<String> = scan_connected_devices();
|
||||
let mut tasks: Vec<JoinHandle<()>> = Vec::new();
|
||||
for device in devices {
|
||||
tasks.push(tokio::spawn(create_new_reader(device, config.clone())));
|
||||
}
|
||||
start_monitoring_udev(config.clone(), tasks).await;
|
||||
}
|
||||
Loading…
Add table
Reference in a new issue