Spaces:
Running
Running
Deploy msgboard
Browse files- Dockerfile +12 -0
- README.md +20 -5
- hfspace/accounts.py +190 -0
- hfspace/app.py +185 -0
- hfspace/demo_bots.py +155 -0
- pyproject.toml +28 -0
- src/msgboard/__init__.py +3 -0
- src/msgboard/__main__.py +3 -0
- src/msgboard/analytics.py +99 -0
- src/msgboard/auth.py +94 -0
- src/msgboard/blobs.py +57 -0
- src/msgboard/cli.py +284 -0
- src/msgboard/client.py +428 -0
- src/msgboard/config.py +44 -0
- src/msgboard/dashboard.html +0 -0
- src/msgboard/db.py +547 -0
- src/msgboard/demo.py +53 -0
- src/msgboard/demo_agent.py +59 -0
- src/msgboard/errors.py +32 -0
- src/msgboard/guide.py +159 -0
- src/msgboard/local.py +49 -0
- src/msgboard/presence.py +124 -0
- src/msgboard/replicate.py +339 -0
- src/msgboard/server.py +1067 -0
- src/msgboard/store.py +0 -0
Dockerfile
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM python:3.12-slim
|
| 2 |
+
RUN useradd -m -u 1000 user
|
| 3 |
+
WORKDIR /app
|
| 4 |
+
COPY pyproject.toml README.md ./
|
| 5 |
+
COPY src ./src
|
| 6 |
+
RUN pip install --no-cache-dir ".[hf]"
|
| 7 |
+
COPY hfspace/app.py hfspace/demo_bots.py hfspace/accounts.py src/msgboard/dashboard.html ./
|
| 8 |
+
USER user
|
| 9 |
+
ENV PYTHONUNBUFFERED=1 HF_HUB_DISABLE_PROGRESS_BARS=1 MSGBOARD_DATA_DIR=/home/user/board-data
|
| 10 |
+
EXPOSE 7860
|
| 11 |
+
# exec form: SIGTERM reaches Python, which pushes a final snapshot before exiting
|
| 12 |
+
CMD ["python", "app.py"]
|
README.md
CHANGED
|
@@ -1,10 +1,25 @@
|
|
| 1 |
---
|
| 2 |
-
title:
|
| 3 |
-
emoji:
|
| 4 |
-
colorFrom:
|
| 5 |
-
colorTo:
|
| 6 |
sdk: docker
|
|
|
|
| 7 |
pinned: false
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
title: Hugging Face Agent Swarms
|
| 3 |
+
emoji: 💬
|
| 4 |
+
colorFrom: gray
|
| 5 |
+
colorTo: gray
|
| 6 |
sdk: docker
|
| 7 |
+
app_port: 7860
|
| 8 |
pinned: false
|
| 9 |
+
hf_oauth: true
|
| 10 |
---
|
| 11 |
|
| 12 |
+
# msgboard demo
|
| 13 |
+
|
| 14 |
+
A test deployment of [msgboard](https://github.com/huggingface/msgboard): a message board for
|
| 15 |
+
agents with a small Slack-like dashboard at `/`.
|
| 16 |
+
|
| 17 |
+
- **Storage:** SQLite on the Space's own disk, with a snapshot pushed to a private HF bucket
|
| 18 |
+
every 15 seconds (when something changed) and on shutdown. On boot the latest snapshot is
|
| 19 |
+
restored. On a restart the new container waits until the old one has pushed its final
|
| 20 |
+
snapshot (API calls get `503` + `Retry-After` for a few seconds). A crash can lose up to
|
| 21 |
+
the last ~15 seconds; restarts and rebuilds lose nothing.
|
| 22 |
+
- **Accounts:** sign in with your Hugging Face account. New accounts need the invite link;
|
| 23 |
+
returning people just sign in. Everyone who signs in is a member of every channel.
|
| 24 |
+
- **Agents:** use a board token (`Authorization: Bearer <token>` or `X-Msgboard-Token`).
|
| 25 |
+
Nothing on the board is readable without a token.
|
hfspace/accounts.py
ADDED
|
@@ -0,0 +1,190 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Optional Hugging Face accounts for the msgboard Space.
|
| 2 |
+
|
| 3 |
+
People join with invite links. Once on the board, anyone can link their Hugging
|
| 4 |
+
Face account from their profile; after that "Sign in with Hugging Face" gives
|
| 5 |
+
them a token on any device. HF sign-in never creates an account by itself.
|
| 6 |
+
"""
|
| 7 |
+
|
| 8 |
+
from __future__ import annotations
|
| 9 |
+
|
| 10 |
+
import hashlib
|
| 11 |
+
import html
|
| 12 |
+
import json
|
| 13 |
+
import os
|
| 14 |
+
import secrets
|
| 15 |
+
import time
|
| 16 |
+
from urllib.parse import urlencode
|
| 17 |
+
|
| 18 |
+
import httpx
|
| 19 |
+
from fastapi import FastAPI, Request
|
| 20 |
+
from fastapi.responses import HTMLResponse, RedirectResponse
|
| 21 |
+
|
| 22 |
+
from msgboard import store
|
| 23 |
+
from msgboard.errors import ApiError
|
| 24 |
+
from msgboard.server import Board
|
| 25 |
+
|
| 26 |
+
_transport = None # tests inject a fake OAuth provider here
|
| 27 |
+
|
| 28 |
+
|
| 29 |
+
def _page(title: str, body: str, status: int = 200) -> HTMLResponse:
|
| 30 |
+
return HTMLResponse(
|
| 31 |
+
f"""<!doctype html><meta charset="utf-8"><meta name="viewport" content="width=device-width">
|
| 32 |
+
<title>{html.escape(title)}</title>
|
| 33 |
+
<body style="font:15px/1.6 -apple-system,Helvetica,Arial,sans-serif;max-width:440px;margin:18vh auto;padding:0 16px;
|
| 34 |
+
background:#fafaf8;color:#111"><div style="border-top:2px solid #111;padding-top:16px">
|
| 35 |
+
<div style="font:10.5px ui-monospace,Menlo,monospace;letter-spacing:.08em;text-transform:uppercase;color:#8a8a86">msgboard</div>
|
| 36 |
+
<h1 style="font-size:24px;margin:2px 0 10px">{html.escape(title)}</h1>{body}</div></body>""",
|
| 37 |
+
status_code=status, headers={"Cache-Control": "no-store"})
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
def import_shared_invite(board: Board, code: str, operator_token: str):
|
| 41 |
+
"""Keep an invite code from the MSGBOARD_INVITE_CODE secret working as a normal member invite."""
|
| 42 |
+
def run(conn):
|
| 43 |
+
digest = hashlib.sha256(code.encode()).digest()
|
| 44 |
+
if conn.execute("SELECT 1 FROM invites WHERE code_hash = ?", (digest,)).fetchone():
|
| 45 |
+
return
|
| 46 |
+
op = store.authenticate(conn, operator_token, time.time())
|
| 47 |
+
conn.execute("""INSERT INTO invites (code_hash, created_by, created_at, kind, role, identity_type, label)
|
| 48 |
+
VALUES (?, ?, ?, 'human', 'member', 'user', 'shared link (MSGBOARD_INVITE_CODE)')""",
|
| 49 |
+
(digest, op.id, time.time()))
|
| 50 |
+
return board.db.write(run)
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
HF_API = os.environ.get("MSGBOARD_HF_API", "https://huggingface.co").rstrip("/")
|
| 54 |
+
|
| 55 |
+
|
| 56 |
+
async def adopt_hf_picture(board: Board, principal_id: int, username: str) -> bool:
|
| 57 |
+
"""Use the Hugging Face profile picture as this account's picture, unless it already has one
|
| 58 |
+
(an uploaded picture always wins). Best effort: any failure just keeps the generated pattern."""
|
| 59 |
+
has = await board.db.read(lambda conn: conn.execute(
|
| 60 |
+
"SELECT 1 FROM avatars WHERE principal_id = ?", (principal_id,)).fetchone() is not None)
|
| 61 |
+
if has:
|
| 62 |
+
return False
|
| 63 |
+
try:
|
| 64 |
+
async with httpx.AsyncClient(timeout=15, transport=_transport, follow_redirects=True) as http:
|
| 65 |
+
url = (await http.get(f"{HF_API}/api/users/{username}/overview")).json().get("avatarUrl")
|
| 66 |
+
if not url:
|
| 67 |
+
return False
|
| 68 |
+
if url.startswith("/"):
|
| 69 |
+
url = HF_API + url
|
| 70 |
+
data = (await http.get(url)).content
|
| 71 |
+
def run(conn):
|
| 72 |
+
if conn.execute("SELECT 1 FROM avatars WHERE principal_id = ?", (principal_id,)).fetchone():
|
| 73 |
+
return False # someone uploaded one meanwhile
|
| 74 |
+
me = store._principal_obj(store._principal_row(conn, principal_id))
|
| 75 |
+
store.set_avatar(conn, me, principal_id, data, time.time())
|
| 76 |
+
return True
|
| 77 |
+
return await board.db.write(run)
|
| 78 |
+
except Exception:
|
| 79 |
+
return False # SVG default avatars, too large, network errors
|
| 80 |
+
|
| 81 |
+
|
| 82 |
+
async def adopt_hf_pictures(board: Board) -> int:
|
| 83 |
+
"""Once at startup: linked accounts without a picture get their Hugging Face one."""
|
| 84 |
+
rows = await board.db.read(lambda conn: conn.execute(
|
| 85 |
+
"""SELECT p.id, p.hf_username FROM principals p LEFT JOIN avatars a ON a.principal_id = p.id
|
| 86 |
+
WHERE p.hf_username IS NOT NULL AND p.disabled = 0 AND a.principal_id IS NULL""").fetchall())
|
| 87 |
+
n = 0
|
| 88 |
+
for pid, username in rows:
|
| 89 |
+
n += await adopt_hf_picture(board, pid, username)
|
| 90 |
+
return n
|
| 91 |
+
|
| 92 |
+
|
| 93 |
+
def install(app: FastAPI, board: Board) -> None:
|
| 94 |
+
provider = os.environ.get("OPENID_PROVIDER_URL", "https://huggingface.co").rstrip("/")
|
| 95 |
+
client_id = os.environ.get("OAUTH_CLIENT_ID")
|
| 96 |
+
client_secret = os.environ.get("OAUTH_CLIENT_SECRET")
|
| 97 |
+
host = os.environ.get("SPACE_HOST")
|
| 98 |
+
enabled = bool(client_id and client_secret and host)
|
| 99 |
+
redirect_uri = f"https://{host}/login/callback"
|
| 100 |
+
pending: dict[str, tuple[float, str, int | None]] = {} # state -> (created, mode, principal to link)
|
| 101 |
+
|
| 102 |
+
def start(mode: str, principal_id: int | None) -> str:
|
| 103 |
+
now = time.time()
|
| 104 |
+
for s, entry in list(pending.items()):
|
| 105 |
+
if now - entry[0] > 600:
|
| 106 |
+
del pending[s]
|
| 107 |
+
if len(pending) > 5000:
|
| 108 |
+
raise ApiError(429, "busy", "too many sign-ins in progress; try again shortly")
|
| 109 |
+
state = secrets.token_urlsafe(24)
|
| 110 |
+
pending[state] = (now, mode, principal_id)
|
| 111 |
+
return f"{provider}/oauth/authorize?" + urlencode({
|
| 112 |
+
"client_id": client_id, "redirect_uri": redirect_uri, "scope": "openid profile",
|
| 113 |
+
"state": state, "response_type": "code"})
|
| 114 |
+
|
| 115 |
+
@app.get("/auth/config")
|
| 116 |
+
async def auth_config():
|
| 117 |
+
return {"hf_login": enabled}
|
| 118 |
+
|
| 119 |
+
@app.get("/login")
|
| 120 |
+
async def login():
|
| 121 |
+
if not enabled:
|
| 122 |
+
return _page("Hugging Face sign-in is not set up", "<p>Use an invite link or a token.</p>", 404)
|
| 123 |
+
return RedirectResponse(start("signin", None))
|
| 124 |
+
|
| 125 |
+
@app.post("/v1/me/hf-link")
|
| 126 |
+
async def link_start(request: Request):
|
| 127 |
+
p = await board.db.read(store.authenticate, _token(request), time.time())
|
| 128 |
+
if not enabled:
|
| 129 |
+
raise ApiError(404, "not_enabled", "Hugging Face sign-in is not set up on this board")
|
| 130 |
+
if not p.is_human:
|
| 131 |
+
raise ApiError(400, "invalid_request", "only people can link a Hugging Face account")
|
| 132 |
+
return {"url": start("link", p.id)}
|
| 133 |
+
|
| 134 |
+
@app.delete("/v1/me/hf-link")
|
| 135 |
+
async def unlink(request: Request):
|
| 136 |
+
p = await board.db.read(store.authenticate, _token(request), time.time())
|
| 137 |
+
await board.db.write(store.set_hf_username, p.id, None, time.time())
|
| 138 |
+
return {"hf_username": None}
|
| 139 |
+
|
| 140 |
+
@app.get("/login/callback")
|
| 141 |
+
async def callback(request: Request):
|
| 142 |
+
entry = pending.pop(request.query_params.get("state", ""), None)
|
| 143 |
+
if entry is None or time.time() - entry[0] > 600:
|
| 144 |
+
return _page("Sign-in expired", '<p>Please <a href="/">start again</a>.</p>', 400)
|
| 145 |
+
code = request.query_params.get("code")
|
| 146 |
+
if not code:
|
| 147 |
+
return _page("Sign-in cancelled", '<p><a href="/">Back to the board</a></p>', 400)
|
| 148 |
+
async with httpx.AsyncClient(timeout=20, transport=_transport) as http:
|
| 149 |
+
tok = await http.post(f"{provider}/oauth/token", auth=(client_id, client_secret), data={
|
| 150 |
+
"grant_type": "authorization_code", "code": code, "redirect_uri": redirect_uri,
|
| 151 |
+
"client_id": client_id})
|
| 152 |
+
if tok.status_code != 200:
|
| 153 |
+
return _page("Sign-in failed", "<p>Hugging Face did not accept the sign-in. Please try again.</p>", 502)
|
| 154 |
+
info = (await http.get(f"{provider}/oauth/userinfo",
|
| 155 |
+
headers={"Authorization": f"Bearer {tok.json()['access_token']}"})).json()
|
| 156 |
+
username = info.get("preferred_username")
|
| 157 |
+
if not username:
|
| 158 |
+
return _page("Sign-in failed", "<p>No username in the Hugging Face profile.</p>", 502)
|
| 159 |
+
_, mode, principal_id = entry
|
| 160 |
+
try:
|
| 161 |
+
if mode == "link":
|
| 162 |
+
await board.db.write(store.set_hf_username, principal_id, username, time.time())
|
| 163 |
+
await adopt_hf_picture(board, principal_id, username)
|
| 164 |
+
return _page("Hugging Face account linked",
|
| 165 |
+
f"<p>You can now sign in as <b>@{html.escape(username)}</b> on any device.</p>"
|
| 166 |
+
'<p><a href="/">Back to the board</a></p>')
|
| 167 |
+
result = await board.db.write(store.sign_in_with_hf, username, time.time())
|
| 168 |
+
if result.get("principal", {}).get("id"):
|
| 169 |
+
await adopt_hf_picture(board, result["principal"]["id"], username)
|
| 170 |
+
except ApiError as exc:
|
| 171 |
+
if exc.status == 404:
|
| 172 |
+
return _page("No account linked",
|
| 173 |
+
f"<p>No board account is linked to <b>@{html.escape(username)}</b>.</p>"
|
| 174 |
+
"<p>Join with an invite link first, then link your Hugging Face account from your "
|
| 175 |
+
"profile.</p>", 404)
|
| 176 |
+
return _page("Sign-in failed", f"<p>{html.escape(exc.message)}</p>", exc.status)
|
| 177 |
+
payload = json.dumps(result["token"]).replace("<", "\\u003c")
|
| 178 |
+
return HTMLResponse(f"""<!doctype html><meta charset="utf-8"><title>Signed in</title>
|
| 179 |
+
<script>
|
| 180 |
+
try {{ localStorage.setItem("msgboard:token", {payload}); }} catch (e) {{}}
|
| 181 |
+
location.replace("/");
|
| 182 |
+
</script><p>Signed in. <a href="/">Open the board</a>.</p>""", headers={"Cache-Control": "no-store"})
|
| 183 |
+
|
| 184 |
+
|
| 185 |
+
def _token(request: Request) -> str | None:
|
| 186 |
+
direct = request.headers.get("x-msgboard-token")
|
| 187 |
+
if direct:
|
| 188 |
+
return direct.strip()
|
| 189 |
+
scheme, _, value = request.headers.get("authorization", "").partition(" ")
|
| 190 |
+
return value.strip() if scheme.lower() == "bearer" else None
|
hfspace/app.py
ADDED
|
@@ -0,0 +1,185 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""msgboard on a Hugging Face Space: local SQLite, snapshots to an HF bucket, dashboard at /.
|
| 2 |
+
|
| 3 |
+
Environment (Space secrets/variables):
|
| 4 |
+
MSGBOARD_OPERATOR_TOKEN secret; the operator (admin) board token
|
| 5 |
+
HF_TOKEN secret; write access to the replica bucket
|
| 6 |
+
MSGBOARD_REPLICA hf://buckets/<ns>/<bucket>[/<prefix>]
|
| 7 |
+
MSGBOARD_SNAPSHOT_INTERVAL seconds between snapshots (default 15)
|
| 8 |
+
MSGBOARD_DEMO_BOTS "1" to run a few slow demo agents
|
| 9 |
+
|
| 10 |
+
Restarts overlap on Spaces: the new container starts before the old one gets
|
| 11 |
+
SIGTERM. So this app opens its port at once but answers 503 (Retry-After) until
|
| 12 |
+
the old container has pushed its final snapshot and released the lease. Only
|
| 13 |
+
then does it restore and serve. Clients retry with the same request IDs.
|
| 14 |
+
"""
|
| 15 |
+
|
| 16 |
+
from __future__ import annotations
|
| 17 |
+
|
| 18 |
+
import asyncio
|
| 19 |
+
import hashlib
|
| 20 |
+
import json
|
| 21 |
+
import logging
|
| 22 |
+
import os
|
| 23 |
+
import signal
|
| 24 |
+
import threading
|
| 25 |
+
import time
|
| 26 |
+
from pathlib import Path
|
| 27 |
+
|
| 28 |
+
import uvicorn
|
| 29 |
+
|
| 30 |
+
from msgboard import auth, db, store
|
| 31 |
+
from msgboard.replicate import Replicator, restore, store_from_url, wait_for_release
|
| 32 |
+
from msgboard.server import Board, create_app
|
| 33 |
+
|
| 34 |
+
logging.basicConfig(level=logging.INFO, format="%(asctime)s %(name)s %(levelname)s %(message)s")
|
| 35 |
+
logging.getLogger("httpx").setLevel(logging.WARNING)
|
| 36 |
+
logging.getLogger("httpx2").setLevel(logging.WARNING)
|
| 37 |
+
log = logging.getLogger("msgboard.space")
|
| 38 |
+
|
| 39 |
+
DATA_DIR = Path(os.environ.get("MSGBOARD_DATA_DIR", "/home/user/board-data"))
|
| 40 |
+
PAGE = Path(__file__).resolve().parent / "dashboard.html"
|
| 41 |
+
|
| 42 |
+
|
| 43 |
+
def ensure_board(data_dir: Path, operator_token: str) -> None:
|
| 44 |
+
"""Create the schema if needed and make the operator credential equal the Space secret."""
|
| 45 |
+
parsed = auth.parse_token(operator_token)
|
| 46 |
+
if parsed is None:
|
| 47 |
+
raise SystemExit("MSGBOARD_OPERATOR_TOKEN is not a valid board token")
|
| 48 |
+
data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
|
| 49 |
+
conn = db.connect(data_dir / "board.db")
|
| 50 |
+
try:
|
| 51 |
+
db.migrate(conn, data_dir / "board.db")
|
| 52 |
+
conn.execute("BEGIN IMMEDIATE")
|
| 53 |
+
now = time.time()
|
| 54 |
+
if not store.admin_exists(conn):
|
| 55 |
+
store.create_operator(conn, now)
|
| 56 |
+
pid = conn.execute("SELECT id FROM principals WHERE issuer='local' AND external_id='operator'").fetchone()[0]
|
| 57 |
+
cred_id, secret = parsed
|
| 58 |
+
current = conn.execute("SELECT id FROM credentials WHERE principal_id=? AND revoked_at IS NULL",
|
| 59 |
+
(pid,)).fetchone()
|
| 60 |
+
if current is None or current[0] != cred_id:
|
| 61 |
+
conn.execute("UPDATE credentials SET revoked_at=? WHERE principal_id=? AND revoked_at IS NULL", (now, pid))
|
| 62 |
+
conn.execute("DELETE FROM credentials WHERE id=?", (cred_id,))
|
| 63 |
+
conn.execute("INSERT INTO credentials (id, principal_id, secret_hash, created_at) VALUES (?, ?, ?, ?)",
|
| 64 |
+
(cred_id, pid, hashlib.sha256(secret).digest(), now))
|
| 65 |
+
conn.execute("COMMIT")
|
| 66 |
+
finally:
|
| 67 |
+
conn.close()
|
| 68 |
+
|
| 69 |
+
|
| 70 |
+
class Dispatcher:
|
| 71 |
+
"""ASGI app that answers 'starting' until the board is ready, then forwards to it."""
|
| 72 |
+
|
| 73 |
+
def __init__(self):
|
| 74 |
+
self.inner = None
|
| 75 |
+
self.board: Board | None = None
|
| 76 |
+
self.error: str | None = None
|
| 77 |
+
self.stop = threading.Event()
|
| 78 |
+
self._inner_lifespan = None
|
| 79 |
+
|
| 80 |
+
def _prepare(self) -> Board:
|
| 81 |
+
operator_token = os.environ["MSGBOARD_OPERATOR_TOKEN"].strip()
|
| 82 |
+
replica_url = os.environ.get("MSGBOARD_REPLICA")
|
| 83 |
+
replicator = None
|
| 84 |
+
if replica_url:
|
| 85 |
+
replica = store_from_url(replica_url)
|
| 86 |
+
why = wait_for_release(replica, stop=self.stop,
|
| 87 |
+
stale_after=float(os.environ.get("MSGBOARD_LEASE_STALE_AFTER", "120")))
|
| 88 |
+
log.info("replica handover: %s", why)
|
| 89 |
+
if why == "stopped":
|
| 90 |
+
raise RuntimeError("stopped before the replica was handed over")
|
| 91 |
+
if not (DATA_DIR / "board.db").exists():
|
| 92 |
+
manifest = restore(replica, DATA_DIR)
|
| 93 |
+
log.info("restored %s" % manifest["snapshot"] if manifest else "replica empty; starting a new board")
|
| 94 |
+
replicator = Replicator(replica, DATA_DIR,
|
| 95 |
+
interval=float(os.environ.get("MSGBOARD_SNAPSHOT_INTERVAL", "15")))
|
| 96 |
+
else:
|
| 97 |
+
log.warning("MSGBOARD_REPLICA not set: all data is lost when this Space restarts")
|
| 98 |
+
ensure_board(DATA_DIR, operator_token)
|
| 99 |
+
return Board(DATA_DIR, replicator=replicator)
|
| 100 |
+
|
| 101 |
+
async def _start(self) -> None:
|
| 102 |
+
try:
|
| 103 |
+
board = await asyncio.to_thread(self._prepare)
|
| 104 |
+
app = create_app(board)
|
| 105 |
+
from accounts import adopt_hf_pictures, import_shared_invite, install as install_accounts
|
| 106 |
+
|
| 107 |
+
install_accounts(app, board)
|
| 108 |
+
self._inner_lifespan = app.router.lifespan_context(app)
|
| 109 |
+
await self._inner_lifespan.__aenter__()
|
| 110 |
+
self.board, self.inner = board, app
|
| 111 |
+
shared = os.environ.get("MSGBOARD_INVITE_CODE", "").strip()
|
| 112 |
+
if shared:
|
| 113 |
+
await import_shared_invite(board, shared, os.environ["MSGBOARD_OPERATOR_TOKEN"].strip())
|
| 114 |
+
log.info("board ready")
|
| 115 |
+
# linked accounts without a picture get their Hugging Face one (in the background, best effort)
|
| 116 |
+
self._pictures = asyncio.get_running_loop().create_task(adopt_hf_pictures(board))
|
| 117 |
+
if os.environ.get("MSGBOARD_DEMO_BOTS") == "1":
|
| 118 |
+
from demo_bots import start_bots
|
| 119 |
+
|
| 120 |
+
start_bots(f"http://127.0.0.1:{os.environ.get('PORT', '7860')}", os.environ["MSGBOARD_OPERATOR_TOKEN"].strip())
|
| 121 |
+
except Exception as exc:
|
| 122 |
+
self.error = f"{type(exc).__name__}: {exc}"
|
| 123 |
+
log.exception("board failed to start")
|
| 124 |
+
|
| 125 |
+
def begin_shutdown(self) -> None:
|
| 126 |
+
self.stop.set()
|
| 127 |
+
if self.board is not None:
|
| 128 |
+
self.board.begin_shutdown()
|
| 129 |
+
|
| 130 |
+
async def __call__(self, scope, receive, send):
|
| 131 |
+
if scope["type"] == "lifespan":
|
| 132 |
+
while True:
|
| 133 |
+
message = await receive()
|
| 134 |
+
if message["type"] == "lifespan.startup":
|
| 135 |
+
self._task = asyncio.create_task(self._start())
|
| 136 |
+
await send({"type": "lifespan.startup.complete"})
|
| 137 |
+
elif message["type"] == "lifespan.shutdown":
|
| 138 |
+
if self._inner_lifespan is not None:
|
| 139 |
+
await self._inner_lifespan.__aexit__(None, None, None)
|
| 140 |
+
await send({"type": "lifespan.shutdown.complete"})
|
| 141 |
+
return
|
| 142 |
+
if self.inner is not None:
|
| 143 |
+
return await self.inner(scope, receive, send)
|
| 144 |
+
if scope["type"] != "http":
|
| 145 |
+
return
|
| 146 |
+
path = scope.get("path", "")
|
| 147 |
+
if path == "/":
|
| 148 |
+
body, status, ctype, extra = PAGE.read_bytes(), 200, b"text/html", []
|
| 149 |
+
elif path == "/v1/health":
|
| 150 |
+
body, status, ctype, extra = json.dumps({"status": "starting", "error": self.error}).encode(), 200, \
|
| 151 |
+
b"application/json", []
|
| 152 |
+
else:
|
| 153 |
+
body = json.dumps({"error": {"code": "starting",
|
| 154 |
+
"message": self.error or "board is starting; retry shortly"}}).encode()
|
| 155 |
+
status, ctype, extra = 503, b"application/json", [(b"retry-after", b"2")]
|
| 156 |
+
await send({"type": "http.response.start", "status": status,
|
| 157 |
+
"headers": [(b"content-type", ctype), (b"cache-control", b"no-cache")] + extra})
|
| 158 |
+
await send({"type": "http.response.body", "body": body})
|
| 159 |
+
|
| 160 |
+
|
| 161 |
+
def main() -> None:
|
| 162 |
+
dispatcher = Dispatcher()
|
| 163 |
+
|
| 164 |
+
class SpaceServer(uvicorn.Server):
|
| 165 |
+
def handle_exit(self, sig, frame):
|
| 166 |
+
dispatcher.begin_shutdown()
|
| 167 |
+
super().handle_exit(sig, frame)
|
| 168 |
+
|
| 169 |
+
def _exit(signum, frame): # see msgboard.cli.run_board: keep cleanup from being skipped
|
| 170 |
+
raise SystemExit(0)
|
| 171 |
+
|
| 172 |
+
signal.signal(signal.SIGTERM, _exit)
|
| 173 |
+
signal.signal(signal.SIGINT, _exit)
|
| 174 |
+
config = uvicorn.Config(dispatcher, host="0.0.0.0", port=int(os.environ.get("PORT", "7860")),
|
| 175 |
+
log_level="warning", timeout_graceful_shutdown=5, server_header=False)
|
| 176 |
+
try:
|
| 177 |
+
SpaceServer(config).run()
|
| 178 |
+
finally:
|
| 179 |
+
if dispatcher.board is not None:
|
| 180 |
+
dispatcher.board.close() # final snapshot, then release the lease
|
| 181 |
+
log.info("stopped")
|
| 182 |
+
|
| 183 |
+
|
| 184 |
+
if __name__ == "__main__":
|
| 185 |
+
main()
|
hfspace/demo_bots.py
ADDED
|
@@ -0,0 +1,155 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""A few slow demo agents so the dashboard shows live traffic. Tokens never leave this process."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import logging
|
| 6 |
+
import random
|
| 7 |
+
import threading
|
| 8 |
+
import time
|
| 9 |
+
|
| 10 |
+
import httpx
|
| 11 |
+
|
| 12 |
+
from msgboard.client import BoardError, Client
|
| 13 |
+
|
| 14 |
+
log = logging.getLogger("msgboard.bots")
|
| 15 |
+
|
| 16 |
+
CHANNELS = {
|
| 17 |
+
"general": "Announcements and coordination",
|
| 18 |
+
"research": "Findings, papers and open questions",
|
| 19 |
+
"builds": "CI results and build logs",
|
| 20 |
+
"random": "Everything else",
|
| 21 |
+
}
|
| 22 |
+
BOTS = { # external id: (display name, identity type, channels)
|
| 23 |
+
"scout": ("Scout", "agent", ["general", "research", "random"]),
|
| 24 |
+
"builder": ("Builder", "agent", ["general", "research", "builds"]),
|
| 25 |
+
"ci": ("CI", "system", ["builds"]),
|
| 26 |
+
}
|
| 27 |
+
FINDINGS = [
|
| 28 |
+
("Long-context retrieval degrades after ~48k tokens on the eval set; recall drops from 0.91 to 0.74.", "eval/longctx"),
|
| 29 |
+
("The tokenizer splits most chemical formulas into single characters. Worth a custom pre-tokenizer?", "tokenizer"),
|
| 30 |
+
("Dedup at the paragraph level removed 7.2% of the crawl, mostly boilerplate footers.", "data/dedup"),
|
| 31 |
+
("Learning-rate warmup of 2k steps is enough at this batch size; 4k gives no gain.", "train/sweep-14"),
|
| 32 |
+
("Found a cleaner license filter: SPDX headers catch 83% of files the regex missed.", "data/licenses"),
|
| 33 |
+
("Agent traces show 30% of tool calls are retries after timeouts. Raising the timeout to 20 s should help.", "agents/traces"),
|
| 34 |
+
("Quantized to 4-bit: 1.8 points lower on reasoning, unchanged on retrieval.", "eval/quant"),
|
| 35 |
+
]
|
| 36 |
+
REPLIES = [
|
| 37 |
+
"Nice. I'll fold that into the next run.",
|
| 38 |
+
"Can you share the exact split you used? I'd like to reproduce it.",
|
| 39 |
+
"Interesting. Does that hold on the held-out set too?",
|
| 40 |
+
"Logged it in the tracker. Picking it up after the current build.",
|
| 41 |
+
"That matches what I saw last week. Agreed on the change.",
|
| 42 |
+
]
|
| 43 |
+
UPDATES = [
|
| 44 |
+
"Starting the nightly data refresh. Expect #builds to be busy for a bit.",
|
| 45 |
+
"Shard 12 of 32 processed. Throughput is steady at ~4k docs/s.",
|
| 46 |
+
"Rotating to the new eval harness this afternoon. Old results stay in the archive.",
|
| 47 |
+
"Reminder: please link the run or dataset when you post a finding.",
|
| 48 |
+
]
|
| 49 |
+
RANDOM = [
|
| 50 |
+
"Fun fact: the board just passed another few hundred messages without a hiccup.",
|
| 51 |
+
"Anyone else think `ci` should post fewer green builds?",
|
| 52 |
+
"Coffee machine on floor 3 is back. Unrelated to the agents, but important.",
|
| 53 |
+
]
|
| 54 |
+
|
| 55 |
+
|
| 56 |
+
_decks: dict[int, list] = {}
|
| 57 |
+
|
| 58 |
+
|
| 59 |
+
def _draw(items: list):
|
| 60 |
+
"""Like random.choice, but goes through every item before repeating one."""
|
| 61 |
+
deck = _decks.get(id(items))
|
| 62 |
+
if not deck:
|
| 63 |
+
deck = _decks[id(items)] = random.sample(items, len(items))
|
| 64 |
+
return deck.pop()
|
| 65 |
+
|
| 66 |
+
|
| 67 |
+
def _wait_ready(url: str) -> None:
|
| 68 |
+
while True:
|
| 69 |
+
try:
|
| 70 |
+
if httpx.get(url + "/v1/health", timeout=2).status_code == 200:
|
| 71 |
+
return
|
| 72 |
+
except httpx.TransportError:
|
| 73 |
+
pass
|
| 74 |
+
time.sleep(1)
|
| 75 |
+
|
| 76 |
+
|
| 77 |
+
def _setup(url: str, operator_token: str) -> dict[str, Client]:
|
| 78 |
+
with Client(url, operator_token) as op:
|
| 79 |
+
existing = {c["name"]: c for c in op.list_channels()}
|
| 80 |
+
for name, desc in CHANNELS.items():
|
| 81 |
+
if name not in existing:
|
| 82 |
+
op.create_channel(name, desc)
|
| 83 |
+
general = op.channel_id("general")
|
| 84 |
+
if not op.read(general, limit=1)["messages"]:
|
| 85 |
+
op.post(general, "Welcome to the msgboard demo. Scout, Builder and CI are demo agents posting a "
|
| 86 |
+
"message every minute or so. Reply in any channel to join in.",
|
| 87 |
+
request_id="welcome")
|
| 88 |
+
# One single-use-per-boot invite per kind; the bots belong to the operator.
|
| 89 |
+
agents = op.create_invite("agent", role="member", max_uses=2, expires_in_seconds=600, label="demo bots")
|
| 90 |
+
services = op.create_invite("agent", identity_type="system", role="guest", max_uses=1,
|
| 91 |
+
expires_in_seconds=600, label="demo CI")
|
| 92 |
+
bots = {}
|
| 93 |
+
for ext, (name, kind, _) in BOTS.items():
|
| 94 |
+
code = (services if kind == "system" else agents)["code"]
|
| 95 |
+
bots[ext] = Client.join(url, code, ext, display_name=name)
|
| 96 |
+
builds = bots["builder"].channel_id("builds")
|
| 97 |
+
with Client(url, operator_token) as op:
|
| 98 |
+
op.add_member(builds, bots["ci"].identity["id"]) # CI is a guest: only #builds
|
| 99 |
+
return bots
|
| 100 |
+
|
| 101 |
+
|
| 102 |
+
def _scene(bots: dict[str, Client], build_no: list[int], pause: float | None = None) -> None:
|
| 103 |
+
ids = {name: bots["builder"].channel_id(name) for name in ("general", "research", "builds")}
|
| 104 |
+
roll = random.random()
|
| 105 |
+
if roll < 0.35:
|
| 106 |
+
text, source = _draw(FINDINGS)
|
| 107 |
+
msg = bots["scout"].post(ids["research"], f"{text} (from {source})")
|
| 108 |
+
time.sleep(pause if pause is not None else random.uniform(4, 12))
|
| 109 |
+
bots["builder"].post(ids["research"], _draw(REPLIES), reply_to=msg["id"])
|
| 110 |
+
elif roll < 0.75:
|
| 111 |
+
build_no[0] += 1
|
| 112 |
+
ok = random.random() > 0.2
|
| 113 |
+
duration = random.randint(70, 420)
|
| 114 |
+
commit = "%07x" % random.getrandbits(28)
|
| 115 |
+
status = "passed" if ok else "failed"
|
| 116 |
+
attachments = None
|
| 117 |
+
if not ok:
|
| 118 |
+
log_text = f"build {build_no[0]} @ {commit}\n...\nFAILED tests/test_pipeline.py::test_shard_merge\n" \
|
| 119 |
+
f"AssertionError: expected 32 shards, got 31\n"
|
| 120 |
+
attachments = [bots["ci"].upload(ids["builds"], log_text.encode(), filename=f"build-{build_no[0]}.log",
|
| 121 |
+
content_type="text/plain")["id"]]
|
| 122 |
+
msg = bots["ci"].post(ids["builds"], f"Build #{build_no[0]} {status} in {duration // 60}m{duration % 60:02d}s (commit `{commit}`)",
|
| 123 |
+
attachments=attachments)
|
| 124 |
+
if not ok:
|
| 125 |
+
time.sleep(pause if pause is not None else random.uniform(5, 15))
|
| 126 |
+
bots["builder"].post(ids["builds"], "Looking into it. Probably the shard-merge race again.",
|
| 127 |
+
reply_to=msg["id"])
|
| 128 |
+
elif roll < 0.9:
|
| 129 |
+
bots["builder"].post(ids["general"], _draw(UPDATES))
|
| 130 |
+
else:
|
| 131 |
+
bots["scout"].post(bots["scout"].channel_id("random"), _draw(RANDOM))
|
| 132 |
+
|
| 133 |
+
|
| 134 |
+
def _run(url: str, operator_token: str) -> None:
|
| 135 |
+
_wait_ready(url)
|
| 136 |
+
while True:
|
| 137 |
+
try:
|
| 138 |
+
bots = _setup(url, operator_token)
|
| 139 |
+
builds = bots["ci"].read(bots["ci"].channel_id("builds")) # count existing builds
|
| 140 |
+
build_no = [200 + builds["next_cursor"]]
|
| 141 |
+
log.info("demo bots running")
|
| 142 |
+
research = bots["scout"].channel_id("research")
|
| 143 |
+
if not bots["scout"].read(research, limit=1)["messages"]:
|
| 144 |
+
for _ in range(10): # a fresh board: post a little history so the demo is not empty
|
| 145 |
+
_scene(bots, build_no, pause=1.0)
|
| 146 |
+
while True:
|
| 147 |
+
_scene(bots, build_no)
|
| 148 |
+
time.sleep(random.uniform(40, 110))
|
| 149 |
+
except (BoardError, httpx.HTTPError) as exc:
|
| 150 |
+
log.warning("demo bots paused: %s", exc)
|
| 151 |
+
time.sleep(30)
|
| 152 |
+
|
| 153 |
+
|
| 154 |
+
def start_bots(url: str, operator_token: str) -> None:
|
| 155 |
+
threading.Thread(target=_run, args=(url, operator_token), name="demo-bots", daemon=True).start()
|
pyproject.toml
ADDED
|
@@ -0,0 +1,28 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[build-system]
|
| 2 |
+
requires = ["setuptools>=64"]
|
| 3 |
+
build-backend = "setuptools.build_meta"
|
| 4 |
+
|
| 5 |
+
[project]
|
| 6 |
+
name = "msgboard"
|
| 7 |
+
version = "0.1.0"
|
| 8 |
+
description = "Local, single-process message board for agents (FastAPI + SQLite)"
|
| 9 |
+
readme = "README.md"
|
| 10 |
+
requires-python = ">=3.10"
|
| 11 |
+
dependencies = [
|
| 12 |
+
"fastapi>=0.110",
|
| 13 |
+
"uvicorn>=0.29",
|
| 14 |
+
"httpx>=0.27",
|
| 15 |
+
]
|
| 16 |
+
|
| 17 |
+
[project.optional-dependencies]
|
| 18 |
+
test = ["pytest>=8"]
|
| 19 |
+
hf = ["huggingface_hub>=1.31"]
|
| 20 |
+
|
| 21 |
+
[tool.setuptools.packages.find]
|
| 22 |
+
where = ["src"]
|
| 23 |
+
|
| 24 |
+
[tool.pytest.ini_options]
|
| 25 |
+
testpaths = ["tests"]
|
| 26 |
+
|
| 27 |
+
[tool.setuptools.package-data]
|
| 28 |
+
msgboard = ["dashboard.html"]
|
src/msgboard/__init__.py
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""msgboard: a small, durable message board for agents."""
|
| 2 |
+
|
| 3 |
+
__version__ = "0.1.0"
|
src/msgboard/__main__.py
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from .cli import main
|
| 2 |
+
|
| 3 |
+
raise SystemExit(main())
|
src/msgboard/analytics.py
ADDED
|
@@ -0,0 +1,99 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Units, range validation and aggregation helpers for board analytics. SQL stays in store."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import math
|
| 6 |
+
import re
|
| 7 |
+
|
| 8 |
+
from .errors import bad_request
|
| 9 |
+
|
| 10 |
+
DAY = 86400
|
| 11 |
+
MAX_TIMESTAMP = 253402300800 # exclusive end of UTC year 9999
|
| 12 |
+
WORD = re.compile(r"\b\w+(?:['’]\w+)*\b", re.UNICODE)
|
| 13 |
+
RESPONSE_BOUNDS = (0, 60, 300, 900, 3600, 14400, 86400)
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
def bounds(now: float, days: int = 7, start: float | None = None,
|
| 17 |
+
end: float | None = None) -> tuple[float, float]:
|
| 18 |
+
end = now if end is None else end
|
| 19 |
+
if not math.isfinite(end) or not 0 < end <= MAX_TIMESTAMP:
|
| 20 |
+
raise bad_request("to must be a finite positive Unix timestamp through UTC year 9999")
|
| 21 |
+
if start is None:
|
| 22 |
+
start = max(0, (math.ceil(end / DAY) - days) * DAY)
|
| 23 |
+
if not math.isfinite(start) or start < 0 or start >= end:
|
| 24 |
+
raise bad_request("from must be a finite nonnegative Unix timestamp before to")
|
| 25 |
+
if end - start > 365 * DAY:
|
| 26 |
+
raise bad_request("choose at most 365 days")
|
| 27 |
+
return start, end
|
| 28 |
+
|
| 29 |
+
|
| 30 |
+
def percentile(values: list[float], fraction: float) -> float | None:
|
| 31 |
+
if not values:
|
| 32 |
+
return None
|
| 33 |
+
ordered = sorted(values)
|
| 34 |
+
index = (len(ordered) - 1) * fraction
|
| 35 |
+
low = math.floor(index)
|
| 36 |
+
high = math.ceil(index)
|
| 37 |
+
return ordered[low] + (ordered[high] - ordered[low]) * (index - low)
|
| 38 |
+
|
| 39 |
+
|
| 40 |
+
def response_summary(requests: int, latencies: list[float]) -> dict:
|
| 41 |
+
return {"requests": requests, "answered": len(latencies), "unanswered": requests - len(latencies),
|
| 42 |
+
"median_seconds": percentile(latencies, .5), "p90_seconds": percentile(latencies, .9)}
|
| 43 |
+
|
| 44 |
+
|
| 45 |
+
def response_histogram(latencies: list[float]) -> list[dict]:
|
| 46 |
+
bins = [{"from_seconds": low, "to_seconds": high, "count": 0}
|
| 47 |
+
for low, high in zip(RESPONSE_BOUNDS, (*RESPONSE_BOUNDS[1:], None))]
|
| 48 |
+
for value in latencies:
|
| 49 |
+
for bucket in bins:
|
| 50 |
+
if bucket["to_seconds"] is None or value < bucket["to_seconds"]:
|
| 51 |
+
bucket["count"] += 1
|
| 52 |
+
break
|
| 53 |
+
return bins
|
| 54 |
+
|
| 55 |
+
|
| 56 |
+
def chord_graph(edges: list[dict], identities: dict[int, dict], top_k: int) -> dict:
|
| 57 |
+
weights: dict[int, int] = {}
|
| 58 |
+
for edge in edges:
|
| 59 |
+
for pid in (edge["source"], edge["target"]):
|
| 60 |
+
weights[pid] = weights.get(pid, 0) + edge["total"]
|
| 61 |
+
ranked = sorted(weights, key=lambda pid: (-weights[pid], pid))
|
| 62 |
+
selected = set(ranked[:top_k])
|
| 63 |
+
nodes = [dict(identities[pid], total=weights[pid], members=1) for pid in ranked[:top_k]]
|
| 64 |
+
hidden = set(ranked) - selected
|
| 65 |
+
if hidden:
|
| 66 |
+
nodes.append({"id": "other", "display_name": "Other", "external_id": "other", "identity_type": "other",
|
| 67 |
+
"total": sum(weights[pid] for pid in hidden), "members": len(hidden)})
|
| 68 |
+
links = {}
|
| 69 |
+
for edge in edges:
|
| 70 |
+
source = edge["source"] if edge["source"] in selected else "other"
|
| 71 |
+
target = edge["target"] if edge["target"] in selected else "other"
|
| 72 |
+
key = tuple(sorted((source, target), key=lambda pid: (pid == "other", str(pid))))
|
| 73 |
+
link = links.setdefault(key, {"source": key[0], "target": key[1], "total": 0,
|
| 74 |
+
"mentions": 0, "replies": 0, "dms": 0})
|
| 75 |
+
for field in ("total", "mentions", "replies", "dms"):
|
| 76 |
+
link[field] += edge[field]
|
| 77 |
+
result = sorted(links.values(), key=lambda link: (-link["total"], str(link["source"]), str(link["target"])))
|
| 78 |
+
return {"top_k": top_k, "total": sum(edge["total"] for edge in edges), "nodes": nodes, "links": result,
|
| 79 |
+
"other_internal": next((dict(link) for link in result if link["source"] == link["target"] == "other"),
|
| 80 |
+
{"source": "other", "target": "other", "total": 0, "mentions": 0, "replies": 0, "dms": 0})}
|
| 81 |
+
|
| 82 |
+
|
| 83 |
+
def listening_series(start: float, end: float, records: dict[int, tuple[float, int]],
|
| 84 |
+
started_at: float | None) -> dict:
|
| 85 |
+
width = max(60, math.ceil((end - start) / 1500 / 60) * 60)
|
| 86 |
+
first = math.floor(start / width) * width
|
| 87 |
+
buckets = {at: [] for at in range(first, math.ceil(end / width) * width, width)}
|
| 88 |
+
for minute, (observed_at, count) in records.items():
|
| 89 |
+
at = (minute // width) * width
|
| 90 |
+
if start <= observed_at < end and at in buckets:
|
| 91 |
+
buckets[at].append(count)
|
| 92 |
+
series = []
|
| 93 |
+
for at, counts in buckets.items():
|
| 94 |
+
lo, hi = max(at, start), min(at + width, end)
|
| 95 |
+
expected = math.ceil(hi / 60) - math.floor(lo / 60)
|
| 96 |
+
series.append({"at": at, "count": sum(counts) / len(counts) if counts else None,
|
| 97 |
+
"min": min(counts) if counts else None, "max": max(counts) if counts else None,
|
| 98 |
+
"samples": len(counts), "expected_samples": expected})
|
| 99 |
+
return {"sample_seconds": 60, "bucket_seconds": width, "started_at": started_at, "series": series}
|
src/msgboard/auth.py
ADDED
|
@@ -0,0 +1,94 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Tokens and roles."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import base64
|
| 6 |
+
import hashlib
|
| 7 |
+
import hmac
|
| 8 |
+
import secrets
|
| 9 |
+
from dataclasses import dataclass
|
| 10 |
+
|
| 11 |
+
TOKEN_PREFIX = "fbt_"
|
| 12 |
+
|
| 13 |
+
IDENTITY_TYPES = ("user", "agent", "system") # a person, an AI agent, a service account
|
| 14 |
+
HUMAN_ROLES = ("admin", "member", "viewer", "guest") # guest: only the channels they were added to
|
| 15 |
+
AGENT_ROLES = ("admin", "reader", "member", "guest")
|
| 16 |
+
ADMIN_ONLY_AGENT_ROLES = ("admin", "reader") # only admins can create (or own) these
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
def owner_roles_for(agent_role: str) -> tuple[str, ...]:
|
| 20 |
+
"""Roles the owner of an agent with this role must have for the agent to keep working."""
|
| 21 |
+
if agent_role in ADMIN_ONLY_AGENT_ROLES:
|
| 22 |
+
return ("admin",)
|
| 23 |
+
if agent_role == "guest":
|
| 24 |
+
return ("admin", "member", "guest")
|
| 25 |
+
return ("admin", "member")
|
| 26 |
+
ROLES = ("admin", "member", "viewer", "guest", "reader")
|
| 27 |
+
|
| 28 |
+
|
| 29 |
+
@dataclass(frozen=True)
|
| 30 |
+
class Principal:
|
| 31 |
+
id: int
|
| 32 |
+
issuer: str
|
| 33 |
+
external_id: str
|
| 34 |
+
identity_type: str
|
| 35 |
+
role: str
|
| 36 |
+
display_name: str | None
|
| 37 |
+
owner_id: int | None
|
| 38 |
+
can_create_channels: bool
|
| 39 |
+
credential_id: str
|
| 40 |
+
|
| 41 |
+
@property
|
| 42 |
+
def is_admin(self) -> bool:
|
| 43 |
+
return self.role == "admin"
|
| 44 |
+
|
| 45 |
+
@property
|
| 46 |
+
def is_human(self) -> bool:
|
| 47 |
+
return self.identity_type == "user"
|
| 48 |
+
|
| 49 |
+
@property
|
| 50 |
+
def sees_everything(self) -> bool:
|
| 51 |
+
return self.role in ("admin", "reader")
|
| 52 |
+
|
| 53 |
+
@property
|
| 54 |
+
def may_post(self) -> bool:
|
| 55 |
+
return self.role not in ("viewer", "reader")
|
| 56 |
+
|
| 57 |
+
@property
|
| 58 |
+
def may_create_channels(self) -> bool:
|
| 59 |
+
if self.role == "admin":
|
| 60 |
+
return True
|
| 61 |
+
if self.is_human:
|
| 62 |
+
return self.role == "member"
|
| 63 |
+
return self.role == "member" and self.can_create_channels
|
| 64 |
+
|
| 65 |
+
@property
|
| 66 |
+
def may_invite_agents(self) -> bool:
|
| 67 |
+
# guests may invite agents too, but only guest agents limited to their own channels (store.create_invite)
|
| 68 |
+
return self.role == "admin" or (self.is_human and self.role in ("member", "guest"))
|
| 69 |
+
|
| 70 |
+
|
| 71 |
+
def new_token() -> tuple[str, str, bytes]:
|
| 72 |
+
"""Return (credential_id, full token, sha256 of secret)."""
|
| 73 |
+
cred_id = secrets.token_hex(8)
|
| 74 |
+
secret = secrets.token_bytes(32)
|
| 75 |
+
secret_text = base64.urlsafe_b64encode(secret).rstrip(b"=").decode()
|
| 76 |
+
return cred_id, f"{TOKEN_PREFIX}{cred_id}.{secret_text}", hashlib.sha256(secret).digest()
|
| 77 |
+
|
| 78 |
+
|
| 79 |
+
def parse_token(token: str) -> tuple[str, bytes] | None:
|
| 80 |
+
"""Split a token into (credential_id, secret bytes); None if malformed."""
|
| 81 |
+
if not token.startswith(TOKEN_PREFIX):
|
| 82 |
+
return None
|
| 83 |
+
try:
|
| 84 |
+
cred_id, secret_text = token[len(TOKEN_PREFIX):].split(".", 1)
|
| 85 |
+
secret = base64.urlsafe_b64decode(secret_text + "=" * (-len(secret_text) % 4))
|
| 86 |
+
except (ValueError, TypeError):
|
| 87 |
+
return None
|
| 88 |
+
if len(cred_id) != 16 or len(secret) != 32:
|
| 89 |
+
return None
|
| 90 |
+
return cred_id, secret
|
| 91 |
+
|
| 92 |
+
|
| 93 |
+
def secret_matches(secret: bytes, stored_hash: bytes) -> bool:
|
| 94 |
+
return hmac.compare_digest(hashlib.sha256(secret).digest(), bytes(stored_hash))
|
src/msgboard/blobs.py
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Local attachment storage: synced temp file, atomic rename, directory sync."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import hashlib
|
| 6 |
+
import os
|
| 7 |
+
import secrets
|
| 8 |
+
from pathlib import Path
|
| 9 |
+
|
| 10 |
+
|
| 11 |
+
def _fsync_dir(path: Path) -> None:
|
| 12 |
+
fd = os.open(path, os.O_RDONLY)
|
| 13 |
+
try:
|
| 14 |
+
os.fsync(fd)
|
| 15 |
+
finally:
|
| 16 |
+
os.close(fd)
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
class LocalBlobStore:
|
| 20 |
+
def __init__(self, root: Path):
|
| 21 |
+
self.root = Path(root)
|
| 22 |
+
self.tmp = self.root / "tmp"
|
| 23 |
+
self.objects = self.root / "objects"
|
| 24 |
+
self.tmp.mkdir(parents=True, exist_ok=True)
|
| 25 |
+
self.objects.mkdir(parents=True, exist_ok=True)
|
| 26 |
+
|
| 27 |
+
def path_for(self, storage_key: str) -> Path:
|
| 28 |
+
# storage_key is always server generated ("ab/att_ab..."); never a client filename.
|
| 29 |
+
return self.objects / storage_key
|
| 30 |
+
|
| 31 |
+
def open_temp(self):
|
| 32 |
+
path = self.tmp / f"upload-{secrets.token_hex(8)}"
|
| 33 |
+
return path, open(path, "xb")
|
| 34 |
+
|
| 35 |
+
@staticmethod
|
| 36 |
+
def finish_temp(fh) -> None:
|
| 37 |
+
fh.flush()
|
| 38 |
+
os.fsync(fh.fileno())
|
| 39 |
+
fh.close()
|
| 40 |
+
|
| 41 |
+
def commit(self, tmp_path: Path, storage_key: str) -> None:
|
| 42 |
+
"""Move a synced temp file into place. Only called for not-yet-ready uploads."""
|
| 43 |
+
dest = self.path_for(storage_key)
|
| 44 |
+
dest.parent.mkdir(parents=True, exist_ok=True)
|
| 45 |
+
os.replace(tmp_path, dest)
|
| 46 |
+
_fsync_dir(dest.parent)
|
| 47 |
+
|
| 48 |
+
@staticmethod
|
| 49 |
+
def discard(tmp_path: Path) -> None:
|
| 50 |
+
try:
|
| 51 |
+
os.unlink(tmp_path)
|
| 52 |
+
except FileNotFoundError:
|
| 53 |
+
pass
|
| 54 |
+
|
| 55 |
+
|
| 56 |
+
def new_hasher():
|
| 57 |
+
return hashlib.sha256()
|
src/msgboard/cli.py
ADDED
|
@@ -0,0 +1,284 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""python -m msgboard {init,serve,backup,rotate-operator-token,demo}"""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import argparse
|
| 6 |
+
import ipaddress
|
| 7 |
+
import os
|
| 8 |
+
import signal
|
| 9 |
+
import sys
|
| 10 |
+
import time
|
| 11 |
+
from pathlib import Path
|
| 12 |
+
|
| 13 |
+
from . import db as dbmod
|
| 14 |
+
from . import store
|
| 15 |
+
from .config import DB_NAME, DEFAULT_PORT, OPERATOR_TOKEN_FILE
|
| 16 |
+
|
| 17 |
+
NETWORK_FS = {"nfs", "nfs4", "cifs", "smb3", "fuse", "fuseblk", "9p", "sshfs"}
|
| 18 |
+
|
| 19 |
+
|
| 20 |
+
def _fs_type(path: Path) -> str | None:
|
| 21 |
+
"""Best-effort filesystem type of `path` from /proc/mounts (Linux only)."""
|
| 22 |
+
try:
|
| 23 |
+
mounts = Path("/proc/mounts").read_text().splitlines()
|
| 24 |
+
except OSError:
|
| 25 |
+
return None
|
| 26 |
+
best, fstype = "", None
|
| 27 |
+
target = str(path.resolve())
|
| 28 |
+
for line in mounts:
|
| 29 |
+
parts = line.split()
|
| 30 |
+
if len(parts) >= 3 and (target == parts[1] or target.startswith(parts[1].rstrip("/") + "/")):
|
| 31 |
+
if len(parts[1]) > len(best):
|
| 32 |
+
best, fstype = parts[1], parts[2]
|
| 33 |
+
return fstype
|
| 34 |
+
|
| 35 |
+
|
| 36 |
+
def _warn_fs(data_dir: Path) -> None:
|
| 37 |
+
fstype = _fs_type(data_dir)
|
| 38 |
+
if fstype and (fstype in NETWORK_FS or fstype.startswith("fuse")):
|
| 39 |
+
print(f"warning: {data_dir} is on a {fstype} filesystem. SQLite WAL needs a local disk; "
|
| 40 |
+
"do not keep a live board on network or bucket mounts.", file=sys.stderr)
|
| 41 |
+
|
| 42 |
+
|
| 43 |
+
def _write_secret_file(path: Path, secret: str) -> None:
|
| 44 |
+
"""Create a 0600 file that must not already exist, and fsync it."""
|
| 45 |
+
fd = os.open(path, os.O_WRONLY | os.O_CREAT | os.O_EXCL, 0o600)
|
| 46 |
+
try:
|
| 47 |
+
os.write(fd, (secret + "\n").encode())
|
| 48 |
+
os.fsync(fd)
|
| 49 |
+
finally:
|
| 50 |
+
os.close(fd)
|
| 51 |
+
|
| 52 |
+
|
| 53 |
+
def cmd_init(args) -> int:
|
| 54 |
+
data_dir = Path(args.data_dir).expanduser()
|
| 55 |
+
data_dir.mkdir(parents=True, exist_ok=True)
|
| 56 |
+
os.chmod(data_dir, 0o700)
|
| 57 |
+
_warn_fs(data_dir)
|
| 58 |
+
db_path = data_dir / DB_NAME
|
| 59 |
+
token_path = data_dir / OPERATOR_TOKEN_FILE
|
| 60 |
+
existed = db_path.exists()
|
| 61 |
+
conn = dbmod.connect(db_path)
|
| 62 |
+
try:
|
| 63 |
+
before = dbmod.schema_version(conn)
|
| 64 |
+
try:
|
| 65 |
+
version = dbmod.migrate(conn, db_path)
|
| 66 |
+
except dbmod.SchemaError as exc:
|
| 67 |
+
print(f"error: {exc}", file=sys.stderr)
|
| 68 |
+
return 1
|
| 69 |
+
if store.admin_exists(conn):
|
| 70 |
+
note = "migrated" if before != version else "unchanged"
|
| 71 |
+
print(f"Board already initialized in {data_dir} (schema v{version}, {note}).")
|
| 72 |
+
tmp_path = data_dir / (OPERATOR_TOKEN_FILE + ".tmp")
|
| 73 |
+
if not token_path.exists() and tmp_path.exists():
|
| 74 |
+
# An earlier init committed but stopped before moving the file into place.
|
| 75 |
+
try:
|
| 76 |
+
store.authenticate(conn, tmp_path.read_text().strip(), time.time())
|
| 77 |
+
os.link(tmp_path, token_path)
|
| 78 |
+
tmp_path.unlink()
|
| 79 |
+
except Exception:
|
| 80 |
+
pass
|
| 81 |
+
if token_path.exists():
|
| 82 |
+
print(f"Operator token file: {token_path}")
|
| 83 |
+
else:
|
| 84 |
+
print("Operator token file is missing; existing credentials were left untouched. "
|
| 85 |
+
"Run `python -m msgboard rotate-operator-token --data-dir ...` to issue a new one.")
|
| 86 |
+
return 0
|
| 87 |
+
if token_path.exists():
|
| 88 |
+
print(f"error: {token_path} exists but the database has no operator; refusing to overwrite it. "
|
| 89 |
+
"Move the file away if it is stale.", file=sys.stderr)
|
| 90 |
+
return 1
|
| 91 |
+
# Write the secret to a temp file first, then commit, then move it into place.
|
| 92 |
+
tmp_path = data_dir / (OPERATOR_TOKEN_FILE + ".tmp")
|
| 93 |
+
if tmp_path.exists():
|
| 94 |
+
tmp_path.unlink()
|
| 95 |
+
conn.execute("BEGIN IMMEDIATE")
|
| 96 |
+
try:
|
| 97 |
+
_, token = store.create_operator(conn, time.time())
|
| 98 |
+
_write_secret_file(tmp_path, token)
|
| 99 |
+
conn.execute("COMMIT")
|
| 100 |
+
except BaseException:
|
| 101 |
+
conn.execute("ROLLBACK")
|
| 102 |
+
if tmp_path.exists():
|
| 103 |
+
tmp_path.unlink()
|
| 104 |
+
raise
|
| 105 |
+
os.link(tmp_path, token_path) # fails rather than overwrite
|
| 106 |
+
tmp_path.unlink()
|
| 107 |
+
finally:
|
| 108 |
+
conn.close()
|
| 109 |
+
print(f"{'Initialized schema in existing' if existed else 'Created'} board in {data_dir} (schema v{version}).")
|
| 110 |
+
print(f"Operator token saved to {token_path} (mode 0600). Keep it secret; open the dashboard with it to create invite links.")
|
| 111 |
+
return 0
|
| 112 |
+
|
| 113 |
+
|
| 114 |
+
def cmd_rotate_operator(args) -> int:
|
| 115 |
+
data_dir = Path(args.data_dir).expanduser()
|
| 116 |
+
db_path = data_dir / DB_NAME
|
| 117 |
+
token_path = data_dir / OPERATOR_TOKEN_FILE
|
| 118 |
+
if not db_path.exists():
|
| 119 |
+
print(f"error: no board in {data_dir}", file=sys.stderr)
|
| 120 |
+
return 1
|
| 121 |
+
conn = dbmod.connect(db_path)
|
| 122 |
+
try:
|
| 123 |
+
if dbmod.schema_version(conn) != dbmod.SCHEMA_VERSION:
|
| 124 |
+
print("error: run init first to migrate the schema", file=sys.stderr)
|
| 125 |
+
return 1
|
| 126 |
+
new_path = data_dir / (OPERATOR_TOKEN_FILE + ".new")
|
| 127 |
+
if new_path.exists():
|
| 128 |
+
new_path.unlink()
|
| 129 |
+
conn.execute("BEGIN IMMEDIATE")
|
| 130 |
+
try:
|
| 131 |
+
token = store.rotate_operator(conn, time.time())
|
| 132 |
+
_write_secret_file(new_path, token)
|
| 133 |
+
conn.execute("COMMIT")
|
| 134 |
+
except BaseException:
|
| 135 |
+
conn.execute("ROLLBACK")
|
| 136 |
+
if new_path.exists():
|
| 137 |
+
new_path.unlink()
|
| 138 |
+
raise
|
| 139 |
+
os.replace(new_path, token_path)
|
| 140 |
+
finally:
|
| 141 |
+
conn.close()
|
| 142 |
+
print(f"Operator token rotated; the old one is revoked. New token saved to {token_path}.")
|
| 143 |
+
return 0
|
| 144 |
+
|
| 145 |
+
|
| 146 |
+
def cmd_backup(args) -> int:
|
| 147 |
+
db_path = Path(args.data_dir).expanduser() / DB_NAME
|
| 148 |
+
if not db_path.exists():
|
| 149 |
+
print(f"error: no board at {db_path}", file=sys.stderr)
|
| 150 |
+
return 1
|
| 151 |
+
conn = dbmod.connect(db_path)
|
| 152 |
+
try:
|
| 153 |
+
dbmod.backup(conn, args.out)
|
| 154 |
+
finally:
|
| 155 |
+
conn.close()
|
| 156 |
+
print(f"Backup written to {args.out}. Attachments live in {db_path.parent / 'blobs'}; copy them too.")
|
| 157 |
+
return 0
|
| 158 |
+
|
| 159 |
+
|
| 160 |
+
def _is_loopback(host: str) -> bool:
|
| 161 |
+
if host == "localhost":
|
| 162 |
+
return True
|
| 163 |
+
try:
|
| 164 |
+
return ipaddress.ip_address(host).is_loopback
|
| 165 |
+
except ValueError:
|
| 166 |
+
return False
|
| 167 |
+
|
| 168 |
+
|
| 169 |
+
def cmd_serve(args) -> int:
|
| 170 |
+
from .server import AlreadyRunning, Board, NotInitialized
|
| 171 |
+
|
| 172 |
+
data_dir = Path(args.data_dir).expanduser()
|
| 173 |
+
_warn_fs(data_dir)
|
| 174 |
+
replicator = None
|
| 175 |
+
if args.replica:
|
| 176 |
+
from .replicate import Replicator, restore, store_from_url
|
| 177 |
+
|
| 178 |
+
store = store_from_url(args.replica)
|
| 179 |
+
if not (data_dir / DB_NAME).exists():
|
| 180 |
+
manifest = restore(store, data_dir)
|
| 181 |
+
print(f"Restored {manifest['snapshot']} from replica." if manifest
|
| 182 |
+
else "Replica is empty; nothing to restore.", file=sys.stderr)
|
| 183 |
+
replicator = Replicator(store, data_dir, interval=args.snapshot_interval)
|
| 184 |
+
try:
|
| 185 |
+
board = Board(data_dir, replicator=replicator)
|
| 186 |
+
except (NotInitialized, AlreadyRunning, dbmod.SchemaError) as exc:
|
| 187 |
+
print(f"error: {exc}", file=sys.stderr)
|
| 188 |
+
return 1
|
| 189 |
+
if not _is_loopback(args.host):
|
| 190 |
+
print(f"warning: listening on {args.host}. Tokens travel in plain HTTP; "
|
| 191 |
+
"put HTTPS in front of this outside isolated local development.", file=sys.stderr)
|
| 192 |
+
|
| 193 |
+
run_board(board, args.host, args.port, args.log_level)
|
| 194 |
+
return 0
|
| 195 |
+
|
| 196 |
+
|
| 197 |
+
def run_board(board, host: str, port: int, log_level: str = "info", app=None) -> None:
|
| 198 |
+
"""Serve `board` in the foreground until SIGINT/SIGTERM, then close it (final snapshot included)."""
|
| 199 |
+
import uvicorn
|
| 200 |
+
|
| 201 |
+
from .server import create_app
|
| 202 |
+
|
| 203 |
+
class BoardServer(uvicorn.Server):
|
| 204 |
+
def handle_exit(self, sig, frame):
|
| 205 |
+
board.begin_shutdown()
|
| 206 |
+
super().handle_exit(sig, frame)
|
| 207 |
+
|
| 208 |
+
config = uvicorn.Config(
|
| 209 |
+
app or create_app(board), host=host, port=port, log_level=log_level,
|
| 210 |
+
timeout_graceful_shutdown=5, server_header=False,
|
| 211 |
+
)
|
| 212 |
+
|
| 213 |
+
# uvicorn re-raises the signal it caught after shutting down. With the
|
| 214 |
+
# default SIGTERM action that kills the process before board.close() runs,
|
| 215 |
+
# so the final snapshot would be lost. Turn it into a normal exit instead.
|
| 216 |
+
def _exit(signum, frame):
|
| 217 |
+
raise SystemExit(0)
|
| 218 |
+
|
| 219 |
+
signal.signal(signal.SIGTERM, _exit)
|
| 220 |
+
signal.signal(signal.SIGINT, _exit)
|
| 221 |
+
try:
|
| 222 |
+
BoardServer(config).run()
|
| 223 |
+
finally:
|
| 224 |
+
board.close()
|
| 225 |
+
print("Board stopped cleanly.", file=sys.stderr)
|
| 226 |
+
|
| 227 |
+
|
| 228 |
+
def cmd_restore(args) -> int:
|
| 229 |
+
from .replicate import restore, store_from_url
|
| 230 |
+
|
| 231 |
+
manifest = restore(store_from_url(args.replica), Path(args.data_dir).expanduser())
|
| 232 |
+
if manifest is None:
|
| 233 |
+
print("error: replica has no snapshot", file=sys.stderr)
|
| 234 |
+
return 1
|
| 235 |
+
print(f"Restored {manifest['snapshot']} ({manifest['size']} bytes) into {args.data_dir}.")
|
| 236 |
+
return 0
|
| 237 |
+
|
| 238 |
+
|
| 239 |
+
def cmd_demo(args) -> int:
|
| 240 |
+
from .demo import run_demo
|
| 241 |
+
|
| 242 |
+
return run_demo(args.data_dir, args.port)
|
| 243 |
+
|
| 244 |
+
|
| 245 |
+
def main(argv: list[str] | None = None) -> int:
|
| 246 |
+
parser = argparse.ArgumentParser(prog="python -m msgboard", description=__doc__)
|
| 247 |
+
sub = parser.add_subparsers(dest="command", required=True)
|
| 248 |
+
|
| 249 |
+
p = sub.add_parser("init", help="create or migrate a board; never overwrites existing credentials")
|
| 250 |
+
p.add_argument("--data-dir", required=True)
|
| 251 |
+
p.set_defaults(func=cmd_init)
|
| 252 |
+
|
| 253 |
+
p = sub.add_parser("serve", help="run the HTTP server in the foreground (Ctrl-C to stop)")
|
| 254 |
+
p.add_argument("--data-dir", required=True)
|
| 255 |
+
p.add_argument("--host", default="127.0.0.1")
|
| 256 |
+
p.add_argument("--port", type=int, default=DEFAULT_PORT)
|
| 257 |
+
p.add_argument("--log-level", default="info")
|
| 258 |
+
p.add_argument("--replica", default=None,
|
| 259 |
+
help="snapshot target: hf://buckets/<ns>/<bucket>[/<prefix>] or a directory; "
|
| 260 |
+
"an empty data dir is restored from it first")
|
| 261 |
+
p.add_argument("--snapshot-interval", type=float, default=10.0, help="seconds between snapshots")
|
| 262 |
+
p.set_defaults(func=cmd_serve)
|
| 263 |
+
|
| 264 |
+
p = sub.add_parser("restore", help="restore an empty data dir from a replica")
|
| 265 |
+
p.add_argument("--data-dir", required=True)
|
| 266 |
+
p.add_argument("--replica", required=True)
|
| 267 |
+
p.set_defaults(func=cmd_restore)
|
| 268 |
+
|
| 269 |
+
p = sub.add_parser("backup", help="write a consistent SQLite snapshot")
|
| 270 |
+
p.add_argument("--data-dir", required=True)
|
| 271 |
+
p.add_argument("--out", required=True)
|
| 272 |
+
p.set_defaults(func=cmd_backup)
|
| 273 |
+
|
| 274 |
+
p = sub.add_parser("rotate-operator-token", help="replace the operator token (local file access required)")
|
| 275 |
+
p.add_argument("--data-dir", required=True)
|
| 276 |
+
p.set_defaults(func=cmd_rotate_operator)
|
| 277 |
+
|
| 278 |
+
p = sub.add_parser("demo", help="start a throwaway board and run two agents against it")
|
| 279 |
+
p.add_argument("--data-dir", default=None, help="defaults to a new temporary directory")
|
| 280 |
+
p.add_argument("--port", type=int, default=0, help="0 picks a free port")
|
| 281 |
+
p.set_defaults(func=cmd_demo)
|
| 282 |
+
|
| 283 |
+
args = parser.parse_args(argv)
|
| 284 |
+
return args.func(args)
|
src/msgboard/client.py
ADDED
|
@@ -0,0 +1,428 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Small synchronous Python SDK.
|
| 2 |
+
|
| 3 |
+
Every call that changes state is safe to retry: posts reuse their request ID,
|
| 4 |
+
uploads reuse their upload ID, registration is idempotent and token issue is a
|
| 5 |
+
rotation. The client retries transient failures with jittered backoff and
|
| 6 |
+
honours Retry-After.
|
| 7 |
+
"""
|
| 8 |
+
|
| 9 |
+
from __future__ import annotations
|
| 10 |
+
|
| 11 |
+
import os
|
| 12 |
+
import random
|
| 13 |
+
import time
|
| 14 |
+
import uuid
|
| 15 |
+
from urllib.parse import quote
|
| 16 |
+
from pathlib import Path
|
| 17 |
+
from typing import Any, Iterator
|
| 18 |
+
|
| 19 |
+
import httpx
|
| 20 |
+
|
| 21 |
+
RETRY_STATUS = {429, 502, 503, 504}
|
| 22 |
+
|
| 23 |
+
|
| 24 |
+
class BoardError(Exception):
|
| 25 |
+
def __init__(self, status: int, code: str, message: str):
|
| 26 |
+
super().__init__(f"{status} {code}: {message}")
|
| 27 |
+
self.status = status
|
| 28 |
+
self.code = code
|
| 29 |
+
self.message = message
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
class Client:
|
| 33 |
+
def __init__(self, base_url: str, token: str, *, timeout: float = 10.0,
|
| 34 |
+
max_retries: int = 5, backoff: float = 0.25, transport: httpx.BaseTransport | None = None,
|
| 35 |
+
headers: dict | None = None, token_header: str | None = None):
|
| 36 |
+
"""`token_header="X-Msgboard-Token"` sends the board token there, leaving
|
| 37 |
+
Authorization free for a proxy login passed in `headers` (e.g. a private HF Space)."""
|
| 38 |
+
self.base_url = base_url.rstrip("/")
|
| 39 |
+
self.token = token
|
| 40 |
+
self.timeout = timeout
|
| 41 |
+
self.max_retries = max_retries
|
| 42 |
+
self.backoff = backoff
|
| 43 |
+
self._http = httpx.Client(
|
| 44 |
+
base_url=self.base_url,
|
| 45 |
+
headers={**(headers or {}),
|
| 46 |
+
**(({token_header: token} if token_header else {"Authorization": f"Bearer {token}"})
|
| 47 |
+
if token else {})},
|
| 48 |
+
timeout=timeout,
|
| 49 |
+
transport=transport,
|
| 50 |
+
)
|
| 51 |
+
|
| 52 |
+
@classmethod
|
| 53 |
+
def from_env(cls, **kwargs) -> "Client":
|
| 54 |
+
"""Build a client from MSGBOARD_URL and MSGBOARD_TOKEN (what a launcher injects).
|
| 55 |
+
For a private HF Space also set MSGBOARD_HF_TOKEN."""
|
| 56 |
+
if os.environ.get("MSGBOARD_HF_TOKEN"):
|
| 57 |
+
kwargs.setdefault("headers", {"Authorization": f"Bearer {os.environ['MSGBOARD_HF_TOKEN']}"})
|
| 58 |
+
kwargs.setdefault("token_header", "X-Msgboard-Token")
|
| 59 |
+
return cls(os.environ["MSGBOARD_URL"], os.environ["MSGBOARD_TOKEN"], **kwargs)
|
| 60 |
+
|
| 61 |
+
def close(self) -> None:
|
| 62 |
+
self._http.close()
|
| 63 |
+
|
| 64 |
+
def __enter__(self):
|
| 65 |
+
return self
|
| 66 |
+
|
| 67 |
+
def __exit__(self, *exc):
|
| 68 |
+
self.close()
|
| 69 |
+
|
| 70 |
+
# ---------------------------------------------------------------- transport
|
| 71 |
+
|
| 72 |
+
def _request(self, method: str, path: str, *, timeout: float | None = None, **kwargs) -> httpx.Response:
|
| 73 |
+
attempt = 0
|
| 74 |
+
while True:
|
| 75 |
+
try:
|
| 76 |
+
resp = self._http.request(method, "/v1" + path, timeout=timeout or self.timeout, **kwargs)
|
| 77 |
+
except httpx.TransportError:
|
| 78 |
+
# Includes timeouts and dropped connections: the server may or may
|
| 79 |
+
# not have committed. Retrying with the same IDs is safe.
|
| 80 |
+
if attempt >= self.max_retries:
|
| 81 |
+
raise
|
| 82 |
+
self._sleep(attempt, None)
|
| 83 |
+
attempt += 1
|
| 84 |
+
continue
|
| 85 |
+
if resp.status_code in RETRY_STATUS and attempt < self.max_retries:
|
| 86 |
+
self._sleep(attempt, resp.headers.get("retry-after"))
|
| 87 |
+
attempt += 1
|
| 88 |
+
continue
|
| 89 |
+
if resp.status_code >= 400:
|
| 90 |
+
try:
|
| 91 |
+
err = resp.json()["error"]
|
| 92 |
+
raise BoardError(resp.status_code, err["code"], err["message"])
|
| 93 |
+
except (ValueError, KeyError, TypeError):
|
| 94 |
+
raise BoardError(resp.status_code, "http_error", resp.text[:200])
|
| 95 |
+
return resp
|
| 96 |
+
|
| 97 |
+
def _sleep(self, attempt: int, retry_after: str | None) -> None:
|
| 98 |
+
delay = self.backoff * (2 ** attempt) * (0.5 + random.random())
|
| 99 |
+
if retry_after:
|
| 100 |
+
try:
|
| 101 |
+
delay = max(delay, float(retry_after))
|
| 102 |
+
except ValueError:
|
| 103 |
+
pass
|
| 104 |
+
time.sleep(min(delay, 30.0))
|
| 105 |
+
|
| 106 |
+
# ---------------------------------------------------------------- identity
|
| 107 |
+
|
| 108 |
+
@classmethod
|
| 109 |
+
def join(cls, base_url: str, code: str, name: str | None = None, display_name: str | None = None, **kwargs) -> "Client":
|
| 110 |
+
"""Redeem an invite code and return a client using the new identity's token.
|
| 111 |
+
|
| 112 |
+
For agent invites, joining again with the same name returns the same agent with a fresh
|
| 113 |
+
token (the old one stops working), so a launcher can call this every time it starts an agent.
|
| 114 |
+
The token is in `client.token` and the identity in `client.identity`. Tokens do not expire.
|
| 115 |
+
"""
|
| 116 |
+
body = {"code": code}
|
| 117 |
+
if name is not None: # a named agent invite already decides the name
|
| 118 |
+
body["name"] = name
|
| 119 |
+
if display_name is not None:
|
| 120 |
+
body["display_name"] = display_name
|
| 121 |
+
bootstrap = cls(base_url, "", **kwargs)
|
| 122 |
+
try:
|
| 123 |
+
result = bootstrap._request("POST", "/join", json=body).json()
|
| 124 |
+
finally:
|
| 125 |
+
bootstrap.close()
|
| 126 |
+
client = cls(base_url, result["token"], **kwargs)
|
| 127 |
+
client.identity = result["principal"]
|
| 128 |
+
return client
|
| 129 |
+
|
| 130 |
+
def me(self) -> dict:
|
| 131 |
+
return self._request("GET", "/me").json()
|
| 132 |
+
|
| 133 |
+
def people(self) -> list[dict]:
|
| 134 |
+
"""Everyone you can see on the board: people, agents and service accounts."""
|
| 135 |
+
return self._request("GET", "/principals").json()["principals"]
|
| 136 |
+
|
| 137 |
+
def presence(self) -> list[dict]:
|
| 138 |
+
"""Agent listening state: admins see all, others see their own agents (and themselves if agents)."""
|
| 139 |
+
return self._request("GET", "/presence").json()["principals"]
|
| 140 |
+
|
| 141 |
+
def update_identity(self, principal_id: int, **changes) -> dict:
|
| 142 |
+
"""display_name; for admins or an agent's owner also role, disabled, can_create_channels."""
|
| 143 |
+
return self._request("PATCH", f"/principals/{principal_id}", json=changes).json()
|
| 144 |
+
|
| 145 |
+
def create_invite(self, kind: str = "agent", *, role: str = "member", identity_type: str = "agent",
|
| 146 |
+
can_create_channels: bool = False, max_uses: int | None = None,
|
| 147 |
+
expires_in_seconds: float | None = None, label: str | None = None,
|
| 148 |
+
channels: list[int | str] = (), agent_name: str | None = None,
|
| 149 |
+
join_open_channels: bool = False) -> dict:
|
| 150 |
+
"""kind="agent" (members and admins): agents you own. kind="human" (admins): people with `role`.
|
| 151 |
+
`channels` (ids or names) are joined automatically; that is how guest agents get their channels.
|
| 152 |
+
`agent_name` makes a link for one named agent (its handle is derived from the name).
|
| 153 |
+
The returned `code` is shown only once."""
|
| 154 |
+
body = {"kind": kind, "role": role, "max_uses": max_uses, "expires_in_seconds": expires_in_seconds,
|
| 155 |
+
"label": label, "channels": list(channels)}
|
| 156 |
+
if kind == "agent":
|
| 157 |
+
body.update(identity_type=identity_type, can_create_channels=can_create_channels)
|
| 158 |
+
if agent_name:
|
| 159 |
+
body["agent_name"] = agent_name
|
| 160 |
+
if join_open_channels:
|
| 161 |
+
body["join_open_channels"] = True
|
| 162 |
+
return self._request("POST", "/invites", json=body).json()
|
| 163 |
+
|
| 164 |
+
def invites(self) -> list[dict]:
|
| 165 |
+
return self._request("GET", "/invites").json()["invites"]
|
| 166 |
+
|
| 167 |
+
def revoke_invite(self, invite_id: int) -> dict:
|
| 168 |
+
return self._request("DELETE", f"/invites/{invite_id}").json()
|
| 169 |
+
|
| 170 |
+
def mark_all_read(self) -> dict:
|
| 171 |
+
"""Mark every channel, DM and thread you can read as read."""
|
| 172 |
+
return self._request("POST", "/me/read-all").json()
|
| 173 |
+
|
| 174 |
+
def suspend(self, principal_id: int, reason: str | None = None) -> dict:
|
| 175 |
+
"""Admins only: block an account (a person's agents go with them); the reason is shown to them."""
|
| 176 |
+
return self._request("POST", f"/principals/{principal_id}/suspend", json={"reason": reason}).json()
|
| 177 |
+
|
| 178 |
+
def unsuspend(self, principal_id: int) -> dict:
|
| 179 |
+
return self._request("POST", f"/principals/{principal_id}/unsuspend").json()
|
| 180 |
+
|
| 181 |
+
def suspended(self) -> list[dict]:
|
| 182 |
+
"""Admins only: who is suspended, by whom, when and why."""
|
| 183 |
+
return self._request("GET", "/suspended").json()["suspended"]
|
| 184 |
+
|
| 185 |
+
def rotate_token(self, principal_id: int) -> dict:
|
| 186 |
+
"""New token for an agent you own (or any agent, for admins); the old one stops working."""
|
| 187 |
+
return self._request("POST", f"/principals/{principal_id}/token").json()
|
| 188 |
+
|
| 189 |
+
def revoke_token(self, token_id: str) -> dict:
|
| 190 |
+
return self._request("DELETE", f"/tokens/{token_id}").json()
|
| 191 |
+
|
| 192 |
+
def settings(self) -> dict:
|
| 193 |
+
return self._request("GET", "/settings").json()
|
| 194 |
+
|
| 195 |
+
def update_settings(self, **changes) -> dict:
|
| 196 |
+
return self._request("PATCH", "/settings", json=changes).json()
|
| 197 |
+
|
| 198 |
+
# ---------------------------------------------------------------- channels
|
| 199 |
+
|
| 200 |
+
def list_channels(self) -> list[dict]:
|
| 201 |
+
return self._request("GET", "/channels").json()["channels"]
|
| 202 |
+
|
| 203 |
+
def channel_id(self, name: str) -> int:
|
| 204 |
+
channels = self._request("GET", "/channels", params={"name": name}).json()["channels"]
|
| 205 |
+
if not channels:
|
| 206 |
+
raise BoardError(404, "not_found", f"channel {name!r} not found or not accessible")
|
| 207 |
+
return channels[0]["id"]
|
| 208 |
+
|
| 209 |
+
def create_channel(self, name: str, description: str | None = None, private: bool = False) -> dict:
|
| 210 |
+
return self._request("POST", "/channels",
|
| 211 |
+
json={"name": name, "description": description, "private": private}).json()
|
| 212 |
+
|
| 213 |
+
def channel_info(self, channel_id: int) -> dict:
|
| 214 |
+
"""Creator, counts, explicit members and who posts in a channel."""
|
| 215 |
+
return self._request("GET", f"/channels/{channel_id}/info").json()
|
| 216 |
+
|
| 217 |
+
def update_channel(self, channel_id: int, **changes) -> dict:
|
| 218 |
+
"""Change `name`, `description` (admins and the creator) or `archived` (admins)."""
|
| 219 |
+
return self._request("PATCH", f"/channels/{channel_id}", json=changes).json()
|
| 220 |
+
|
| 221 |
+
def delete_channel(self, channel_id: int, confirm: str) -> dict:
|
| 222 |
+
"""Delete a channel and everything in it, for good (admins). `confirm` is the channel's name."""
|
| 223 |
+
return self._request("DELETE", f"/channels/{channel_id}", params={"confirm": confirm}).json()
|
| 224 |
+
|
| 225 |
+
def members(self, channel_id: int) -> list[dict]:
|
| 226 |
+
return self._request("GET", f"/channels/{channel_id}/members").json()["members"]
|
| 227 |
+
|
| 228 |
+
def add_member(self, channel_id: int, principal_id: int) -> list[dict]:
|
| 229 |
+
return self._request("POST", f"/channels/{channel_id}/members",
|
| 230 |
+
json={"principal_id": principal_id}).json()["members"]
|
| 231 |
+
|
| 232 |
+
def remove_member(self, channel_id: int, principal_id: int) -> dict:
|
| 233 |
+
return self._request("DELETE", f"/channels/{channel_id}/members/{principal_id}").json()
|
| 234 |
+
|
| 235 |
+
def mark_read(self, channel_id: int, seq: int) -> dict:
|
| 236 |
+
"""Move your read position forward; returns the channel's remaining unread counts."""
|
| 237 |
+
return self._request("POST", f"/channels/{channel_id}/read", json={"seq": seq}).json()
|
| 238 |
+
|
| 239 |
+
def mark_thread_read(self, channel_id: int, thread_id: int, seq: int) -> dict:
|
| 240 |
+
"""Mark a thread read up to `seq`; things for you in a thread stay unread until you do."""
|
| 241 |
+
return self._request("POST", f"/channels/{channel_id}/threads/{thread_id}/read", json={"seq": seq}).json()
|
| 242 |
+
|
| 243 |
+
def activity(self, limit: int = 50) -> list[dict]:
|
| 244 |
+
"""Messages for you, newest first: mentions, replies in your threads, DMs (each with `reason`, `unread`)."""
|
| 245 |
+
return self._request("GET", "/activity", params={"limit": limit}).json()["messages"]
|
| 246 |
+
|
| 247 |
+
def activity_page(self, after: int | None = None, *, wait: float = 0, channels: list[int] | str = (),
|
| 248 |
+
limit: int = 100) -> dict:
|
| 249 |
+
"""Everything for you since cursor `after`, oldest first: mentions, replies in your threads, DMs,
|
| 250 |
+
and every message in `channels`. Without `after`, recent items and a `next_cursor` meaning "now".
|
| 251 |
+
With `wait`, long-poll until something arrives. Persist `next_cursor` to resume after a restart."""
|
| 252 |
+
params: dict[str, Any] = {"limit": limit}
|
| 253 |
+
if after is not None:
|
| 254 |
+
params["after"] = after
|
| 255 |
+
params["wait"] = wait
|
| 256 |
+
if channels: # channel ids, or "joined" for every channel you are a member of
|
| 257 |
+
params["channels"] = channels if isinstance(channels, str) else ",".join(str(c) for c in channels)
|
| 258 |
+
return self._request("GET", "/activity", params=params, timeout=self.timeout + wait).json()
|
| 259 |
+
|
| 260 |
+
def follow_activity(self, after: int | None = None, *, channels: list[int] | str = "joined", wait: float = 25) -> Iterator[dict]:
|
| 261 |
+
"""Yield messages meant for you forever (see `activity_page`). Persist message['id'] as your cursor.
|
| 262 |
+
With no cursor, starts from now."""
|
| 263 |
+
cursor = after if after is not None else self.activity_page(limit=1, channels=channels)["next_cursor"]
|
| 264 |
+
while True:
|
| 265 |
+
page = self.activity_page(cursor, wait=wait, channels=channels)
|
| 266 |
+
for message in page["messages"]:
|
| 267 |
+
yield message
|
| 268 |
+
cursor = page["next_cursor"]
|
| 269 |
+
|
| 270 |
+
def search(self, q: str = "", *, channel: int | None = None, sender: int | None = None, after: str | None = None,
|
| 271 |
+
before: str | None = None, sort: str = "newest", limit: int = 20, offset: int = 0) -> dict:
|
| 272 |
+
"""Search messages you can read. `q` takes words, "phrases" and filters such as from:@name in:#channel
|
| 273 |
+
after:2026-10-01 has:file is:reply by:agent. Returns messages (with `snippet`, `channel`), matching
|
| 274 |
+
channels and people, `total`, and `next_offset` for the next page."""
|
| 275 |
+
params = {"q": q, "sort": sort, "limit": limit, "offset": offset}
|
| 276 |
+
params.update({k: v for k, v in {"channel": channel, "sender": sender, "after": after, "before": before}.items() if v is not None})
|
| 277 |
+
return self._request("GET", "/search", params=params).json()
|
| 278 |
+
|
| 279 |
+
def doc(self, channel_id: int) -> dict:
|
| 280 |
+
"""The channel's shared Markdown doc: {"body", "version", "updated_by", "updated_at", "can_edit"}."""
|
| 281 |
+
return self._request("GET", f"/channels/{channel_id}/doc").json()
|
| 282 |
+
|
| 283 |
+
def save_doc(self, channel_id: int, body: str, base_version: int) -> dict:
|
| 284 |
+
"""Replace the doc. `base_version` is the version you edited; if someone saved since, this raises
|
| 285 |
+
BoardError 409 (code doc_conflict): read the doc again, merge, and save with the new version."""
|
| 286 |
+
return self._request("PUT", f"/channels/{channel_id}/doc", json={"body": body, "base_version": base_version}).json()
|
| 287 |
+
|
| 288 |
+
def set_runtime(self, harness: str | None = None, model: str | None = None, *, tools: list[str] = (),
|
| 289 |
+
host: str | None = None, **more) -> dict:
|
| 290 |
+
"""Say how and where you run (shown on your profile). Call again whenever it changes."""
|
| 291 |
+
runtime = {"harness": harness, "model": model, "tools": list(tools) or None, "host": host, **more}
|
| 292 |
+
me = getattr(self, "identity", None) or self.me() # clients made from a token have no `identity` yet
|
| 293 |
+
return self.update_identity(me["id"],
|
| 294 |
+
runtime={k: v for k, v in runtime.items() if v is not None})
|
| 295 |
+
|
| 296 |
+
# ---------------------------------------------------------------- storage folders
|
| 297 |
+
|
| 298 |
+
def _my_handle(self) -> str:
|
| 299 |
+
me = getattr(self, "identity", None) or self.me()
|
| 300 |
+
return me["external_id"]
|
| 301 |
+
|
| 302 |
+
def storage_put(self, path: str, data: bytes | str | Path, *, handle: str | None = None,
|
| 303 |
+
content_type: str | None = None) -> dict:
|
| 304 |
+
"""Upload a file to your storage folder (or `handle`'s, if you may write there). Folders are created
|
| 305 |
+
as needed. Returns {path, size, sha256, url}; put `url` in a message as  to show an image."""
|
| 306 |
+
if isinstance(data, (str, Path)):
|
| 307 |
+
data = Path(data).read_bytes()
|
| 308 |
+
import hashlib
|
| 309 |
+
headers = {"X-Checksum-Sha256": hashlib.sha256(data).hexdigest()}
|
| 310 |
+
if content_type:
|
| 311 |
+
headers["Content-Type"] = content_type
|
| 312 |
+
return self._request("PUT", f"/storage/{quote(handle or self._my_handle(), safe=':')}/{quote(path.strip('/'))}",
|
| 313 |
+
content=data, headers=headers, timeout=max(self.timeout, 300)).json()
|
| 314 |
+
|
| 315 |
+
def storage_get(self, path: str, *, handle: str | None = None) -> bytes:
|
| 316 |
+
return self._request("GET", f"/storage/{quote(handle or self._my_handle(), safe=':')}/{quote(path.strip('/'))}",
|
| 317 |
+
timeout=max(self.timeout, 300)).content
|
| 318 |
+
|
| 319 |
+
def storage_list(self, prefix: str = "", *, handle: str | None = None) -> dict:
|
| 320 |
+
"""Files in a storage folder (optionally under `prefix`), with used and quota bytes."""
|
| 321 |
+
return self._request("GET", f"/storage/{quote(handle or self._my_handle(), safe=':')}", params={"prefix": prefix}).json()
|
| 322 |
+
|
| 323 |
+
def storage_delete(self, path: str, *, handle: str | None = None) -> dict:
|
| 324 |
+
"""Delete a file, or a folder with everything in it."""
|
| 325 |
+
return self._request("DELETE", f"/storage/{quote(handle or self._my_handle(), safe=':')}/{quote(path.strip('/'))}").json()
|
| 326 |
+
|
| 327 |
+
def guide(self) -> str:
|
| 328 |
+
"""How to use this board, as markdown, ending with the workspace guidelines its admins wrote."""
|
| 329 |
+
return self._request("GET", "/guide").text
|
| 330 |
+
|
| 331 |
+
def analytics(self, days: int = 7, *, start: float | None = None, end: float | None = None,
|
| 332 |
+
top_k: int = 10) -> dict:
|
| 333 |
+
"""Analytics for admins/readers. Optional Unix-second bounds select [start, end)."""
|
| 334 |
+
params = {"days": days, "top_k": top_k}
|
| 335 |
+
if start is not None:
|
| 336 |
+
params["from"] = start
|
| 337 |
+
if end is not None:
|
| 338 |
+
params["to"] = end
|
| 339 |
+
return self._request("GET", "/analytics", params=params).json()
|
| 340 |
+
|
| 341 |
+
def open_dm(self, principal_id: int) -> dict:
|
| 342 |
+
"""The direct-message channel with someone (created on first use); post to its `id`."""
|
| 343 |
+
return self._request("POST", "/dms", json={"with": principal_id}).json()
|
| 344 |
+
|
| 345 |
+
# ---------------------------------------------------------------- messages
|
| 346 |
+
|
| 347 |
+
def post(self, channel_id: int, body: str, *, reply_to: int | None = None,
|
| 348 |
+
attachments: list[str] | None = None, request_id: str | None = None) -> dict:
|
| 349 |
+
"""Append a message. Pass and persist `request_id` to make retries survive your own restarts."""
|
| 350 |
+
request_id = request_id or uuid.uuid4().hex
|
| 351 |
+
payload: dict[str, Any] = {"body": body}
|
| 352 |
+
if reply_to is not None:
|
| 353 |
+
payload["reply_to"] = reply_to
|
| 354 |
+
if attachments:
|
| 355 |
+
payload["attachments"] = attachments
|
| 356 |
+
return self._request("POST", f"/channels/{channel_id}/messages", json=payload,
|
| 357 |
+
headers={"Idempotency-Key": request_id}).json()
|
| 358 |
+
|
| 359 |
+
def edit(self, channel_id: int, message_id: int, body: str) -> dict:
|
| 360 |
+
"""Replace the body of one of your own messages (people only; agents get 403). Earlier versions are kept."""
|
| 361 |
+
return self._request("PATCH", f"/channels/{channel_id}/messages/{message_id}", json={"body": body}).json()
|
| 362 |
+
|
| 363 |
+
def react(self, channel_id: int, message_id: int, emoji: str) -> dict:
|
| 364 |
+
"""Add your reaction (an emoji like "👍" or "✅"). Safe to repeat. Returns the message with `reactions`."""
|
| 365 |
+
return self._request("PUT", f"/channels/{channel_id}/messages/{message_id}/reactions/{quote(emoji, safe='')}").json()
|
| 366 |
+
|
| 367 |
+
def unreact(self, channel_id: int, message_id: int, emoji: str) -> dict:
|
| 368 |
+
"""Remove your reaction. Safe to repeat."""
|
| 369 |
+
return self._request("DELETE", f"/channels/{channel_id}/messages/{message_id}/reactions/{quote(emoji, safe='')}").json()
|
| 370 |
+
|
| 371 |
+
def delete_message(self, channel_id: int, message_id: int) -> dict:
|
| 372 |
+
"""Delete a message for good (admins); a thread's first message takes its replies with it."""
|
| 373 |
+
return self._request("DELETE", f"/channels/{channel_id}/messages/{message_id}").json()
|
| 374 |
+
|
| 375 |
+
def revisions(self, channel_id: int, message_id: int) -> list[dict]:
|
| 376 |
+
"""Every version of a message, oldest first (the last is the current body). Edits never lose history."""
|
| 377 |
+
return self._request("GET", f"/channels/{channel_id}/messages/{message_id}/revisions").json()["versions"]
|
| 378 |
+
|
| 379 |
+
def read(self, channel_id: int, after: int = 0, limit: int = 100, wait: float = 0) -> dict:
|
| 380 |
+
"""One page of history after cursor `after`; with `wait`, long-poll for new messages."""
|
| 381 |
+
return self._request("GET", f"/channels/{channel_id}/messages",
|
| 382 |
+
params={"after": after, "limit": limit, "wait": wait},
|
| 383 |
+
timeout=self.timeout + wait).json()
|
| 384 |
+
|
| 385 |
+
def follow(self, channel_id: int, after: int = 0, wait: float = 25) -> Iterator[dict]:
|
| 386 |
+
"""Yield messages forever, long-polling between pages. Persist message['seq'] as your cursor."""
|
| 387 |
+
cursor = after
|
| 388 |
+
while True:
|
| 389 |
+
page = self.read(channel_id, after=cursor, wait=wait)
|
| 390 |
+
for message in page["messages"]:
|
| 391 |
+
yield message
|
| 392 |
+
cursor = page["next_cursor"]
|
| 393 |
+
|
| 394 |
+
# ------------------------------------------------------------- attachments
|
| 395 |
+
|
| 396 |
+
def upload(self, channel_id: int, data: bytes | str | Path, *, upload_id: str | None = None,
|
| 397 |
+
filename: str | None = None, content_type: str = "application/octet-stream") -> dict:
|
| 398 |
+
if isinstance(data, (str, Path)):
|
| 399 |
+
path = Path(data)
|
| 400 |
+
filename = filename or path.name
|
| 401 |
+
data = path.read_bytes()
|
| 402 |
+
import hashlib
|
| 403 |
+
|
| 404 |
+
headers = {"Content-Type": content_type, "X-Checksum-Sha256": hashlib.sha256(data).hexdigest()}
|
| 405 |
+
if filename:
|
| 406 |
+
headers["X-Filename"] = filename
|
| 407 |
+
upload_id = upload_id or uuid.uuid4().hex
|
| 408 |
+
return self._request("PUT", f"/channels/{channel_id}/attachments/{upload_id}",
|
| 409 |
+
content=data, headers=headers).json()
|
| 410 |
+
|
| 411 |
+
def set_avatar(self, image: bytes | str | Path, principal_id: int | None = None) -> dict:
|
| 412 |
+
"""Upload a profile picture (PNG, JPEG, GIF or WebP, at most 256 KB). Defaults to your own."""
|
| 413 |
+
data = Path(image).read_bytes() if isinstance(image, (str, Path)) else image
|
| 414 |
+
target = principal_id if principal_id is not None else self.me()["id"]
|
| 415 |
+
return self._request("PUT", f"/principals/{target}/avatar", content=data).json()
|
| 416 |
+
|
| 417 |
+
def remove_avatar(self, principal_id: int | None = None) -> dict:
|
| 418 |
+
"""Go back to the generated pattern."""
|
| 419 |
+
target = principal_id if principal_id is not None else self.me()["id"]
|
| 420 |
+
return self._request("DELETE", f"/principals/{target}/avatar").json()
|
| 421 |
+
|
| 422 |
+
def set_pattern(self, seed: int, principal_id: int | None = None) -> dict:
|
| 423 |
+
"""Choose a generated pattern by seed (0 to 2**32 - 1); removes an uploaded picture."""
|
| 424 |
+
target = principal_id if principal_id is not None else self.me()["id"]
|
| 425 |
+
return self._request("PUT", f"/principals/{target}/pattern", json={"seed": seed}).json()
|
| 426 |
+
|
| 427 |
+
def download(self, attachment_id: str) -> bytes:
|
| 428 |
+
return self._request("GET", f"/attachments/{attachment_id}").content
|
src/msgboard/config.py
ADDED
|
@@ -0,0 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Fixed limits. Kept in one place so they are easy to find and tune."""
|
| 2 |
+
|
| 3 |
+
from dataclasses import dataclass
|
| 4 |
+
|
| 5 |
+
|
| 6 |
+
@dataclass(frozen=True)
|
| 7 |
+
class Limits:
|
| 8 |
+
max_body_bytes: int = 64 * 1024
|
| 9 |
+
max_attachments_per_message: int = 16
|
| 10 |
+
max_page_size: int = 500
|
| 11 |
+
default_page_size: int = 100
|
| 12 |
+
max_page_bytes: int = 4 * 1024 * 1024
|
| 13 |
+
max_wait_seconds: float = 30.0
|
| 14 |
+
max_open_polls: int = 2000
|
| 15 |
+
max_pending_writes: int = 512
|
| 16 |
+
# Reads hold the GIL while building rows, so more threads than CPUs only add contention:
|
| 17 |
+
# on the Space's 2 vCPUs, 2 readers do ~35% more work per second than 4.
|
| 18 |
+
reader_threads: int = 2
|
| 19 |
+
# Load shedding: past this many queued reads, answer 503 + Retry-After at once instead of queueing;
|
| 20 |
+
# a read that waited longer than read_queue_timeout is dropped (its client has given up or will retry).
|
| 21 |
+
max_queued_reads: int = 1000
|
| 22 |
+
read_queue_timeout: float = 15.0
|
| 23 |
+
# A single read that runs longer than this is interrupted, so one bad query cannot hold a reader for minutes.
|
| 24 |
+
max_read_seconds: float = 10.0
|
| 25 |
+
# Activity long-polls: after a wakeup, wait this long so several posts are answered with one query
|
| 26 |
+
# (with hundreds of agents following one busy channel, every post used to wake every poller at once).
|
| 27 |
+
poll_coalesce_seconds: float = 1.5
|
| 28 |
+
# Bearer tokens are checked against the database at most this often (any write that is not a
|
| 29 |
+
# post/reaction/read marker clears the cache, so suspensions and revocations apply at once).
|
| 30 |
+
auth_cache_seconds: float = 30.0
|
| 31 |
+
max_upload_bytes: int = 50 * 1024 * 1024
|
| 32 |
+
max_avatar_bytes: int = 256 * 1024
|
| 33 |
+
max_concurrent_uploads: int = 4
|
| 34 |
+
max_storage_file_bytes: int = 2 * 1024 ** 3 # one file in someone's storage folder
|
| 35 |
+
max_storage_bytes: int = 20 * 1024 ** 3 # one storage folder in total
|
| 36 |
+
storage_cache_bytes: int = 5 * 1024 ** 3 # local copies of bucket files, for downloads
|
| 37 |
+
default_token_ttl: int = 24 * 3600
|
| 38 |
+
max_admin_token_ttl: int = 365 * 24 * 3600
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
LIMITS = Limits()
|
| 42 |
+
DEFAULT_PORT = 8765
|
| 43 |
+
DB_NAME = "board.db"
|
| 44 |
+
OPERATOR_TOKEN_FILE = "operator.token"
|
src/msgboard/dashboard.html
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
src/msgboard/db.py
ADDED
|
@@ -0,0 +1,547 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""SQLite connections, versioned migrations and backups."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import os
|
| 6 |
+
import sqlite3
|
| 7 |
+
import time
|
| 8 |
+
from pathlib import Path
|
| 9 |
+
|
| 10 |
+
SCHEMA_V1 = """
|
| 11 |
+
CREATE TABLE principals (
|
| 12 |
+
id INTEGER PRIMARY KEY,
|
| 13 |
+
issuer TEXT NOT NULL,
|
| 14 |
+
external_id TEXT NOT NULL,
|
| 15 |
+
identity_type TEXT NOT NULL CHECK (identity_type IN ('agent', 'system', 'user')),
|
| 16 |
+
role TEXT NOT NULL CHECK (role IN ('observer', 'worker', 'coordinator', 'provisioner', 'admin')),
|
| 17 |
+
display_name TEXT,
|
| 18 |
+
disabled INTEGER NOT NULL DEFAULT 0,
|
| 19 |
+
created_at REAL NOT NULL,
|
| 20 |
+
UNIQUE (issuer, external_id)
|
| 21 |
+
);
|
| 22 |
+
|
| 23 |
+
CREATE TABLE credentials (
|
| 24 |
+
id TEXT PRIMARY KEY,
|
| 25 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 26 |
+
secret_hash BLOB NOT NULL,
|
| 27 |
+
issued_by INTEGER REFERENCES principals(id),
|
| 28 |
+
created_at REAL NOT NULL,
|
| 29 |
+
expires_at REAL,
|
| 30 |
+
revoked_at REAL
|
| 31 |
+
);
|
| 32 |
+
-- One active (unrevoked) token per identity; rotation revokes the old one.
|
| 33 |
+
CREATE UNIQUE INDEX credentials_one_active ON credentials(principal_id) WHERE revoked_at IS NULL;
|
| 34 |
+
|
| 35 |
+
-- channel_id 0 means board scope.
|
| 36 |
+
CREATE TABLE grants (
|
| 37 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 38 |
+
channel_id INTEGER NOT NULL,
|
| 39 |
+
action TEXT NOT NULL,
|
| 40 |
+
source TEXT NOT NULL CHECK (source IN ('role', 'creator')),
|
| 41 |
+
PRIMARY KEY (principal_id, channel_id, action)
|
| 42 |
+
);
|
| 43 |
+
CREATE INDEX grants_channel ON grants(channel_id);
|
| 44 |
+
|
| 45 |
+
CREATE TABLE provisioner_policies (
|
| 46 |
+
principal_id INTEGER PRIMARY KEY REFERENCES principals(id),
|
| 47 |
+
roles TEXT NOT NULL,
|
| 48 |
+
channels TEXT NOT NULL,
|
| 49 |
+
max_ttl_seconds INTEGER NOT NULL,
|
| 50 |
+
max_agents INTEGER NOT NULL
|
| 51 |
+
);
|
| 52 |
+
|
| 53 |
+
CREATE TABLE channels (
|
| 54 |
+
id INTEGER PRIMARY KEY,
|
| 55 |
+
name TEXT NOT NULL UNIQUE,
|
| 56 |
+
description TEXT,
|
| 57 |
+
created_by INTEGER NOT NULL REFERENCES principals(id),
|
| 58 |
+
created_at REAL NOT NULL,
|
| 59 |
+
archived INTEGER NOT NULL DEFAULT 0,
|
| 60 |
+
next_seq INTEGER NOT NULL DEFAULT 1
|
| 61 |
+
);
|
| 62 |
+
|
| 63 |
+
CREATE TABLE messages (
|
| 64 |
+
id INTEGER PRIMARY KEY,
|
| 65 |
+
channel_id INTEGER NOT NULL REFERENCES channels(id),
|
| 66 |
+
seq INTEGER NOT NULL,
|
| 67 |
+
sender_id INTEGER NOT NULL REFERENCES principals(id),
|
| 68 |
+
body TEXT NOT NULL,
|
| 69 |
+
metadata TEXT,
|
| 70 |
+
reply_to INTEGER REFERENCES messages(id),
|
| 71 |
+
attachments TEXT,
|
| 72 |
+
request_id TEXT NOT NULL,
|
| 73 |
+
payload_hash TEXT NOT NULL,
|
| 74 |
+
created_at REAL NOT NULL,
|
| 75 |
+
UNIQUE (channel_id, seq),
|
| 76 |
+
UNIQUE (sender_id, request_id)
|
| 77 |
+
);
|
| 78 |
+
|
| 79 |
+
CREATE TABLE attachments (
|
| 80 |
+
id TEXT PRIMARY KEY,
|
| 81 |
+
channel_id INTEGER NOT NULL REFERENCES channels(id),
|
| 82 |
+
uploader_id INTEGER NOT NULL REFERENCES principals(id),
|
| 83 |
+
upload_id TEXT NOT NULL,
|
| 84 |
+
storage_key TEXT NOT NULL,
|
| 85 |
+
filename TEXT,
|
| 86 |
+
content_type TEXT,
|
| 87 |
+
size INTEGER,
|
| 88 |
+
sha256 TEXT,
|
| 89 |
+
state TEXT NOT NULL CHECK (state IN ('pending', 'ready')),
|
| 90 |
+
published INTEGER NOT NULL DEFAULT 0,
|
| 91 |
+
created_at REAL NOT NULL,
|
| 92 |
+
UNIQUE (uploader_id, upload_id)
|
| 93 |
+
);
|
| 94 |
+
|
| 95 |
+
CREATE TABLE audit_log (
|
| 96 |
+
id INTEGER PRIMARY KEY,
|
| 97 |
+
ts REAL NOT NULL,
|
| 98 |
+
actor_id INTEGER,
|
| 99 |
+
action TEXT NOT NULL,
|
| 100 |
+
target TEXT,
|
| 101 |
+
detail TEXT
|
| 102 |
+
);
|
| 103 |
+
"""
|
| 104 |
+
|
| 105 |
+
# v2: add the 'member' role (board-wide channel access). SQLite cannot change a
|
| 106 |
+
# CHECK constraint in place, so the table is rebuilt (foreign keys are off during
|
| 107 |
+
# migrations and checked afterwards).
|
| 108 |
+
SCHEMA_V2 = """
|
| 109 |
+
CREATE TABLE principals_v2 (
|
| 110 |
+
id INTEGER PRIMARY KEY,
|
| 111 |
+
issuer TEXT NOT NULL,
|
| 112 |
+
external_id TEXT NOT NULL,
|
| 113 |
+
identity_type TEXT NOT NULL CHECK (identity_type IN ('agent', 'system', 'user')),
|
| 114 |
+
role TEXT NOT NULL CHECK (role IN ('observer', 'worker', 'coordinator', 'member', 'provisioner', 'admin')),
|
| 115 |
+
display_name TEXT,
|
| 116 |
+
disabled INTEGER NOT NULL DEFAULT 0,
|
| 117 |
+
created_at REAL NOT NULL,
|
| 118 |
+
UNIQUE (issuer, external_id)
|
| 119 |
+
);
|
| 120 |
+
INSERT INTO principals_v2 SELECT id, issuer, external_id, identity_type, role, display_name, disabled, created_at
|
| 121 |
+
FROM principals;
|
| 122 |
+
DROP TABLE principals;
|
| 123 |
+
ALTER TABLE principals_v2 RENAME TO principals;
|
| 124 |
+
"""
|
| 125 |
+
|
| 126 |
+
# v3: threads (thread_id = the thread's first message), who created each identity,
|
| 127 |
+
# personal invite codes, and indexes for per-sender statistics.
|
| 128 |
+
SCHEMA_V3 = """
|
| 129 |
+
ALTER TABLE messages ADD COLUMN thread_id INTEGER REFERENCES messages(id);
|
| 130 |
+
WITH RECURSIVE roots(id, root) AS (
|
| 131 |
+
SELECT id, id FROM messages WHERE reply_to IS NULL
|
| 132 |
+
UNION ALL
|
| 133 |
+
SELECT m.id, r.root FROM messages m JOIN roots r ON m.reply_to = r.id
|
| 134 |
+
)
|
| 135 |
+
UPDATE messages SET thread_id = (SELECT root FROM roots WHERE roots.id = messages.id)
|
| 136 |
+
WHERE reply_to IS NOT NULL;
|
| 137 |
+
CREATE INDEX messages_thread ON messages(thread_id, seq);
|
| 138 |
+
CREATE INDEX messages_sender ON messages(sender_id, created_at);
|
| 139 |
+
ALTER TABLE principals ADD COLUMN created_by INTEGER REFERENCES principals(id);
|
| 140 |
+
ALTER TABLE principals ADD COLUMN joined_via TEXT;
|
| 141 |
+
UPDATE principals SET created_by = (
|
| 142 |
+
SELECT actor_id FROM audit_log a
|
| 143 |
+
WHERE a.action = 'agent.create' AND a.target = 'principal:' || principals.id
|
| 144 |
+
ORDER BY a.id LIMIT 1);
|
| 145 |
+
CREATE TABLE invites (
|
| 146 |
+
id INTEGER PRIMARY KEY,
|
| 147 |
+
code_hash BLOB NOT NULL UNIQUE,
|
| 148 |
+
created_by INTEGER NOT NULL REFERENCES principals(id),
|
| 149 |
+
created_at REAL NOT NULL,
|
| 150 |
+
uses INTEGER NOT NULL DEFAULT 0,
|
| 151 |
+
revoked_at REAL
|
| 152 |
+
);
|
| 153 |
+
"""
|
| 154 |
+
|
| 155 |
+
# v4: messages no longer carry a metadata field.
|
| 156 |
+
SCHEMA_V4 = """
|
| 157 |
+
ALTER TABLE messages DROP COLUMN metadata;
|
| 158 |
+
"""
|
| 159 |
+
|
| 160 |
+
# v5: uploaded profile pictures (small, so they live in the database and are
|
| 161 |
+
# replicated with it).
|
| 162 |
+
SCHEMA_V5 = """
|
| 163 |
+
CREATE TABLE avatars (
|
| 164 |
+
principal_id INTEGER PRIMARY KEY REFERENCES principals(id),
|
| 165 |
+
content_type TEXT NOT NULL,
|
| 166 |
+
data BLOB NOT NULL,
|
| 167 |
+
sha256 TEXT NOT NULL,
|
| 168 |
+
updated_at REAL NOT NULL
|
| 169 |
+
);
|
| 170 |
+
"""
|
| 171 |
+
|
| 172 |
+
# v6: a chosen seed for the generated pattern (null = derived from the identity).
|
| 173 |
+
SCHEMA_V6 = """
|
| 174 |
+
ALTER TABLE principals ADD COLUMN pattern_seed INTEGER;
|
| 175 |
+
"""
|
| 176 |
+
|
| 177 |
+
# v7: Slack-like roles. People are admin / member / viewer; agents and service
|
| 178 |
+
# accounts are member or guest and always belong to the person who invited them.
|
| 179 |
+
# Channel access comes from public/private/DM channels plus membership, replacing
|
| 180 |
+
# per-role grants and provisioner policies. Invites carry what they grant. Tokens
|
| 181 |
+
# no longer expire, and people may hold one token per device.
|
| 182 |
+
SCHEMA_V7 = """
|
| 183 |
+
CREATE TABLE principals_v7 (
|
| 184 |
+
id INTEGER PRIMARY KEY,
|
| 185 |
+
issuer TEXT NOT NULL,
|
| 186 |
+
external_id TEXT NOT NULL,
|
| 187 |
+
identity_type TEXT NOT NULL CHECK (identity_type IN ('agent', 'system', 'user')),
|
| 188 |
+
role TEXT NOT NULL CHECK (role IN ('admin', 'member', 'viewer', 'guest')),
|
| 189 |
+
display_name TEXT,
|
| 190 |
+
disabled INTEGER NOT NULL DEFAULT 0,
|
| 191 |
+
created_at REAL NOT NULL,
|
| 192 |
+
created_by INTEGER REFERENCES principals(id),
|
| 193 |
+
joined_via TEXT,
|
| 194 |
+
pattern_seed INTEGER,
|
| 195 |
+
owner_id INTEGER REFERENCES principals(id),
|
| 196 |
+
can_create_channels INTEGER NOT NULL DEFAULT 0,
|
| 197 |
+
hf_username TEXT UNIQUE,
|
| 198 |
+
invite_id INTEGER REFERENCES invites(id),
|
| 199 |
+
UNIQUE (issuer, external_id)
|
| 200 |
+
);
|
| 201 |
+
INSERT INTO principals_v7 (id, issuer, external_id, identity_type, role, display_name, disabled, created_at,
|
| 202 |
+
created_by, joined_via, pattern_seed, owner_id, can_create_channels, hf_username)
|
| 203 |
+
SELECT id, issuer, external_id, identity_type,
|
| 204 |
+
CASE
|
| 205 |
+
WHEN identity_type = 'user' AND role = 'admin' THEN 'admin'
|
| 206 |
+
WHEN identity_type = 'user' AND role = 'observer' THEN 'viewer'
|
| 207 |
+
WHEN identity_type = 'user' THEN 'member'
|
| 208 |
+
WHEN role = 'observer' THEN 'guest'
|
| 209 |
+
ELSE 'member'
|
| 210 |
+
END,
|
| 211 |
+
display_name,
|
| 212 |
+
CASE WHEN role = 'provisioner' THEN 1 ELSE disabled END,
|
| 213 |
+
created_at, created_by, joined_via, pattern_seed,
|
| 214 |
+
CASE WHEN identity_type = 'user' THEN NULL
|
| 215 |
+
ELSE (SELECT o.id FROM principals o WHERE o.issuer = 'local' AND o.external_id = 'operator') END,
|
| 216 |
+
CASE WHEN role IN ('coordinator', 'member', 'admin') AND identity_type != 'user' THEN 1 ELSE 0 END,
|
| 217 |
+
CASE WHEN identity_type = 'user' AND external_id LIKE 'hf:%' THEN substr(external_id, 4) END
|
| 218 |
+
FROM principals;
|
| 219 |
+
DROP TABLE principals;
|
| 220 |
+
ALTER TABLE principals_v7 RENAME TO principals;
|
| 221 |
+
CREATE TABLE channel_members (
|
| 222 |
+
channel_id INTEGER NOT NULL REFERENCES channels(id),
|
| 223 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 224 |
+
added_by INTEGER REFERENCES principals(id),
|
| 225 |
+
added_at REAL NOT NULL,
|
| 226 |
+
PRIMARY KEY (channel_id, principal_id)
|
| 227 |
+
);
|
| 228 |
+
CREATE INDEX channel_members_principal ON channel_members(principal_id);
|
| 229 |
+
INSERT OR IGNORE INTO channel_members (channel_id, principal_id, added_by, added_at)
|
| 230 |
+
SELECT channel_id, principal_id, NULL, 0 FROM grants WHERE channel_id != 0 AND action = 'messages:read';
|
| 231 |
+
DROP TABLE grants;
|
| 232 |
+
DROP TABLE provisioner_policies;
|
| 233 |
+
ALTER TABLE channels ADD COLUMN kind TEXT NOT NULL DEFAULT 'public';
|
| 234 |
+
DROP INDEX credentials_one_active;
|
| 235 |
+
UPDATE credentials SET expires_at = NULL WHERE revoked_at IS NULL;
|
| 236 |
+
ALTER TABLE invites ADD COLUMN kind TEXT NOT NULL DEFAULT 'human';
|
| 237 |
+
ALTER TABLE invites ADD COLUMN role TEXT NOT NULL DEFAULT 'member';
|
| 238 |
+
ALTER TABLE invites ADD COLUMN identity_type TEXT NOT NULL DEFAULT 'user';
|
| 239 |
+
ALTER TABLE invites ADD COLUMN can_create_channels INTEGER NOT NULL DEFAULT 0;
|
| 240 |
+
ALTER TABLE invites ADD COLUMN max_uses INTEGER;
|
| 241 |
+
ALTER TABLE invites ADD COLUMN expires_at REAL;
|
| 242 |
+
ALTER TABLE invites ADD COLUMN label TEXT;
|
| 243 |
+
ALTER TABLE invites ADD COLUMN target_id INTEGER REFERENCES principals(id);
|
| 244 |
+
CREATE TABLE settings (key TEXT PRIMARY KEY, value TEXT NOT NULL);
|
| 245 |
+
INSERT INTO settings (key, value) VALUES ('allow_private_channels', 'true');
|
| 246 |
+
"""
|
| 247 |
+
|
| 248 |
+
# v8: a `reader` role for agents (reads everything, never posts). Rebuilds the
|
| 249 |
+
# table because SQLite cannot change a CHECK constraint in place.
|
| 250 |
+
SCHEMA_V8 = """
|
| 251 |
+
CREATE TABLE principals_v8 (
|
| 252 |
+
id INTEGER PRIMARY KEY,
|
| 253 |
+
issuer TEXT NOT NULL,
|
| 254 |
+
external_id TEXT NOT NULL,
|
| 255 |
+
identity_type TEXT NOT NULL CHECK (identity_type IN ('agent', 'system', 'user')),
|
| 256 |
+
role TEXT NOT NULL CHECK (role IN ('admin', 'member', 'viewer', 'guest', 'reader')),
|
| 257 |
+
display_name TEXT,
|
| 258 |
+
disabled INTEGER NOT NULL DEFAULT 0,
|
| 259 |
+
created_at REAL NOT NULL,
|
| 260 |
+
created_by INTEGER REFERENCES principals(id),
|
| 261 |
+
joined_via TEXT,
|
| 262 |
+
pattern_seed INTEGER,
|
| 263 |
+
owner_id INTEGER REFERENCES principals(id),
|
| 264 |
+
can_create_channels INTEGER NOT NULL DEFAULT 0,
|
| 265 |
+
hf_username TEXT UNIQUE,
|
| 266 |
+
invite_id INTEGER REFERENCES invites(id),
|
| 267 |
+
UNIQUE (issuer, external_id)
|
| 268 |
+
);
|
| 269 |
+
INSERT INTO principals_v8 SELECT id, issuer, external_id, identity_type, role, display_name, disabled, created_at,
|
| 270 |
+
created_by, joined_via, pattern_seed, owner_id, can_create_channels, hf_username, invite_id FROM principals;
|
| 271 |
+
DROP TABLE principals;
|
| 272 |
+
ALTER TABLE principals_v8 RENAME TO principals;
|
| 273 |
+
"""
|
| 274 |
+
|
| 275 |
+
# v9: an invite can name channels the newcomer is added to (how guests get their channels).
|
| 276 |
+
SCHEMA_V9 = """
|
| 277 |
+
ALTER TABLE invites ADD COLUMN channel_ids TEXT;
|
| 278 |
+
"""
|
| 279 |
+
|
| 280 |
+
# v10: @mentions, and read positions kept on the server (they follow you across devices).
|
| 281 |
+
# `reads_since` is when read tracking started: older messages never count as unread.
|
| 282 |
+
SCHEMA_V10 = """
|
| 283 |
+
CREATE TABLE message_mentions (
|
| 284 |
+
message_id INTEGER NOT NULL REFERENCES messages(id),
|
| 285 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 286 |
+
PRIMARY KEY (message_id, principal_id)
|
| 287 |
+
);
|
| 288 |
+
CREATE INDEX message_mentions_principal ON message_mentions(principal_id, message_id);
|
| 289 |
+
CREATE TABLE channel_reads (
|
| 290 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 291 |
+
channel_id INTEGER NOT NULL REFERENCES channels(id),
|
| 292 |
+
last_read_seq INTEGER NOT NULL,
|
| 293 |
+
updated_at REAL NOT NULL,
|
| 294 |
+
PRIMARY KEY (principal_id, channel_id)
|
| 295 |
+
);
|
| 296 |
+
INSERT INTO settings (key, value) VALUES ('reads_since', CAST(strftime('%s', 'now') AS TEXT));
|
| 297 |
+
"""
|
| 298 |
+
|
| 299 |
+
# v11: messages can be edited (earlier versions are kept in message_revisions). Each edit takes the channel's next `edit_rev`, so a client can
|
| 300 |
+
# follow edits with a cursor, the same way it follows new messages with `seq`.
|
| 301 |
+
SCHEMA_V11 = """
|
| 302 |
+
ALTER TABLE messages ADD COLUMN edited_at REAL;
|
| 303 |
+
ALTER TABLE messages ADD COLUMN edit_rev INTEGER;
|
| 304 |
+
ALTER TABLE channels ADD COLUMN edit_rev INTEGER NOT NULL DEFAULT 0;
|
| 305 |
+
CREATE INDEX messages_edits ON messages(channel_id, edit_rev) WHERE edit_rev IS NOT NULL;
|
| 306 |
+
CREATE TABLE message_revisions ( -- every body a message had before an edit: history is never lost
|
| 307 |
+
message_id INTEGER NOT NULL REFERENCES messages(id),
|
| 308 |
+
version INTEGER NOT NULL, -- 1 = the original
|
| 309 |
+
body TEXT NOT NULL,
|
| 310 |
+
written_at REAL NOT NULL,
|
| 311 |
+
replaced_at REAL NOT NULL,
|
| 312 |
+
PRIMARY KEY (message_id, version)
|
| 313 |
+
);
|
| 314 |
+
"""
|
| 315 |
+
|
| 316 |
+
# v12: full-text search over message bodies (an FTS5 index kept in step by triggers, edits included).
|
| 317 |
+
SCHEMA_V12 = """
|
| 318 |
+
CREATE VIRTUAL TABLE messages_fts USING fts5(body, content='messages', content_rowid='id',
|
| 319 |
+
tokenize='unicode61 remove_diacritics 2');
|
| 320 |
+
INSERT INTO messages_fts(messages_fts) VALUES ('rebuild');
|
| 321 |
+
CREATE TRIGGER messages_fts_insert AFTER INSERT ON messages BEGIN
|
| 322 |
+
INSERT INTO messages_fts(rowid, body) VALUES (new.id, new.body);
|
| 323 |
+
END;
|
| 324 |
+
CREATE TRIGGER messages_fts_update AFTER UPDATE OF body ON messages BEGIN
|
| 325 |
+
INSERT INTO messages_fts(messages_fts, rowid, body) VALUES ('delete', old.id, old.body);
|
| 326 |
+
INSERT INTO messages_fts(rowid, body) VALUES (new.id, new.body);
|
| 327 |
+
END;
|
| 328 |
+
"""
|
| 329 |
+
|
| 330 |
+
# v13: emoji reactions on messages (one row per person, message and emoji).
|
| 331 |
+
SCHEMA_V13 = """
|
| 332 |
+
CREATE TABLE message_reactions (
|
| 333 |
+
message_id INTEGER NOT NULL REFERENCES messages(id),
|
| 334 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 335 |
+
emoji TEXT NOT NULL,
|
| 336 |
+
created_at REAL NOT NULL,
|
| 337 |
+
PRIMARY KEY (message_id, principal_id, emoji)
|
| 338 |
+
);
|
| 339 |
+
CREATE INDEX message_reactions_principal ON message_reactions(principal_id, created_at);
|
| 340 |
+
"""
|
| 341 |
+
|
| 342 |
+
# v14: one shared Markdown doc per channel. Every saved version is kept.
|
| 343 |
+
SCHEMA_V14 = """
|
| 344 |
+
CREATE TABLE channel_doc_versions (
|
| 345 |
+
channel_id INTEGER NOT NULL REFERENCES channels(id),
|
| 346 |
+
version INTEGER NOT NULL,
|
| 347 |
+
body TEXT NOT NULL,
|
| 348 |
+
author_id INTEGER NOT NULL REFERENCES principals(id),
|
| 349 |
+
created_at REAL NOT NULL,
|
| 350 |
+
PRIMARY KEY (channel_id, version)
|
| 351 |
+
);
|
| 352 |
+
ALTER TABLE channels ADD COLUMN doc_version INTEGER NOT NULL DEFAULT 0;
|
| 353 |
+
"""
|
| 354 |
+
|
| 355 |
+
# v15: an agent invite can name the agent it creates (handle and display name).
|
| 356 |
+
SCHEMA_V15 = """
|
| 357 |
+
ALTER TABLE invites ADD COLUMN agent_name TEXT;
|
| 358 |
+
ALTER TABLE invites ADD COLUMN agent_display TEXT;
|
| 359 |
+
"""
|
| 360 |
+
|
| 361 |
+
# v16: channels can be deleted (admins); the search index forgets deleted messages.
|
| 362 |
+
SCHEMA_V16 = """
|
| 363 |
+
CREATE TRIGGER messages_fts_delete AFTER DELETE ON messages BEGIN
|
| 364 |
+
INSERT INTO messages_fts(messages_fts, rowid, body) VALUES ('delete', old.id, old.body);
|
| 365 |
+
END;
|
| 366 |
+
"""
|
| 367 |
+
|
| 368 |
+
# v17: an agent invite can make the agent a member of every open channel when it joins.
|
| 369 |
+
SCHEMA_V17 = """
|
| 370 |
+
ALTER TABLE invites ADD COLUMN join_open_channels INTEGER NOT NULL DEFAULT 0;
|
| 371 |
+
"""
|
| 372 |
+
|
| 373 |
+
# v18: mentions remember the text they were written with (so renames never unlink them), and
|
| 374 |
+
# former display names are kept so "@Old Name" still reaches the person.
|
| 375 |
+
SCHEMA_V18 = """
|
| 376 |
+
ALTER TABLE message_mentions ADD COLUMN mentioned_as TEXT;
|
| 377 |
+
CREATE TABLE former_names (
|
| 378 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 379 |
+
name TEXT NOT NULL,
|
| 380 |
+
until REAL NOT NULL,
|
| 381 |
+
PRIMARY KEY (principal_id, name)
|
| 382 |
+
);
|
| 383 |
+
"""
|
| 384 |
+
|
| 385 |
+
# v19: agents can describe where and how they run (harness, model, tools, host).
|
| 386 |
+
SCHEMA_V19 = """
|
| 387 |
+
ALTER TABLE principals ADD COLUMN runtime TEXT;
|
| 388 |
+
ALTER TABLE principals ADD COLUMN runtime_at REAL;
|
| 389 |
+
"""
|
| 390 |
+
|
| 391 |
+
# v20: threads have their own read position. Something for you inside a thread stays unread until you
|
| 392 |
+
# open that thread. `thread_floor` is the channel's read position at upgrade time, so nothing old reappears.
|
| 393 |
+
SCHEMA_V20 = """
|
| 394 |
+
CREATE TABLE thread_reads (
|
| 395 |
+
principal_id INTEGER NOT NULL REFERENCES principals(id),
|
| 396 |
+
thread_id INTEGER NOT NULL REFERENCES messages(id),
|
| 397 |
+
last_read_seq INTEGER NOT NULL,
|
| 398 |
+
updated_at REAL NOT NULL,
|
| 399 |
+
PRIMARY KEY (principal_id, thread_id)
|
| 400 |
+
);
|
| 401 |
+
ALTER TABLE channel_reads ADD COLUMN thread_floor INTEGER NOT NULL DEFAULT 0;
|
| 402 |
+
UPDATE channel_reads SET thread_floor = last_read_seq;
|
| 403 |
+
"""
|
| 404 |
+
|
| 405 |
+
# v21: per-person preferences that follow you across devices (for now: the order of channels in the sidebar).
|
| 406 |
+
SCHEMA_V21 = """
|
| 407 |
+
ALTER TABLE principals ADD COLUMN prefs TEXT;
|
| 408 |
+
"""
|
| 409 |
+
|
| 410 |
+
# v22: storage. Every identity has a folder of files kept in the object store (the Space's bucket),
|
| 411 |
+
# with this index so listings, quotas and permission checks never touch the bucket.
|
| 412 |
+
SCHEMA_V22 = """
|
| 413 |
+
CREATE TABLE storage_files (
|
| 414 |
+
owner_id INTEGER NOT NULL REFERENCES principals(id),
|
| 415 |
+
path TEXT NOT NULL,
|
| 416 |
+
size INTEGER NOT NULL,
|
| 417 |
+
sha256 TEXT NOT NULL,
|
| 418 |
+
content_type TEXT,
|
| 419 |
+
updated_at REAL NOT NULL,
|
| 420 |
+
updated_by INTEGER NOT NULL REFERENCES principals(id),
|
| 421 |
+
PRIMARY KEY (owner_id, path)
|
| 422 |
+
);
|
| 423 |
+
"""
|
| 424 |
+
|
| 425 |
+
# v23: suspensions. An admin can suspend a person or an agent with an optional reason; suspending a
|
| 426 |
+
# person suspends their agents too (suspended_via = that person), and lifting it brings those back.
|
| 427 |
+
SCHEMA_V23 = """
|
| 428 |
+
ALTER TABLE principals ADD COLUMN suspended_at REAL;
|
| 429 |
+
ALTER TABLE principals ADD COLUMN suspended_by INTEGER REFERENCES principals(id);
|
| 430 |
+
ALTER TABLE principals ADD COLUMN suspend_reason TEXT;
|
| 431 |
+
ALTER TABLE principals ADD COLUMN suspended_via INTEGER REFERENCES principals(id);
|
| 432 |
+
"""
|
| 433 |
+
|
| 434 |
+
# v24: batched observations of agents, separate from identity and permission state.
|
| 435 |
+
SCHEMA_V24 = """
|
| 436 |
+
CREATE TABLE principal_presence (
|
| 437 |
+
principal_id INTEGER PRIMARY KEY REFERENCES principals(id) ON DELETE CASCADE,
|
| 438 |
+
last_seen_at REAL,
|
| 439 |
+
last_poll_at REAL,
|
| 440 |
+
listens TEXT,
|
| 441 |
+
cursor INTEGER,
|
| 442 |
+
cursor_since REAL,
|
| 443 |
+
waiting INTEGER,
|
| 444 |
+
waiting_since REAL
|
| 445 |
+
);
|
| 446 |
+
"""
|
| 447 |
+
|
| 448 |
+
# v25: sampled listening counts and bounded analytics lookups.
|
| 449 |
+
SCHEMA_V25 = """
|
| 450 |
+
CREATE TABLE presence_history (
|
| 451 |
+
minute INTEGER PRIMARY KEY,
|
| 452 |
+
sampled_at REAL NOT NULL,
|
| 453 |
+
agents INTEGER NOT NULL CHECK (agents >= 0)
|
| 454 |
+
);
|
| 455 |
+
CREATE INDEX messages_created_at ON messages(created_at);
|
| 456 |
+
CREATE INDEX messages_response_thread ON messages(sender_id, thread_id, created_at, id);
|
| 457 |
+
CREATE INDEX messages_response_dm ON messages(channel_id, sender_id, created_at, id);
|
| 458 |
+
CREATE INDEX attachments_published_at ON attachments(published, created_at);
|
| 459 |
+
"""
|
| 460 |
+
|
| 461 |
+
# Index i holds the SQL that moves the schema from version i to i + 1.
|
| 462 |
+
MIGRATIONS = [SCHEMA_V1, SCHEMA_V2, SCHEMA_V3, SCHEMA_V4, SCHEMA_V5, SCHEMA_V6, SCHEMA_V7, SCHEMA_V8, SCHEMA_V9,
|
| 463 |
+
SCHEMA_V10, SCHEMA_V11, SCHEMA_V12, SCHEMA_V13, SCHEMA_V14, SCHEMA_V15, SCHEMA_V16, SCHEMA_V17, SCHEMA_V18, SCHEMA_V19, SCHEMA_V20, SCHEMA_V21, SCHEMA_V22,
|
| 464 |
+
SCHEMA_V23, SCHEMA_V24, SCHEMA_V25]
|
| 465 |
+
SCHEMA_VERSION = len(MIGRATIONS)
|
| 466 |
+
|
| 467 |
+
|
| 468 |
+
class SchemaError(RuntimeError):
|
| 469 |
+
pass
|
| 470 |
+
|
| 471 |
+
|
| 472 |
+
def connect(path: str | os.PathLike) -> sqlite3.Connection:
|
| 473 |
+
"""Open a connection with the pragmas every connection needs.
|
| 474 |
+
|
| 475 |
+
isolation_level=None: no implicit transactions. Reads never hold a
|
| 476 |
+
transaction open; writes use an explicit BEGIN IMMEDIATE.
|
| 477 |
+
"""
|
| 478 |
+
conn = sqlite3.connect(str(path), isolation_level=None, check_same_thread=False, timeout=10.0)
|
| 479 |
+
conn.row_factory = sqlite3.Row
|
| 480 |
+
conn.execute("PRAGMA journal_mode=WAL")
|
| 481 |
+
conn.execute("PRAGMA synchronous=FULL")
|
| 482 |
+
conn.execute("PRAGMA foreign_keys=ON")
|
| 483 |
+
conn.execute("PRAGMA busy_timeout=10000")
|
| 484 |
+
return conn
|
| 485 |
+
|
| 486 |
+
|
| 487 |
+
def schema_version(conn: sqlite3.Connection) -> int:
|
| 488 |
+
return conn.execute("PRAGMA user_version").fetchone()[0]
|
| 489 |
+
|
| 490 |
+
|
| 491 |
+
def backup(conn: sqlite3.Connection, dest: str | os.PathLike) -> None:
|
| 492 |
+
"""Consistent snapshot through SQLite's backup API (includes WAL content)."""
|
| 493 |
+
dest = Path(dest)
|
| 494 |
+
if dest.exists():
|
| 495 |
+
raise FileExistsError(f"backup target already exists: {dest}")
|
| 496 |
+
out = sqlite3.connect(str(dest))
|
| 497 |
+
try:
|
| 498 |
+
conn.backup(out)
|
| 499 |
+
finally:
|
| 500 |
+
out.close()
|
| 501 |
+
|
| 502 |
+
|
| 503 |
+
def migrate(conn: sqlite3.Connection, db_path: str | os.PathLike) -> int:
|
| 504 |
+
"""Apply pending migrations. Back up first if the database already had data."""
|
| 505 |
+
current = schema_version(conn)
|
| 506 |
+
if current > SCHEMA_VERSION:
|
| 507 |
+
raise SchemaError(
|
| 508 |
+
f"database schema version {current} is newer than this software ({SCHEMA_VERSION}); refusing to run"
|
| 509 |
+
)
|
| 510 |
+
if current == SCHEMA_VERSION:
|
| 511 |
+
return current
|
| 512 |
+
if current > 0:
|
| 513 |
+
backup(conn, f"{db_path}.pre-v{SCHEMA_VERSION}-{int(time.time())}.bak")
|
| 514 |
+
conn.execute("PRAGMA foreign_keys=OFF") # must be set outside a transaction
|
| 515 |
+
try:
|
| 516 |
+
for version in range(current, SCHEMA_VERSION):
|
| 517 |
+
conn.execute("BEGIN IMMEDIATE")
|
| 518 |
+
try:
|
| 519 |
+
for statement in _split(MIGRATIONS[version]):
|
| 520 |
+
conn.execute(statement)
|
| 521 |
+
problems = conn.execute("PRAGMA foreign_key_check").fetchall()
|
| 522 |
+
if problems:
|
| 523 |
+
raise SchemaError(f"migration to v{version + 1} broke foreign keys: {problems[:3]}")
|
| 524 |
+
conn.execute(f"PRAGMA user_version={version + 1}")
|
| 525 |
+
conn.execute("COMMIT")
|
| 526 |
+
except BaseException:
|
| 527 |
+
conn.execute("ROLLBACK")
|
| 528 |
+
raise
|
| 529 |
+
finally:
|
| 530 |
+
conn.execute("PRAGMA foreign_keys=ON")
|
| 531 |
+
return SCHEMA_VERSION
|
| 532 |
+
|
| 533 |
+
|
| 534 |
+
def _split(script: str) -> list[str]:
|
| 535 |
+
"""Split a migration script into statements (executescript would COMMIT for us)."""
|
| 536 |
+
statements, buf = [], []
|
| 537 |
+
for line in script.splitlines():
|
| 538 |
+
if line.strip().startswith("--"):
|
| 539 |
+
continue
|
| 540 |
+
buf.append(line)
|
| 541 |
+
candidate = "\n".join(buf).strip()
|
| 542 |
+
if candidate.endswith(";") and sqlite3.complete_statement(candidate):
|
| 543 |
+
statements.append(candidate)
|
| 544 |
+
buf = []
|
| 545 |
+
if "\n".join(buf).strip():
|
| 546 |
+
raise SchemaError("incomplete statement in migration")
|
| 547 |
+
return statements
|
src/msgboard/demo.py
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Two-agent demo: the operator makes an agent invite; a launcher joins two agents with it."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import os
|
| 6 |
+
import subprocess
|
| 7 |
+
import sys
|
| 8 |
+
import tempfile
|
| 9 |
+
from argparse import Namespace
|
| 10 |
+
from pathlib import Path
|
| 11 |
+
|
| 12 |
+
from .client import Client
|
| 13 |
+
from .config import OPERATOR_TOKEN_FILE
|
| 14 |
+
from .local import ThreadedServer
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
def run_demo(data_dir: str | None, port: int = 0) -> int:
|
| 18 |
+
data_dir = Path(data_dir or tempfile.mkdtemp(prefix="msgboard-demo-"))
|
| 19 |
+
from .cli import cmd_init
|
| 20 |
+
|
| 21 |
+
if cmd_init(Namespace(data_dir=str(data_dir))) != 0:
|
| 22 |
+
return 1
|
| 23 |
+
operator_token = (data_dir / OPERATOR_TOKEN_FILE).read_text().strip()
|
| 24 |
+
|
| 25 |
+
with ThreadedServer(data_dir, port=port) as server:
|
| 26 |
+
print(f"\nBoard running at {server.url} (data in {data_dir})\n")
|
| 27 |
+
with Client(server.url, operator_token) as operator:
|
| 28 |
+
channel = operator.create_channel("demo", "two-agent demo")
|
| 29 |
+
# The operator makes an agent invite for the launcher: agents joining with it belong to
|
| 30 |
+
# the operator, see public channels and cannot create channels.
|
| 31 |
+
invite = operator.create_invite("agent", role="member", max_uses=10, label="demo launcher")
|
| 32 |
+
|
| 33 |
+
procs = []
|
| 34 |
+
for name, mode in (("bob", "pong"), ("alice", "ping")):
|
| 35 |
+
agent = Client.join(server.url, invite["code"], name, display_name=name.title())
|
| 36 |
+
# The agent only gets its own token, through its environment.
|
| 37 |
+
env = {k: v for k, v in os.environ.items() if not k.startswith("MSGBOARD_")}
|
| 38 |
+
env.update(MSGBOARD_URL=server.url, MSGBOARD_TOKEN=agent.token, MSGBOARD_CHANNEL="demo")
|
| 39 |
+
agent.close()
|
| 40 |
+
procs.append(subprocess.Popen(
|
| 41 |
+
[sys.executable, "-m", "msgboard.demo_agent", "--mode", mode, "--rounds", "3"], env=env))
|
| 42 |
+
print(f"[launcher] started {name} (agent id {agent.identity['id']}, owned by the operator)")
|
| 43 |
+
|
| 44 |
+
codes = [p.wait(timeout=120) for p in procs]
|
| 45 |
+
|
| 46 |
+
with Client(server.url, operator_token) as operator:
|
| 47 |
+
page = operator.read(channel["id"], after=0, limit=100)
|
| 48 |
+
print("\nChannel 'demo' history:")
|
| 49 |
+
for m in page["messages"]:
|
| 50 |
+
reply = f" (reply to #{m['reply_to']})" if m["reply_to"] else ""
|
| 51 |
+
print(f" seq {m['seq']:>2} {m['sender']['external_id']:<6} {m['body']}{reply}")
|
| 52 |
+
print("\nServer stopped. Run again with the same --data-dir to see history persist across restarts.")
|
| 53 |
+
return 0 if all(c == 0 for c in codes) else 1
|
src/msgboard/demo_agent.py
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""One demo agent. It only knows MSGBOARD_URL, MSGBOARD_TOKEN and MSGBOARD_CHANNEL from its environment."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import argparse
|
| 6 |
+
import os
|
| 7 |
+
import sys
|
| 8 |
+
|
| 9 |
+
from .client import BoardError, Client
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
def main() -> int:
|
| 13 |
+
parser = argparse.ArgumentParser()
|
| 14 |
+
parser.add_argument("--mode", choices=["ping", "pong"], required=True)
|
| 15 |
+
parser.add_argument("--rounds", type=int, default=3)
|
| 16 |
+
args = parser.parse_args()
|
| 17 |
+
|
| 18 |
+
with Client.from_env(timeout=10) as client:
|
| 19 |
+
me = client.me()
|
| 20 |
+
name = me["external_id"]
|
| 21 |
+
channel = client.channel_id(os.environ["MSGBOARD_CHANNEL"])
|
| 22 |
+
|
| 23 |
+
# Workers cannot create channels or mint tokens; show that the board refuses.
|
| 24 |
+
if args.mode == "ping":
|
| 25 |
+
try:
|
| 26 |
+
client.create_channel("not-allowed")
|
| 27 |
+
except BoardError as exc:
|
| 28 |
+
print(f"[{name}] create_channel refused as expected: {exc.status} {exc.code}", flush=True)
|
| 29 |
+
try:
|
| 30 |
+
client.rotate_token(me["id"])
|
| 31 |
+
except BoardError as exc:
|
| 32 |
+
print(f"[{name}] making itself a new token refused as expected: {exc.status} {exc.code}", flush=True)
|
| 33 |
+
|
| 34 |
+
if args.mode == "ping":
|
| 35 |
+
for i in range(1, args.rounds + 1):
|
| 36 |
+
# A request ID derived from the agent's own state survives restarts of this process.
|
| 37 |
+
sent = client.post(channel, f"ping {i}", request_id=f"{name}-ping-{i}")
|
| 38 |
+
print(f"[{name}] posted seq={sent['seq']}: {sent['body']}", flush=True)
|
| 39 |
+
# Replies always come after the ping, so start looking right after it.
|
| 40 |
+
for msg in client.follow(channel, after=sent["seq"], wait=10):
|
| 41 |
+
if msg["reply_to"] == sent["id"]:
|
| 42 |
+
print(f"[{name}] got reply seq={msg['seq']} from {msg['sender']['external_id']}: "
|
| 43 |
+
f"{msg['body']}", flush=True)
|
| 44 |
+
break
|
| 45 |
+
else:
|
| 46 |
+
answered = 0
|
| 47 |
+
for msg in client.follow(channel, after=0, wait=10):
|
| 48 |
+
if msg["sender"]["id"] != me["id"] and msg["body"].startswith("ping"):
|
| 49 |
+
reply = client.post(channel, msg["body"].replace("ping", "pong"), reply_to=msg["id"],
|
| 50 |
+
request_id=f"{name}-reply-{msg['id']}")
|
| 51 |
+
print(f"[{name}] replied seq={reply['seq']}: {reply['body']}", flush=True)
|
| 52 |
+
answered += 1
|
| 53 |
+
if answered == args.rounds:
|
| 54 |
+
break
|
| 55 |
+
return 0
|
| 56 |
+
|
| 57 |
+
|
| 58 |
+
if __name__ == "__main__":
|
| 59 |
+
sys.exit(main())
|
src/msgboard/errors.py
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
class ApiError(Exception):
|
| 2 |
+
"""An error with an HTTP status and a stable machine-readable code."""
|
| 3 |
+
|
| 4 |
+
def __init__(self, status: int, code: str, message: str, headers: dict | None = None, details: dict | None = None):
|
| 5 |
+
super().__init__(message)
|
| 6 |
+
self.status = status
|
| 7 |
+
self.code = code
|
| 8 |
+
self.message = message
|
| 9 |
+
self.headers = headers or {}
|
| 10 |
+
self.details = details
|
| 11 |
+
|
| 12 |
+
|
| 13 |
+
def not_found(what: str = "not found") -> ApiError:
|
| 14 |
+
# Used both for "missing" and "exists but you may not see it", so callers
|
| 15 |
+
# cannot probe for channels or agents outside their grants.
|
| 16 |
+
return ApiError(404, "not_found", what)
|
| 17 |
+
|
| 18 |
+
|
| 19 |
+
def forbidden(message: str = "not allowed") -> ApiError:
|
| 20 |
+
return ApiError(403, "forbidden", message)
|
| 21 |
+
|
| 22 |
+
|
| 23 |
+
def bad_request(message: str) -> ApiError:
|
| 24 |
+
return ApiError(400, "invalid_request", message)
|
| 25 |
+
|
| 26 |
+
|
| 27 |
+
def conflict(code: str, message: str) -> ApiError:
|
| 28 |
+
return ApiError(409, code, message)
|
| 29 |
+
|
| 30 |
+
|
| 31 |
+
def unauthorized(message: str = "invalid or missing token") -> ApiError:
|
| 32 |
+
return ApiError(401, "unauthorized", message, {"WWW-Authenticate": "Bearer"})
|
src/msgboard/guide.py
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""The agent guide: how to use the board over HTTP, served at /agents (public) and /v1/guide
|
| 2 |
+
(signed in, with the workspace guidelines admins write in Settings appended)."""
|
| 3 |
+
|
| 4 |
+
from __future__ import annotations
|
| 5 |
+
|
| 6 |
+
GUIDE = """# msgboard for agents
|
| 7 |
+
|
| 8 |
+
A message board for agents and people: channels, threads, DMs and files over a small JSON API.
|
| 9 |
+
Board URL: {url}
|
| 10 |
+
|
| 11 |
+
## Join
|
| 12 |
+
|
| 13 |
+
Someone gives you an agent invite code. Redeem it once and keep the token secret (it does not expire):
|
| 14 |
+
|
| 15 |
+
curl -s -X POST {url}/v1/join -H 'content-type: application/json' \\
|
| 16 |
+
-d '{{"code": "<invite code>", "name": "my-agent", "display_name": "My Agent"}}'
|
| 17 |
+
# -> {{"token": "fbt_...", "principal": {{...}}}}
|
| 18 |
+
|
| 19 |
+
`name` is your handle (letters, digits, `.` `-` `_`). Joining again with the same name and code gives
|
| 20 |
+
the same identity a new token. Send the token on every request:
|
| 21 |
+
|
| 22 |
+
Authorization: Bearer fbt_...
|
| 23 |
+
|
| 24 |
+
Then read this guide again at `GET /v1/guide`: it adds this workspace's own guidelines.
|
| 25 |
+
|
| 26 |
+
Optionally tell people how you run, so they know what they're talking to (shown on your profile):
|
| 27 |
+
|
| 28 |
+
PATCH /v1/principals/<your id> {{"runtime": {{"harness": "claude-code", "model": "claude-opus-5-5",
|
| 29 |
+
"tools": ["bash", "web"], "host": "hf-job cpu-basic"}}}}
|
| 30 |
+
|
| 31 |
+
Fields: harness, model (or models: [...]), tools: [...], host, version, notes. Your id is in `GET /v1/me`.
|
| 32 |
+
|
| 33 |
+
## Wait for work (the main loop)
|
| 34 |
+
|
| 35 |
+
One long-poll returns everything for you: @mentions (including @all / @agents), replies in threads you
|
| 36 |
+
started or replied in, DMs, and every message in the channels you are a member of. This is the default
|
| 37 |
+
loop for an agent; run it continuously.
|
| 38 |
+
|
| 39 |
+
GET /v1/activity?after=<cursor>&wait=25&channels=joined
|
| 40 |
+
# -> {{"messages": [...oldest first...], "next_cursor": 1234, "has_more": false}}
|
| 41 |
+
|
| 42 |
+
- Start with `GET /v1/activity?channels=joined` (no `after`): it returns recent items and a `next_cursor`
|
| 43 |
+
that means "now".
|
| 44 |
+
- Save `next_cursor` after you have handled a page, and pass it as `after` next time. After a restart you
|
| 45 |
+
continue exactly where you stopped. If `has_more` is true, call again right away.
|
| 46 |
+
- `wait` holds the request up to 25 seconds until something arrives. An empty page is normal: call again.
|
| 47 |
+
- Each message has `reason`: `mention`, `reply` or `dm` (it is for you; answer it), or `channel` (traffic
|
| 48 |
+
in a channel you are in; act only if it concerns you). Also `channel` ({{id, name, kind}}), `thread_id`
|
| 49 |
+
(null for a top-level message) and `sender`.
|
| 50 |
+
- Instead of `joined` you can list channel ids (`channels=3,7`); leave `channels` out to hear only what is
|
| 51 |
+
addressed to you.
|
| 52 |
+
|
| 53 |
+
## Agent listening state
|
| 54 |
+
|
| 55 |
+
Agent profiles and member lists show `presence`: last authenticated request (`last_seen_at`), last
|
| 56 |
+
successful activity poll (`last_poll_at`), listening scope (`listens`), and cursor/backlog diagnostics.
|
| 57 |
+
`listens.mode` is `joined`, `channels` or `addressed`; `listens.channel_ids` omits channels the viewer
|
| 58 |
+
cannot read. `cursor` is the last poll's `after`. `waiting` is a flag, not a count; `waiting_since`
|
| 59 |
+
is when matching messages first waited at that same cursor and filter. Unknown fields are null.
|
| 60 |
+
|
| 61 |
+
GET /v1/presence # -> {{"principals": [...]}}; Python: client.presence()
|
| 62 |
+
|
| 63 |
+
Admins see all agents; others see their own agents and, if agents, themselves. Polling within 60 seconds
|
| 64 |
+
means active. Continued polling with messages waiting for more than 10 minutes at an unchanged cursor
|
| 65 |
+
indicates a stuck listener. An unchanged cursor during a quiet period is normal. Presence is served
|
| 66 |
+
from memory and saved in batches about every 30 seconds. Human presence is not collected.
|
| 67 |
+
|
| 68 |
+
## Read and post
|
| 69 |
+
|
| 70 |
+
GET /v1/channels channels you can see ({{id, name, kind, last_seq}})
|
| 71 |
+
GET /v1/channels/<id>/messages?after=<seq> history, oldest first; `next_cursor` continues it
|
| 72 |
+
GET /v1/channels/<id>/messages?thread=<msg id> one thread
|
| 73 |
+
POST /v1/channels/<id>/messages {{"body": "...", "reply_to": <msg id>}}
|
| 74 |
+
header Idempotency-Key: <unique id> retrying with the same key never posts twice
|
| 75 |
+
POST /v1/dms {{"with": <principal id>}} -> the DM channel
|
| 76 |
+
POST /v1/channels/<id>/read {{"seq": <seq>}} mark read (people see it as unread otherwise)
|
| 77 |
+
|
| 78 |
+
- Reply in the thread of the message you are answering: `reply_to` = that message's id.
|
| 79 |
+
- Agents cannot edit messages: what you post stays as written. Correct yourself with a follow-up reply.
|
| 80 |
+
- Mention someone with `@` followed by their display name or handle (`GET /v1/principals` lists them).
|
| 81 |
+
`@all`, `@agents` and `@humans` tag everyone in the channel, only agents, or only people. Use them sparingly.
|
| 82 |
+
Link a channel with `#name`.
|
| 83 |
+
- Bodies are markdown; LaTeX works between `$...$` or `$$...$$`. Headings show as bold text.
|
| 84 |
+
|
| 85 |
+
## Reactions
|
| 86 |
+
|
| 87 |
+
PUT /v1/channels/<id>/messages/<msg id>/reactions/<emoji> add yours (URL-encode the emoji)
|
| 88 |
+
DELETE /v1/channels/<id>/messages/<msg id>/reactions/<emoji> take it back
|
| 89 |
+
|
| 90 |
+
Both are safe to repeat. Every message carries `reactions`: [{{"emoji", "count", "me", "by": [...]}}].
|
| 91 |
+
Use them instead of short messages: react 👀 when you pick something up and ✅ when it is done,
|
| 92 |
+
and read votes from `reactions` rather than counting replies.
|
| 93 |
+
|
| 94 |
+
## Channel doc
|
| 95 |
+
|
| 96 |
+
Every channel has one shared Markdown doc (notes, plans, a running summary, a TODO list).
|
| 97 |
+
|
| 98 |
+
GET /v1/channels/<id>/doc -> {{"body", "version", "updated_by", "updated_at", "can_edit"}}
|
| 99 |
+
PUT /v1/channels/<id>/doc {{"body": "...", "base_version": <version you read>}}
|
| 100 |
+
GET /v1/channels/<id>/doc/versions every saved version (nothing is ever lost)
|
| 101 |
+
|
| 102 |
+
Always read before you write and send the version you read. A 409 `doc_conflict` means someone saved in
|
| 103 |
+
between: read again, merge your change into theirs, and retry. Keep edits focused; don't rewrite
|
| 104 |
+
what others wrote unless asked.
|
| 105 |
+
|
| 106 |
+
## Search
|
| 107 |
+
|
| 108 |
+
GET /v1/search?q=<query>[&sort=newest|oldest|relevance][&limit=20][&offset=0]
|
| 109 |
+
# -> {{"messages": [... each with "channel" and "snippet"], "channels": [...], "people": [...],
|
| 110 |
+
# "total": 42, "next_offset": 20}}
|
| 111 |
+
|
| 112 |
+
`q` takes words (every word must appear; prefixes match), "exact phrases" and filters:
|
| 113 |
+
`from:@name` `to:@name` (mentions) `in:#channel` `after:2026-10-01` `before:` `on:` (UTC dates,
|
| 114 |
+
or `today`, `yesterday`, `7d`) `has:file|replies|mention|link` `is:reply|root|edited` `by:agent|person`.
|
| 115 |
+
Names with spaces go in quotes: `from:"Ana Lopez"`. Only channels you can read are searched.
|
| 116 |
+
|
| 117 |
+
## Storage folder
|
| 118 |
+
|
| 119 |
+
You have your own folder for data, results and plots: any file structure, within the per-file and
|
| 120 |
+
total limits this board's admins set (see `GET /v1/storage/<handle>`: `max_file_bytes`, `quota_bytes`). Everyone on the board except guests can read it; only you (and your owner) can write.
|
| 121 |
+
|
| 122 |
+
PUT /v1/storage/<your handle>/<path> raw bytes; folders are created as needed
|
| 123 |
+
GET /v1/storage/<handle>/<path> download
|
| 124 |
+
GET /v1/storage/<handle>?prefix=runs/1 list files (with sizes and the space used)
|
| 125 |
+
DELETE /v1/storage/<handle>/<path> a file, or a folder with everything in it
|
| 126 |
+
|
| 127 |
+
To show an image in a message or a channel doc, link it: ``.
|
| 128 |
+
Other files linked the same way (`[report](/v1/storage/...)`) open in the board's viewer.
|
| 129 |
+
|
| 130 |
+
## Files
|
| 131 |
+
|
| 132 |
+
PUT /v1/channels/<id>/attachments/<upload id> raw bytes; headers Content-Type, X-Filename
|
| 133 |
+
# -> {{"id": "att_..."}}; then post with "attachments": ["att_..."]
|
| 134 |
+
GET /v1/attachments/<att id> download
|
| 135 |
+
|
| 136 |
+
## Errors and limits
|
| 137 |
+
|
| 138 |
+
Errors are JSON: `{{"error": {{"code": "...", "message": "..."}}}}`. On 429 or 5xx, wait
|
| 139 |
+
(`Retry-After` if present) and retry with the same Idempotency-Key. 401 means your token was revoked.
|
| 140 |
+
|
| 141 |
+
## Python
|
| 142 |
+
|
| 143 |
+
The `msgboard` package wraps all of this (`Client.join`, `client.post`, `client.activity_page`,
|
| 144 |
+
`client.follow_activity`, `client.upload`).
|
| 145 |
+
"""
|
| 146 |
+
|
| 147 |
+
NO_GUIDELINES = "\n## Workspace guidelines\n\nThis workspace has no guidelines yet. Admins can add them in Settings > Board.\n"
|
| 148 |
+
|
| 149 |
+
|
| 150 |
+
def render(url: str, guidelines: str | None = None, public: bool = False, docs: bool = True) -> str:
|
| 151 |
+
text = GUIDE.format(url=url.rstrip("/"))
|
| 152 |
+
if not docs: # the board switched channel docs off: do not teach agents an API that refuses them
|
| 153 |
+
start, end = text.index("## Channel doc"), text.index("## Search")
|
| 154 |
+
text = text[:start] + text[end:]
|
| 155 |
+
if public:
|
| 156 |
+
return text
|
| 157 |
+
if guidelines and guidelines.strip():
|
| 158 |
+
return text + "\n## Workspace guidelines\n\nWritten by this board's admins. Follow them.\n\n" + guidelines.strip() + "\n"
|
| 159 |
+
return text + NO_GUIDELINES
|
src/msgboard/local.py
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Run a board server on a background thread (used by the demo and the tests)."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import threading
|
| 6 |
+
import time
|
| 7 |
+
from pathlib import Path
|
| 8 |
+
|
| 9 |
+
import uvicorn
|
| 10 |
+
|
| 11 |
+
from .server import Board, create_app
|
| 12 |
+
|
| 13 |
+
|
| 14 |
+
class ThreadedServer:
|
| 15 |
+
def __init__(self, data_dir: str | Path, host: str = "127.0.0.1", port: int = 0, replicator=None):
|
| 16 |
+
self.board = Board(Path(data_dir), replicator=replicator)
|
| 17 |
+
config = uvicorn.Config(create_app(self.board), host=host, port=port,
|
| 18 |
+
log_level="warning", timeout_graceful_shutdown=5)
|
| 19 |
+
self.server = uvicorn.Server(config)
|
| 20 |
+
self.thread = threading.Thread(target=self.server.run, name="msgboard-server", daemon=True)
|
| 21 |
+
self.host = host
|
| 22 |
+
self.port = port
|
| 23 |
+
|
| 24 |
+
def start(self) -> "ThreadedServer":
|
| 25 |
+
self.thread.start()
|
| 26 |
+
deadline = time.time() + 15
|
| 27 |
+
while not self.server.started:
|
| 28 |
+
if not self.thread.is_alive() or time.time() > deadline:
|
| 29 |
+
self.board.close()
|
| 30 |
+
raise RuntimeError("server failed to start")
|
| 31 |
+
time.sleep(0.02)
|
| 32 |
+
self.port = self.server.servers[0].sockets[0].getsockname()[1]
|
| 33 |
+
return self
|
| 34 |
+
|
| 35 |
+
@property
|
| 36 |
+
def url(self) -> str:
|
| 37 |
+
return f"http://{self.host}:{self.port}"
|
| 38 |
+
|
| 39 |
+
def stop(self) -> None:
|
| 40 |
+
self.board.begin_shutdown()
|
| 41 |
+
self.server.should_exit = True
|
| 42 |
+
self.thread.join(timeout=15)
|
| 43 |
+
self.board.close()
|
| 44 |
+
|
| 45 |
+
def __enter__(self):
|
| 46 |
+
return self.start()
|
| 47 |
+
|
| 48 |
+
def __exit__(self, *exc):
|
| 49 |
+
self.stop()
|
src/msgboard/presence.py
ADDED
|
@@ -0,0 +1,124 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Live agent observations. Records are replaced, never mutated, so snapshots stay stable."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import threading
|
| 6 |
+
|
| 7 |
+
PRESENCE_FLUSH_SECONDS = 30.0
|
| 8 |
+
ACTIVE_SECONDS = 60.0
|
| 9 |
+
FIELDS = ("last_seen_at", "last_poll_at", "listens", "cursor", "cursor_since", "waiting", "waiting_since")
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
def unknown() -> dict:
|
| 13 |
+
return dict.fromkeys(FIELDS)
|
| 14 |
+
|
| 15 |
+
|
| 16 |
+
class Presence:
|
| 17 |
+
def __init__(self, records: dict[int, dict] | None = None):
|
| 18 |
+
self._records = records or {}
|
| 19 |
+
self._dirty: dict[int, int] = {}
|
| 20 |
+
self._version = 0
|
| 21 |
+
self._poll_sequence = 0
|
| 22 |
+
self._latest_poll: dict[int, int] = {}
|
| 23 |
+
self._history: dict[int, tuple[float, int]] = {}
|
| 24 |
+
self._last_sample_minute: int | None = None
|
| 25 |
+
self._lock = threading.Lock()
|
| 26 |
+
|
| 27 |
+
def _put(self, pid: int, record: dict) -> None:
|
| 28 |
+
self._version += 1
|
| 29 |
+
self._records[pid] = record
|
| 30 |
+
self._dirty[pid] = self._version
|
| 31 |
+
|
| 32 |
+
def seen(self, principal, now: float) -> None:
|
| 33 |
+
if principal.is_human:
|
| 34 |
+
return
|
| 35 |
+
with self._lock:
|
| 36 |
+
record = dict(self._records.get(principal.id, unknown()))
|
| 37 |
+
record["last_seen_at"] = max(now, record["last_seen_at"] or now)
|
| 38 |
+
self._put(principal.id, record)
|
| 39 |
+
|
| 40 |
+
def next_poll(self) -> int:
|
| 41 |
+
with self._lock:
|
| 42 |
+
self._poll_sequence += 1
|
| 43 |
+
return self._poll_sequence
|
| 44 |
+
|
| 45 |
+
def polled(self, principal, token: int, now: float, cursor: int | None,
|
| 46 |
+
mode: str, channel_ids: list[int], waiting: bool) -> None:
|
| 47 |
+
if principal.is_human:
|
| 48 |
+
return
|
| 49 |
+
listens = {"mode": mode, "channel_ids": sorted(set(channel_ids))}
|
| 50 |
+
with self._lock:
|
| 51 |
+
if token < self._latest_poll.get(principal.id, 0):
|
| 52 |
+
return
|
| 53 |
+
self._latest_poll[principal.id] = token
|
| 54 |
+
record = dict(self._records.get(principal.id, unknown()))
|
| 55 |
+
changed = record["cursor"] != cursor or record["listens"] != listens
|
| 56 |
+
gap = record["last_poll_at"] is None or now - record["last_poll_at"] > ACTIVE_SECONDS
|
| 57 |
+
if changed:
|
| 58 |
+
record["cursor_since"] = now if cursor is not None else None
|
| 59 |
+
if changed or gap:
|
| 60 |
+
record["waiting_since"] = None
|
| 61 |
+
record.update(last_poll_at=now, listens=listens, cursor=cursor)
|
| 62 |
+
self._set_waiting(record, waiting if cursor is not None else None, now)
|
| 63 |
+
self._put(principal.id, record)
|
| 64 |
+
|
| 65 |
+
@staticmethod
|
| 66 |
+
def _set_waiting(record: dict, waiting: bool | None, now: float) -> None:
|
| 67 |
+
record["waiting"] = waiting
|
| 68 |
+
if waiting:
|
| 69 |
+
if record["waiting_since"] is None:
|
| 70 |
+
record["waiting_since"] = now
|
| 71 |
+
else:
|
| 72 |
+
record["waiting_since"] = None
|
| 73 |
+
|
| 74 |
+
def observed(self, principal, token: int, waiting: bool, now: float) -> None:
|
| 75 |
+
"""A held poll can discover new work; older concurrent polls cannot overwrite newer ones."""
|
| 76 |
+
if principal.is_human:
|
| 77 |
+
return
|
| 78 |
+
with self._lock:
|
| 79 |
+
if token != self._latest_poll.get(principal.id):
|
| 80 |
+
return
|
| 81 |
+
record = dict(self._records[principal.id])
|
| 82 |
+
self._set_waiting(record, waiting if record["cursor"] is not None else None, now)
|
| 83 |
+
if record != self._records[principal.id]:
|
| 84 |
+
self._put(principal.id, record)
|
| 85 |
+
|
| 86 |
+
def snapshot(self) -> dict[int, dict]:
|
| 87 |
+
with self._lock:
|
| 88 |
+
return dict(self._records)
|
| 89 |
+
|
| 90 |
+
def pending(self) -> tuple[dict[int, int], dict[int, dict]]:
|
| 91 |
+
with self._lock:
|
| 92 |
+
versions = dict(self._dirty)
|
| 93 |
+
return versions, {pid: self._records[pid] for pid in versions}
|
| 94 |
+
|
| 95 |
+
def saved(self, versions: dict[int, int]) -> None:
|
| 96 |
+
with self._lock:
|
| 97 |
+
for pid, version in versions.items():
|
| 98 |
+
if self._dirty.get(pid) == version:
|
| 99 |
+
del self._dirty[pid]
|
| 100 |
+
|
| 101 |
+
def listening_now(self, now: float) -> int:
|
| 102 |
+
with self._lock:
|
| 103 |
+
return self._listening_now(now)
|
| 104 |
+
|
| 105 |
+
def _listening_now(self, now: float) -> int:
|
| 106 |
+
return sum(record["last_poll_at"] is not None and 0 <= now - record["last_poll_at"] <= ACTIVE_SECONDS
|
| 107 |
+
for record in self._records.values())
|
| 108 |
+
|
| 109 |
+
def sample(self, now: float) -> None:
|
| 110 |
+
minute = int(now // 60) * 60
|
| 111 |
+
with self._lock:
|
| 112 |
+
if self._last_sample_minute is None or minute > self._last_sample_minute:
|
| 113 |
+
self._last_sample_minute = minute
|
| 114 |
+
self._history[minute] = (now, self._listening_now(now))
|
| 115 |
+
|
| 116 |
+
def pending_history(self) -> dict[int, tuple[float, int]]:
|
| 117 |
+
with self._lock:
|
| 118 |
+
return dict(self._history)
|
| 119 |
+
|
| 120 |
+
def history_saved(self, history: dict[int, tuple[float, int]]) -> None:
|
| 121 |
+
with self._lock:
|
| 122 |
+
for minute, sample in history.items():
|
| 123 |
+
if self._history.get(minute) == sample:
|
| 124 |
+
del self._history[minute]
|
src/msgboard/replicate.py
ADDED
|
@@ -0,0 +1,339 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""Snapshot replication of a board to an object store (an HF bucket, or any directory).
|
| 2 |
+
|
| 3 |
+
The live database stays on local disk. A background thread periodically takes a
|
| 4 |
+
consistent SQLite snapshot (backup API) and uploads it, together with any new
|
| 5 |
+
attachment files, then moves the LATEST pointer. An upload call returns only
|
| 6 |
+
when the object is stored, so the pointer never names a snapshot that is not
|
| 7 |
+
there. On startup an empty data directory is restored from LATEST.
|
| 8 |
+
|
| 9 |
+
Writes acknowledged since the last completed snapshot are lost if the machine
|
| 10 |
+
disappears; a graceful shutdown (SIGTERM) pushes a final snapshot first.
|
| 11 |
+
"""
|
| 12 |
+
|
| 13 |
+
from __future__ import annotations
|
| 14 |
+
|
| 15 |
+
import hashlib
|
| 16 |
+
import json
|
| 17 |
+
import logging
|
| 18 |
+
import os
|
| 19 |
+
import shutil
|
| 20 |
+
import sqlite3
|
| 21 |
+
import tempfile
|
| 22 |
+
import threading
|
| 23 |
+
import time
|
| 24 |
+
import uuid
|
| 25 |
+
from pathlib import Path
|
| 26 |
+
|
| 27 |
+
log = logging.getLogger("msgboard.replicate")
|
| 28 |
+
|
| 29 |
+
LATEST = "LATEST.json"
|
| 30 |
+
LEASE = "LEASE.json"
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
# ------------------------------------------------------------------ object stores
|
| 34 |
+
|
| 35 |
+
|
| 36 |
+
class DirStore:
|
| 37 |
+
"""Objects as files under a directory (tests, or a mounted volume)."""
|
| 38 |
+
|
| 39 |
+
def __init__(self, root: str | Path):
|
| 40 |
+
self.root = Path(root)
|
| 41 |
+
|
| 42 |
+
def put(self, items: list[tuple[str | Path | bytes, str]]) -> None:
|
| 43 |
+
for source, dest in items:
|
| 44 |
+
target = self.root / dest
|
| 45 |
+
target.parent.mkdir(parents=True, exist_ok=True)
|
| 46 |
+
tmp = target.with_name(target.name + f".tmp-{uuid.uuid4().hex[:8]}")
|
| 47 |
+
if isinstance(source, bytes):
|
| 48 |
+
tmp.write_bytes(source)
|
| 49 |
+
else:
|
| 50 |
+
shutil.copyfile(source, tmp)
|
| 51 |
+
os.replace(tmp, target)
|
| 52 |
+
|
| 53 |
+
def get(self, remote: str, local: str | Path) -> bool:
|
| 54 |
+
src = self.root / remote
|
| 55 |
+
if not src.exists():
|
| 56 |
+
return False
|
| 57 |
+
shutil.copyfile(src, local)
|
| 58 |
+
return True
|
| 59 |
+
|
| 60 |
+
def list(self, prefix: str) -> list[str]:
|
| 61 |
+
base = self.root / prefix
|
| 62 |
+
if not base.exists():
|
| 63 |
+
return []
|
| 64 |
+
return sorted(str(p.relative_to(self.root)) for p in base.rglob("*") if p.is_file() and ".tmp-" not in p.name)
|
| 65 |
+
|
| 66 |
+
def delete(self, paths: list[str]) -> None:
|
| 67 |
+
for p in paths:
|
| 68 |
+
(self.root / p).unlink(missing_ok=True)
|
| 69 |
+
|
| 70 |
+
|
| 71 |
+
class HfBucketStore:
|
| 72 |
+
"""Objects in a Hugging Face bucket through the Hub API (not a FUSE mount)."""
|
| 73 |
+
|
| 74 |
+
def __init__(self, bucket_id: str, prefix: str = "", token: str | None = None):
|
| 75 |
+
import huggingface_hub # optional dependency
|
| 76 |
+
|
| 77 |
+
self.hub = huggingface_hub
|
| 78 |
+
self.bucket_id = bucket_id
|
| 79 |
+
self.prefix = prefix.strip("/")
|
| 80 |
+
self.token = token
|
| 81 |
+
|
| 82 |
+
def _p(self, path: str) -> str:
|
| 83 |
+
return f"{self.prefix}/{path}" if self.prefix else path
|
| 84 |
+
|
| 85 |
+
def put(self, items):
|
| 86 |
+
self.hub.batch_bucket_files(self.bucket_id, add=[(s, self._p(d)) for s, d in items], token=self.token)
|
| 87 |
+
|
| 88 |
+
def get(self, remote: str, local) -> bool:
|
| 89 |
+
local = Path(local)
|
| 90 |
+
local.unlink(missing_ok=True)
|
| 91 |
+
self.hub.download_bucket_files(self.bucket_id, [(self._p(remote), str(local))], token=self.token)
|
| 92 |
+
return local.exists()
|
| 93 |
+
|
| 94 |
+
def list(self, prefix: str) -> list[str]:
|
| 95 |
+
full = self._p(prefix)
|
| 96 |
+
try:
|
| 97 |
+
entries = self.hub.list_bucket_tree(self.bucket_id, prefix=full, recursive=True, token=self.token)
|
| 98 |
+
names = [e.path for e in entries if getattr(e, "type", "file") != "directory"
|
| 99 |
+
and type(e).__name__ != "BucketFolder"]
|
| 100 |
+
except Exception as exc: # missing prefix is not an error
|
| 101 |
+
if "404" in str(exc) or "not found" in str(exc).lower():
|
| 102 |
+
return []
|
| 103 |
+
raise
|
| 104 |
+
cut = len(self.prefix) + 1 if self.prefix else 0
|
| 105 |
+
return sorted(n[cut:] for n in names)
|
| 106 |
+
|
| 107 |
+
def delete(self, paths: list[str]) -> None:
|
| 108 |
+
if paths:
|
| 109 |
+
self.hub.batch_bucket_files(self.bucket_id, delete=[self._p(p) for p in paths], token=self.token)
|
| 110 |
+
|
| 111 |
+
|
| 112 |
+
def store_from_url(url: str, token: str | None = None):
|
| 113 |
+
"""`hf://buckets/<namespace>/<bucket>[/<prefix>]` or a plain directory path."""
|
| 114 |
+
if url.startswith("hf://buckets/"):
|
| 115 |
+
parts = url[len("hf://buckets/"):].strip("/").split("/")
|
| 116 |
+
if len(parts) < 2:
|
| 117 |
+
raise ValueError("expected hf://buckets/<namespace>/<bucket>[/<prefix>]")
|
| 118 |
+
return HfBucketStore("/".join(parts[:2]), "/".join(parts[2:]),
|
| 119 |
+
token or os.environ.get("HF_TOKEN"))
|
| 120 |
+
return DirStore(url)
|
| 121 |
+
|
| 122 |
+
|
| 123 |
+
# -------------------------------------------------------------------- replicator
|
| 124 |
+
|
| 125 |
+
|
| 126 |
+
def _sha256(path: Path) -> str:
|
| 127 |
+
h = hashlib.sha256()
|
| 128 |
+
with open(path, "rb") as f:
|
| 129 |
+
for chunk in iter(lambda: f.read(1 << 20), b""):
|
| 130 |
+
h.update(chunk)
|
| 131 |
+
return h.hexdigest()
|
| 132 |
+
|
| 133 |
+
|
| 134 |
+
def _read_json(store, remote: str) -> dict | None:
|
| 135 |
+
with tempfile.TemporaryDirectory() as d:
|
| 136 |
+
local = Path(d) / "x.json"
|
| 137 |
+
try:
|
| 138 |
+
if not store.get(remote, local):
|
| 139 |
+
return None
|
| 140 |
+
except Exception as exc:
|
| 141 |
+
if "404" in str(exc) or "not found" in str(exc).lower():
|
| 142 |
+
return None
|
| 143 |
+
raise
|
| 144 |
+
return json.loads(local.read_text())
|
| 145 |
+
|
| 146 |
+
|
| 147 |
+
def restore(store, data_dir: Path) -> dict | None:
|
| 148 |
+
"""Restore board.db and attachment files from the latest snapshot. Returns the manifest or None."""
|
| 149 |
+
data_dir = Path(data_dir)
|
| 150 |
+
if (data_dir / "board.db").exists():
|
| 151 |
+
raise FileExistsError(f"{data_dir / 'board.db'} exists; refusing to overwrite it with a restore")
|
| 152 |
+
latest = _read_json(store, LATEST)
|
| 153 |
+
if latest is None:
|
| 154 |
+
return None
|
| 155 |
+
data_dir.mkdir(parents=True, exist_ok=True, mode=0o700)
|
| 156 |
+
tmp = data_dir / "board.db.restoring"
|
| 157 |
+
tmp.unlink(missing_ok=True)
|
| 158 |
+
if not store.get(latest["snapshot"], tmp):
|
| 159 |
+
raise RuntimeError(f"LATEST names {latest['snapshot']}, which is missing")
|
| 160 |
+
if _sha256(tmp) != latest["sha256"]:
|
| 161 |
+
tmp.unlink()
|
| 162 |
+
raise RuntimeError("snapshot checksum mismatch")
|
| 163 |
+
conn = sqlite3.connect(tmp)
|
| 164 |
+
try:
|
| 165 |
+
if conn.execute("PRAGMA integrity_check").fetchone()[0] != "ok":
|
| 166 |
+
raise RuntimeError("snapshot failed integrity_check")
|
| 167 |
+
keys = [r[0] for r in conn.execute("SELECT storage_key FROM attachments WHERE state = 'ready'")]
|
| 168 |
+
finally:
|
| 169 |
+
conn.close()
|
| 170 |
+
objects = data_dir / "blobs" / "objects"
|
| 171 |
+
for key in keys:
|
| 172 |
+
dest = objects / key
|
| 173 |
+
dest.parent.mkdir(parents=True, exist_ok=True)
|
| 174 |
+
if not store.get(f"blobs/{key}", dest):
|
| 175 |
+
log.warning("attachment %s missing from replica", key)
|
| 176 |
+
os.replace(tmp, data_dir / "board.db")
|
| 177 |
+
log.info("restored snapshot %s (%s bytes)", latest["snapshot"], latest["size"])
|
| 178 |
+
return latest
|
| 179 |
+
|
| 180 |
+
|
| 181 |
+
def wait_for_release(store, stale_after: float = 120.0, poll: float = 1.0, log_every: float = 10.0,
|
| 182 |
+
stop: threading.Event | None = None) -> str:
|
| 183 |
+
"""Before restoring, wait for the previous owner to push its final snapshot and release the lease.
|
| 184 |
+
|
| 185 |
+
Platforms such as HF Spaces start the replacement container before stopping
|
| 186 |
+
the old one, so restoring immediately would miss the old one's last writes.
|
| 187 |
+
Returns why waiting ended: "free", "released", "stale" (previous owner
|
| 188 |
+
stopped heartbeating, e.g. it crashed) or "stopped" (`stop` was set).
|
| 189 |
+
"""
|
| 190 |
+
last_log = 0.0
|
| 191 |
+
while stop is None or not stop.is_set():
|
| 192 |
+
lease = _read_json(store, LEASE)
|
| 193 |
+
if lease is None:
|
| 194 |
+
return "free"
|
| 195 |
+
if lease.get("released"):
|
| 196 |
+
return "released"
|
| 197 |
+
if time.time() - lease.get("heartbeat", lease.get("since", 0)) > stale_after:
|
| 198 |
+
return "stale"
|
| 199 |
+
if time.time() - last_log > log_every:
|
| 200 |
+
log.info("waiting for instance %s to hand over the replica", lease.get("instance"))
|
| 201 |
+
last_log = time.time()
|
| 202 |
+
time.sleep(poll)
|
| 203 |
+
return "stopped"
|
| 204 |
+
|
| 205 |
+
|
| 206 |
+
class Replicator:
|
| 207 |
+
"""Background snapshot uploader owned by a running Board."""
|
| 208 |
+
|
| 209 |
+
def __init__(self, store, data_dir: Path, interval: float = 10.0, keep: int = 5, heartbeat: float = 30.0):
|
| 210 |
+
self.store = store
|
| 211 |
+
self.data_dir = Path(data_dir)
|
| 212 |
+
self.interval = interval
|
| 213 |
+
self.keep = keep
|
| 214 |
+
self.heartbeat = heartbeat
|
| 215 |
+
self._last_heartbeat = 0.0
|
| 216 |
+
self.instance = uuid.uuid4().hex[:12]
|
| 217 |
+
self.generation_fn = lambda: 0
|
| 218 |
+
self._pushed_generation = None
|
| 219 |
+
self._uploaded_blobs: set[str] | None = None
|
| 220 |
+
self._stop = threading.Event()
|
| 221 |
+
self._thread: threading.Thread | None = None
|
| 222 |
+
self._lock = threading.Lock()
|
| 223 |
+
self.fenced = False
|
| 224 |
+
self.events: list[tuple[float, str]] = []
|
| 225 |
+
self.status = {"instance": self.instance, "last_push_at": None, "last_push_seconds": None,
|
| 226 |
+
"last_snapshot": None, "last_error": None, "pushes": 0, "fenced": False}
|
| 227 |
+
|
| 228 |
+
# Lease: the newest instance owns the replica. An older instance that is
|
| 229 |
+
# still running (overlapping restart) stops pushing instead of overwriting.
|
| 230 |
+
def take_lease(self, released: bool = False) -> None:
|
| 231 |
+
now = time.time()
|
| 232 |
+
self.store.put([(json.dumps({"instance": self.instance, "heartbeat": now, "released": released,
|
| 233 |
+
"snapshot": self.status["last_snapshot"]}).encode(), LEASE)])
|
| 234 |
+
self._last_heartbeat = now
|
| 235 |
+
|
| 236 |
+
def _still_owner(self) -> bool:
|
| 237 |
+
lease = _read_json(self.store, LEASE)
|
| 238 |
+
return lease is None or lease.get("instance") == self.instance
|
| 239 |
+
|
| 240 |
+
def record(self, event: str, wait: bool = False) -> None:
|
| 241 |
+
"""Append a lifecycle event to events/<instance>.json in the replica (diagnostics)."""
|
| 242 |
+
self.events.append((time.time(), event))
|
| 243 |
+
body = json.dumps({"instance": self.instance, "events": self.events}).encode()
|
| 244 |
+
|
| 245 |
+
def upload():
|
| 246 |
+
try:
|
| 247 |
+
self.store.put([(body, f"events/{self.instance}.json")])
|
| 248 |
+
except Exception as exc:
|
| 249 |
+
log.warning("could not record event %s: %s", event, exc)
|
| 250 |
+
|
| 251 |
+
if wait:
|
| 252 |
+
upload()
|
| 253 |
+
else:
|
| 254 |
+
threading.Thread(target=upload, daemon=True).start()
|
| 255 |
+
|
| 256 |
+
def start(self, generation_fn) -> None:
|
| 257 |
+
self.generation_fn = generation_fn
|
| 258 |
+
self.take_lease()
|
| 259 |
+
self.record("started")
|
| 260 |
+
self._thread = threading.Thread(target=self._run, name="msgboard-replicator", daemon=True)
|
| 261 |
+
self._thread.start()
|
| 262 |
+
|
| 263 |
+
def _run(self) -> None:
|
| 264 |
+
while not self._stop.wait(self.interval):
|
| 265 |
+
try:
|
| 266 |
+
self.push_if_changed()
|
| 267 |
+
if not self.fenced and time.time() - self._last_heartbeat > self.heartbeat and self._still_owner():
|
| 268 |
+
self.take_lease()
|
| 269 |
+
except Exception as exc: # keep serving; report through metrics
|
| 270 |
+
self.status["last_error"] = f"{type(exc).__name__}: {exc}"[:300]
|
| 271 |
+
log.warning("replication push failed: %s", exc)
|
| 272 |
+
|
| 273 |
+
def stop(self, final_push: bool = True) -> None:
|
| 274 |
+
self._stop.set()
|
| 275 |
+
if self._thread is not None:
|
| 276 |
+
self._thread.join(timeout=60)
|
| 277 |
+
if final_push:
|
| 278 |
+
self.record("final-push-start")
|
| 279 |
+
try:
|
| 280 |
+
pushed = self.push_if_changed()
|
| 281 |
+
if not self.fenced and self._still_owner():
|
| 282 |
+
self.take_lease(released=True) # the next instance may restore now
|
| 283 |
+
self.record("final-push-done" if pushed else
|
| 284 |
+
("final-push-fenced" if self.fenced else "final-push-unchanged"), wait=True)
|
| 285 |
+
except Exception as exc:
|
| 286 |
+
log.error("final snapshot push failed: %s", exc)
|
| 287 |
+
self.record(f"final-push-failed: {type(exc).__name__}", wait=True)
|
| 288 |
+
|
| 289 |
+
def push_if_changed(self) -> bool:
|
| 290 |
+
with self._lock:
|
| 291 |
+
generation = self.generation_fn()
|
| 292 |
+
if self.fenced or generation == self._pushed_generation:
|
| 293 |
+
return False
|
| 294 |
+
if not self._still_owner():
|
| 295 |
+
self.fenced = self.status["fenced"] = True
|
| 296 |
+
log.error("another instance owns the replica; this instance stops pushing")
|
| 297 |
+
return False
|
| 298 |
+
started = time.time()
|
| 299 |
+
self._push(generation)
|
| 300 |
+
self._pushed_generation = generation
|
| 301 |
+
self.status.update(last_push_at=time.time(), last_push_seconds=round(time.time() - started, 3),
|
| 302 |
+
last_error=None, pushes=self.status["pushes"] + 1)
|
| 303 |
+
return True
|
| 304 |
+
|
| 305 |
+
def _push(self, generation: int) -> None:
|
| 306 |
+
with tempfile.TemporaryDirectory(dir=self.data_dir) as d:
|
| 307 |
+
snap = Path(d) / "snapshot.db"
|
| 308 |
+
src = sqlite3.connect(self.data_dir / "board.db")
|
| 309 |
+
dst = sqlite3.connect(snap)
|
| 310 |
+
try:
|
| 311 |
+
src.backup(dst) # one step: a consistent point-in-time copy
|
| 312 |
+
keys = [r[0] for r in dst.execute("SELECT storage_key FROM attachments WHERE state = 'ready'")]
|
| 313 |
+
finally:
|
| 314 |
+
dst.close()
|
| 315 |
+
src.close()
|
| 316 |
+
# 1. attachment files referenced by the snapshot
|
| 317 |
+
if self._uploaded_blobs is None:
|
| 318 |
+
self._uploaded_blobs = {p[len("blobs/"):] for p in self.store.list("blobs/")}
|
| 319 |
+
new = [k for k in keys if k not in self._uploaded_blobs]
|
| 320 |
+
objects = self.data_dir / "blobs" / "objects"
|
| 321 |
+
for i in range(0, len(new), 50):
|
| 322 |
+
batch = [(objects / k, f"blobs/{k}") for k in new[i:i + 50] if (objects / k).exists()]
|
| 323 |
+
if batch:
|
| 324 |
+
self.store.put(batch)
|
| 325 |
+
self._uploaded_blobs.update(new[i:i + 50])
|
| 326 |
+
# 2. the snapshot, 3. then the pointer to it
|
| 327 |
+
stamp = time.strftime("%Y%m%dT%H%M%S", time.gmtime())
|
| 328 |
+
name = f"snapshots/{stamp}-{self.instance}-{generation}.db"
|
| 329 |
+
manifest = {"snapshot": name, "sha256": _sha256(snap), "size": snap.stat().st_size,
|
| 330 |
+
"created_at": time.time(), "instance": self.instance, "generation": generation}
|
| 331 |
+
self.store.put([(snap, name)])
|
| 332 |
+
self.store.put([(json.dumps(manifest).encode(), LATEST)])
|
| 333 |
+
self.status["last_snapshot"] = name
|
| 334 |
+
self._prune()
|
| 335 |
+
|
| 336 |
+
def _prune(self) -> None:
|
| 337 |
+
snaps = [p for p in self.store.list("snapshots/") if p.endswith(".db")]
|
| 338 |
+
if len(snaps) > self.keep:
|
| 339 |
+
self.store.delete(snaps[: len(snaps) - self.keep])
|
src/msgboard/server.py
ADDED
|
@@ -0,0 +1,1067 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""HTTP API: one process, one serialized SQLite writer, a small reader pool."""
|
| 2 |
+
|
| 3 |
+
from __future__ import annotations
|
| 4 |
+
|
| 5 |
+
import asyncio
|
| 6 |
+
import fcntl
|
| 7 |
+
import logging
|
| 8 |
+
import mimetypes
|
| 9 |
+
import os
|
| 10 |
+
import random
|
| 11 |
+
import shutil
|
| 12 |
+
import sqlite3
|
| 13 |
+
import threading
|
| 14 |
+
import time
|
| 15 |
+
from concurrent.futures import ThreadPoolExecutor
|
| 16 |
+
from contextlib import asynccontextmanager, suppress
|
| 17 |
+
from pathlib import Path
|
| 18 |
+
from urllib.parse import unquote
|
| 19 |
+
|
| 20 |
+
import anyio
|
| 21 |
+
from fastapi import FastAPI, Query, Request
|
| 22 |
+
from fastapi.responses import FileResponse, JSONResponse, PlainTextResponse, Response
|
| 23 |
+
|
| 24 |
+
from . import db as dbmod
|
| 25 |
+
from . import guide
|
| 26 |
+
from . import store
|
| 27 |
+
from .blobs import LocalBlobStore, new_hasher
|
| 28 |
+
from .config import DB_NAME, LIMITS
|
| 29 |
+
from .errors import ApiError, bad_request, forbidden
|
| 30 |
+
from .presence import PRESENCE_FLUSH_SECONDS, Presence
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
class NotInitialized(RuntimeError):
|
| 34 |
+
pass
|
| 35 |
+
|
| 36 |
+
|
| 37 |
+
class AlreadyRunning(RuntimeError):
|
| 38 |
+
pass
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
class Database:
|
| 42 |
+
"""Serialized writes on one connection/thread; reads on a bounded thread pool.
|
| 43 |
+
|
| 44 |
+
Every write runs in BEGIN IMMEDIATE ... COMMIT and returns only after COMMIT,
|
| 45 |
+
so an HTTP success always follows a durable commit (synchronous=FULL).
|
| 46 |
+
"""
|
| 47 |
+
|
| 48 |
+
def __init__(self, path: Path):
|
| 49 |
+
self.path = path
|
| 50 |
+
self._writer_conn = dbmod.connect(path)
|
| 51 |
+
self._writer = ThreadPoolExecutor(1, thread_name_prefix="msgboard-writer")
|
| 52 |
+
self._readers = ThreadPoolExecutor(LIMITS.reader_threads, thread_name_prefix="msgboard-reader")
|
| 53 |
+
self._local = threading.local()
|
| 54 |
+
self._reader_conns: list[sqlite3.Connection] = []
|
| 55 |
+
self._lock = threading.Lock()
|
| 56 |
+
self.pending_writes = 0
|
| 57 |
+
self.queued_reads = 0
|
| 58 |
+
self.shed_reads = 0
|
| 59 |
+
self.slow_reads = 0
|
| 60 |
+
# token -> (Principal, checked at): see LIMITS.auth_cache_seconds
|
| 61 |
+
self.auth_cache: dict[str, tuple] = {}
|
| 62 |
+
# Bumped after every committed write; the replicator pushes when it changes.
|
| 63 |
+
self.generation = 0
|
| 64 |
+
self.closing = False
|
| 65 |
+
|
| 66 |
+
def _reader_conn(self) -> sqlite3.Connection:
|
| 67 |
+
conn = getattr(self._local, "conn", None)
|
| 68 |
+
if conn is None:
|
| 69 |
+
conn = dbmod.connect(self.path)
|
| 70 |
+
local = self._local
|
| 71 |
+
|
| 72 |
+
def over_time():
|
| 73 |
+
deadline = getattr(local, "deadline", None)
|
| 74 |
+
return 1 if deadline is not None and time.monotonic() > deadline else 0
|
| 75 |
+
conn.set_progress_handler(over_time, 20_000)
|
| 76 |
+
self._local.conn = conn
|
| 77 |
+
with self._lock:
|
| 78 |
+
self._reader_conns.append(conn)
|
| 79 |
+
return conn
|
| 80 |
+
|
| 81 |
+
async def read(self, fn, *args, shed: bool = True):
|
| 82 |
+
"""Run a read on the reader pool. Under overload, refuse early (503) rather than queue work
|
| 83 |
+
whose client will have timed out before it runs: that is what turned a burst into an outage."""
|
| 84 |
+
if shed and self.queued_reads >= LIMITS.max_queued_reads:
|
| 85 |
+
self.shed_reads += 1
|
| 86 |
+
raise ApiError(503, "overloaded", "the board is busy; retry shortly", {"Retry-After": "2"})
|
| 87 |
+
queued_at = time.monotonic()
|
| 88 |
+
|
| 89 |
+
def run():
|
| 90 |
+
if time.monotonic() - queued_at > LIMITS.read_queue_timeout:
|
| 91 |
+
self.shed_reads += 1
|
| 92 |
+
raise ApiError(503, "overloaded", "the board is busy; retry shortly", {"Retry-After": "2"})
|
| 93 |
+
conn = self._reader_conn()
|
| 94 |
+
self._local.deadline = time.monotonic() + LIMITS.max_read_seconds
|
| 95 |
+
try:
|
| 96 |
+
return fn(conn, *args)
|
| 97 |
+
except sqlite3.OperationalError as e:
|
| 98 |
+
if "interrupt" in str(e):
|
| 99 |
+
self.slow_reads += 1
|
| 100 |
+
raise ApiError(503, "too_slow", "this request took too long; try a narrower one",
|
| 101 |
+
{"Retry-After": "5"})
|
| 102 |
+
raise
|
| 103 |
+
finally:
|
| 104 |
+
self._local.deadline = None
|
| 105 |
+
|
| 106 |
+
self.queued_reads += 1 # queued or running; only touched on the event loop
|
| 107 |
+
try:
|
| 108 |
+
return await asyncio.get_running_loop().run_in_executor(self._readers, run)
|
| 109 |
+
finally:
|
| 110 |
+
self.queued_reads -= 1
|
| 111 |
+
|
| 112 |
+
def _run_write(self, fn, args):
|
| 113 |
+
conn = self._writer_conn
|
| 114 |
+
conn.execute("BEGIN IMMEDIATE")
|
| 115 |
+
try:
|
| 116 |
+
result = fn(conn, *args)
|
| 117 |
+
except BaseException:
|
| 118 |
+
conn.execute("ROLLBACK")
|
| 119 |
+
raise
|
| 120 |
+
conn.execute("COMMIT")
|
| 121 |
+
return result
|
| 122 |
+
|
| 123 |
+
async def write(self, fn, *args):
|
| 124 |
+
if self.closing:
|
| 125 |
+
raise ApiError(503, "shutting_down", "server is shutting down; retry shortly", {"Retry-After": "2"})
|
| 126 |
+
if self.pending_writes >= LIMITS.max_pending_writes:
|
| 127 |
+
raise ApiError(503, "overloaded", "write queue is full", {"Retry-After": "1"})
|
| 128 |
+
self.pending_writes += 1
|
| 129 |
+
try:
|
| 130 |
+
loop = asyncio.get_running_loop()
|
| 131 |
+
result = await loop.run_in_executor(self._writer, self._run_write, fn, args)
|
| 132 |
+
self.generation += 1
|
| 133 |
+
if fn not in AUTH_SAFE_WRITES:
|
| 134 |
+
self.auth_cache.clear()
|
| 135 |
+
return result
|
| 136 |
+
finally:
|
| 137 |
+
self.pending_writes -= 1
|
| 138 |
+
|
| 139 |
+
def close(self) -> None:
|
| 140 |
+
self._writer.shutdown(wait=True)
|
| 141 |
+
self._readers.shutdown(wait=True)
|
| 142 |
+
for conn in self._reader_conns:
|
| 143 |
+
conn.close()
|
| 144 |
+
try:
|
| 145 |
+
self._writer_conn.execute("PRAGMA wal_checkpoint(TRUNCATE)")
|
| 146 |
+
finally:
|
| 147 |
+
self._writer_conn.close()
|
| 148 |
+
|
| 149 |
+
|
| 150 |
+
ANY_CHANNEL = -1 # notifier key woken by a post in any channel (activity long-polls wait on it)
|
| 151 |
+
# Writes that cannot change who a token belongs to or what it may do; every other write clears the auth cache.
|
| 152 |
+
AUTH_SAFE_WRITES = frozenset({store.post_message, store.react, store.mark_read, store.mark_thread_read,
|
| 153 |
+
store.mark_all_read, store.save_presence})
|
| 154 |
+
|
| 155 |
+
|
| 156 |
+
class Notifier:
|
| 157 |
+
"""In-memory per-channel wakeups. Hints only: waiters always re-query the database."""
|
| 158 |
+
|
| 159 |
+
def __init__(self):
|
| 160 |
+
self._waiters: dict[int, set[asyncio.Future]] = {}
|
| 161 |
+
self.loop: asyncio.AbstractEventLoop | None = None
|
| 162 |
+
|
| 163 |
+
def register(self, channel_id: int) -> asyncio.Future:
|
| 164 |
+
fut = asyncio.get_running_loop().create_future()
|
| 165 |
+
self._waiters.setdefault(channel_id, set()).add(fut)
|
| 166 |
+
return fut
|
| 167 |
+
|
| 168 |
+
def unregister(self, channel_id: int, fut: asyncio.Future) -> None:
|
| 169 |
+
waiters = self._waiters.get(channel_id)
|
| 170 |
+
if waiters is not None:
|
| 171 |
+
waiters.discard(fut)
|
| 172 |
+
if not waiters:
|
| 173 |
+
del self._waiters[channel_id]
|
| 174 |
+
|
| 175 |
+
def notify(self, channel_id: int) -> None:
|
| 176 |
+
for fut in self._waiters.pop(channel_id, ()):
|
| 177 |
+
if not fut.done():
|
| 178 |
+
fut.set_result(None)
|
| 179 |
+
|
| 180 |
+
def wake_all(self) -> None:
|
| 181 |
+
for channel_id in list(self._waiters):
|
| 182 |
+
self.notify(channel_id)
|
| 183 |
+
|
| 184 |
+
|
| 185 |
+
class Board:
|
| 186 |
+
"""Everything one running server owns for a data directory."""
|
| 187 |
+
|
| 188 |
+
def __init__(self, data_dir: Path, replicator=None):
|
| 189 |
+
self.data_dir = Path(data_dir)
|
| 190 |
+
db_path = self.data_dir / DB_NAME
|
| 191 |
+
if not db_path.exists():
|
| 192 |
+
raise NotInitialized(f"no board in {self.data_dir}; run `python -m msgboard init` first")
|
| 193 |
+
self._lock_fh = open(self.data_dir / "server.lock", "a")
|
| 194 |
+
try:
|
| 195 |
+
fcntl.flock(self._lock_fh, fcntl.LOCK_EX | fcntl.LOCK_NB)
|
| 196 |
+
except BlockingIOError:
|
| 197 |
+
self._lock_fh.close()
|
| 198 |
+
raise AlreadyRunning(f"another server is already using {self.data_dir}")
|
| 199 |
+
check = dbmod.connect(db_path)
|
| 200 |
+
try:
|
| 201 |
+
version = dbmod.schema_version(check)
|
| 202 |
+
if version > dbmod.SCHEMA_VERSION:
|
| 203 |
+
raise dbmod.SchemaError(f"database schema {version} is newer than this software")
|
| 204 |
+
if version < dbmod.SCHEMA_VERSION:
|
| 205 |
+
raise NotInitialized("database schema is outdated; run `python -m msgboard init` to migrate")
|
| 206 |
+
if not store.admin_exists(check):
|
| 207 |
+
raise NotInitialized("board has no operator; run `python -m msgboard init`")
|
| 208 |
+
except BaseException:
|
| 209 |
+
check.close()
|
| 210 |
+
self._release_lock()
|
| 211 |
+
raise
|
| 212 |
+
check.close()
|
| 213 |
+
self.db = Database(db_path)
|
| 214 |
+
# one-time maintenance on the writer connection before serving
|
| 215 |
+
with self.db._lock:
|
| 216 |
+
conn = self.db._writer_conn
|
| 217 |
+
conn.execute("BEGIN IMMEDIATE")
|
| 218 |
+
try:
|
| 219 |
+
store.backfill_mentions(conn)
|
| 220 |
+
store.backfill_mentioned_as(conn)
|
| 221 |
+
conn.execute("COMMIT")
|
| 222 |
+
except BaseException:
|
| 223 |
+
conn.execute("ROLLBACK")
|
| 224 |
+
raise
|
| 225 |
+
self.presence = Presence(store.load_presence(conn))
|
| 226 |
+
self.blobs = LocalBlobStore(self.data_dir / "blobs")
|
| 227 |
+
# storage folders live in the replication store (the Space's bucket) or, locally, under the data dir;
|
| 228 |
+
# downloads are served from a local cache of those objects
|
| 229 |
+
from .replicate import DirStore
|
| 230 |
+
self.storage = replicator.store if replicator is not None else DirStore(self.data_dir)
|
| 231 |
+
self.storage_cache = self.data_dir / "storage-cache"
|
| 232 |
+
self.storage_cache.mkdir(exist_ok=True)
|
| 233 |
+
self.notifier = Notifier()
|
| 234 |
+
self.shutting_down = False
|
| 235 |
+
self.open_polls = 0
|
| 236 |
+
self.open_uploads = 0
|
| 237 |
+
self.commit_lock: asyncio.Lock | None = None
|
| 238 |
+
self.started = time.time()
|
| 239 |
+
self.counters = {"posts": 0, "posts_replayed": 0, "edits": 0, "polls": 0, "errors_5xx": 0}
|
| 240 |
+
self.replicator = replicator
|
| 241 |
+
if replicator is not None:
|
| 242 |
+
replicator.start(lambda: self.db.generation)
|
| 243 |
+
|
| 244 |
+
def begin_shutdown(self) -> None:
|
| 245 |
+
"""Release held long-polls so a Ctrl-C does not wait for them to time out."""
|
| 246 |
+
if self.replicator is not None and not self.shutting_down:
|
| 247 |
+
self.replicator.record("shutdown-signal")
|
| 248 |
+
self.shutting_down = True
|
| 249 |
+
if self.replicator is not None:
|
| 250 |
+
# With replication, stop taking writes now: the final snapshot must
|
| 251 |
+
# include every acknowledged write, and the next instance is waiting.
|
| 252 |
+
self.db.closing = True
|
| 253 |
+
loop = self.notifier.loop
|
| 254 |
+
if loop is not None and not loop.is_closed():
|
| 255 |
+
loop.call_soon_threadsafe(self.notifier.wake_all)
|
| 256 |
+
|
| 257 |
+
def _release_lock(self) -> None:
|
| 258 |
+
try:
|
| 259 |
+
fcntl.flock(self._lock_fh, fcntl.LOCK_UN)
|
| 260 |
+
finally:
|
| 261 |
+
self._lock_fh.close()
|
| 262 |
+
|
| 263 |
+
async def flush_presence(self) -> None:
|
| 264 |
+
versions, records = self.presence.pending()
|
| 265 |
+
history = self.presence.pending_history()
|
| 266 |
+
if records or history:
|
| 267 |
+
await self.db.write(store.save_presence, records, history)
|
| 268 |
+
self.presence.saved(versions)
|
| 269 |
+
self.presence.history_saved(history)
|
| 270 |
+
|
| 271 |
+
async def presence_loop(self) -> None:
|
| 272 |
+
while True:
|
| 273 |
+
await asyncio.sleep(PRESENCE_FLUSH_SECONDS)
|
| 274 |
+
if self.shutting_down:
|
| 275 |
+
return
|
| 276 |
+
self.presence.sample(time.time())
|
| 277 |
+
try:
|
| 278 |
+
await self.flush_presence()
|
| 279 |
+
except Exception:
|
| 280 |
+
# Observations remain dirty and retry on the next interval. Presence
|
| 281 |
+
# must not cause regular requests to fail when the writer is busy.
|
| 282 |
+
logging.getLogger(__name__).exception("could not persist agent presence")
|
| 283 |
+
|
| 284 |
+
def close(self) -> None:
|
| 285 |
+
if getattr(self, "_closed", False):
|
| 286 |
+
return
|
| 287 |
+
self._closed = True
|
| 288 |
+
# The event loop has stopped. Queue the last observations after in-flight
|
| 289 |
+
# writes, before the replicator takes its final snapshot. This also works
|
| 290 |
+
# when begin_shutdown has closed the normal HTTP writer entry point.
|
| 291 |
+
versions, records = self.presence.pending()
|
| 292 |
+
history = self.presence.pending_history()
|
| 293 |
+
if records or history:
|
| 294 |
+
try:
|
| 295 |
+
self.db._writer.submit(self.db._run_write, store.save_presence, (records, history)).result()
|
| 296 |
+
self.db.generation += 1
|
| 297 |
+
self.presence.saved(versions)
|
| 298 |
+
self.presence.history_saved(history)
|
| 299 |
+
except Exception:
|
| 300 |
+
# Ephemeral presence must never prevent the final snapshot of
|
| 301 |
+
# acknowledged messages or leave the database/lease open.
|
| 302 |
+
logging.getLogger(__name__).exception("could not persist final agent presence")
|
| 303 |
+
if self.replicator is not None:
|
| 304 |
+
self.replicator.stop(final_push=True) # after in-flight writes, before closing
|
| 305 |
+
self.db.close()
|
| 306 |
+
self._release_lock()
|
| 307 |
+
|
| 308 |
+
|
| 309 |
+
def _bearer(request: Request) -> str | None:
|
| 310 |
+
# X-Msgboard-Token lets clients reach the board through a proxy that uses
|
| 311 |
+
# Authorization for its own login (for example a private HF Space).
|
| 312 |
+
direct = request.headers.get("x-msgboard-token")
|
| 313 |
+
if direct:
|
| 314 |
+
return direct.strip()
|
| 315 |
+
header = request.headers.get("authorization", "")
|
| 316 |
+
scheme, _, value = header.partition(" ")
|
| 317 |
+
return value.strip() if scheme.lower() == "bearer" else None
|
| 318 |
+
|
| 319 |
+
|
| 320 |
+
async def _json_body(request: Request, max_bytes: int = 256 * 1024):
|
| 321 |
+
import json
|
| 322 |
+
|
| 323 |
+
data = bytearray()
|
| 324 |
+
async for chunk in request.stream():
|
| 325 |
+
data += chunk
|
| 326 |
+
if len(data) > max_bytes:
|
| 327 |
+
raise ApiError(413, "too_large", "request body too large")
|
| 328 |
+
if not data:
|
| 329 |
+
return {}
|
| 330 |
+
try:
|
| 331 |
+
return json.loads(data)
|
| 332 |
+
except ValueError:
|
| 333 |
+
raise bad_request("request body is not valid JSON")
|
| 334 |
+
|
| 335 |
+
|
| 336 |
+
def create_app(board: Board, dashboard: bool = True) -> FastAPI:
|
| 337 |
+
@asynccontextmanager
|
| 338 |
+
async def lifespan(app):
|
| 339 |
+
board.notifier.loop = asyncio.get_running_loop()
|
| 340 |
+
board.commit_lock = asyncio.Lock()
|
| 341 |
+
board.presence.sample(time.time())
|
| 342 |
+
task = asyncio.create_task(board.presence_loop())
|
| 343 |
+
try:
|
| 344 |
+
yield
|
| 345 |
+
finally:
|
| 346 |
+
board.begin_shutdown()
|
| 347 |
+
task.cancel()
|
| 348 |
+
with suppress(asyncio.CancelledError):
|
| 349 |
+
await task
|
| 350 |
+
|
| 351 |
+
app = FastAPI(title="msgboard", version="0.1.0", lifespan=lifespan,
|
| 352 |
+
docs_url=None, redoc_url=None, openapi_url="/v1/openapi.json")
|
| 353 |
+
app.state.board = board
|
| 354 |
+
db = board.db
|
| 355 |
+
|
| 356 |
+
@app.exception_handler(ApiError)
|
| 357 |
+
async def api_error(request: Request, exc: ApiError):
|
| 358 |
+
if exc.status >= 500:
|
| 359 |
+
board.counters["errors_5xx"] += 1
|
| 360 |
+
body = {"code": exc.code, "message": exc.message}
|
| 361 |
+
if exc.details:
|
| 362 |
+
body["details"] = exc.details
|
| 363 |
+
return JSONResponse({"error": body}, status_code=exc.status, headers=exc.headers)
|
| 364 |
+
|
| 365 |
+
async def authn(request: Request, *, observe: bool = True):
|
| 366 |
+
token = _bearer(request)
|
| 367 |
+
now = time.monotonic()
|
| 368 |
+
hit = db.auth_cache.get(token) if token else None
|
| 369 |
+
if hit is not None and now - hit[1] < LIMITS.auth_cache_seconds:
|
| 370 |
+
if observe:
|
| 371 |
+
board.presence.seen(hit[0], time.time())
|
| 372 |
+
return hit[0]
|
| 373 |
+
p = await db.read(store.authenticate, token, time.time())
|
| 374 |
+
if len(db.auth_cache) > 20_000:
|
| 375 |
+
db.auth_cache.clear()
|
| 376 |
+
db.auth_cache[token] = (p, now)
|
| 377 |
+
if observe:
|
| 378 |
+
board.presence.seen(p, time.time())
|
| 379 |
+
return p
|
| 380 |
+
|
| 381 |
+
async def with_presence(p, items):
|
| 382 |
+
return await db.read(store.attach_presence, p, items, board.presence.snapshot())
|
| 383 |
+
|
| 384 |
+
# ------------------------------------------------------------- health
|
| 385 |
+
|
| 386 |
+
@app.get("/v1/health")
|
| 387 |
+
async def health():
|
| 388 |
+
return {"status": "ok"}
|
| 389 |
+
|
| 390 |
+
@app.get("/v1/ready")
|
| 391 |
+
async def ready():
|
| 392 |
+
await db.read(lambda c: c.execute("SELECT 1").fetchone())
|
| 393 |
+
return {"status": "ready", "shutting_down": board.shutting_down}
|
| 394 |
+
|
| 395 |
+
@app.get("/v1/metrics")
|
| 396 |
+
async def metrics(request: Request):
|
| 397 |
+
p = await authn(request)
|
| 398 |
+
if not p.is_admin:
|
| 399 |
+
raise forbidden("admin only")
|
| 400 |
+
return health_summary()
|
| 401 |
+
|
| 402 |
+
def health_summary():
|
| 403 |
+
db_path = board.data_dir / DB_NAME
|
| 404 |
+
wal = Path(str(db_path) + "-wal")
|
| 405 |
+
return {
|
| 406 |
+
**board.counters,
|
| 407 |
+
"open_polls": board.open_polls,
|
| 408 |
+
"open_uploads": board.open_uploads,
|
| 409 |
+
"pending_writes": db.pending_writes,
|
| 410 |
+
"queued_reads": db.queued_reads,
|
| 411 |
+
"shed_reads": db.shed_reads,
|
| 412 |
+
"slow_reads": db.slow_reads,
|
| 413 |
+
"db_bytes": db_path.stat().st_size,
|
| 414 |
+
"wal_bytes": wal.stat().st_size if wal.exists() else 0,
|
| 415 |
+
"disk_free_bytes": shutil.disk_usage(board.data_dir).free,
|
| 416 |
+
"uptime_seconds": time.time() - board.started,
|
| 417 |
+
"replication": board.replicator.status if board.replicator else None,
|
| 418 |
+
}
|
| 419 |
+
|
| 420 |
+
# ------------------------------------------------------- identities
|
| 421 |
+
|
| 422 |
+
def body_dict(body):
|
| 423 |
+
if not isinstance(body, dict):
|
| 424 |
+
raise bad_request("request body must be a JSON object")
|
| 425 |
+
return body
|
| 426 |
+
|
| 427 |
+
@app.get("/v1/me")
|
| 428 |
+
async def get_me(request: Request):
|
| 429 |
+
p = await authn(request)
|
| 430 |
+
return (await with_presence(p, [await db.read(store.me, p)]))[0]
|
| 431 |
+
|
| 432 |
+
@app.post("/v1/me/read-all")
|
| 433 |
+
async def mark_all_read(request: Request):
|
| 434 |
+
"""Mark everything you can read as read (channels, DMs and threads)."""
|
| 435 |
+
p = await authn(request)
|
| 436 |
+
return await db.write(store.mark_all_read, p, time.time())
|
| 437 |
+
|
| 438 |
+
@app.put("/v1/me/prefs")
|
| 439 |
+
async def put_prefs(request: Request):
|
| 440 |
+
p = await authn(request)
|
| 441 |
+
return await db.write(store.update_prefs, p, await _json_body(request, 64 * 1024), time.time())
|
| 442 |
+
|
| 443 |
+
@app.get("/v1/principals")
|
| 444 |
+
async def list_principals(request: Request):
|
| 445 |
+
p = await authn(request)
|
| 446 |
+
return {"principals": await with_presence(p, await db.read(store.list_principals, p))}
|
| 447 |
+
|
| 448 |
+
@app.get("/v1/presence")
|
| 449 |
+
async def list_presence(request: Request):
|
| 450 |
+
p = await authn(request)
|
| 451 |
+
return {"principals": await db.read(store.list_presence, p, board.presence.snapshot())}
|
| 452 |
+
|
| 453 |
+
@app.patch("/v1/principals/{principal_id}")
|
| 454 |
+
async def update_principal(principal_id: int, request: Request):
|
| 455 |
+
p = await authn(request)
|
| 456 |
+
body = body_dict(await _json_body(request))
|
| 457 |
+
result = await db.write(store.update_principal, p, principal_id, body, time.time())
|
| 458 |
+
return (await with_presence(p, [result]))[0]
|
| 459 |
+
|
| 460 |
+
@app.post("/v1/principals/{principal_id}/suspend")
|
| 461 |
+
async def suspend(principal_id: int, request: Request):
|
| 462 |
+
p = await authn(request)
|
| 463 |
+
raw = await request.body()
|
| 464 |
+
body = body_dict(await _json_body(request)) if raw.strip() else {}
|
| 465 |
+
return await db.write(store.suspend, p, principal_id, body.get("reason"), time.time())
|
| 466 |
+
|
| 467 |
+
@app.post("/v1/principals/{principal_id}/unsuspend")
|
| 468 |
+
async def unsuspend(principal_id: int, request: Request):
|
| 469 |
+
p = await authn(request)
|
| 470 |
+
return await db.write(store.unsuspend, p, principal_id, time.time())
|
| 471 |
+
|
| 472 |
+
@app.get("/v1/suspended")
|
| 473 |
+
async def list_suspended(request: Request):
|
| 474 |
+
p = await authn(request)
|
| 475 |
+
return {"suspended": await db.read(store.list_suspended, p)}
|
| 476 |
+
|
| 477 |
+
@app.post("/v1/principals/{principal_id}/token")
|
| 478 |
+
async def rotate_token(principal_id: int, request: Request):
|
| 479 |
+
p = await authn(request)
|
| 480 |
+
result = await db.write(store.rotate_agent_token, p, principal_id, time.time())
|
| 481 |
+
return JSONResponse(result, status_code=201, headers={"Cache-Control": "no-store"})
|
| 482 |
+
|
| 483 |
+
@app.post("/v1/principals/{principal_id}/signin-link")
|
| 484 |
+
async def signin_link(principal_id: int, request: Request):
|
| 485 |
+
p = await authn(request)
|
| 486 |
+
result = await db.write(store.create_signin_link, p, principal_id, time.time())
|
| 487 |
+
return JSONResponse(result, status_code=201, headers={"Cache-Control": "no-store"})
|
| 488 |
+
|
| 489 |
+
@app.delete("/v1/tokens/{token_id}")
|
| 490 |
+
async def revoke_token(token_id: str, request: Request):
|
| 491 |
+
p = await authn(request)
|
| 492 |
+
return await db.write(store.revoke_token, p, token_id, time.time())
|
| 493 |
+
|
| 494 |
+
# ------------------------------------------------- invites and joining
|
| 495 |
+
|
| 496 |
+
@app.post("/v1/invites")
|
| 497 |
+
async def create_invite(request: Request):
|
| 498 |
+
p = await authn(request)
|
| 499 |
+
body = body_dict(await _json_body(request))
|
| 500 |
+
return JSONResponse(await db.write(store.create_invite, p, body, time.time()), status_code=201,
|
| 501 |
+
headers={"Cache-Control": "no-store"})
|
| 502 |
+
|
| 503 |
+
@app.get("/v1/invites")
|
| 504 |
+
async def list_invites(request: Request):
|
| 505 |
+
p = await authn(request)
|
| 506 |
+
return {"invites": await db.read(store.list_invites, p, time.time())}
|
| 507 |
+
|
| 508 |
+
@app.delete("/v1/invites/{invite_id}")
|
| 509 |
+
async def revoke_invite(invite_id: int, request: Request):
|
| 510 |
+
p = await authn(request)
|
| 511 |
+
return await db.write(store.revoke_invite, p, invite_id, time.time())
|
| 512 |
+
|
| 513 |
+
@app.get("/v1/join/{code}")
|
| 514 |
+
async def lookup_invite(code: str):
|
| 515 |
+
return await db.read(store.lookup_invite, code, time.time())
|
| 516 |
+
|
| 517 |
+
@app.post("/v1/join")
|
| 518 |
+
async def join(request: Request):
|
| 519 |
+
body = body_dict(await _json_body(request, 4096))
|
| 520 |
+
result = await db.write(store.join, body.get("code"), body.get("name"), body.get("display_name"), time.time())
|
| 521 |
+
return JSONResponse(result, status_code=201 if result["created"] else 200, headers={"Cache-Control": "no-store"})
|
| 522 |
+
|
| 523 |
+
# ------------------------------------------------------------ settings
|
| 524 |
+
|
| 525 |
+
@app.get("/v1/settings")
|
| 526 |
+
async def get_settings(request: Request):
|
| 527 |
+
await authn(request)
|
| 528 |
+
return await db.read(store.get_settings)
|
| 529 |
+
|
| 530 |
+
@app.patch("/v1/settings")
|
| 531 |
+
async def update_settings(request: Request):
|
| 532 |
+
p = await authn(request)
|
| 533 |
+
body = body_dict(await _json_body(request))
|
| 534 |
+
return await db.write(store.update_settings, p, body, time.time())
|
| 535 |
+
|
| 536 |
+
# --------------------------------------------------------- channels
|
| 537 |
+
|
| 538 |
+
@app.get("/v1/channels")
|
| 539 |
+
async def list_channels(request: Request, name: str | None = None):
|
| 540 |
+
p = await authn(request)
|
| 541 |
+
return {"channels": await db.read(store.list_channels, p, name)}
|
| 542 |
+
|
| 543 |
+
@app.post("/v1/channels")
|
| 544 |
+
async def create_channel(request: Request):
|
| 545 |
+
p = await authn(request)
|
| 546 |
+
body = body_dict(await _json_body(request))
|
| 547 |
+
channel, created = await db.write(store.create_channel, p, body.get("name"), body.get("description"),
|
| 548 |
+
body.get("private", False), time.time())
|
| 549 |
+
return JSONResponse(channel, status_code=201 if created else 200)
|
| 550 |
+
|
| 551 |
+
@app.get("/v1/messages/{message_id}")
|
| 552 |
+
async def get_message(message_id: int, request: Request):
|
| 553 |
+
p = await authn(request)
|
| 554 |
+
return await db.read(store.get_message, p, message_id)
|
| 555 |
+
|
| 556 |
+
@app.get("/v1/channels/{channel_id}/doc")
|
| 557 |
+
async def get_doc(channel_id: int, request: Request):
|
| 558 |
+
p = await authn(request)
|
| 559 |
+
return await db.read(store.get_doc, p, channel_id)
|
| 560 |
+
|
| 561 |
+
@app.put("/v1/channels/{channel_id}/doc")
|
| 562 |
+
async def put_doc(channel_id: int, request: Request):
|
| 563 |
+
p = await authn(request)
|
| 564 |
+
doc, changed = await db.write(store.save_doc, p, channel_id,
|
| 565 |
+
await _json_body(request, store.MAX_DOC_BYTES * 2 + 4096), time.time())
|
| 566 |
+
return doc
|
| 567 |
+
|
| 568 |
+
@app.get("/v1/channels/{channel_id}/doc/versions")
|
| 569 |
+
async def doc_versions(channel_id: int, request: Request):
|
| 570 |
+
p = await authn(request)
|
| 571 |
+
return await db.read(store.doc_versions, p, channel_id)
|
| 572 |
+
|
| 573 |
+
@app.get("/v1/channels/{channel_id}/doc/versions/{version}")
|
| 574 |
+
async def doc_version(channel_id: int, version: int, request: Request):
|
| 575 |
+
p = await authn(request)
|
| 576 |
+
return await db.read(store.doc_version, p, channel_id, version)
|
| 577 |
+
|
| 578 |
+
@app.delete("/v1/channels/{channel_id}")
|
| 579 |
+
async def delete_channel(channel_id: int, request: Request, confirm: str = ""):
|
| 580 |
+
p = await authn(request)
|
| 581 |
+
result = await db.write(store.delete_channel, p, channel_id, confirm, time.time())
|
| 582 |
+
for key in result.pop("storage_keys"): # the rows are gone; now the files
|
| 583 |
+
try:
|
| 584 |
+
board.blobs.path_for(key).unlink()
|
| 585 |
+
except FileNotFoundError:
|
| 586 |
+
pass
|
| 587 |
+
board.notifier.notify(channel_id)
|
| 588 |
+
return result
|
| 589 |
+
|
| 590 |
+
@app.get("/v1/channels/{channel_id}/info")
|
| 591 |
+
async def channel_info(channel_id: int, request: Request):
|
| 592 |
+
p = await authn(request)
|
| 593 |
+
result = await db.read(store.channel_info, p, channel_id)
|
| 594 |
+
result["members"] = await with_presence(p, result["members"])
|
| 595 |
+
result["posters"] = await with_presence(p, result["posters"])
|
| 596 |
+
return result
|
| 597 |
+
|
| 598 |
+
@app.patch("/v1/channels/{channel_id}")
|
| 599 |
+
async def update_channel(channel_id: int, request: Request):
|
| 600 |
+
p = await authn(request)
|
| 601 |
+
return await db.write(store.update_channel, p, channel_id, await _json_body(request, 4096), time.time())
|
| 602 |
+
|
| 603 |
+
@app.post("/v1/channels/{channel_id}/archive")
|
| 604 |
+
async def archive_channel(channel_id: int, request: Request):
|
| 605 |
+
p = await authn(request)
|
| 606 |
+
return await db.write(store.archive_channel, p, channel_id, time.time())
|
| 607 |
+
|
| 608 |
+
@app.post("/v1/channels/{channel_id}/threads/{thread_id}/read")
|
| 609 |
+
async def mark_thread_read(channel_id: int, thread_id: int, request: Request):
|
| 610 |
+
p = await authn(request)
|
| 611 |
+
body = body_dict(await _json_body(request, 1024))
|
| 612 |
+
return await db.write(store.mark_thread_read, p, channel_id, thread_id, body.get("seq"), time.time())
|
| 613 |
+
|
| 614 |
+
@app.post("/v1/channels/{channel_id}/read")
|
| 615 |
+
async def mark_read(channel_id: int, request: Request):
|
| 616 |
+
p = await authn(request)
|
| 617 |
+
body = body_dict(await _json_body(request, 1024))
|
| 618 |
+
return await db.write(store.mark_read, p, channel_id, body.get("seq"), time.time())
|
| 619 |
+
|
| 620 |
+
@app.get("/v1/activity")
|
| 621 |
+
async def get_activity(request: Request, limit: int = Query(50, ge=1, le=200), after: int | None = Query(None, ge=0),
|
| 622 |
+
wait: float = Query(0, ge=0, le=LIMITS.max_wait_seconds), channels: str | None = None,
|
| 623 |
+
reactions: bool = False):
|
| 624 |
+
"""Everything for you (mentions, replies, DMs, plus `channels` you follow). With `after`, a
|
| 625 |
+
cursor read, oldest first; add `wait` to long-poll it."""
|
| 626 |
+
p = await authn(request)
|
| 627 |
+
parts = [x.strip() for x in channels.split(",") if x.strip()] if channels else []
|
| 628 |
+
joined = "joined" in parts # every channel you are a member of
|
| 629 |
+
try:
|
| 630 |
+
follow = [int(x) for x in parts if x != "joined"]
|
| 631 |
+
except ValueError:
|
| 632 |
+
raise bad_request("channels must be comma-separated channel ids, or `joined`")
|
| 633 |
+
if joined:
|
| 634 |
+
follow += await db.read(store.joined_channel_ids, p)
|
| 635 |
+
if len(follow) > 100:
|
| 636 |
+
raise bad_request("follow at most 100 channels")
|
| 637 |
+
if wait > 0 and after is None:
|
| 638 |
+
raise bad_request("wait needs a cursor: pass `after` (start from the next_cursor of a plain read)")
|
| 639 |
+
token = board.presence.next_poll()
|
| 640 |
+
polled_at = time.time()
|
| 641 |
+
mode = "joined" if joined else ("channels" if follow else "addressed")
|
| 642 |
+
|
| 643 |
+
def observe(result):
|
| 644 |
+
board.presence.polled(p, token, polled_at, after, mode, follow, bool(result["messages"]))
|
| 645 |
+
|
| 646 |
+
if wait <= 0:
|
| 647 |
+
result = await db.read(store.activity, p, limit, after, follow, reactions)
|
| 648 |
+
observe(result)
|
| 649 |
+
return result
|
| 650 |
+
if board.open_polls >= LIMITS.max_open_polls:
|
| 651 |
+
raise ApiError(429, "too_many_polls", "too many open long-polls", {"Retry-After": "1"})
|
| 652 |
+
board.open_polls += 1
|
| 653 |
+
board.counters["polls"] += 1
|
| 654 |
+
fut = board.notifier.register(ANY_CHANNEL)
|
| 655 |
+
try:
|
| 656 |
+
result = await db.read(store.activity, p, limit, after, follow)
|
| 657 |
+
observe(result)
|
| 658 |
+
if result["messages"] or board.shutting_down:
|
| 659 |
+
return result
|
| 660 |
+
deadline = time.monotonic() + wait
|
| 661 |
+
while True: # any post wakes us; keep waiting until one is actually for this caller
|
| 662 |
+
left = deadline - time.monotonic()
|
| 663 |
+
if left <= 0:
|
| 664 |
+
break
|
| 665 |
+
done, _ = await asyncio.wait({fut}, timeout=left)
|
| 666 |
+
if done and not board.shutting_down:
|
| 667 |
+
# let a burst of posts land, then answer them with one query instead of one per post
|
| 668 |
+
await asyncio.sleep(min(LIMITS.poll_coalesce_seconds * (0.75 + random.random() / 2),
|
| 669 |
+
max(0.0, deadline - time.monotonic())))
|
| 670 |
+
if await request.is_disconnected():
|
| 671 |
+
return result
|
| 672 |
+
board.notifier.unregister(ANY_CHANNEL, fut)
|
| 673 |
+
fut = board.notifier.register(ANY_CHANNEL)
|
| 674 |
+
p = await authn(request, observe=False)
|
| 675 |
+
result = await db.read(store.activity, p, limit, after, follow, shed=False)
|
| 676 |
+
board.presence.observed(p, token, bool(result["messages"]), time.time())
|
| 677 |
+
if result["messages"] or board.shutting_down:
|
| 678 |
+
return result
|
| 679 |
+
return result
|
| 680 |
+
finally:
|
| 681 |
+
board.notifier.unregister(ANY_CHANNEL, fut)
|
| 682 |
+
board.open_polls -= 1
|
| 683 |
+
|
| 684 |
+
@app.get("/v1/search")
|
| 685 |
+
async def search(request: Request, q: str = "", channel: int | None = None, sender: int | None = None,
|
| 686 |
+
after: str | None = None, before: str | None = None, sort: str = "newest",
|
| 687 |
+
limit: int = Query(20, ge=1, le=100), offset: int = Query(0, ge=0, le=10_000)):
|
| 688 |
+
"""Full-text search over messages you can read, plus matching channel and people names.
|
| 689 |
+
`q` takes words, "phrases" and filters: from:@name in:#channel after:/before:/on:DATE has:file ..."""
|
| 690 |
+
p = await authn(request)
|
| 691 |
+
if len(q) > 500:
|
| 692 |
+
raise bad_request("q is at most 500 characters")
|
| 693 |
+
return await db.read(lambda conn: store.search(conn, p, q, channel_id=channel, sender_id=sender, after=after,
|
| 694 |
+
before=before, sort=sort, limit=limit, offset=offset,
|
| 695 |
+
now=time.time()))
|
| 696 |
+
|
| 697 |
+
@app.get("/v1/analytics")
|
| 698 |
+
async def get_analytics(request: Request, days: int = Query(7, ge=1, le=365),
|
| 699 |
+
start: float | None = Query(None, alias="from"),
|
| 700 |
+
end: float | None = Query(None, alias="to"),
|
| 701 |
+
top_k: int = Query(10, ge=1, le=100)):
|
| 702 |
+
p = await authn(request)
|
| 703 |
+
result = await db.read(store.analytics, p, time.time(), days, start, end, top_k,
|
| 704 |
+
board.presence.pending_history())
|
| 705 |
+
result["listening"]["now"] = board.presence.listening_now(time.time())
|
| 706 |
+
result["health"] = health_summary()
|
| 707 |
+
return result
|
| 708 |
+
|
| 709 |
+
def _origin(request: Request) -> str:
|
| 710 |
+
proto = request.headers.get("x-forwarded-proto") or request.url.scheme
|
| 711 |
+
host = request.headers.get("x-forwarded-host") or request.headers.get("host") or request.url.netloc
|
| 712 |
+
return f"{proto}://{host}{request.scope.get('root_path', '')}"
|
| 713 |
+
|
| 714 |
+
@app.get("/agents", include_in_schema=False)
|
| 715 |
+
async def agents_guide(request: Request):
|
| 716 |
+
return PlainTextResponse(guide.render(_origin(request), public=True), media_type="text/markdown; charset=utf-8")
|
| 717 |
+
|
| 718 |
+
@app.get("/v1/guide")
|
| 719 |
+
async def get_guide(request: Request):
|
| 720 |
+
await authn(request)
|
| 721 |
+
settings = await db.read(store.get_settings)
|
| 722 |
+
return PlainTextResponse(guide.render(_origin(request), settings["guidelines"], docs=settings["channel_docs"]),
|
| 723 |
+
media_type="text/markdown; charset=utf-8")
|
| 724 |
+
|
| 725 |
+
@app.get("/v1/channels/{channel_id}/members")
|
| 726 |
+
async def list_members(channel_id: int, request: Request):
|
| 727 |
+
p = await authn(request)
|
| 728 |
+
return {"members": await with_presence(p, await db.read(store.list_members, p, channel_id))}
|
| 729 |
+
|
| 730 |
+
@app.post("/v1/channels/{channel_id}/members")
|
| 731 |
+
async def add_member(channel_id: int, request: Request):
|
| 732 |
+
p = await authn(request)
|
| 733 |
+
body = body_dict(await _json_body(request))
|
| 734 |
+
members = await db.write(store.add_member, p, channel_id, body.get("principal_id"), time.time())
|
| 735 |
+
return {"members": await with_presence(p, members)}
|
| 736 |
+
|
| 737 |
+
@app.delete("/v1/channels/{channel_id}/members/{principal_id}")
|
| 738 |
+
async def remove_member(channel_id: int, principal_id: int, request: Request):
|
| 739 |
+
p = await authn(request)
|
| 740 |
+
result = await db.write(store.remove_member, p, channel_id, principal_id, time.time())
|
| 741 |
+
return result if isinstance(result, dict) else {"members": await with_presence(p, result)}
|
| 742 |
+
|
| 743 |
+
@app.post("/v1/dms")
|
| 744 |
+
async def open_dm(request: Request):
|
| 745 |
+
p = await authn(request)
|
| 746 |
+
body = body_dict(await _json_body(request))
|
| 747 |
+
channel, created = await db.write(store.open_dm, p, body.get("with"), time.time())
|
| 748 |
+
return JSONResponse(channel, status_code=201 if created else 200)
|
| 749 |
+
|
| 750 |
+
# --------------------------------------------------------- messages
|
| 751 |
+
|
| 752 |
+
@app.post("/v1/channels/{channel_id}/messages")
|
| 753 |
+
async def post_message(channel_id: int, request: Request):
|
| 754 |
+
p = await authn(request)
|
| 755 |
+
request_id = request.headers.get("idempotency-key")
|
| 756 |
+
if not request_id or not store.CLIENT_ID.match(request_id):
|
| 757 |
+
raise bad_request("Idempotency-Key header is required and must match " + store.CLIENT_ID.pattern)
|
| 758 |
+
payload = await _json_body(request, LIMITS.max_body_bytes * 2 + 64 * 1024)
|
| 759 |
+
normalized, digest = store.normalize_payload(channel_id, payload)
|
| 760 |
+
message, created = await db.write(
|
| 761 |
+
store.post_message, p, channel_id, request_id, normalized, digest, time.time())
|
| 762 |
+
if created:
|
| 763 |
+
board.counters["posts"] += 1
|
| 764 |
+
board.notifier.notify(channel_id)
|
| 765 |
+
board.notifier.notify(ANY_CHANNEL)
|
| 766 |
+
return JSONResponse(message, status_code=201)
|
| 767 |
+
board.counters["posts_replayed"] += 1
|
| 768 |
+
return JSONResponse(message, status_code=200, headers={"Idempotent-Replayed": "true"})
|
| 769 |
+
|
| 770 |
+
@app.patch("/v1/channels/{channel_id}/messages/{message_id}")
|
| 771 |
+
async def edit_message(channel_id: int, message_id: int, request: Request):
|
| 772 |
+
p = await authn(request)
|
| 773 |
+
payload = await _json_body(request, LIMITS.max_body_bytes * 2 + 1024)
|
| 774 |
+
message, changed = await db.write(store.edit_message, p, channel_id, message_id, payload, time.time())
|
| 775 |
+
if changed:
|
| 776 |
+
board.counters["edits"] += 1
|
| 777 |
+
board.notifier.notify(channel_id)
|
| 778 |
+
return message
|
| 779 |
+
|
| 780 |
+
@app.put("/v1/channels/{channel_id}/messages/{message_id}/reactions/{emoji}")
|
| 781 |
+
async def add_reaction(channel_id: int, message_id: int, emoji: str, request: Request):
|
| 782 |
+
p = await authn(request)
|
| 783 |
+
message, changed = await db.write(store.react, p, channel_id, message_id, unquote(emoji), True, time.time())
|
| 784 |
+
if changed:
|
| 785 |
+
board.notifier.notify(channel_id)
|
| 786 |
+
return message
|
| 787 |
+
|
| 788 |
+
@app.delete("/v1/channels/{channel_id}/messages/{message_id}/reactions/{emoji}")
|
| 789 |
+
async def remove_reaction(channel_id: int, message_id: int, emoji: str, request: Request):
|
| 790 |
+
p = await authn(request)
|
| 791 |
+
message, changed = await db.write(store.react, p, channel_id, message_id, unquote(emoji), False, time.time())
|
| 792 |
+
if changed:
|
| 793 |
+
board.notifier.notify(channel_id)
|
| 794 |
+
return message
|
| 795 |
+
|
| 796 |
+
@app.delete("/v1/channels/{channel_id}/messages/{message_id}")
|
| 797 |
+
async def delete_message(channel_id: int, message_id: int, request: Request):
|
| 798 |
+
p = await authn(request)
|
| 799 |
+
result = await db.write(store.delete_message, p, channel_id, message_id, time.time())
|
| 800 |
+
for key in result.pop("storage_keys"):
|
| 801 |
+
try:
|
| 802 |
+
board.blobs.path_for(key).unlink()
|
| 803 |
+
except FileNotFoundError:
|
| 804 |
+
pass
|
| 805 |
+
board.notifier.notify(channel_id)
|
| 806 |
+
return result
|
| 807 |
+
|
| 808 |
+
@app.get("/v1/channels/{channel_id}/messages/{message_id}/revisions")
|
| 809 |
+
async def message_revisions(channel_id: int, message_id: int, request: Request):
|
| 810 |
+
p = await authn(request)
|
| 811 |
+
return await db.read(store.message_revisions, p, channel_id, message_id)
|
| 812 |
+
|
| 813 |
+
@app.get("/v1/channels/{channel_id}/messages")
|
| 814 |
+
async def read_messages(
|
| 815 |
+
channel_id: int,
|
| 816 |
+
request: Request,
|
| 817 |
+
after: int = Query(0, ge=0),
|
| 818 |
+
limit: int = Query(LIMITS.default_page_size, ge=1, le=LIMITS.max_page_size),
|
| 819 |
+
wait: float = Query(0, ge=0, le=LIMITS.max_wait_seconds),
|
| 820 |
+
before: int | None = Query(None, ge=1),
|
| 821 |
+
top_level: bool = False,
|
| 822 |
+
thread: int | None = Query(None, ge=1),
|
| 823 |
+
edits_after: int | None = Query(None, ge=0),
|
| 824 |
+
):
|
| 825 |
+
p = await authn(request)
|
| 826 |
+
if before is not None or top_level or thread is not None:
|
| 827 |
+
if wait > 0 or edits_after is not None:
|
| 828 |
+
raise bad_request("wait and edits_after only work with plain `after` reads")
|
| 829 |
+
return await db.read(store.read_messages, p, channel_id, after, limit, before, top_level, thread)
|
| 830 |
+
|
| 831 |
+
async def page(p):
|
| 832 |
+
result = await db.read(store.read_messages, p, channel_id, after, limit)
|
| 833 |
+
if edits_after is not None: # also hand back messages edited since this edit cursor
|
| 834 |
+
result.update(await db.read(store.read_edits, p, channel_id, edits_after, limit))
|
| 835 |
+
return result
|
| 836 |
+
|
| 837 |
+
if wait <= 0:
|
| 838 |
+
return await page(p)
|
| 839 |
+
if board.open_polls >= LIMITS.max_open_polls:
|
| 840 |
+
raise ApiError(429, "too_many_polls", "too many open long-polls", {"Retry-After": "1"})
|
| 841 |
+
board.open_polls += 1
|
| 842 |
+
board.counters["polls"] += 1
|
| 843 |
+
# Register before checking history so a commit between the query and
|
| 844 |
+
# the wait cannot be missed.
|
| 845 |
+
fut = board.notifier.register(channel_id)
|
| 846 |
+
try:
|
| 847 |
+
result = await page(p)
|
| 848 |
+
if result["messages"] or result.get("edits") or board.shutting_down:
|
| 849 |
+
return result
|
| 850 |
+
await asyncio.wait({fut}, timeout=wait)
|
| 851 |
+
# Re-authenticate and re-authorize: the token may have been revoked
|
| 852 |
+
# or the grant removed while this request was held.
|
| 853 |
+
p = await authn(request)
|
| 854 |
+
return await page(p)
|
| 855 |
+
finally:
|
| 856 |
+
board.notifier.unregister(channel_id, fut)
|
| 857 |
+
board.open_polls -= 1
|
| 858 |
+
|
| 859 |
+
# ------------------------------------------------- profiles, invites
|
| 860 |
+
|
| 861 |
+
@app.get("/v1/principals/{principal_id}/profile")
|
| 862 |
+
async def get_profile(principal_id: int, request: Request):
|
| 863 |
+
p = await authn(request)
|
| 864 |
+
result = await db.read(store.profile, p, principal_id, time.time())
|
| 865 |
+
result = (await with_presence(p, [result]))[0]
|
| 866 |
+
result["agents"] = await with_presence(p, result["agents"])
|
| 867 |
+
return result
|
| 868 |
+
|
| 869 |
+
@app.put("/v1/principals/{principal_id}/avatar")
|
| 870 |
+
async def put_avatar(principal_id: int, request: Request):
|
| 871 |
+
p = await authn(request)
|
| 872 |
+
data = bytearray()
|
| 873 |
+
async for chunk in request.stream():
|
| 874 |
+
data += chunk
|
| 875 |
+
if len(data) > LIMITS.max_avatar_bytes:
|
| 876 |
+
raise ApiError(413, "too_large", f"pictures are limited to {LIMITS.max_avatar_bytes // 1024} KB")
|
| 877 |
+
return await db.write(store.set_avatar, p, principal_id, bytes(data), time.time())
|
| 878 |
+
|
| 879 |
+
@app.delete("/v1/principals/{principal_id}/avatar")
|
| 880 |
+
async def delete_avatar(principal_id: int, request: Request):
|
| 881 |
+
p = await authn(request)
|
| 882 |
+
return await db.write(store.delete_avatar, p, principal_id, time.time())
|
| 883 |
+
|
| 884 |
+
@app.put("/v1/principals/{principal_id}/pattern")
|
| 885 |
+
async def put_pattern(principal_id: int, request: Request):
|
| 886 |
+
p = await authn(request)
|
| 887 |
+
body = await _json_body(request, 1024)
|
| 888 |
+
seed = body.get("seed") if isinstance(body, dict) else None
|
| 889 |
+
return await db.write(store.set_pattern, p, principal_id, seed, time.time())
|
| 890 |
+
|
| 891 |
+
@app.get("/v1/principals/{principal_id}/avatar")
|
| 892 |
+
async def get_avatar(principal_id: int, request: Request):
|
| 893 |
+
await authn(request) # any member of the board may see pictures
|
| 894 |
+
ctype, data, digest = await db.read(store.get_avatar, principal_id)
|
| 895 |
+
return Response(data, media_type=ctype, headers={
|
| 896 |
+
"ETag": f'"{digest[:16]}"', "Cache-Control": "private, max-age=86400",
|
| 897 |
+
"X-Content-Type-Options": "nosniff", "Content-Security-Policy": "default-src 'none'"})
|
| 898 |
+
|
| 899 |
+
# ------------------------------------------------------ attachments
|
| 900 |
+
|
| 901 |
+
@app.put("/v1/channels/{channel_id}/attachments/{upload_id}")
|
| 902 |
+
async def upload(channel_id: int, upload_id: str, request: Request):
|
| 903 |
+
p = await authn(request)
|
| 904 |
+
declared = request.headers.get("content-length")
|
| 905 |
+
if declared and declared.isdigit() and int(declared) > LIMITS.max_upload_bytes:
|
| 906 |
+
raise ApiError(413, "too_large", f"uploads are limited to {LIMITS.max_upload_bytes} bytes")
|
| 907 |
+
if board.open_uploads >= LIMITS.max_concurrent_uploads:
|
| 908 |
+
raise ApiError(429, "too_many_uploads", "too many concurrent uploads", {"Retry-After": "1"})
|
| 909 |
+
# Browsers can only send ASCII headers, so the filename may be percent-encoded.
|
| 910 |
+
filename = os.path.basename(unquote(request.headers.get("x-filename", "")))[:255] or None
|
| 911 |
+
if filename and any(ord(ch) < 32 for ch in filename):
|
| 912 |
+
raise bad_request("invalid filename")
|
| 913 |
+
content_type = request.headers.get("content-type", "application/octet-stream")[:128]
|
| 914 |
+
board.open_uploads += 1
|
| 915 |
+
tmp_path = None
|
| 916 |
+
try:
|
| 917 |
+
row = await db.write(store.begin_upload, p, channel_id, upload_id, filename, content_type, time.time())
|
| 918 |
+
tmp_path, fh = await anyio.to_thread.run_sync(board.blobs.open_temp)
|
| 919 |
+
hasher, size = new_hasher(), 0
|
| 920 |
+
try:
|
| 921 |
+
async for chunk in request.stream():
|
| 922 |
+
size += len(chunk)
|
| 923 |
+
if size > LIMITS.max_upload_bytes:
|
| 924 |
+
raise ApiError(413, "too_large", f"uploads are limited to {LIMITS.max_upload_bytes} bytes")
|
| 925 |
+
hasher.update(chunk)
|
| 926 |
+
await anyio.to_thread.run_sync(fh.write, chunk)
|
| 927 |
+
await anyio.to_thread.run_sync(board.blobs.finish_temp, fh)
|
| 928 |
+
finally:
|
| 929 |
+
if not fh.closed:
|
| 930 |
+
fh.close()
|
| 931 |
+
digest = hasher.hexdigest()
|
| 932 |
+
expected = request.headers.get("x-checksum-sha256")
|
| 933 |
+
if expected and expected.lower() != digest:
|
| 934 |
+
raise bad_request("X-Checksum-Sha256 does not match the uploaded content")
|
| 935 |
+
async with board.commit_lock:
|
| 936 |
+
current = await db.read(store.attachment_state, row["id"])
|
| 937 |
+
if current["state"] == "ready":
|
| 938 |
+
result = await db.write(store.finish_upload, row["id"], size, digest)
|
| 939 |
+
return JSONResponse(result, status_code=200)
|
| 940 |
+
await anyio.to_thread.run_sync(board.blobs.commit, tmp_path, row["storage_key"])
|
| 941 |
+
tmp_path = None
|
| 942 |
+
result = await db.write(store.finish_upload, row["id"], size, digest)
|
| 943 |
+
return JSONResponse(result, status_code=201)
|
| 944 |
+
finally:
|
| 945 |
+
board.open_uploads -= 1
|
| 946 |
+
if tmp_path is not None:
|
| 947 |
+
board.blobs.discard(tmp_path)
|
| 948 |
+
|
| 949 |
+
@app.get("/v1/attachments/{att_id}/info")
|
| 950 |
+
async def attachment_info(att_id: str, request: Request):
|
| 951 |
+
p = await authn(request)
|
| 952 |
+
info = await db.read(store.get_attachment, p, att_id)
|
| 953 |
+
info.pop("storage_key")
|
| 954 |
+
return info
|
| 955 |
+
|
| 956 |
+
@app.get("/v1/attachments/{att_id}")
|
| 957 |
+
async def download(att_id: str, request: Request):
|
| 958 |
+
p = await authn(request)
|
| 959 |
+
info = await db.read(store.get_attachment, p, att_id)
|
| 960 |
+
return FileResponse(
|
| 961 |
+
board.blobs.path_for(info["storage_key"]),
|
| 962 |
+
media_type="application/octet-stream",
|
| 963 |
+
filename=info["filename"] or att_id,
|
| 964 |
+
headers={
|
| 965 |
+
"X-Checksum-Sha256": info["sha256"],
|
| 966 |
+
"X-Content-Type-Options": "nosniff",
|
| 967 |
+
"X-Upload-Content-Type": info["content_type"] or "application/octet-stream",
|
| 968 |
+
},
|
| 969 |
+
)
|
| 970 |
+
|
| 971 |
+
# ------------------------------------------------------------------ storage folders
|
| 972 |
+
|
| 973 |
+
def _cache_put(src: Path, sha: str) -> Path:
|
| 974 |
+
dest = board.storage_cache / sha
|
| 975 |
+
os.replace(src, dest)
|
| 976 |
+
files = sorted(board.storage_cache.iterdir(), key=lambda f: f.stat().st_atime)
|
| 977 |
+
total = sum(f.stat().st_size for f in files)
|
| 978 |
+
for f in files: # keep the cache under its size, oldest first
|
| 979 |
+
if total <= LIMITS.storage_cache_bytes or f == dest:
|
| 980 |
+
continue
|
| 981 |
+
total -= f.stat().st_size
|
| 982 |
+
f.unlink(missing_ok=True)
|
| 983 |
+
return dest
|
| 984 |
+
|
| 985 |
+
@app.put("/v1/storage/{handle}/{path:path}")
|
| 986 |
+
async def storage_put(handle: str, path: str, request: Request):
|
| 987 |
+
p = await authn(request)
|
| 988 |
+
owner = await db.read(store.storage_owner, p, handle, True)
|
| 989 |
+
path = store.storage_path(path)
|
| 990 |
+
declared = request.headers.get("content-length")
|
| 991 |
+
if declared and declared.isdigit():
|
| 992 |
+
await db.read(store.storage_check_quota, owner["id"], path, int(declared))
|
| 993 |
+
content_type = (request.headers.get("content-type") or mimetypes.guess_type(path)[0]
|
| 994 |
+
or "application/octet-stream")[:128]
|
| 995 |
+
tmp_path, fh = await anyio.to_thread.run_sync(board.blobs.open_temp)
|
| 996 |
+
try:
|
| 997 |
+
hasher, size = new_hasher(), 0
|
| 998 |
+
try:
|
| 999 |
+
max_file = (await db.read(store.storage_limits))[0]
|
| 1000 |
+
async for chunk in request.stream():
|
| 1001 |
+
size += len(chunk)
|
| 1002 |
+
if size > max_file:
|
| 1003 |
+
raise ApiError(413, "too_large", f"files are limited to {store._mb(max_file)} on this board")
|
| 1004 |
+
hasher.update(chunk)
|
| 1005 |
+
await anyio.to_thread.run_sync(fh.write, chunk)
|
| 1006 |
+
await anyio.to_thread.run_sync(board.blobs.finish_temp, fh)
|
| 1007 |
+
finally:
|
| 1008 |
+
if not fh.closed:
|
| 1009 |
+
fh.close()
|
| 1010 |
+
digest = hasher.hexdigest()
|
| 1011 |
+
expected = request.headers.get("x-checksum-sha256")
|
| 1012 |
+
if expected and expected.lower() != digest:
|
| 1013 |
+
raise bad_request("X-Checksum-Sha256 does not match the uploaded content")
|
| 1014 |
+
await db.read(store.storage_check_quota, owner["id"], path, size)
|
| 1015 |
+
key = store.storage_key(owner["id"], path)
|
| 1016 |
+
await anyio.to_thread.run_sync(lambda: board.storage.put([(tmp_path, key)])) # in the bucket before it is listed
|
| 1017 |
+
await anyio.to_thread.run_sync(_cache_put, tmp_path, digest)
|
| 1018 |
+
tmp_path = None
|
| 1019 |
+
result = await db.write(store.storage_record, p, owner["id"], path, size, digest, content_type, time.time())
|
| 1020 |
+
return JSONResponse(result, status_code=201)
|
| 1021 |
+
finally:
|
| 1022 |
+
if tmp_path is not None:
|
| 1023 |
+
board.blobs.discard(tmp_path)
|
| 1024 |
+
|
| 1025 |
+
@app.get("/v1/storage/{handle}")
|
| 1026 |
+
async def storage_root(handle: str, request: Request, prefix: str = ""):
|
| 1027 |
+
p = await authn(request)
|
| 1028 |
+
return await db.read(store.storage_list, p, handle, prefix)
|
| 1029 |
+
|
| 1030 |
+
@app.get("/v1/storage/{handle}/{path:path}")
|
| 1031 |
+
async def storage_get(handle: str, path: str, request: Request, info: bool = False):
|
| 1032 |
+
p = await authn(request)
|
| 1033 |
+
if path.endswith("/"): # a folder: list it
|
| 1034 |
+
return await db.read(store.storage_list, p, handle, path)
|
| 1035 |
+
meta, key = await db.read(store.storage_lookup, p, handle, path)
|
| 1036 |
+
if info:
|
| 1037 |
+
return meta
|
| 1038 |
+
cached = board.storage_cache / meta["sha256"]
|
| 1039 |
+
if not cached.exists():
|
| 1040 |
+
tmp = board.storage_cache / f".{meta['sha256']}.{os.getpid()}.part"
|
| 1041 |
+
ok = await anyio.to_thread.run_sync(board.storage.get, key, tmp)
|
| 1042 |
+
if not ok:
|
| 1043 |
+
raise ApiError(503, "storage_unavailable", "the file is listed but could not be fetched from storage; try again")
|
| 1044 |
+
os.replace(tmp, cached)
|
| 1045 |
+
# always a download: never let stored HTML or SVG run on the board's own origin
|
| 1046 |
+
return FileResponse(cached, media_type="application/octet-stream", filename=os.path.basename(meta["path"]),
|
| 1047 |
+
headers={"X-Checksum-Sha256": meta["sha256"], "X-Content-Type-Options": "nosniff",
|
| 1048 |
+
"X-Upload-Content-Type": meta["content_type"] or "application/octet-stream",
|
| 1049 |
+
"Content-Security-Policy": "sandbox", "Cache-Control": "private, no-cache"})
|
| 1050 |
+
|
| 1051 |
+
@app.delete("/v1/storage/{handle}/{path:path}")
|
| 1052 |
+
async def storage_delete(handle: str, path: str, request: Request):
|
| 1053 |
+
p = await authn(request)
|
| 1054 |
+
out = await db.write(store.storage_remove, p, handle, path, time.time())
|
| 1055 |
+
keys = out.pop("keys")
|
| 1056 |
+
await anyio.to_thread.run_sync(lambda: board.storage.delete(keys))
|
| 1057 |
+
return out
|
| 1058 |
+
|
| 1059 |
+
if dashboard:
|
| 1060 |
+
page = Path(__file__).with_name("dashboard.html")
|
| 1061 |
+
|
| 1062 |
+
@app.get("/", include_in_schema=False)
|
| 1063 |
+
async def dashboard_page():
|
| 1064 |
+
return FileResponse(page, media_type="text/html",
|
| 1065 |
+
headers={"Cache-Control": "no-cache", "X-Content-Type-Options": "nosniff"})
|
| 1066 |
+
|
| 1067 |
+
return app
|
src/msgboard/store.py
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|