#!/bin/bash

# Nomad Local 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
    
    if ! command -v docker &> /dev/null; then
        log_error "docker 未找到，请安装 Docker"
        exit 1
    fi
    
    log_success "依赖检查完成"
}

# 部署 Nomad
deploy_nomad() {
    log_info "开始部署 Nomad (本地单节点)..."
    
    local playbook_file="$PROJECT_ROOT/configuration/playbooks/applications/nomad-local.yml"
    
    if [[ ! -f "$playbook_file" ]]; then
        log_error "Playbook 文件不存在: $playbook_file"
        exit 1
    fi
    
    # 运行 Ansible playbook
    if ansible-playbook "$playbook_file" -v; then
        log_success "Nomad 本地部署完成"
    else
        log_error "Nomad 本地部署失败"
        exit 1
    fi
}

# 验证部署
verify_deployment() {
    log_info "验证 Nomad 部署..."
    
    # 等待服务启动
    sleep 5
    
    # 检查服务状态
    log_info "检查 Nomad 服务状态..."
    systemctl status nomad --no-pager || true
    
    # 检查 Nomad 版本
    log_info "检查 Nomad 版本..."
    /usr/local/bin/nomad version || true
    
    # 检查节点状态
    log_info "检查节点状态..."
    /usr/local/bin/nomad node status || true
    
    # 检查服务器状态
    log_info "检查服务器状态..."
    /usr/local/bin/nomad server members || true
}

# 显示访问信息
show_access_info() {
    local current_ip=$(hostname -I | awk '{print $1}')
    
    log_info "Nomad 访问信息："
    echo ""
    echo "Web UI 访问地址："
    echo "  - http://localhost:4646"
    echo "  - http://${current_ip}:4646"
    echo ""
    echo "API 访问地址："
    echo "  - http://localhost:4646/v1/"
    echo "  - http://${current_ip}:4646/v1/"
    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
    deploy_nomad
    verify_deployment
    show_access_info
    
    log_success "Nomad 本地部署流程完成！"
}

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