#!/usr/bin/env python3
"""herdr live: socket 연결 하나를 열어 둔 채 pane 상태를 브라우저에 실시간으로 그리고, 브라우저에서 입력을 보낸다.
session.snapshot 처음 상태 (요청 하나, 응답 하나)
events.subscribe 상태 변화 (연결을 열어 둔 채 NDJSON이 계속 온다)
agent.prompt agent pane에 프롬프트
pane.send_text shell pane에 글자 + Enter
실행: herdr pane 안에서 python3 live.py, 브라우저로 http://127.0.0.1:8765
"""
import json, os, socket
from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
SOCKET = os.environ["HERDR_SOCKET_PATH"]
PORT = int(os.environ.get("PORT", "8765"))
RESYNC = {"pane_created", "pane_closed"}
def open_request(method, params):
conn = socket.socket(socket.AF_UNIX)
conn.connect(SOCKET)
conn.sendall((json.dumps({"id": method, "method": method, "params": params}) + "\n").encode())
return conn, conn.makefile(encoding="utf-8")
def call(method, params=None):
conn, lines = open_request(method, params or {})
with conn:
return json.loads(lines.readline())
def subscribe(pane_ids):
subs = [{"type": "pane.created"}, {"type": "pane.closed"}]
subs += [{"type": "pane.agent_status_changed", "pane_id": p} for p in pane_ids]
return open_request("events.subscribe", {"subscriptions": subs})
def send(pane, text):
if pane.get("agent"):
return call("agent.prompt", {"target": pane["pane_id"], "text": text})
return call("pane.send_text", {"pane_id": pane["pane_id"], "text": text + "\r"})
class Handler(BaseHTTPRequestHandler):
def do_GET(self):
if self.path == "/events":
return self.stream()
self.reply(200, "text/html; charset=utf-8", PAGE.encode())
def do_POST(self):
body = json.loads(self.rfile.read(int(self.headers["Content-Length"])))
self.reply(200, "application/json", json.dumps(send(body["pane"], body["text"])).encode())
def stream(self):
self.send_response(200)
self.send_header("Content-Type", "text/event-stream")
self.send_header("Cache-Control", "no-cache")
self.end_headers()
try:
while True: # pane이 생기거나 닫히면 snapshot부터 다시
snapshot = call("session.snapshot")["result"]["snapshot"]
self.push("snapshot", snapshot)
conn, lines = subscribe([p["pane_id"] for p in snapshot["panes"]])
with conn:
for line in lines:
message = json.loads(line)
self.push("event", message)
if message.get("event") in RESYNC:
break
except (BrokenPipeError, ConnectionResetError):
pass
def push(self, kind, data):
self.wfile.write(f"event: {kind}\ndata: {json.dumps(data)}\n\n".encode())
self.wfile.flush()
def reply(self, code, kind, body):
self.send_response(code)
self.send_header("Content-Type", kind)
self.end_headers()
self.wfile.write(body)
def log_message(self, *args):
pass
PAGE = """herdr live
herdr live
session.snapshot 한 번, 그다음은 events.subscribe가 계속 밀어 준다.
"""
if __name__ == "__main__":
print(f"herdr live: http://127.0.0.1:{PORT}")
ThreadingHTTPServer(("127.0.0.1", PORT), Handler).serve_forever()