#!/bin/bash
set -e

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

SWARM_MANAGER_IP="144.22.229.35"
SWARM_MANAGER_PORT="2377"

echo -e "${YELLOW}=== Provisionamento de Nó Worker — Docker Swarm Sittax ===${NC}\n"

# Elevação de privilégio
if [ "$EUID" -ne 0 ]; then
  SUDO="sudo"
else
  SUDO=""
fi

# ─────────────────────────────────────────
# [1/4] Dependências do SO
# ─────────────────────────────────────────
echo -e "${YELLOW}[1/4] Instalando dependências...${NC}"
$SUDO apt-get update -qq
$SUDO apt-get install -y -qq \
  ca-certificates curl gnupg lsb-release \
  jq git unzip wget apt-transport-https

# ─────────────────────────────────────────
# [2/4] Docker Engine
# ─────────────────────────────────────────
if ! command -v docker &>/dev/null; then
  echo -e "${YELLOW}[2/4] Instalando Docker Engine...${NC}"
  $SUDO install -m 0755 -d /etc/apt/keyrings
  $SUDO curl -fsSL https://download.docker.com/linux/ubuntu/gpg \
    -o /etc/apt/keyrings/docker.asc
  $SUDO chmod a+r /etc/apt/keyrings/docker.asc
  echo \
    "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.asc] \
    https://download.docker.com/linux/ubuntu \
    $(. /etc/os-release && echo "$VERSION_CODENAME") stable" | \
    $SUDO tee /etc/apt/sources.list.d/docker.list > /dev/null
  $SUDO apt-get update -qq
  $SUDO apt-get install -y \
    docker-ce docker-ce-cli containerd.io \
    docker-buildx-plugin docker-compose-plugin
  $SUDO systemctl enable docker
  $SUDO systemctl start docker
  echo -e "${GREEN}Docker instalado com sucesso.${NC}"
else
  echo -e "${GREEN}[2/4] Docker já instalado. $(docker --version)${NC}"
fi

# ─────────────────────────────────────────
# [3/4] Daemon do Docker
# ─────────────────────────────────────────
echo -e "${YELLOW}[3/4] Configurando daemon do Docker...${NC}"
$SUDO tee /etc/docker/daemon.json > /dev/null << 'DAEMON_EOF'
{
  "log-driver": "local",
  "log-opts": {
    "max-size": "50m",
    "max-file": "3"
  },
  "max-concurrent-downloads": 10,
  "max-concurrent-uploads": 5
}
DAEMON_EOF
$SUDO systemctl restart docker
echo -e "${GREEN}Daemon configurado.${NC}"

# ─────────────────────────────────────────
# [4/4] Entrada no Swarm
# ─────────────────────────────────────────
echo -e "${YELLOW}[4/4] Configurando entrada no Swarm...${NC}"

# Verifica se já está em um swarm
SWARM_STATE=$($SUDO docker info --format '{{.Swarm.LocalNodeState}}' 2>/dev/null || echo "inactive")

if [ "$SWARM_STATE" = "active" ]; then
  echo -e "${YELLOW}Atenção: este nó já faz parte de um Swarm.${NC}"
  read -p "Deseja sair do Swarm atual e entrar no novo? (s/n) [default: n]: " LEAVE_CONFIRM </dev/tty
  if [[ "$LEAVE_CONFIRM" =~ ^[SsYy]$ ]]; then
    $SUDO docker swarm leave --force
    echo -e "${GREEN}Saiu do Swarm anterior.${NC}"
  else
    echo -e "${RED}Operação cancelada.${NC}"
    exit 0
  fi
fi

# ─────────────────────────────────────────
# Instrução para o manager
# ─────────────────────────────────────────
echo ""
echo -e "${CYAN}┌──────────────────────────────────────────────────────────────────┐${NC}"
echo -e "${CYAN}│   AÇÃO NECESSÁRIA NA MÁQUINA PRINCIPAL (${SWARM_MANAGER_IP})          │${NC}"
echo -e "${CYAN}├──────────────────────────────────────────────────────────────────┤${NC}"
echo -e "${CYAN}│                                                                  │${NC}"
echo -e "${CYAN}│  Abra um terminal na máquina principal e execute:               │${NC}"
echo -e "${CYAN}│                                                                  │${NC}"
echo -e "${CYAN}│    ${YELLOW}docker swarm join-token worker${CYAN}                             │${NC}"
echo -e "${CYAN}│                                                                  │${NC}"
echo -e "${CYAN}│  O retorno será algo como:                                       │${NC}"
echo -e "${CYAN}│                                                                  │${NC}"
echo -e "${CYAN}│    docker swarm join --token ${YELLOW}SWMTKN-1-xxxxxx...${CYAN} \            │${NC}"
echo -e "${CYAN}│      ${SWARM_MANAGER_IP}:${SWARM_MANAGER_PORT}                                     │${NC}"
echo -e "${CYAN}│                                                                  │${NC}"
echo -e "${CYAN}│  Copie APENAS o token (a parte SWMTKN-1-...)                    │${NC}"
echo -e "${CYAN}└──────────────────────────────────────────────────────────────────┘${NC}"
echo ""

# ─────────────────────────────────────────
# Leitura do token
# ─────────────────────────────────────────
while true; do
  read -p "Cole o token aqui (SWMTKN-1-...): " SWARM_TOKEN </dev/tty

  if [ -z "$SWARM_TOKEN" ]; then
    echo -e "${RED}Token não informado. Tente novamente.${NC}"
    continue
  fi

  if [[ ! "$SWARM_TOKEN" =~ ^SWMTKN-[0-9]+-[a-z0-9]+-[a-z0-9]+$ ]]; then
    echo -e "${RED}Formato inválido. O token deve ter o formato SWMTKN-1-<hash>-<hash>.${NC}"
    read -p "Tentar novamente? (s/n): " RETRY </dev/tty
    if [[ "$RETRY" =~ ^[SsYy]$ ]]; then
      continue
    else
      exit 1
    fi
  fi

  break
done

# ─────────────────────────────────────────
# Join no Swarm
# ─────────────────────────────────────────
echo -e "\n${YELLOW}Entrando no Swarm (${SWARM_MANAGER_IP}:${SWARM_MANAGER_PORT})...${NC}"

if $SUDO docker swarm join \
    --token "$SWARM_TOKEN" \
    "${SWARM_MANAGER_IP}:${SWARM_MANAGER_PORT}"; then
  echo ""
  echo -e "${GREEN}╔══════════════════════════════════════════════════════════════╗${NC}"
  echo -e "${GREEN}║   Nó adicionado ao Swarm com sucesso!                        ║${NC}"
  echo -e "${GREEN}╚══════════════════════════════════════════════════════════════╝${NC}"
  echo ""
  echo -e "Para confirmar na máquina principal:"
  echo -e "  ${YELLOW}docker node ls${NC}"
  echo ""
  echo -e "Para ver os serviços rodando:"
  echo -e "  ${YELLOW}docker service ls${NC}"
else
  echo ""
  echo -e "${RED}Falha ao entrar no Swarm. Verifique:${NC}"
  echo -e "  • O token está correto e não expirou"
  echo -e "  • A porta ${SWARM_MANAGER_PORT} está acessível em ${SWARM_MANAGER_IP}"
  echo -e "  • Firewall: ${YELLOW}sudo ufw allow 2377/tcp${NC} (na máquina principal)"
  echo -e "  • Firewall: ${YELLOW}sudo ufw allow 7946/tcp && sudo ufw allow 7946/udp${NC} (em ambas)"
  exit 1
fi