mirror of
https://github.com/agessaman/meshcore-bot.git
synced 2026-10-30 17:42:50 +00:00
agessaman
727909ba8e
setup_logging's formatter, handler and library-logger wiring is now configure_bot_logging(config, bot_root); MeshCoreBot.setup_logging calls it and still installs the signal handlers. The meshcore logger list that setup_logging and _configure_meshcore_debug_logging each spelled out is one MESHCORE_LOGGER_NAMES, and both use configure_meshcore_loggers. _JsonFormatter moves along as JsonFormatter. Statement order inside setup is unchanged, so the startup log lines come out as before. Also fixes mypy on the admin_server module from the previous commit: as a new module it is type-checked, and its Flask view annotations needed loosening.
62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
"""Loopback HTTP admin API for the bot: config reload and a health check."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import threading
|
|
from typing import Any
|
|
|
|
|
|
class BotAdminServer(threading.Thread):
|
|
"""Minimal Flask HTTP server exposing bot admin endpoints.
|
|
|
|
Runs in a daemon thread alongside the bot's asyncio loop.
|
|
Configured via ``[Admin]`` section in config.ini:
|
|
|
|
[Admin]
|
|
enabled = true
|
|
port = 5001
|
|
token = <secret> ; required; requests without matching Bearer token are rejected
|
|
"""
|
|
|
|
def __init__(self, bot: Any, port: int, token: str) -> None:
|
|
super().__init__(daemon=True, name="BotAdminServer")
|
|
self._bot = bot
|
|
self._port = port
|
|
self._token = token
|
|
|
|
def run(self) -> None:
|
|
try:
|
|
from flask import Flask, jsonify
|
|
from flask import request as flask_request
|
|
|
|
app = Flask("bot_admin")
|
|
# Suppress Flask startup banner and request logs
|
|
import logging as _logging
|
|
_logging.getLogger("werkzeug").setLevel(_logging.ERROR)
|
|
|
|
def _check_auth() -> Any:
|
|
auth = flask_request.headers.get("Authorization", "")
|
|
if not auth.startswith("Bearer ") or auth[7:] != self._token:
|
|
return jsonify({"error": "unauthorized"}), 401
|
|
return None
|
|
|
|
@app.post("/api/admin/reload")
|
|
def reload_config() -> Any:
|
|
denied = _check_auth()
|
|
if denied is not None:
|
|
return denied
|
|
success, msg = self._bot.reload_config()
|
|
status = 200 if success else 409
|
|
return jsonify({"success": success, "message": msg}), status
|
|
|
|
@app.get("/api/admin/health")
|
|
def health() -> Any:
|
|
denied = _check_auth()
|
|
if denied is not None:
|
|
return denied
|
|
return jsonify({"status": "ok"})
|
|
|
|
app.run(host="127.0.0.1", port=self._port, threaded=True)
|
|
except Exception as exc: # noqa: BLE001
|
|
self._bot.logger.error("BotAdminServer failed to start: %s", exc)
|