mirror of
https://github.com/kuhyx/utils.git
synced 2026-11-14 08:30:58 +01:00
Krzysztof kuhy Rudnicki
7af6afe51a
Some checks failed
dependency freshness / dependencies on newest stable (push) Has been cancelled
file length / file-length (push) Has been cancelled
md-naming / markdown naming convention (push) Has been cancelled
omniroute_offload pre-commit / pre-commit (push) Has been cancelled
palette drift / palette (push) Has been cancelled
repo-contract / repo contract (push) Has been cancelled
shared gate tests / gate unit tests (push) Has been cancelled
omniroute-gateway start|health|update|backup|combo keeps the local OmniRoute gateway up. systemd --user timers: health every 10 min (real free-best chat; restart <=1/30 min, reactivate providers whose cooldown ended, notify once per outage), update weekly Sun 12:00 (backup, newest stable -web tag, rollback of image + volume on a failed health check). Failure modes: health/update fail loudly (exit 1 + desktop + Signal); all providers cooling = degraded (exit 0, one notice/day); a missing signal-bot .env logs and exits non-zero on alert paths; health skips while update holds the lock. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01WzxdWZSdJmNjGUvLnHJ4WG
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""omniroute-gateway: keep the local OmniRoute gateway up, checked and current.
|
|
|
|
start (re)create the container on the pinned image and wait for its API
|
|
health real chat probe + fix ladder (systemd timer, every 10 min)
|
|
update backup + newest stable tag with rollback (systemd timer, weekly)
|
|
backup stop, tar the data volume, start
|
|
combo rebuild free-best from ~/.config/omniroute/order.tsv
|
|
Instance: OMNIROUTE_NAME / OMNIROUTE_VOLUME / OMNIROUTE_PORT (default: live).
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import sys
|
|
from typing import TYPE_CHECKING
|
|
|
|
from omniroute_offload import _gw_api as api
|
|
from omniroute_offload import _gw_combo as combo
|
|
from omniroute_offload import _gw_docker as dock
|
|
from omniroute_offload import _gw_health as health
|
|
from omniroute_offload import _gw_update as update
|
|
from omniroute_offload._gw_config import Config
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Callable
|
|
|
|
|
|
def _start(cfg: Config) -> int:
|
|
dock.recreate(cfg, cfg.image(), health.secret(cfg, "admin_password"))
|
|
if health.wait_up(cfg):
|
|
print(f"omniroute up: {cfg.base_url} ({cfg.image()})")
|
|
return 0
|
|
print(f"{cfg.name} did not come up; see: docker logs {cfg.name}", file=sys.stderr)
|
|
return 1
|
|
|
|
|
|
def _backup(cfg: Config) -> int:
|
|
print(dock.backup(cfg, cfg.image().rsplit(":", 1)[-1]))
|
|
return 0
|
|
|
|
|
|
def _combo(cfg: Config) -> int:
|
|
return 0 if combo.refresh(cfg) else 1
|
|
|
|
|
|
def main(argv: list[str] | None = None) -> int:
|
|
"""Run one subcommand; non-zero on failure."""
|
|
parser = argparse.ArgumentParser(
|
|
prog="omniroute-gateway",
|
|
description=__doc__,
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
)
|
|
parser.add_argument(
|
|
"command", choices=["start", "health", "update", "backup", "combo"]
|
|
)
|
|
args = parser.parse_args(argv)
|
|
cfg = Config.from_env()
|
|
actions: dict[str, Callable[[Config], int]] = {
|
|
"start": _start,
|
|
"health": health.run,
|
|
"update": update.run,
|
|
"backup": _backup,
|
|
"combo": _combo,
|
|
}
|
|
try:
|
|
return actions[args.command](cfg)
|
|
except (dock.DockerError, api.ApiError, OSError) as err:
|
|
print(f"omniroute-gateway {args.command}: {err}", file=sys.stderr)
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|