137 lines
4.3 KiB
Python
137 lines
4.3 KiB
Python
"""Input utilities for HarmonyOS device text input."""
|
|
|
|
import base64
|
|
import subprocess
|
|
from typing import Optional
|
|
|
|
from phone_agent.hdc.connection import _run_hdc_command
|
|
|
|
|
|
def type_text(text: str, device_id: str | None = None, x: int = None, y: int = None) -> None:
|
|
"""
|
|
Type text into the currently focused input field.
|
|
|
|
Args:
|
|
text: The text to type.
|
|
device_id: Optional HDC device ID for multi-device setups.
|
|
x: Optional X coordinate for input field (deprecated, kept for compatibility).
|
|
y: Optional Y coordinate for input field (deprecated, kept for compatibility).
|
|
|
|
Note:
|
|
HarmonyOS uses: hdc shell uitest uiInput text "文本内容"
|
|
This command works without coordinates when input field is focused.
|
|
Recommendation: Click on the input field first to focus it, then use this function.
|
|
"""
|
|
hdc_prefix = _get_hdc_prefix(device_id)
|
|
|
|
# Escape special characters for shell (keep quotes for proper text handling)
|
|
# The text will be wrapped in quotes in the command
|
|
escaped_text = text.replace('"', '\\"').replace("$", "\\$")
|
|
|
|
try:
|
|
# HarmonyOS uitest uiInput text command
|
|
# Format: hdc shell uitest uiInput text "文本内容"
|
|
_run_hdc_command(
|
|
hdc_prefix + ["shell", "uitest", "uiInput", "text", escaped_text],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
except Exception as e:
|
|
print(f"[HDC] Text input failed: {e}")
|
|
# Fallback: try with coordinates if provided (for older HarmonyOS versions)
|
|
if x is not None and y is not None:
|
|
try:
|
|
_run_hdc_command(
|
|
hdc_prefix + ["shell", "uitest", "uiInput", "inputText", str(x), str(y), escaped_text],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def clear_text(device_id: str | None = None) -> None:
|
|
"""
|
|
Clear text in the currently focused input field.
|
|
|
|
Args:
|
|
device_id: Optional HDC device ID for multi-device setups.
|
|
|
|
Note:
|
|
This method uses repeated delete key events to clear text.
|
|
For HarmonyOS, you might also use select all + delete for better efficiency.
|
|
"""
|
|
hdc_prefix = _get_hdc_prefix(device_id)
|
|
# Ctrl+A to select all (key code 2072 for Ctrl, 2017 for A)
|
|
# Then delete
|
|
_run_hdc_command(
|
|
hdc_prefix + ["shell", "uitest", "uiInput", "keyEvent", "2072", "2017"],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
_run_hdc_command(
|
|
hdc_prefix + ["shell", "uitest", "uiInput", "keyEvent", "2055"], # Delete key
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
|
|
def detect_and_set_adb_keyboard(device_id: str | None = None) -> str:
|
|
"""
|
|
Detect current keyboard and switch to ADB Keyboard if available.
|
|
|
|
Args:
|
|
device_id: Optional HDC device ID for multi-device setups.
|
|
|
|
Returns:
|
|
The original keyboard IME identifier for later restoration.
|
|
|
|
Note:
|
|
This is a placeholder. HarmonyOS may not support ADB Keyboard.
|
|
If there's a similar tool for HarmonyOS, integrate it here.
|
|
"""
|
|
hdc_prefix = _get_hdc_prefix(device_id)
|
|
|
|
# Get current IME (if HarmonyOS supports this)
|
|
try:
|
|
result = _run_hdc_command(
|
|
hdc_prefix + ["shell", "settings", "get", "secure", "default_input_method"],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
current_ime = (result.stdout + result.stderr).strip()
|
|
|
|
# If ADB Keyboard equivalent exists for HarmonyOS, switch to it
|
|
# For now, we'll just return the current IME
|
|
return current_ime
|
|
except Exception:
|
|
return ""
|
|
|
|
|
|
def restore_keyboard(ime: str, device_id: str | None = None) -> None:
|
|
"""
|
|
Restore the original keyboard IME.
|
|
|
|
Args:
|
|
ime: The IME identifier to restore.
|
|
device_id: Optional HDC device ID for multi-device setups.
|
|
"""
|
|
if not ime:
|
|
return
|
|
|
|
hdc_prefix = _get_hdc_prefix(device_id)
|
|
|
|
try:
|
|
_run_hdc_command(
|
|
hdc_prefix + ["shell", "ime", "set", ime], capture_output=True, text=True
|
|
)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def _get_hdc_prefix(device_id: str | None) -> list:
|
|
"""Get HDC command prefix with optional device specifier."""
|
|
if device_id:
|
|
return ["hdc", "-t", device_id]
|
|
return ["hdc"]
|