From 93c4c0d49e54927345696624cc122e4ee62a2bcc Mon Sep 17 00:00:00 2001
From: hornet <hornet@lainlounge.xyz>
Date: Fri, 4 Oct 2024 18:58:46 +0000
Subject: [PATCH] uploaded sou

---
 lainmonitor.py   | 75 ++++++++++++++++++++++++++++++++++++++++++++++++
 requirements.txt |  1 +
 2 files changed, 76 insertions(+)
 create mode 100644 lainmonitor.py
 create mode 100644 requirements.txt

diff --git a/lainmonitor.py b/lainmonitor.py
new file mode 100644
index 0000000..5386e5a
--- /dev/null
+++ b/lainmonitor.py
@@ -0,0 +1,75 @@
+#description: telegram bot for monitoring the system
+#dependencies: telebot
+#usage: python3 lainmonitor.py | or run it as a service
+#author: hornetmaidan
+import subprocess
+import telebot
+#define the variables
+status, hostname, uptime, zerotier, prosody, postgres, tailscale, disk, ping = 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown'
+#telegram bot token
+TOKEN = 'PLACE_YOUR_TOKEN_HERE'
+
+#bot init
+bot = telebot.TeleBot(TOKEN)
+
+#get system info
+def getinfo():
+    global status, hostname, uptime, zerotier, prosody, postgres, tailscale, disk
+    hostname = subprocess.check_output(['hostname']).decode().strip()
+    uptime = subprocess.check_output(['uptime', '-p']).decode().strip()
+    #systemd-only services
+    zerotier = subprocess.Popen("sudo systemctl status zerotier-one | grep 'Active'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
+    prosody = subprocess.Popen("sudo systemctl status prosody | grep 'Active'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
+    postgres = subprocess.Popen("sudo systemctl status postgresql | grep 'Active'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
+    tailscale = subprocess.Popen("sudo systemctl status tailscaled | grep 'Active'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
+    disk = subprocess.check_output(['df', '-h']).decode().strip()
+    if hostname == 'unknown':
+        status = 'offline'
+    else:
+        status = 'online'
+    return hostname, uptime, zerotier, prosody, postgres, tailscale, disk
+
+#ping tailscale (change the IP address to the one you want or add more)
+def check_tailscale():
+    global ping
+    ping = subprocess.Popen("ping TAILSCALE_IP -c 1 | grep '1 packets'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
+    if '1 received' in ping:
+        ping = 'connected'
+    else:
+        ping = 'unreachable'
+    return ping
+
+#debug handler
+def check():
+    global status, hostname, uptime, zerotier, prosody, postgres, tailscale, disk
+    getinfo()
+    print('system status:', status)
+    print('hostname:', hostname)
+    print('uptime:', uptime)
+    print('zerotier:', zerotier)
+    print('prosody:', prosody)
+    print('postgres:', postgres)
+    print('tailscale:', tailscale)
+    print('disk:', disk)
+    return status, hostname, uptime, zerotier, prosody, postgres, tailscale, disk
+
+#message handling
+@bot.message_handler(commands=['start', 'help', 'status', 'reboot', 'ping'])
+def handle(message):
+    if message.text == '/start':
+        bot.reply_to(message, 'lainmonitor v1.0 --- standing by...')
+    elif message.text == '/help':
+        bot.reply_to(message, 'commands: /start, /help, /status, /reboot, /ping')
+    elif message.text == '/status':
+        check()
+        status_message = f'hostname: {hostname}\nsystem status: {status}\nuptime: {uptime}\nzerotier: {zerotier}\nprosody: {prosody}\npostgres: {postgres}\ntailscale: {tailscale}'
+        bot.reply_to(message, status_message)
+        bot.reply_to(message, f'filesystem info for {hostname}: \n\n{disk}')
+    elif message.text == '/reboot':
+        bot.reply_to(message, 'work in progress...')
+    elif message.text == '/ping':
+        check_tailscale()
+        bot.reply_to(message, f'ping status: {ping}')
+
+#polling
+bot.polling()
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..b1d932a
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1 @@
+telebot