62 lines
2.5 KiB
Python
62 lines
2.5 KiB
Python
|
"""Noise WebSocket protocol gateway transport."""
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import subprocess
|
||
|
|
import json
|
||
|
|
from typing import Dict, Any, Optional, List, Tuple
|
||
|
|
|
||
|
|
class NoiseGateway:
|
||
|
|
"""Primary fast gateway transport connecting via Noise protocol."""
|
||
|
|
|
||
|
|
def __init__(self, node: str = "muse"):
|
||
|
|
self.node = node
|
||
|
|
self.netvm_bin = os.path.expanduser("~/Projects/NetVM/bin")
|
||
|
|
|
||
|
|
def _call_hybrid(self, func_name: str, *args, **kwargs) -> Tuple[Any, Optional[str]]:
|
||
|
|
"""Try calling muse_hybrid directly if available."""
|
||
|
|
try:
|
||
|
|
if self.netvm_bin not in sys.path:
|
||
|
|
sys.path.insert(0, self.netvm_bin)
|
||
|
|
import muse_hybrid
|
||
|
|
func = getattr(muse_hybrid, func_name, None)
|
||
|
|
if func:
|
||
|
|
return func(self.node, *args, **kwargs)
|
||
|
|
except Exception as e:
|
||
|
|
return None, str(e)
|
||
|
|
return None, "muse_hybrid not found"
|
||
|
|
|
||
|
|
def list_threads(self) -> Tuple[Optional[List[Dict[str, Any]]], Optional[str]]:
|
||
|
|
res, err = self._call_hybrid("get_threads")
|
||
|
|
if res is not None:
|
||
|
|
return res, None
|
||
|
|
|
||
|
|
# CLI fallback via muse-cli-node
|
||
|
|
cli_path = os.path.join(self.netvm_bin, "muse-cli-node")
|
||
|
|
if os.path.exists(cli_path):
|
||
|
|
try:
|
||
|
|
proc = subprocess.run([cli_path, self.node, "threads", "--json"],
|
||
|
|
capture_output=True, text=True, timeout=10)
|
||
|
|
if proc.returncode == 0:
|
||
|
|
return json.loads(proc.stdout), None
|
||
|
|
except Exception as e:
|
||
|
|
return None, str(e)
|
||
|
|
return None, err or "gateway unavailable"
|
||
|
|
|
||
|
|
def send_message(self, message: str, thread_id: Optional[str] = None) -> Tuple[Optional[Dict[str, Any]], Optional[str]]:
|
||
|
|
res, err = self._call_hybrid("send_message", message=message, thread_id=thread_id)
|
||
|
|
if res is not None:
|
||
|
|
return res, None
|
||
|
|
|
||
|
|
cli_path = os.path.join(self.netvm_bin, "muse-cli-node")
|
||
|
|
if os.path.exists(cli_path):
|
||
|
|
try:
|
||
|
|
cmd = [cli_path, self.node, "send", message]
|
||
|
|
if thread_id:
|
||
|
|
cmd += ["--thread", thread_id]
|
||
|
|
proc = subprocess.run(cmd, capture_output=True, text=True, timeout=15)
|
||
|
|
if proc.returncode == 0:
|
||
|
|
return {"ok": True, "output": proc.stdout}, None
|
||
|
|
except Exception as e:
|
||
|
|
return None, str(e)
|
||
|
|
return None, err or "gateway unavailable"
|