From 4eaf147daf04d0bbc3e40a0ab1b06064ee825ebe Mon Sep 17 00:00:00 2001
From: earthonion <garlic@duck.com>
Date: Sun, 21 Dec 2025 12:22:21 -0500
Subject: [PATCH] change to readline intead of aioconsole, for command history

---
 src/ws.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/ws.py b/src/ws.py
index 5b16ac6..20e3a1c 100644
--- a/src/ws.py
+++ b/src/ws.py
@@ -3,7 +3,7 @@
 import asyncio
 import pathlib
 import argparse
-import aioconsole
+import readline
 import websockets
 
 
@@ -40,8 +40,17 @@ async def send_file(ws: websockets.ClientConnection, file_path: str):
 
 
 async def command(ws: websockets.ClientConnection):
+    loop = asyncio.get_event_loop()
     while ws.state == websockets.protocol.State.OPEN:
-        cmd = await aioconsole.ainput()
+        try:
+            cmd = await loop.run_in_executor(None, input, "> ")
+        except (EOFError, KeyboardInterrupt):
+            print("\n[*] Disconnecting...")
+            await ws.close()
+            global retry
+            retry = False
+            break
+
         parts = cmd.split(maxsplit=1)
 
         if len(parts) == 2 and parts[0].lower() == "send":
@@ -49,7 +58,6 @@ async def command(ws: websockets.ClientConnection):
         elif cmd.lower() in ("quit", "exit", "disconnect"):
             print("[*] Disconnecting...")
             await ws.close()
-            global retry
             retry = False
             break
         else: