mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-10-31 04:45:23 +08:00
52 lines
2.2 KiB
Python
52 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
from langbot_plugin.api.definition.components.page import Page, PageRequest, PageResponse
|
|
|
|
|
|
class ManagePage(Page):
|
|
"""Management UI backend for the Daily Limit plugin.
|
|
|
|
Endpoints (called from index.html via the Page SDK ``langbot.api``):
|
|
|
|
GET /state -> full snapshot (settings + sessions)
|
|
PUT /settings {..} -> update global settings / new-session default
|
|
PUT /session {id, limit} -> set per-session limit override (null = use default)
|
|
POST /reset {id} -> reset one session's counter to 0
|
|
POST /reset-all -> reset every session's counter
|
|
DELETE /session {id} -> stop tracking a session (removes the row)
|
|
"""
|
|
|
|
async def handle_api(self, request: PageRequest) -> PageResponse:
|
|
plugin = self.plugin
|
|
body = request.body or {}
|
|
ep, method = request.endpoint, request.method
|
|
|
|
if ep == "/state" and method == "GET":
|
|
return PageResponse.ok(await plugin.snapshot())
|
|
|
|
if ep == "/settings" and method == "PUT":
|
|
await plugin.update_settings(body)
|
|
return PageResponse.ok(await plugin.snapshot())
|
|
|
|
if ep == "/session" and method == "PUT":
|
|
session_id = body.get("id", "")
|
|
if not await plugin.set_session_limit(session_id, body.get("limit")):
|
|
return PageResponse.fail("session not found")
|
|
return PageResponse.ok(await plugin.snapshot())
|
|
|
|
if ep == "/session" and method == "DELETE":
|
|
if not await plugin.delete_session(body.get("id", "")):
|
|
return PageResponse.fail("session not found")
|
|
return PageResponse.ok(await plugin.snapshot())
|
|
|
|
if ep == "/reset" and method == "POST":
|
|
if not await plugin.reset_session(body.get("id", "")):
|
|
return PageResponse.fail("session not found")
|
|
return PageResponse.ok(await plugin.snapshot())
|
|
|
|
if ep == "/reset-all" and method == "POST":
|
|
n = await plugin.reset_all()
|
|
return PageResponse.ok({**(await plugin.snapshot()), "reset_count": n})
|
|
|
|
return PageResponse.fail(f"Unknown endpoint: {method} {ep}")
|