main.py

import signal
import sys
import atexit

from detect import detect_wifi_interface, detect_hotspot_interface, ip_of
from firewall import setup_captive_portal, reset_firewall, start_hotspot, stop_hotspot
from server import start_server, set_hotspot_iface
from sessions import delete_sessions


def cleanup():
    print("[+] Limpiando firewall…")
    reset_firewall()
    delete_sessions()
    print("[+] Apagando hotspot…")
    stop_hotspot()

    sys.exit(0)

def signal_handler(sig, frame):
    cleanup()

signal.signal(signal.SIGINT, signal_handler)
signal.signal(signal.SIGTERM, signal_handler)

atexit.register(cleanup)


def main():
    # ARRANCA hostapd, dnsmasq y crea wlan1
    start_hotspot()   

    # Detectar interfaz del hotspot (AP) y su IP (típicamente 10.x.x.x o 192.168.x.x)
    hotspot_iface = detect_hotspot_interface()
    hotspot_ip = ip_of(hotspot_iface)

    # Detectar interfaz con salida a internet (WAN) para NAT
    wan_iface = detect_wifi_interface()

    # Guardar la interfaz hotspot en el servidor para usarla al autorizar IPs
    set_hotspot_iface(hotspot_iface)

    # Configurar reglas de portal cautivo (redirigir HTTP y activar NAT)
    setup_captive_portal(hotspot_iface, hotspot_ip, portal_port=8080, wan_iface=wan_iface)

    # Iniciar servidor enlazado a la IP del hotspot
    start_server(hotspot_ip, port=8080)


if __name__ == "__main__":
    main()