feat(hands,jobs): watchdog ring with a neighbour agent, daily heartbeat and silence alarm
This commit is contained in:
1 parent
0475e09c81
commit
f6cb17acaf
9 files changed
+1029
No files matched your search
Executable
+84
@@ -0,0 +1,84 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
"""Войти в комнату вайбграма по коду приглашения и получить токен агента.
|
||||
|
||||
Обычный путь - CLI `vibegram join <code> --nick <ник> --no-hooks` из клона
|
||||
репы комнаты. Сторожевой комнате клон не нужен (в ней только строки `hb` и
|
||||
`ack`), поэтому здесь тот же вызов хаба напрямую: `POST /api/rooms/join`
|
||||
с `joinCode` и `nick`. Ответ печатается целиком - токен из него едет в
|
||||
`.env` как `VIBEGRAM_HB_TOKEN`, после чего gateway надо перезапустить.
|
||||
|
||||
python3 scripts/vibegram_join.py <join-code> --nick <ник>
|
||||
python3 scripts/vibegram_join.py <join-code> --hub https://vibegram.example
|
||||
|
||||
Хаб по умолчанию берётся из `VIBEGRAM_HUB` в окружении или в соседнем `.env`.
|
||||
Токен печатается в терминал - не вставляй вывод в чаты и в задачи.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
import urllib.error
|
||||
import urllib.request
|
||||
from pathlib import Path
|
||||
|
||||
ENV = Path(__file__).resolve().parent.parent / ".env"
|
||||
|
||||
|
||||
def hub_from_env() -> str | None:
|
||||
if os.environ.get("VIBEGRAM_HUB"):
|
||||
return os.environ["VIBEGRAM_HUB"]
|
||||
if not ENV.exists():
|
||||
return None
|
||||
for line in ENV.read_text(encoding="utf-8").splitlines():
|
||||
key, _, value = line.partition("=")
|
||||
if key.strip() == "VIBEGRAM_HUB" and value.strip():
|
||||
return value.strip()
|
||||
return None
|
||||
|
||||
|
||||
def main() -> int:
|
||||
parser = argparse.ArgumentParser(description=__doc__)
|
||||
parser.add_argument("code", help="код приглашения из комнаты")
|
||||
parser.add_argument("--nick", help="ник агента в этой комнате")
|
||||
parser.add_argument("--hub", default=hub_from_env(), help="адрес хаба вайбграма")
|
||||
args = parser.parse_args()
|
||||
if not args.hub:
|
||||
print("не знаю адрес хаба: задай --hub или VIBEGRAM_HUB", file=sys.stderr)
|
||||
return 2
|
||||
|
||||
body: dict[str, str] = {"joinCode": args.code}
|
||||
if args.nick:
|
||||
body["nick"] = args.nick
|
||||
request = urllib.request.Request( # noqa: S310 - адрес хаба задаёт оператор
|
||||
f"{args.hub.rstrip('/')}/api/rooms/join",
|
||||
data=json.dumps(body).encode(),
|
||||
headers={"content-type": "application/json"},
|
||||
method="POST",
|
||||
)
|
||||
try:
|
||||
with urllib.request.urlopen(request, timeout=20) as response: # noqa: S310
|
||||
data = json.loads(response.read())
|
||||
except urllib.error.HTTPError as exc:
|
||||
print(f"хаб отказал: {exc.code} {exc.read().decode()[:400]}", file=sys.stderr)
|
||||
return 1
|
||||
except OSError as exc:
|
||||
print(f"хаб не отвечает: {exc}", file=sys.stderr)
|
||||
return 1
|
||||
|
||||
print(json.dumps(data, ensure_ascii=False, indent=2))
|
||||
token = next(
|
||||
(data[k] for k in ("token", "agentToken", "apiToken") if data.get(k)), None
|
||||
)
|
||||
if token:
|
||||
print(f"\nVIBEGRAM_HB_TOKEN={token}\n\nдальше: строку в .env, затем рестарт")
|
||||
else:
|
||||
print("\nтокена в ответе не видно - смотри поля выше", file=sys.stderr)
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
raise SystemExit(main())
|
||||
Reference in new issue
Block a user