222 lines
12 KiB
Python
222 lines
12 KiB
Python
import logging
|
|
import subprocess
|
|
import os
|
|
from functools import wraps
|
|
from telegram import InlineKeyboardButton, InlineKeyboardMarkup, Update
|
|
from telegram.ext import (
|
|
Application,
|
|
CommandHandler,
|
|
CallbackQueryHandler,
|
|
ContextTypes,
|
|
)
|
|
|
|
# --- CONFIGURAZIONE ---
|
|
BOT_TOKEN = os.environ.get('BOT_TOKEN')
|
|
OWNER_ID = int(os.environ.get('ALLOWED_USER_ID'))
|
|
SSH_USER = "daniely"
|
|
NAS_USER = "daniely"
|
|
MASTER_IP = "192.168.128.80"
|
|
|
|
# --- LISTE DISPOSITIVI ---
|
|
# Core: Dispositivi intelligenti (SSH, CPU, RAM)
|
|
CORE_DEVICES = [
|
|
{"name": "🍓 Pi-1 (Master)", "ip": MASTER_IP, "type": "pi", "user": SSH_USER},
|
|
{"name": "🍓 Pi-2 (Backup)", "ip": "127.0.0.1", "type": "local", "user": ""},
|
|
{"name": "🗄️ NAS 920+", "ip": "192.168.128.100", "type": "nas", "user": NAS_USER},
|
|
{"name": "🗄️ NAS 214", "ip": "192.168.128.90", "type": "nas", "user": NAS_USER}
|
|
]
|
|
|
|
# Infra: Solo Ping e Riavvio (Switch, WiFi)
|
|
INFRA_DEVICES = [
|
|
{"name": "📡 Router", "ip": "192.168.128.1"},
|
|
{"name": "📶 WiFi Sala", "ip": "192.168.128.101"},
|
|
{"name": "📶 WiFi Luca", "ip": "192.168.128.102"},
|
|
{"name": "📶 WiFi Taverna", "ip": "192.168.128.103"},
|
|
{"name": "📶 WiFi Dado", "ip": "192.168.128.104"},
|
|
{"name": "🔌 Sw Sala", "ip": "192.168.128.105"},
|
|
{"name": "🔌 Sw Main", "ip": "192.168.128.106"},
|
|
{"name": "🔌 Sw Lav", "ip": "192.168.128.107"},
|
|
{"name": "🔌 Sw Tav", "ip": "192.168.128.108"}
|
|
]
|
|
|
|
logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# --- FUNZIONI ---
|
|
|
|
def run_cmd(command, ip=None, user=None):
|
|
try:
|
|
if ip == "127.0.0.1" or ip is None:
|
|
return subprocess.check_output(command, shell=True, stderr=subprocess.STDOUT).decode('utf-8').strip()
|
|
else:
|
|
safe_cmd = command.replace("'", "'\\''")
|
|
full_cmd = f"ssh -o LogLevel=ERROR -o BatchMode=yes -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null -o ConnectTimeout=4 {user}@{ip} '{safe_cmd}'"
|
|
return subprocess.check_output(full_cmd, shell=True, stderr=subprocess.STDOUT).decode('utf-8').strip()
|
|
except Exception: return "Err"
|
|
|
|
def get_ping_icon(ip):
|
|
try:
|
|
# Ping ultra-rapido (Timeout 1s)
|
|
subprocess.check_call(["ping", "-c", "1", "-W", "1", ip], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
return "✅"
|
|
except: return "🔴"
|
|
|
|
def get_device_stats(device):
|
|
ip, user, dtype = device['ip'], device['user'], device['type']
|
|
uptime_raw = run_cmd("uptime -p 2>/dev/null || uptime", ip, user)
|
|
if not uptime_raw or "Err" in uptime_raw: return "🔴 **OFFLINE**"
|
|
uptime = uptime_raw.replace("up ", "").split(", load")[0].split(", ")[0]
|
|
|
|
temp = "N/A"
|
|
if dtype in ["pi", "local"]:
|
|
t = run_cmd("cat /sys/class/thermal/thermal_zone0/temp", ip, user)
|
|
if t.isdigit(): temp = f"{int(t)/1000:.1f}°C"
|
|
elif dtype == "nas":
|
|
t = run_cmd("cat /sys/class/hwmon/hwmon0/temp1_input 2>/dev/null || cat /sys/class/thermal/thermal_zone0/temp", ip, user)
|
|
if t.isdigit():
|
|
v = int(t); temp = f"{v/1000:.1f}°C" if v > 1000 else f"{v}°C"
|
|
|
|
if dtype == "nas": ram_cmd = "free | grep Mem | awk '{printf \"%.0f%%\", $3*100/$2}'"
|
|
else: ram_cmd = "free -m | awk 'NR==2{if ($2>0) printf \"%.0f%%\", $3*100/$2; else print \"0%\"}'"
|
|
disk_path = "/" if dtype != "nas" else "/volume1"
|
|
disk_cmd = f"df -h {disk_path} | awk 'NR==2{{print $5}}'"
|
|
|
|
return f"✅ **ONLINE**\n⏱️ Up: {uptime}\n🌡️ Temp: {temp} | 🧠 RAM: {run_cmd(ram_cmd, ip, user)} | 💾 Disk: {run_cmd(disk_cmd, ip, user)}"
|
|
|
|
def read_log_file(filepath, lines=15):
|
|
if not os.path.exists(filepath): return "⚠️ File non trovato."
|
|
try: return subprocess.check_output(['tail', '-n', str(lines), filepath], stderr=subprocess.STDOUT).decode('utf-8')
|
|
except Exception as e: return f"Errore: {str(e)}"
|
|
|
|
def run_speedtest():
|
|
try: return subprocess.check_output("speedtest-cli --simple", shell=True, timeout=50).decode('utf-8')
|
|
except: return "Errore Speedtest"
|
|
|
|
# --- GESTIONE MENU ---
|
|
def restricted(func):
|
|
@wraps(func)
|
|
async def wrapped(update: Update, context: ContextTypes.DEFAULT_TYPE, *args, **kwargs):
|
|
if update.effective_user.id != OWNER_ID: return
|
|
return await func(update, context, *args, **kwargs)
|
|
return wrapped
|
|
|
|
@restricted
|
|
async def start(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
# MENU PRINCIPALE
|
|
keyboard = [
|
|
[InlineKeyboardButton("🖥️ Core Server", callback_data="menu_core"), InlineKeyboardButton("🔍 Scan LAN", callback_data="menu_lan")],
|
|
[InlineKeyboardButton("🛡️ Pi-hole", callback_data="menu_pihole"), InlineKeyboardButton("🌐 Rete", callback_data="menu_net")],
|
|
[InlineKeyboardButton("📜 Logs", callback_data="menu_logs")]
|
|
]
|
|
text = "🎛 **Loogle Control Center**\nSeleziona un pannello:"
|
|
|
|
if update.message:
|
|
await update.message.reply_text(text, reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
else:
|
|
# Se chiamato da un pulsante "Indietro", modifica il messaggio esistente
|
|
await update.callback_query.edit_message_text(text, reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
@restricted
|
|
async def button_handler(update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
|
query = update.callback_query
|
|
await query.answer()
|
|
data = query.data
|
|
|
|
if data == "main_menu": await start(update, context)
|
|
|
|
# --- MENU CORE ---
|
|
elif data == "menu_core":
|
|
keyboard = []
|
|
for i, dev in enumerate(CORE_DEVICES): keyboard.append([InlineKeyboardButton(dev['name'], callback_data=f"stat_{i}")])
|
|
keyboard.append([InlineKeyboardButton("📊 Report Completo", callback_data="stat_all")])
|
|
keyboard.append([InlineKeyboardButton("⬅️ Indietro", callback_data="main_menu")])
|
|
await query.edit_message_text("🖥️ **Core Servers**\nDettagli CPU/RAM/Temp:", reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
elif data == "stat_all":
|
|
await query.edit_message_text("⏳ **Analisi Core Servers...**", parse_mode="Markdown")
|
|
report = "📊 **REPORT CORE**\n"
|
|
for dev in CORE_DEVICES: report += f"\n🔹 **{dev['name']}**\n{get_device_stats(dev)}\n"
|
|
await query.edit_message_text(report, reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("⬅️ Indietro", callback_data="menu_core")]]), parse_mode="Markdown")
|
|
|
|
elif data.startswith("stat_"):
|
|
dev = CORE_DEVICES[int(data.split("_")[1])]
|
|
await query.edit_message_text(f"⏳ Controllo {dev['name']}...", parse_mode="Markdown")
|
|
await query.edit_message_text(f"🔹 **{dev['name']}**\n\n{get_device_stats(dev)}", reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("⬅️ Indietro", callback_data="menu_core")]]), parse_mode="Markdown")
|
|
|
|
# --- MENU LAN (DIAGNOSTICA) ---
|
|
elif data == "menu_lan":
|
|
await query.edit_message_text("⏳ **Ping test in corso...**", parse_mode="Markdown")
|
|
|
|
report = "🔍 **DIAGNOSTICA LAN**\n\n"
|
|
|
|
# Genera tabella compatta
|
|
for dev in INFRA_DEVICES:
|
|
status = get_ping_icon(dev['ip'])
|
|
# Aggiunge riga: Icona Nome (IP)
|
|
report += f"{status} **{dev['name']}**\n`{dev['ip']}`\n"
|
|
|
|
keyboard = [
|
|
[InlineKeyboardButton("⚡ Menu Riavvio Dispositivi", callback_data="menu_reboot")],
|
|
[InlineKeyboardButton("🔄 Aggiorna", callback_data="menu_lan")],
|
|
[InlineKeyboardButton("⬅️ Indietro", callback_data="main_menu")]
|
|
]
|
|
await query.edit_message_text(report, reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
# --- MENU RIAVVIO ---
|
|
elif data == "menu_reboot":
|
|
keyboard = []
|
|
for i, dev in enumerate(INFRA_DEVICES):
|
|
if "Router" not in dev['name']: # Protezione router
|
|
keyboard.append([InlineKeyboardButton(f"⚡ {dev['name']}", callback_data=f"reboot_{i}")])
|
|
keyboard.append([InlineKeyboardButton("⬅️ Indietro", callback_data="menu_lan")])
|
|
await query.edit_message_text("⚠️ **RIAVVIO REMOTO**\nRichiede SSH configurato sul dispositivo target (admin).", reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
elif data.startswith("reboot_"):
|
|
dev = INFRA_DEVICES[int(data.split("_")[1])]
|
|
# Tenta riavvio con utente 'admin' (standard per Zyxel/AP)
|
|
res = run_cmd("reboot", dev['ip'], "admin")
|
|
|
|
await query.edit_message_text(f"⚡ Comando inviato a {dev['name']}...\n\nRisposta:\n`{res}`", reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("⬅️ Indietro", callback_data="menu_reboot")]]), parse_mode="Markdown")
|
|
|
|
# --- ALTRI MENU (Pi-hole, Net, Logs) ---
|
|
elif data == "menu_pihole":
|
|
status_raw = run_cmd("sudo pihole status", MASTER_IP, SSH_USER)
|
|
icon = "✅" if "Enabled" in status_raw or "enabled" in status_raw else "🔴"
|
|
text = f"🛡️ **Pi-hole Master**\nStato: {icon}\n\n`{status_raw}`"
|
|
keyboard = [[InlineKeyboardButton("⏸️ 5m", callback_data="ph_disable_300"), InlineKeyboardButton("⏸️ 30m", callback_data="ph_disable_1800")], [InlineKeyboardButton("▶️ Attiva", callback_data="ph_enable"), InlineKeyboardButton("🔄 Restart", callback_data="ph_restart")], [InlineKeyboardButton("⬅️ Indietro", callback_data="main_menu")]]
|
|
await query.edit_message_text(text, reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
elif data.startswith("ph_"):
|
|
if "disable" in data: run_cmd(f"sudo pihole disable {data.split('_')[2]}s", MASTER_IP, SSH_USER)
|
|
elif "enable" in data: run_cmd("sudo pihole enable", MASTER_IP, SSH_USER)
|
|
elif "restart" in data: run_cmd("sudo systemctl restart pihole-FTL", MASTER_IP, SSH_USER)
|
|
await query.edit_message_text("✅ Comando inviato.", reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("⬅️ Indietro", callback_data="menu_pihole")]]), parse_mode="Markdown")
|
|
|
|
elif data == "menu_net":
|
|
ip = run_cmd("curl -s ifconfig.me")
|
|
keyboard = [[InlineKeyboardButton("🚀 Speedtest", callback_data="net_speedtest")], [InlineKeyboardButton("⬅️ Indietro", callback_data="main_menu")]]
|
|
await query.edit_message_text(f"🌐 **Rete**\n🌍 IP: `{ip}`", reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
elif data == "net_speedtest":
|
|
await query.edit_message_text("🚀 **Speedtest...**", parse_mode="Markdown")
|
|
res = run_speedtest()
|
|
await query.edit_message_text(f"🚀 **Risultato:**\n\n`{res}`", reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("⬅️ Indietro", callback_data="menu_net")]]), parse_mode="Markdown")
|
|
|
|
elif data == "menu_logs":
|
|
keyboard = [[InlineKeyboardButton("🐶 Watchdog", callback_data="log_wd"), InlineKeyboardButton("💾 Backup", callback_data="log_bk")], [InlineKeyboardButton("🔄 NPM Sync", callback_data="log_npm"), InlineKeyboardButton("⬅️ Indietro", callback_data="main_menu")]]
|
|
await query.edit_message_text("📜 **Logs**", reply_markup=InlineKeyboardMarkup(keyboard), parse_mode="Markdown")
|
|
|
|
elif data.startswith("log_"):
|
|
paths = {"log_wd": "/logs/dhcp-watchdog.log", "log_bk": "/logs/raspiBackup.log", "log_npm": "/logs/sync-npm.log"}
|
|
log_c = read_log_file(paths[data])
|
|
await query.edit_message_text(f"📜 **Log:**\n\n`{log_c}`", reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("⬅️ Indietro", callback_data="menu_logs")]]), parse_mode="Markdown")
|
|
|
|
def main():
|
|
application = Application.builder().token(BOT_TOKEN).build()
|
|
application.add_handler(CommandHandler("start", start))
|
|
application.add_handler(CallbackQueryHandler(button_handler))
|
|
application.run_polling()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|