#description: telegram bot for monitoring the system
#dependencies: telebot
#usage: python3 lainmonitor.py | or run it as a service
#author: hornetmaidan
import subprocess
import threading
import telebot
#define the variables
status, hostname, uptime, zerotier, prosody, postgres, tailscale, disk, ping = 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown', 'unknown'
nodes, hostnames, reach, threads = [], [], [], []
#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

#function to ping tailscale nodes
def ping_node(node, hostname):
        ping = subprocess.Popen(f"ping {node} -c 1 | grep '1 packets'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
        if '1 received' in ping:
            reach.append(f'{node}/{hostname} is reachable')
        else:
            reach.append(f'{node}/{hostname} is unreachable')

#ping tailscale nodes
def check_tailscale():
    global nodes, hostnames, reach, threads, ping
    nodes_output = subprocess.Popen("tailscale status | grep '100'", shell=True, stdout=subprocess.PIPE).stdout.read().decode().strip()
    nodes = [line.split()[0] for line in nodes_output.split('\n') if line]
    hostnames = [line.split()[1] for line in nodes_output.split('\n') if line]

    for node, hostname in zip(nodes, hostnames):
            thread = threading.Thread(target=ping_node, args=(node, hostname))
            threads.append(thread)
            thread.start()

    for thread in threads:
        thread.join()

    return reach

#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: \n\n{reach}')

#polling
bot.polling()