86 lines
2.2 KiB
Python
Executable File
86 lines
2.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Human-friendly MCP test client.
|
|
|
|
Usage: ue-wingman.py UserManual
|
|
"""
|
|
|
|
import sys
|
|
import json
|
|
import socket
|
|
|
|
HOST = "localhost"
|
|
PORT = 9851
|
|
TIMEOUT = 120
|
|
|
|
|
|
def main():
|
|
args = sys.argv[1:]
|
|
if not args:
|
|
print("Usage: ue-wingman.py ShowCommands [key=value ...]")
|
|
sys.exit(1)
|
|
|
|
no_args_commands = {"ShowCommands", "UserManual"}
|
|
if len(args) == 1 and args[0] not in no_args_commands:
|
|
# No extra arguments: read JSON object from stdin.
|
|
# Accumulate lines and try to parse after each one.
|
|
decoder = json.JSONDecoder()
|
|
raw = ""
|
|
msg = None
|
|
for line in sys.stdin:
|
|
raw += line
|
|
try:
|
|
msg, _ = decoder.raw_decode(raw.lstrip())
|
|
break
|
|
except json.JSONDecodeError:
|
|
continue
|
|
if msg is None:
|
|
print("Could not parse a complete JSON object from stdin")
|
|
sys.exit(1)
|
|
msg["command"] = args[0]
|
|
else:
|
|
msg = {"command": args[0]}
|
|
for arg in args[1:]:
|
|
key, _, value = arg.partition("=")
|
|
if value.lower() == "true":
|
|
value = True
|
|
elif value.lower() == "false":
|
|
value = False
|
|
else:
|
|
try:
|
|
value = int(value)
|
|
except ValueError:
|
|
pass
|
|
msg[key] = value
|
|
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.settimeout(TIMEOUT)
|
|
try:
|
|
sock.connect((HOST, PORT))
|
|
except (ConnectionRefusedError, socket.timeout, OSError) as e:
|
|
print(f"Cannot connect to {HOST}:{PORT} — is the editor running?")
|
|
sys.exit(1)
|
|
|
|
sock.sendall((json.dumps(msg) + "\n").encode())
|
|
|
|
result = b""
|
|
while True:
|
|
chunk = sock.recv(65536)
|
|
if not chunk:
|
|
break
|
|
result += chunk
|
|
if b"\0" in result:
|
|
break
|
|
|
|
sock.close()
|
|
result = result[:result.index(b"\0")].decode() if b"\0" in result else result.decode()
|
|
|
|
try:
|
|
parsed = json.loads(result)
|
|
print(json.dumps(parsed, indent=2))
|
|
except json.JSONDecodeError:
|
|
print(result)
|
|
|
|
if __name__ == "__main__":
|
|
main()
|