Add auto-discovery support (UDP discovery responder & subnet scanner) and implement virtual screen MCP server

2026-06-03 14:29:49 -04:00
parent 81e7b15b61
commit 070390355e
8 changed files with 2115 additions and 2 deletions
@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+import subprocess
+import json
+import base64
+import time
+import os
+
+def run_test():
+    print("Starting server.py in a subprocess for stdio test...")
+    # Start server in subprocess on port 8089 to avoid port conflicts
+    proc = subprocess.Popen(
+        ["python3", "server.py", "--port", "8089"],
+        stdin=subprocess.PIPE,
+        stdout=subprocess.PIPE,
+        stderr=subprocess.DEVNULL, # ignore logs on stderr
+        text=True,
+        bufsize=1
+    )
+    
+    # Wait a moment for server to bind
+    time.sleep(1.5)
+    
+    def send_cmd(cmd_dict):
+        proc.stdin.write(json.dumps(cmd_dict) + "\n")
+        proc.stdin.flush()
+        line = proc.stdout.readline()
+        if not line:
+            return None
+        return json.loads(line.strip())
+
+    print("\n[1/4] Sending initialize request...")
+    init_resp = send_cmd({
+        "jsonrpc": "2.0",
+        "id": 1,
+        "method": "initialize",
+        "params": {}
+    })
+    print("Response:", json.dumps(init_resp, indent=2))
+    
+    print("\n[2/4] Sending clear_screen tool call...")
+    resp = send_cmd({
+        "jsonrpc": "2.0",
+        "id": 2,
+        "method": "tools/call",
+        "params": {
+            "name": "clear_screen",
+            "arguments": {"color": "#1e1e2e"}
+        }
+    })
+    print("Response:", json.dumps(resp, indent=2))
+    
+    print("\n[3/4] Sending draw_text tool call...")
+    resp = send_cmd({
+        "jsonrpc": "2.0",
+        "id": 3,
+        "method": "tools/call",
+        "params": {
+            "name": "draw_text",
+            "arguments": {
+                "text": "TEST SUCCEEDED",
+                "x": 200,
+                "y": 280,
+                "size": 48,
+                "color": "#10b981"
+            }
+        }
+    })
+    print("Response:", json.dumps(resp, indent=2))
+    
+    print("\n[4/4] Sending get_screenshot tool call...")
+    resp = send_cmd({
+        "jsonrpc": "2.0",
+        "id": 4,
+        "method": "tools/call",
+        "params": {
+            "name": "get_screenshot",
+            "arguments": {}
+        }
+    })
+    
+    try:
+        content = resp["result"]["content"][0]
+        if content["type"] == "image":
+            img_b64 = content["data"]
+            output_file = "test_screenshot.png"
+            with open(output_file, "wb") as f:
+                f.write(base64.b64decode(img_b64))
+            print(f"Success! Captured screenshot saved to: {os.path.abspath(output_file)}")
+        else:
+            print("Failed: content is not an image", content)
+    except Exception as e:
+        print("Error reading screenshot response:", e)
+        print("Raw response:", resp)
+        
+    # Stop subprocess
+    proc.terminate()
+    proc.wait()
+    print("\nVerification script finished.")
+
+if __name__ == "__main__":
+    run_test()