use super::{parse_mouse_mode, MouseMode, PressParams};
use crate::mouse::mouse_input;
use crate::script_engine::instruction::{
InstructionData, InstructionHandler, InstructionMetadata, ScriptError,
};
use crate::script_engine::VMContext;
pub struct PressHandler;
impl InstructionHandler for PressHandler {
fn name(&self) -> &str {
"press"
}
#[inline]
fn parse(&self, args: &[&str]) -> Result<InstructionData, ScriptError> {
let (mode, mode_offset) = parse_mouse_mode(args, MouseMode::Send)?;
let coord_args = &args[..args.len() - mode_offset];
let (x, y) = if coord_args.len() == 2 {
let x = coord_args[0].parse::<i32>().map_err(|e| {
ScriptError::ParseError(format!("Invalid x coordinate '{}': {}", coord_args[0], e))
})?;
let y = coord_args[1].parse::<i32>().map_err(|e| {
ScriptError::ParseError(format!("Invalid y coordinate '{}': {}", coord_args[1], e))
})?;
(Some(x), Some(y))
} else if coord_args.is_empty() {
(None, None)
} else {
return Err(ScriptError::ParseError(
"Press requires either 0 or 2 coordinates. Usage: press [x] [y] [mode]".into(),
));
};
if mode == MouseMode::Post && (x.is_none() || y.is_none()) {
return Err(ScriptError::ParseError(
"PostMessage mode requires coordinates. Usage: press <x> <y> post".into(),
));
}
let send_inputs = if mode == MouseMode::Send {
vec![mouse_input::build_press_left()]
} else {
vec![]
};
Ok(InstructionData::Custom(Box::new(PressParams {
x,
y,
mode,
mode_specified: mode_offset > 0,
send_inputs,
})))
}
#[inline]
fn execute(
&self,
vm: &mut VMContext,
data: &InstructionData,
_metadata: Option<&InstructionMetadata>,
) -> Result<(), ScriptError> {
let params = data.extract_custom::<PressParams>("Invalid press parameters")?;
let effective_mode = if params.mode_specified {
params.mode
} else {
match super::get_input_mode(vm).as_str() {
"post" => MouseMode::Post,
_ => MouseMode::Send,
}
};
match effective_mode {
MouseMode::Send => {
let screen_coords = if let (Some(x), Some(y)) = (params.x, params.y) {
#[cfg(feature = "script_process_context")]
{
if vm.process.has_hwnd() {
Some(super::convert_to_window_coords(vm, x, y)?)
} else {
Some((x, y))
}
}
#[cfg(not(feature = "script_process_context"))]
{
Some((x, y))
}
} else {
None
};
if let Some((screen_x, screen_y)) = screen_coords {
mouse_input::set_cursor_pos(screen_x, screen_y).map_err(|e| {
ScriptError::ExecutionError(format!("SetCursorPos failed: {:?}", e))
})?;
let inputs = vec![mouse_input::build_press_left()];
mouse_input::execute_inputs(&inputs).map_err(|e| {
ScriptError::ExecutionError(format!("Press failed: {:?}", e))
})?;
} else {
mouse_input::execute_inputs(¶ms.send_inputs).map_err(|e| {
ScriptError::ExecutionError(format!("Press failed: {:?}", e))
})?;
}
}
MouseMode::Post => {
#[cfg(feature = "script_process_context")]
{
use crate::mouse::mouse_message;
let window_x = params.x.unwrap_or(0);
let window_y = params.y.unwrap_or(0);
let (client_x, client_y) =
super::convert_to_client_coords(vm, window_x, window_y)?;
mouse_message::post_press_left_atomic(
vm.process.get_hwnd_or_err()?,
client_x,
client_y,
);
}
#[cfg(not(feature = "script_process_context"))]
{
return Err(ScriptError::ExecutionError(
"PostMessage mode requires 'script_process_context' feature. \
Enable it in Cargo.toml: features = [\"scripts_mouse_with_post\"] \
or use SendInput mode (default)."
.into(),
));
}
}
}
Ok(())
}
}