#!/bin/bash

set -e

# CONFIG
TZ="America/Sao_Paulo"
CONFIG_ROOT="$HOME/ha-stack"
DOCKER_COMPOSE_FILE="$CONFIG_ROOT/docker-compose.yml"
Z2M_CONFIG_PATH="$CONFIG_ROOT/zigbee2mqtt/data/configuration.yaml"

echo "🐳 Verificando instalação do Docker..."

if ! command -v docker &> /dev/null; then
  echo "📦 Docker não encontrado. Instalando Docker..."
  curl -sSL https://get.docker.com | sh
  sudo usermod -aG docker $USER
  echo "⚠️ Docker instalado. Faça logout/login para aplicar permissões ou reinicie o Raspberry."
fi

if ! command -v docker-compose &> /dev/null; then
  echo "📦 docker-compose não encontrado. Instalando plugin..."
  mkdir -p ~/.docker/cli-plugins
  curl -SL https://github.com/docker/compose/releases/download/v2.24.7/docker-compose-linux-aarch64     -o ~/.docker/cli-plugins/docker-compose
  chmod +x ~/.docker/cli-plugins/docker-compose
  echo "✅ docker-compose instalado."
fi

echo "🔧 Criando estrutura de pastas em: $CONFIG_ROOT"
mkdir -p "$CONFIG_ROOT/homeassistant"
mkdir -p "$CONFIG_ROOT/mosquitto/config" "$CONFIG_ROOT/mosquitto/data" "$CONFIG_ROOT/mosquitto/log"
mkdir -p "$(dirname "$Z2M_CONFIG_PATH")"

echo "🔍 Detectando dongle Zigbee..."
dongle=$(ls /dev/serial/by-id/ 2>/dev/null || true)

if [ -z "$dongle" ]; then
  echo "❌ Nenhum dongle encontrado em /dev/serial/by-id/"
  exit 1
fi

dongle_path="/dev/serial/by-id/$dongle"

case "$dongle" in
  *"Sonoff"*|*"CC2531"*)
    adapter="zstack"
    ;;
  *"ZBDongle-E"*|*"EZSP"*)
    adapter="ezsp"
    ;;
  *"ConBee"*|*"deCONZ"*)
    adapter="deconz"
    ;;
  *)
    adapter="zstack"
    ;;
esac

echo "✅ Dongle encontrado:"
echo "  Porta:   $dongle_path"
echo "  Adapter: $adapter"

echo "📝 Gerando configuration.yaml para Zigbee2MQTT..."
cat > "$Z2M_CONFIG_PATH" <<EOF
homeassistant: true

mqtt:
  base_topic: zigbee2mqtt
  server: mqtt://mosquitto

serial:
  port: $dongle_path
  adapter: $adapter

frontend:
  port: 8080
EOF

echo "📄 Gerando docker-compose.yml..."
cat > "$DOCKER_COMPOSE_FILE" <<EOF
version: '3.7'

services:
  homeassistant:
    container_name: homeassistant
    image: ghcr.io/home-assistant/home-assistant:stable
    restart: unless-stopped
    privileged: true
    network_mode: host
    environment:
      - TZ=$TZ
    volumes:
      - ./homeassistant:/config

  mosquitto:
    container_name: mosquitto
    image: eclipse-mosquitto
    restart: unless-stopped
    ports:
      - 1883:1883
      - 9001:9001
    volumes:
      - ./mosquitto/config:/mosquitto/config
      - ./mosquitto/data:/mosquitto/data
      - ./mosquitto/log:/mosquitto/log

  zigbee2mqtt:
    container_name: zigbee2mqtt
    image: koenkk/zigbee2mqtt
    restart: unless-stopped
    ports:
      - 8080:8080
    volumes:
      - ./zigbee2mqtt/data:/app/data
    environment:
      - TZ=$TZ
    devices:
      - $dongle_path:$dongle_path
EOF

echo "🚀 Subindo stack com Docker Compose..."
cd "$CONFIG_ROOT"
docker compose up -d

echo
echo "✅ Tudo pronto!"
echo "🔗 Home Assistant: http://$(hostname -I | awk '{print $1}'):8123"
echo "🔗 Zigbee2MQTT:   http://$(hostname -I | awk '{print $1}'):8080"


if ! groups $USER | grep -q docker; then
  echo
  echo "⚠️  O grupo 'docker' ainda não está ativo para o usuário '$USER'."
  echo "🔁 Reiniciando o Raspberry Pi para aplicar permissões..."
  sleep 3
  sudo reboot
fi