#!/bin/bash

# Install Nomad Cluster via APT
# 通过 APT 安装 Nomad 集群

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"

# 颜色定义
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# 日志函数
log_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

log_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

log_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

# 检查依赖
check_dependencies() {
    log_info "检查依赖项..."
    
    if ! command -v ansible-playbook &> /dev/null; then
        log_error "ansible-playbook 未找到，请安装 Ansible"
        exit 1
    fi
    
    log_success "依赖检查完成"
}

# 检查网络连接
check_connectivity() {
    log_info "检查目标主机连接性..."
    
    local inventory_file="$PROJECT_ROOT/configuration/inventories/production/nomad-cluster.ini"
    
    if [[ ! -f "$inventory_file" ]]; then
        log_error "清单文件不存在: $inventory_file"
        exit 1
    fi
    
    # 测试连接
    if ansible nomad_servers -i "$inventory_file" -m ping --one-line; then
        log_success "所有主机连接正常"
    else
        log_warning "部分主机连接失败，但继续安装..."
    fi
}

# 安装 Nomad
install_nomad() {
    log_info "开始在远程主机安装 Nomad..."
    
    local playbook_file="$PROJECT_ROOT/configuration/playbooks/applications/install-nomad-apt.yml"
    local inventory_file="$PROJECT_ROOT/configuration/inventories/production/nomad-cluster.ini"
    
    if [[ ! -f "$playbook_file" ]]; then
        log_error "Playbook 文件不存在: $playbook_file"
        exit 1
    fi
    
    # 运行 Ansible playbook
    if ansible-playbook -i "$inventory_file" "$playbook_file" -v; then
        log_success "Nomad 集群安装完成"
    else
        log_error "Nomad 集群安装失败"
        exit 1
    fi
}

# 验证安装
verify_installation() {
    log_info "验证 Nomad 安装..."
    
    local inventory_file="$PROJECT_ROOT/configuration/inventories/production/nomad-cluster.ini"
    
    # 检查服务状态
    log_info "检查 Nomad 服务状态..."
    ansible nomad_servers -i "$inventory_file" -m shell -a "systemctl status nomad --no-pager" || true
    
    # 检查 Nomad 版本
    log_info "检查 Nomad 版本..."
    ansible nomad_servers -i "$inventory_file" -m shell -a "nomad version" || true
    
    # 检查集群成员
    log_info "检查集群服务器..."
    ansible nomad_servers -i "$inventory_file" -m shell -a "nomad server members" -l "$(head -n1 < <(grep -v '^\[' "$inventory_file" | grep -v '^$' | head -n1))" || true
    
    # 检查节点状态
    log_info "检查节点状态..."
    ansible nomad_servers -i "$inventory_file" -m shell -a "nomad node status" -l "$(head -n1 < <(grep -v '^\[' "$inventory_file" | grep -v '^$' | head -n1))" || true
}

# 显示访问信息
show_access_info() {
    log_info "Nomad 集群访问信息："
    echo ""
    echo "Web UI 访问地址："
    echo "  - http://100.117.106.136:4646 (master)"
    echo "  - http://100.116.158.95:4646 (semaphore)"
    echo ""
    echo "API 访问地址："
    echo "  - http://100.117.106.136:4646/v1/ (master)"
    echo "  - http://100.116.158.95:4646/v1/ (semaphore)"
    echo ""
    echo "常用命令："
    echo "  - 查看集群状态: nomad status"
    echo "  - 查看节点: nomad node status"
    echo "  - 查看服务器: nomad server members"
    echo "  - 提交作业: nomad job run <job-file>"
    echo ""
    echo "示例作业文件位置："
    echo "  - $PROJECT_ROOT/examples/nomad-jobs/"
    echo ""
}

# 主函数
main() {
    log_info "开始 Nomad 集群安装..."
    
    check_dependencies
    check_connectivity
    install_nomad
    verify_installation
    show_access_info
    
    log_success "Nomad 集群安装流程完成！"
}

# 脚本入口
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
    main "$@"
fi