Started repo and finished the bot in one shot
This commit is contained in:
@@ -0,0 +1,6 @@
|
||||
# Discord bot token (Discord Developer Portal -> your application -> Bot)
|
||||
DISCORD_TOKEN=
|
||||
|
||||
# Optional tuning
|
||||
CHECK_INTERVAL=60
|
||||
CHECK_TIMEOUT=10
|
||||
@@ -0,0 +1,5 @@
|
||||
.env
|
||||
venv/
|
||||
__pycache__/
|
||||
*.pyc
|
||||
.status_message.json
|
||||
@@ -0,0 +1 @@
|
||||
you have to use the venv inside the venv folder on this path, dont try to use the system's python
|
||||
@@ -0,0 +1,90 @@
|
||||
import asyncio
|
||||
import json
|
||||
import time
|
||||
|
||||
import aiohttp
|
||||
|
||||
TARGETS = [
|
||||
{
|
||||
'name': 'Gitea',
|
||||
'url': 'https://gitea.rainbow-herring.ts.net/api/healthz',
|
||||
'kind': 'gitea',
|
||||
},
|
||||
{
|
||||
'name': 'Packs Site (Production)',
|
||||
'url': 'https://jakerasp.rainbow-herring.ts.net/healthz/',
|
||||
'kind': 'packs',
|
||||
},
|
||||
{
|
||||
'name': 'Packs Site (Development)',
|
||||
'url': 'https://jakepc.rainbow-herring.ts.net/healthz/',
|
||||
'kind': 'packs',
|
||||
},
|
||||
]
|
||||
|
||||
GITEA_VERSION_URL = 'https://gitea.rainbow-herring.ts.net/api/v1/version'
|
||||
|
||||
|
||||
async def _get(session, url, timeout):
|
||||
started = time.monotonic()
|
||||
async with session.get(
|
||||
url,
|
||||
timeout=aiohttp.ClientTimeout(total=timeout),
|
||||
allow_redirects=False,
|
||||
) as resp:
|
||||
latency_ms = round((time.monotonic() - started) * 1000)
|
||||
body = await resp.text()
|
||||
return resp.status, latency_ms, body
|
||||
|
||||
|
||||
async def check_target(session, target, timeout):
|
||||
result = {
|
||||
'name': target['name'],
|
||||
'kind': target.get('kind', ''),
|
||||
'url': target['url'],
|
||||
'up': False,
|
||||
'status_code': None,
|
||||
'latency_ms': None,
|
||||
'db': None,
|
||||
'version': None,
|
||||
'detail': '',
|
||||
}
|
||||
try:
|
||||
status, latency_ms, body = await _get(session, target['url'], timeout)
|
||||
result['status_code'] = status
|
||||
result['latency_ms'] = latency_ms
|
||||
|
||||
if status == 200:
|
||||
data = {}
|
||||
try:
|
||||
data = json.loads(body)
|
||||
except (json.JSONDecodeError, ValueError):
|
||||
pass
|
||||
result['up'] = data.get('status') == 'pass'
|
||||
if data.get('db') is not None:
|
||||
result['db'] = data['db']
|
||||
else:
|
||||
result['detail'] = f'HTTP {status}'
|
||||
except asyncio.TimeoutError:
|
||||
result['detail'] = 'timeout'
|
||||
except aiohttp.ClientError:
|
||||
result['detail'] = 'connection error'
|
||||
except Exception as exc:
|
||||
result['detail'] = type(exc).__name__
|
||||
|
||||
if result['up'] and result['kind'] == 'gitea':
|
||||
try:
|
||||
status, _, body = await _get(session, GITEA_VERSION_URL, timeout)
|
||||
if status == 200:
|
||||
result['version'] = json.loads(body).get('version')
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
return result
|
||||
|
||||
|
||||
async def run_checks(session, timeout):
|
||||
results = await asyncio.gather(
|
||||
*(check_target(session, target, timeout) for target in TARGETS)
|
||||
)
|
||||
return list(results)
|
||||
@@ -0,0 +1,12 @@
|
||||
import os
|
||||
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv()
|
||||
|
||||
DISCORD_TOKEN = os.getenv('DISCORD_TOKEN', '').strip()
|
||||
|
||||
CHECK_INTERVAL = int(os.getenv('CHECK_INTERVAL', '60'))
|
||||
CHECK_TIMEOUT = float(os.getenv('CHECK_TIMEOUT', '10'))
|
||||
|
||||
STATE_FILE = os.getenv('STATE_FILE', '.status_message.json')
|
||||
@@ -0,0 +1,17 @@
|
||||
[Unit]
|
||||
Description=Health Check Discord Bot
|
||||
After=network-online.target
|
||||
Wants=network-online.target
|
||||
|
||||
[Service]
|
||||
Type=simple
|
||||
User=jake
|
||||
Group=jake
|
||||
WorkingDirectory=/mnt/Disco/Proyects/Python3.13/HealthCheckBot/
|
||||
ExecStart=/mnt/Disco/Proyects/Python3.13/HealthCheckBot/venv/bin/python /mnt/Disco/Proyects/Python3.13/HealthCheckBot/main.py
|
||||
Restart=always
|
||||
RestartSec=5
|
||||
EnvironmentFile=/mnt/Disco/Proyects/Python3.13/HealthCheckBot/.env
|
||||
|
||||
[Install]
|
||||
WantedBy=multi-user.target
|
||||
@@ -0,0 +1,169 @@
|
||||
import asyncio
|
||||
import json
|
||||
import logging
|
||||
|
||||
import aiohttp
|
||||
import discord
|
||||
from discord import app_commands
|
||||
|
||||
import checks
|
||||
import config
|
||||
|
||||
log = logging.getLogger('healthbot')
|
||||
|
||||
|
||||
class HealthBot(discord.Client):
|
||||
def __init__(self):
|
||||
intents = discord.Intents.default()
|
||||
super().__init__(intents=intents)
|
||||
self.tree = app_commands.CommandTree(self)
|
||||
self.session: aiohttp.ClientSession | None = None
|
||||
self.status_messages: list[dict] = []
|
||||
self._monitor_task: asyncio.Task | None = None
|
||||
|
||||
def _load_state(self):
|
||||
try:
|
||||
with open(config.STATE_FILE, 'r') as f:
|
||||
data = json.load(f)
|
||||
self.status_messages = [
|
||||
{'channel_id': int(m['channel_id']), 'message_id': int(m['message_id'])}
|
||||
for m in data.get('messages', [])
|
||||
]
|
||||
except (OSError, ValueError, TypeError, KeyError):
|
||||
self.status_messages = []
|
||||
|
||||
def _save_state(self):
|
||||
try:
|
||||
with open(config.STATE_FILE, 'w') as f:
|
||||
json.dump({'messages': self.status_messages}, f)
|
||||
except OSError:
|
||||
log.warning('Could not save state file %s', config.STATE_FILE)
|
||||
|
||||
def register_status_message(self, channel_id, message_id):
|
||||
if any(m['message_id'] == message_id for m in self.status_messages):
|
||||
return
|
||||
self.status_messages.append({'channel_id': channel_id, 'message_id': message_id})
|
||||
self._save_state()
|
||||
|
||||
def forget_status_message(self, message_id):
|
||||
self.status_messages = [
|
||||
m for m in self.status_messages if m['message_id'] != message_id
|
||||
]
|
||||
self._save_state()
|
||||
|
||||
async def _ensure_session(self):
|
||||
if self.session is None or self.session.closed:
|
||||
self.session = aiohttp.ClientSession()
|
||||
return self.session
|
||||
|
||||
async def run_checks(self):
|
||||
session = await self._ensure_session()
|
||||
return await checks.run_checks(session, config.CHECK_TIMEOUT)
|
||||
|
||||
def build_embed(self, results):
|
||||
all_up = all(r['up'] for r in results)
|
||||
color = discord.Color.green() if all_up else discord.Color.red()
|
||||
embed = discord.Embed(
|
||||
title='Service Health',
|
||||
color=color,
|
||||
timestamp=discord.utils.utcnow(),
|
||||
)
|
||||
for r in results:
|
||||
emoji = '🟢' if r['up'] else '🔴'
|
||||
code = r['status_code'] if r['status_code'] is not None else '—'
|
||||
latency = f"{r['latency_ms']} ms" if r['latency_ms'] is not None else '—'
|
||||
value = f'{emoji} `HTTP {code}` · {latency}'
|
||||
if r.get('db') is not None:
|
||||
value += f' · db {r["db"]}'
|
||||
if r.get('version'):
|
||||
value += f' · v{r["version"]}'
|
||||
if r.get('detail'):
|
||||
value += f'\n*{r["detail"]}*'
|
||||
embed.add_field(name=r['name'], value=value, inline=False)
|
||||
embed.set_footer(text=f'Updates every {config.CHECK_INTERVAL}s')
|
||||
return embed
|
||||
|
||||
async def update_embeds(self):
|
||||
if not self.status_messages:
|
||||
return
|
||||
results = await self.run_checks()
|
||||
embed = self.build_embed(results)
|
||||
|
||||
kept = []
|
||||
for entry in self.status_messages:
|
||||
channel = self.get_channel(entry['channel_id'])
|
||||
if channel is None:
|
||||
try:
|
||||
channel = await self.fetch_channel(entry['channel_id'])
|
||||
except (discord.NotFound, discord.Forbidden, discord.HTTPException):
|
||||
continue
|
||||
try:
|
||||
message = await channel.fetch_message(entry['message_id'])
|
||||
await message.edit(embed=embed)
|
||||
kept.append(entry)
|
||||
except (discord.NotFound, discord.Forbidden, discord.HTTPException):
|
||||
continue
|
||||
|
||||
self.status_messages = kept
|
||||
self._save_state()
|
||||
|
||||
async def _monitor_loop(self):
|
||||
await self.wait_until_ready()
|
||||
log.info('Monitor loop started (interval %ss)', config.CHECK_INTERVAL)
|
||||
while not self.is_closed():
|
||||
try:
|
||||
await self.update_embeds()
|
||||
except Exception as exc:
|
||||
log.exception('Monitor tick failed: %s', exc)
|
||||
await asyncio.sleep(config.CHECK_INTERVAL)
|
||||
|
||||
async def on_ready(self):
|
||||
log.info('Logged in as %s (id %s)', self.user, self.user.id)
|
||||
self._load_state()
|
||||
await self.tree.sync()
|
||||
log.info('Synced slash commands; tracking %d status message(s)', len(self.status_messages))
|
||||
if self._monitor_task is None or self._monitor_task.done():
|
||||
self._monitor_task = self.loop.create_task(self._monitor_loop())
|
||||
|
||||
async def close(self):
|
||||
if self.session is not None and not self.session.closed:
|
||||
await self.session.close()
|
||||
await super().close()
|
||||
|
||||
|
||||
bot = HealthBot()
|
||||
|
||||
|
||||
@bot.tree.command(name='status', description='Post a live-updating health embed for the services')
|
||||
async def status(interaction: discord.Interaction):
|
||||
await interaction.response.defer()
|
||||
results = await bot.run_checks()
|
||||
embed = bot.build_embed(results)
|
||||
|
||||
existing = next(
|
||||
(m for m in bot.status_messages if m['channel_id'] == interaction.channel_id),
|
||||
None,
|
||||
)
|
||||
if existing:
|
||||
channel = interaction.channel
|
||||
try:
|
||||
message = await channel.fetch_message(existing['message_id'])
|
||||
await message.edit(embed=embed)
|
||||
await interaction.followup.send(
|
||||
'Refreshed the existing status embed.', ephemeral=True,
|
||||
)
|
||||
return
|
||||
except (discord.NotFound, discord.Forbidden, discord.HTTPException):
|
||||
bot.forget_status_message(existing['message_id'])
|
||||
|
||||
message = await interaction.followup.send(embed=embed)
|
||||
bot.register_status_message(interaction.channel_id, message.id)
|
||||
|
||||
|
||||
def main():
|
||||
logging.basicConfig(level=logging.INFO, format='%(asctime)s %(name)s %(levelname)s %(message)s')
|
||||
bot.run(config.DISCORD_TOKEN)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
@@ -0,0 +1,3 @@
|
||||
discord.py
|
||||
aiohttp
|
||||
python-dotenv
|
||||
Reference in New Issue
Block a user