83 lines
2.4 KiB
Nix
83 lines
2.4 KiB
Nix
{ pkgs, scripts }:
|
|
let
|
|
notify_script = (pkgs.writers.writePython3Bin "telegram-notify.py"
|
|
{
|
|
libraries = [ pkgs.python3Packages.python-telegram-bot ];
|
|
} ''
|
|
import telegram
|
|
import asyncio
|
|
import sys
|
|
import socket
|
|
import json
|
|
import subprocess
|
|
import tempfile
|
|
|
|
|
|
async def run():
|
|
text = subprocess.check_output(
|
|
[
|
|
"journalctl",
|
|
"--user",
|
|
"-u",
|
|
sys.argv[1],
|
|
"--since",
|
|
"-1h",
|
|
"--no-pager"
|
|
]
|
|
).decode("utf-8")
|
|
|
|
with open("/home/server/.secrets/Telegram/token.json") as f:
|
|
credentials = json.load(f)
|
|
token = credentials["token"]
|
|
chat_id = credentials["chat_id"]
|
|
|
|
with open("/home/server/mail.log", "a") as f:
|
|
f.write(text)
|
|
f.write("\n===========================================\n")
|
|
|
|
bot = telegram.Bot(token=token)
|
|
hostname = socket.gethostname()
|
|
await bot.send_message(
|
|
chat_id,
|
|
f"{hostname} encountered an error in the service: {sys.argv[1]}"
|
|
)
|
|
|
|
with tempfile.NamedTemporaryFile(suffix=".txt") as tmp_file:
|
|
tmp_file.write(text.encode("utf-8"))
|
|
tmp_file.seek(0)
|
|
await bot.send_document(chat_id, document=tmp_file)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(asyncio.wait([loop.create_task(run())]))
|
|
loop.close()
|
|
'');
|
|
|
|
mkTimer = name: cfg: {
|
|
Install.WantedBy = [ "timers.target" ];
|
|
Timer = {
|
|
Persistent = true;
|
|
OnCalendar = cfg.when;
|
|
Unit = "${name}.service";
|
|
};
|
|
};
|
|
mkService = name: cfg: {
|
|
Unit.Description = name;
|
|
Unit.OnFailure = "status_notify@${name}.service";
|
|
# Install = { WantedBy = [ "default.target" ]; };
|
|
Service = { ExecStart = cfg.script; };
|
|
};
|
|
in
|
|
{
|
|
systemd.user.services = pkgs.lib.mapAttrs mkService scripts
|
|
// (pkgs.lib.mapAttrs mkService {
|
|
"status_notify@" = {
|
|
script = "${notify_script}/bin/telegram-notify.py %i";
|
|
};
|
|
});
|
|
systemd.user.timers = pkgs.lib.mapAttrs mkTimer scripts;
|
|
# Don't forget to enable these timers! Or reboot, after which it should also be activated automatically
|
|
# systemctl --user enable --now <script>.timer
|
|
}
|