uploaded sou

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()

requirements.txt

@@ -0,0 +1 @@
+telebot