144 lines
5.5 KiB
Python
144 lines
5.5 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import os
|
|
import time
|
|
import argparse
|
|
from selenium import webdriver
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.common.action_chains import ActionChains
|
|
from selenium.webdriver.common.actions.action_builder import ActionBuilder
|
|
from selenium.webdriver.common.actions.mouse_button import MouseButton
|
|
|
|
class CursorController:
|
|
def __init__(self, debug_port: int, verbose: bool = False):
|
|
"""Initialize connection to Chrome instance"""
|
|
self.verbose = verbose
|
|
options = webdriver.ChromeOptions()
|
|
options.add_experimental_option("debuggerAddress", f"127.0.0.1:{debug_port}")
|
|
self.driver = webdriver.Chrome(options=options)
|
|
|
|
# Load JavaScript files
|
|
js_dir = os.path.join(os.path.dirname(__file__), "..", "js")
|
|
self.js_code = ""
|
|
for filename in ["viewport.js", "mouse.js"]:
|
|
with open(os.path.join(js_dir, filename)) as f:
|
|
self.js_code += f.read() + "\n"
|
|
|
|
def log(self, msg: str):
|
|
"""Log message if verbose mode is enabled"""
|
|
if self.verbose:
|
|
print(f"DEBUG: {msg}", file=sys.stderr)
|
|
|
|
def get_viewport_info(self):
|
|
"""Get viewport dimensions and scroll position"""
|
|
return self.driver.execute_script(self.js_code + "; return getViewportInfo();")
|
|
|
|
def get_mouse_position(self):
|
|
"""Get current mouse coordinates"""
|
|
return self.driver.execute_script(self.js_code + "; return getMousePosition();")
|
|
|
|
def scroll_to_target(self, x: int, y: int):
|
|
"""Scroll viewport to make target coordinates visible"""
|
|
viewport = self.get_viewport_info()
|
|
self.driver.execute_script(
|
|
"window.scrollTo(arguments[0], arguments[1]);",
|
|
max(0, x - viewport['width'] // 2),
|
|
max(0, y - viewport['height'] // 2)
|
|
)
|
|
time.sleep(0.5) # Wait for scroll to complete
|
|
|
|
def get_element_at_position(self, x: int, y: int):
|
|
"""Get the element at specified coordinates"""
|
|
return self.driver.execute_script(
|
|
self.js_code + "; return document.elementFromPoint(arguments[0], arguments[1]);",
|
|
x, y
|
|
)
|
|
|
|
def move_cursor(self, x: int, y: int, perform_click: bool = False,
|
|
double_click: bool = False, right_click: bool = False,
|
|
element_id: str = None) -> bool:
|
|
"""Move cursor to specified coordinates and optionally perform click actions"""
|
|
try:
|
|
self.log(f"Moving cursor to ({x}, {y})")
|
|
|
|
if element_id:
|
|
# Try to find element by ID first
|
|
try:
|
|
element = self.driver.find_element(By.ID, element_id)
|
|
self.log(f"Found element by ID: {element_id}")
|
|
except:
|
|
self.log(f"Failed to find element by ID: {element_id}")
|
|
return False
|
|
else:
|
|
# Scroll to make target visible
|
|
self.scroll_to_target(x, y)
|
|
|
|
# Get element at position
|
|
element = self.get_element_at_position(x, y)
|
|
if not element:
|
|
self.log(f"No element found at coordinates ({x}, {y})")
|
|
return False
|
|
|
|
# Create action chain
|
|
actions = ActionChains(self.driver)
|
|
|
|
if element_id:
|
|
actions.move_to_element(element)
|
|
else:
|
|
actions.move_by_offset(x, y)
|
|
|
|
if perform_click:
|
|
actions.click()
|
|
elif double_click:
|
|
actions.double_click()
|
|
elif right_click:
|
|
actions.context_click()
|
|
|
|
# Execute the action chain
|
|
actions.perform()
|
|
return True
|
|
|
|
except Exception as e:
|
|
self.log(f"Error moving cursor: {str(e)}")
|
|
return False
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Control mouse cursor in Chrome')
|
|
parser.add_argument('debug_port', type=int, help='Chrome debug port')
|
|
parser.add_argument('-x', type=int, help='Target X coordinate')
|
|
parser.add_argument('-y', type=int, help='Target Y coordinate')
|
|
parser.add_argument('--click', action='store_true', help='Perform click')
|
|
parser.add_argument('--double', action='store_true', help='Perform double click')
|
|
parser.add_argument('--right', action='store_true', help='Perform right click')
|
|
parser.add_argument('--id', help='Element ID to target')
|
|
parser.add_argument('-v', '--verbose', action='store_true', help='Enable debug output')
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.double and args.right:
|
|
print("Cannot specify both --double and --right", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
if not args.id and (args.x is None or args.y is None):
|
|
print("Must specify either element ID or both x and y coordinates", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
try:
|
|
controller = CursorController(args.debug_port, args.verbose)
|
|
success = controller.move_cursor(
|
|
x=args.x if args.x is not None else 0,
|
|
y=args.y if args.y is not None else 0,
|
|
perform_click=args.click,
|
|
double_click=args.double,
|
|
right_click=args.right,
|
|
element_id=args.id
|
|
)
|
|
sys.exit(0 if success else 1)
|
|
|
|
except Exception as e:
|
|
if args.verbose:
|
|
print(f"Fatal error: {str(e)}", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
if __name__ == "__main__":
|
|
main() |