use std::str::{from_utf8, Utf8Error};
use std::time::Duration;
use hidapi::{HidDevice, HidError};
use crate::{Kind, StreamDeckError, StreamDeckInput};
pub fn get_feature_report(device: &HidDevice, report_id: u8, length: usize) -> Result<Vec<u8>, HidError> {
let mut buff = vec![0u8; length];
buff.insert(0, report_id);
device.get_feature_report(buff.as_mut_slice())?;
Ok(buff)
}
pub fn send_feature_report(device: &HidDevice, payload: &[u8]) -> Result<(), HidError> {
device.send_feature_report(payload)
}
pub fn read_data(device: &HidDevice, length: usize, timeout: Option<Duration>) -> Result<Vec<u8>, HidError> {
device.set_blocking_mode(timeout.is_some())?;
let mut buf = vec![0u8; length];
match timeout {
Some(timeout) => device.read_timeout(buf.as_mut_slice(), timeout.as_millis() as i32),
None => device.read(buf.as_mut_slice()),
}?;
Ok(buf)
}
pub fn write_data(device: &HidDevice, payload: &[u8]) -> Result<usize, HidError> {
device.write(payload)
}
pub fn extract_str(bytes: &[u8]) -> Result<String, Utf8Error> {
Ok(from_utf8(bytes)?.replace('\0', "").to_string())
}
pub fn flip_key_index(kind: &Kind, key: u8) -> u8 {
let col = key % kind.column_count();
(key - col) + ((kind.column_count() - 1) - col)
}
pub fn read_button_states(kind: &Kind, states: &[u8]) -> Vec<bool> {
if states[0] == 0 {
return vec![];
}
match kind {
Kind::Original => {
let mut bools = vec![];
for i in 0..kind.key_count() {
let flipped_i = flip_key_index(kind, i) as usize;
bools.push(states[flipped_i + 1] != 0);
}
bools
}
Kind::Mini | Kind::MiniMk2 | Kind::MiniDiscord | Kind::MiniMk2Module => states[1..].iter().map(|s| *s != 0).collect(),
_ => states[4..].iter().map(|s| *s != 0).collect(),
}
}
pub fn read_lcd_input(data: &[u8]) -> Result<StreamDeckInput, StreamDeckError> {
let start_x = u16::from_le_bytes([data[6], data[7]]);
let start_y = u16::from_le_bytes([data[8], data[9]]);
match &data[4] {
0x1 => Ok(StreamDeckInput::TouchScreenPress(start_x, start_y)),
0x2 => Ok(StreamDeckInput::TouchScreenLongPress(start_x, start_y)),
0x3 => {
let end_x = u16::from_le_bytes([data[10], data[11]]);
let end_y = u16::from_le_bytes([data[12], data[13]]);
Ok(StreamDeckInput::TouchScreenSwipe((start_x, start_y), (end_x, end_y)))
}
_ => Err(StreamDeckError::BadData),
}
}
pub fn read_encoder_input(kind: &Kind, data: &[u8]) -> Result<StreamDeckInput, StreamDeckError> {
match &data[4] {
0x0 => Ok(StreamDeckInput::EncoderStateChange(data[5..5 + kind.encoder_count() as usize].iter().map(|s| *s != 0).collect())),
0x1 => Ok(StreamDeckInput::EncoderTwist(
data[5..5 + kind.encoder_count() as usize].iter().map(|s| i8::from_le_bytes([*s])).collect(),
)),
_ => Err(StreamDeckError::BadData),
}
}