#!/bin/bash

# Nomad Cluster Deployment Script
# 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_cluster -i "$inventory_file" -m ping --one-line; then
        log_success "所有主机连接正常"
    else
        log_warning "部分主机连接失败，但继续部署..."
    fi
}

# 部署 Nomad 集群
deploy_nomad() {
    log_info "开始部署 Nomad 集群..."
    
    local playbook_file="$PROJECT_ROOT/configuration/playbooks/applications/nomad-cluster.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_cluster() {
    log_info "验证 Nomad 集群状态..."
    
    local inventory_file="$PROJECT_ROOT/configuration/inventories/production/nomad-cluster.ini"
    
    # 检查服务状态
    log_info "检查 Nomad 服务状态..."
    ansible nomad_cluster -i "$inventory_file" -m shell -a "systemctl status nomad --no-pager" || true
    
    # 检查集群成员
    log_info "检查集群服务器..."
    ansible nomad_servers -i "$inventory_file" -m shell -a "/usr/local/bin/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 "/usr/local/bin/nomad node status" -l "$(head -n1 < <(grep -v '^\[' "$inventory_file" | grep -v '^$' | head -n1))" || true
    
    # 显示集群信息
    log_info "集群信息..."
    ansible nomad_servers -i "$inventory_file" -m shell -a "/usr/local/bin/nomad 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://10.0.0.232:4646"
    echo "  - http://10.0.0.179:4646"
    echo ""
    echo "API 访问地址："
    echo "  - http://10.0.0.232:4646/v1/"
    echo "  - http://10.0.0.179:4646/v1/"
    echo ""
    echo "常用命令："
    echo "  - 查看集群状态: nomad status"
    echo "  - 查看节点: nomad node status"
    echo "  - 查看服务器: nomad server members"
    echo "  - 提交作业: nomad job run <job-file>"
    echo ""
}

# 主函数
main() {
    log_info "开始 Nomad 集群部署..."
    
    check_dependencies
    check_connectivity
    deploy_nomad
    verify_cluster
    show_access_info
    
    log_success "Nomad 集群部署流程完成！"
}

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