#!/bin/bash

# ==========================================
# MK-MSG - Instalador Principal Interativo
# ==========================================

GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
BLUE='\033[0;34m'
NC='\033[0m'

log() { echo -e "${GREEN}[INFO]${NC} $1"; }
warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
error() { echo -e "${RED}[ERROR]${NC} $1"; exit 1; }
info() { echo -e "${BLUE}[MENU]${NC} $1"; }

# URLs dos scripts no GitHub
GITHUB_REPO="https://raw.githubusercontent.com/usuariomega/mkmsg/main/install"
SCRIPT_SISTEMA="$GITHUB_REPO/install_mkmsg_sistema.sh"
SCRIPT_WHATSAPP="$GITHUB_REPO/install_whatsapp_api_local.sh"

# Diretório temporário para os scripts
TEMP_DIR=$(mktemp -d)
trap "rm -rf $TEMP_DIR" EXIT

# Função para validar se um IP é válido e privado
validate_private_ip() {
    local ip=$1
    
    # Validar formato: deve ter exatamente 4 octetos separados por pontos
    if ! [[ $ip =~ ^[0-9]+\.[0-9]+\.[0-9]+\.[0-9]+$ ]]; then
        echo "invalid_format"
        return
    fi
    
    # Separar os octetos
    IFS='.' read -r octet1 octet2 octet3 octet4 <<< "$ip"
    
    # Validar se cada octeto está entre 0 e 255
    for octet in $octet1 $octet2 $octet3 $octet4; do
        if ! [[ $octet =~ ^[0-9]+$ ]] || [ "$octet" -lt 0 ] || [ "$octet" -gt 255 ]; then
            echo "invalid_format"
            return
        fi
    done
    
    # Verificar se é IP privado (10.x.x.x, 172.16-31.x.x, 192.168.x.x)
    if [[ $ip =~ ^10\. ]]; then
        echo "private"
        return
    fi
    
    if [[ $ip =~ ^172\.(1[6-9]|2[0-9]|3[0-1])\. ]]; then
        echo "private"
        return
    fi
    
    if [[ $ip =~ ^192\.168\. ]]; then
        echo "private"
        return
    fi
    
    echo "public"
}

# 1. Verificações de Segurança e Ambiente
if [ "$EUID" -ne 0 ]; then 
    error "Por favor, execute como root (use sudo)."
fi

if [ ! -f /etc/debian_version ]; then
    error "Este script é exclusivo para sistemas baseados em Debian (Ubuntu, Mint, etc). Instalação abortada."
fi

if grep -qi "devuan" /etc/os-release; then
    error "INSTALAÇÃO CANCELADA: Este sistema não pode ser instalado dentro do MK-Auth. Use o MK-MSG em uma máquina separada."
fi

LOCAL_IP=$(hostname -I | awk '{print $1}')
IS_PRIVATE=false
if [[ $LOCAL_IP =~ ^10\. ]] || [[ $LOCAL_IP =~ ^172\.(1[6-9]|2[0-9]|3[0-1])\. ]] || [[ $LOCAL_IP =~ ^192\.168\. ]]; then
    IS_PRIVATE=true
fi

if [ "$IS_PRIVATE" = false ]; then
    error "FALHA DE SEGURANÇA: O servidor possui um IP público ($LOCAL_IP). Este sistema só permite instalação em rede local (IP Privado). Abortando."
fi

log "🚀 Bem-vindo ao Instalador MK-MSG!"
log "IP Local Detectado: $LOCAL_IP"
echo ""

# 2. Menu de Seleção com Loop de Validação
while true; do
    info "Escolha o que deseja instalar:"
    echo ""
    echo "  1) Sistema MK-MSG + API WhatsApp"
    echo "  2) Apenas Sistema MK-MSG"
    echo "  3) Apenas API WhatsApp"
    echo "  0) Sair"
    echo ""
    
    read -p "Digite sua escolha (0, 1, 2 ou 3): " CHOICE
    
    # Validar entrada
    if [ "$CHOICE" = "0" ]; then
        log "Instalador cancelado pelo usuário."
        exit 0
    elif [[ "$CHOICE" =~ ^[1-3]$ ]]; then
        break
    else
        warn "❌ Opção inválida. Por favor, escolha 0, 1, 2 ou 3."
        echo ""
    fi
done

echo ""

# 3. Função para baixar scripts do GitHub
download_script() {
    local url=$1
    local output=$2
    local script_name=$(basename "$url")
    
    log "📥 Baixando $script_name do GitHub..."
    
    if ! curl -fsSL "$url" -o "$output" 2>/dev/null; then
        error "Falha ao baixar $script_name. Verifique sua conexão com a internet e se a URL está correta."
    fi
    
    chmod +x "$output"
    log "✅ $script_name baixado com sucesso"
}

# 4. Executar scripts conforme a escolha
case $CHOICE in
    1)
        log "Você escolheu: Sistema MK-MSG + API WhatsApp"
        log "Iniciando instalação completa..."
        echo ""
        
        # Baixar script do sistema
        SISTEMA_SCRIPT="$TEMP_DIR/install_mkmsg_sistema.sh"
        download_script "$SCRIPT_SISTEMA" "$SISTEMA_SCRIPT"
        
        echo ""
        
        # Instalar sistema
        bash "$SISTEMA_SCRIPT"
        SISTEMA_OK=$?
        
        if [ $SISTEMA_OK -eq 0 ]; then
            echo ""
            log "✅ Sistema MK-MSG instalado com sucesso!"
            echo ""
            
            # Tentar obter o token do config.php
            API_TOKEN=""
            if [ -f "/var/www/html/mkmsg/config.php" ]; then
                API_TOKEN=$(grep '\$token' /var/www/html/mkmsg/config.php | grep -oP '"\K[^"]+' | head -1)
                if [ -n "$API_TOKEN" ]; then
                    log "✅ Token obtido do config.php: $API_TOKEN"
                fi
            fi
            
            echo ""
            read -p "Deseja instalar a API WhatsApp agora? [S/n]: " INSTALL_WA
            if [[ "$INSTALL_WA" =~ ^[Ss]$ ]] || [ -z "$INSTALL_WA" ]; then
                echo ""
                log "Iniciando instalação da API WhatsApp..."
                echo ""
                
                # Baixar script do WhatsApp
                WHATSAPP_SCRIPT="$TEMP_DIR/install_whatsapp_api_local.sh"
                download_script "$SCRIPT_WHATSAPP" "$WHATSAPP_SCRIPT"
                
                echo ""
                
                # Passar o token para o instalador do WhatsApp
                if [ -n "$API_TOKEN" ]; then
                    bash "$WHATSAPP_SCRIPT" "$API_TOKEN"
                else
                    bash "$WHATSAPP_SCRIPT"
                fi
            else
                warn "Instalação do WhatsApp ignorada."
                if [ -n "$API_TOKEN" ]; then
                    log "Você pode instalar depois com o token: $API_TOKEN"
                    log "Comando: curl -fsSL $SCRIPT_WHATSAPP | bash -s \"$API_TOKEN\""
                else
                    log "Você pode instalar depois executando: curl -fsSL $SCRIPT_WHATSAPP | bash"
                fi
            fi
        else
            error "Falha na instalação do sistema MK-MSG."
        fi
        ;;
        
    2)
        log "Você escolheu: Apenas Sistema MK-MSG"
        log "Iniciando instalação do sistema..."
        echo ""
        
        # Baixar script do sistema
        SISTEMA_SCRIPT="$TEMP_DIR/install_mkmsg_sistema.sh"
        download_script "$SCRIPT_SISTEMA" "$SISTEMA_SCRIPT"
        
        echo ""
        
        # Instalar sistema
        bash "$SISTEMA_SCRIPT"
        SISTEMA_OK=$?
        
        if [ $SISTEMA_OK -eq 0 ]; then
            echo ""
            log "✅ Sistema MK-MSG instalado com sucesso!"
            
            # Obter e exibir o token gerado
            if [ -f "/var/www/html/mkmsg/config.php" ]; then
                API_TOKEN=$(grep '\$token' /var/www/html/mkmsg/config.php | grep -oP '"\K[^"]+' | head -1)
                if [ -n "$API_TOKEN" ]; then
                    echo ""
                    log "📌 Token gerado: $API_TOKEN"
                    log "   Use este token ao instalar a API WhatsApp em outro servidor"
                    log "   Comando: curl -fsSL $SCRIPT_WHATSAPP | bash -s \"$API_TOKEN\""
                fi
            fi
        fi
        ;;
        
    3)
        log "Você escolheu: Apenas API WhatsApp"
        log "Iniciando instalação da API WhatsApp..."
        echo ""
        
        # Tentar obter token do config.php local se existir
        API_TOKEN=""
        if [ -f "/var/www/html/mkmsg/config.php" ]; then
            API_TOKEN=$(grep '\$token' /var/www/html/mkmsg/config.php | grep -oP '"\K[^"]+' | head -1)
            if [ -n "$API_TOKEN" ]; then
                log "✅ Token encontrado no config.php local: $API_TOKEN"
            fi
        fi
        
        # Se não encontrou token local, perguntar ao usuário
        if [ -z "$API_TOKEN" ]; then
            echo ""
            info "Token não encontrado no sistema local."
            echo ""
            read -p "Digite o token da API (ou deixe em branco para gerar um novo): " USER_TOKEN
            
            if [ -n "$USER_TOKEN" ]; then
                API_TOKEN="$USER_TOKEN"
                log "✅ Token fornecido: $API_TOKEN"
            else
                log "Será gerado um novo token durante a instalação"
            fi
        fi
        
        echo ""
        
        # Baixar script do WhatsApp
        WHATSAPP_SCRIPT="$TEMP_DIR/install_whatsapp_api_local.sh"
        download_script "$SCRIPT_WHATSAPP" "$WHATSAPP_SCRIPT"
        
        echo ""
        
        # Executar instalador do WhatsApp
        if [ -n "$API_TOKEN" ]; then
            bash "$WHATSAPP_SCRIPT" "$API_TOKEN"
        else
            bash "$WHATSAPP_SCRIPT"
        fi
        ;;
esac

log "✅ Instalação finalizada!"