feat: add inserter module with clipboard paste and restore

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
2026-04-06 20:28:18 +02:00
parent 0e9db0b60e
commit 70d2b6d6e4
2 changed files with 114 additions and 0 deletions
+69
View File
@@ -0,0 +1,69 @@
import asyncio
from unittest.mock import AsyncMock, MagicMock, patch, call
import pytest
from whisper_local.inserter import Inserter
class TestInserter:
@pytest.mark.asyncio
@patch("whisper_local.inserter.asyncio.sleep", new_callable=AsyncMock)
@patch("whisper_local.inserter.asyncio.create_subprocess_exec")
async def test_insert_text_calls_tools_in_order(self, mock_exec, mock_sleep):
# Mock für alle subprocess-Aufrufe
mock_proc = AsyncMock()
mock_proc.communicate.return_value = (b"alter clipboard", b"")
mock_proc.returncode = 0
mock_exec.return_value = mock_proc
inserter = Inserter()
await inserter.insert("Hallo Welt")
# Prüfe dass wl-paste, wl-copy, wtype, wl-copy aufgerufen werden
calls = mock_exec.call_args_list
assert len(calls) == 4
assert calls[0][0][0] == "wl-paste" # Clipboard sichern
assert calls[1][0] == ("wl-copy", "Hallo Welt") # Text in Clipboard
assert calls[2][0][0] == "wtype" # Ctrl+V simulieren
assert calls[3][0] == ("wl-copy", "alter clipboard") # Clipboard wiederherstellen
@pytest.mark.asyncio
@patch("whisper_local.inserter.asyncio.create_subprocess_exec")
async def test_insert_empty_text_does_nothing(self, mock_exec):
inserter = Inserter()
await inserter.insert("")
mock_exec.assert_not_called()
@pytest.mark.asyncio
@patch("whisper_local.inserter.asyncio.sleep", new_callable=AsyncMock)
@patch("whisper_local.inserter.asyncio.create_subprocess_exec")
async def test_clipboard_restore_on_paste_failure(self, mock_exec, mock_sleep):
call_count = 0
async def mock_create_proc(*args, **kwargs):
nonlocal call_count
call_count += 1
mock_proc = AsyncMock()
if call_count == 1: # wl-paste
mock_proc.communicate.return_value = (b"original", b"")
mock_proc.returncode = 0
elif call_count == 2: # wl-copy
mock_proc.communicate.return_value = (b"", b"")
mock_proc.returncode = 0
elif call_count == 3: # wtype — fails
mock_proc.communicate.return_value = (b"", b"error")
mock_proc.returncode = 1
else: # wl-copy restore
mock_proc.communicate.return_value = (b"", b"")
mock_proc.returncode = 0
return mock_proc
mock_exec.side_effect = mock_create_proc
inserter = Inserter()
await inserter.insert("Test")
# Clipboard muss trotzdem wiederhergestellt werden
assert call_count == 4
+45
View File
@@ -0,0 +1,45 @@
"""Text-Einfügung via Clipboard und wtype."""
import asyncio
import logging
logger = logging.getLogger(__name__)
PASTE_DELAY = 0.2 # Sekunden Wartezeit vor Clipboard-Restore
class Inserter:
async def _run(self, *cmd: str) -> tuple[bytes, int]:
"""Führt einen Befehl aus, gibt (stdout, returncode) zurück."""
proc = await asyncio.create_subprocess_exec(
*cmd,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
)
stdout, stderr = await proc.communicate()
if proc.returncode != 0:
logger.warning("Befehl fehlgeschlagen: %s (stderr: %s)", cmd, stderr.decode())
return stdout, proc.returncode
async def insert(self, text: str) -> None:
"""Fügt Text ins aktive Fenster ein via Clipboard + Ctrl+V."""
if not text:
return
# 1. Aktuelle Zwischenablage sichern
old_clipboard, _ = await self._run("wl-paste")
try:
# 2. Text in Zwischenablage kopieren
await self._run("wl-copy", text)
# 3. Ctrl+V simulieren
await self._run("wtype", "-M", "ctrl", "v", "-m", "ctrl")
# 4. Kurz warten, dann Clipboard wiederherstellen
await asyncio.sleep(PASTE_DELAY)
finally:
# Clipboard immer wiederherstellen
await self._run("wl-copy", old_clipboard.decode())
logger.info("Text eingefügt (%d Zeichen)", len(text))