#!/bin/bash

# 批量部署安全配置文件脚本
# 使用方法: ./deploy-security-configs.sh [节点名] [配置类型]

set -e

# 配置变量
SECURITY_DIR="/root/mgmt/security"
SECRETS_DIR="$SECURITY_DIR/secrets"
LOGS_DIR="$SECURITY_DIR/logs"
BACKUP_DIR="$SECURITY_DIR/backups"
TEMP_DIR="/tmp/security-deploy"

# 节点列表
NODES=("ch4" "ash3c" "warden" "ash1d" "ash2e" "ch2" "ch3" "de" "onecloud1" "semaphore" "influxdb" "hcp1" "browser" "brother")

# 配置类型
CONFIG_TYPES=("nomad" "consul" "vault" "traefik")

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

# 日志函数
log() {
    echo -e "${BLUE}[$(date '+%Y-%m-%d %H:%M:%S')]${NC} $1"
}

error() {
    echo -e "${RED}[ERROR]${NC} $1" >&2
}

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

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

# 创建必要目录
create_dirs() {
    mkdir -p "$LOGS_DIR" "$BACKUP_DIR" "$TEMP_DIR"
}

# 检查节点是否存在
check_node() {
    local node=$1
    ping -c 1 "$node.tailnet-68f9.ts.net" >/dev/null 2>&1
}

# 备份现有配置
backup_config() {
    local node=$1
    local config_type=$2
    local config_path=$3
    
    local backup_file="$BACKUP_DIR/${node}-${config_type}-$(date +%Y%m%d_%H%M%S).backup"
    
    log "备份 $node 的 $config_type 配置到 $backup_file"
    
    if sshpass -p '3131' ssh -o StrictHostKeyChecking=no -o ConnectTimeout=10 ben@"$node.tailnet-68f9.ts.net" "test -f $config_path"; then
        sshpass -p '3131' ssh -o StrictHostKeyChecking=no -o ConnectTimeout=10 ben@"$node.tailnet-68f9.ts.net" "cat $config_path" > "$backup_file"
        success "备份完成: $backup_file"
    else
        warning "配置文件不存在: $config_path"
    fi
}

# 部署配置文件
deploy_config() {
    local node=$1
    local config_type=$2
    local config_file=$3
    
    log "部署 $config_file 到 $node"
    
    # 确定目标路径
    local target_path
    case $config_type in
        "nomad")
            target_path="/etc/nomad.d/nomad.hcl"
            ;;
        "consul")
            target_path="/etc/consul.d/consul.hcl"
            ;;
        "vault")
            target_path="/etc/vault.d/vault.hcl"
            ;;
        "traefik")
            target_path="/etc/traefik/traefik.yml"
            ;;
        *)
            error "未知配置类型: $config_type"
            return 1
            ;;
    esac
    
    # 备份现有配置
    backup_config "$node" "$config_type" "$target_path"
    
    # 上传配置文件
    log "上传配置文件到 $node:$target_path"
    sshpass -p '3131' scp -o StrictHostKeyChecking=no -o ConnectTimeout=10 "$config_file" ben@"$node.tailnet-68f9.ts.net":/tmp/new-config
    
    # 替换配置文件
    log "替换配置文件"
    sshpass -p '3131' ssh -o StrictHostKeyChecking=no -o ConnectTimeout=10 ben@"$node.tailnet-68f9.ts.net" "
        echo '3131' | sudo -S cp /tmp/new-config $target_path
        echo '3131' | sudo -S chown root:root $target_path
        echo '3131' | sudo -S chmod 644 $target_path
        rm -f /tmp/new-config
    "
    
    success "配置文件部署完成: $node:$target_path"
}

# 重启服务
restart_service() {
    local node=$1
    local config_type=$2
    
    log "重启 $node 的 $config_type 服务"
    
    local service_name
    case $config_type in
        "nomad")
            service_name="nomad"
            ;;
        "consul")
            service_name="consul"
            ;;
        "vault")
            service_name="vault"
            ;;
        "traefik")
            service_name="traefik"
            ;;
        *)
            error "未知服务类型: $config_type"
            return 1
            ;;
    esac
    
    sshpass -p '3131' ssh -o StrictHostKeyChecking=no -o ConnectTimeout=10 ben@"$node.tailnet-68f9.ts.net" "
        echo '3131' | sudo -S systemctl restart $service_name
        sleep 3
        echo '3131' | sudo -S systemctl status $service_name --no-pager
    "
    
    success "服务重启完成: $node:$service_name"
}

# 验证部署
verify_deployment() {
    local node=$1
    local config_type=$2
    
    log "验证 $node 的 $config_type 部署"
    
    case $config_type in
        "nomad")
            sshpass -p '3131' ssh -o StrictHostKeyChecking=no -o ConnectTimeout=10 ben@"$node.tailnet-68f9.ts.net" "
                echo '3131' | sudo -S systemctl is-active nomad
            "
            ;;
        "consul")
            sshpass -p '3131' ssh -o StrictHostKeyChecking=no -o ConnectTimeout=10 ben@"$node.tailnet-68f9.ts.net" "
                echo '3131' | sudo -S systemctl is-active consul
            "
            ;;
        *)
            warning "跳过验证: $config_type"
            ;;
    esac
}

# 主函数
main() {
    local target_node=${1:-"all"}
    local target_type=${2:-"all"}
    
    log "开始批量部署安全配置文件"
    log "目标节点: $target_node"
    log "配置类型: $target_type"
    
    create_dirs
    
    # 处理节点列表
    local nodes_to_process=()
    if [ "$target_node" = "all" ]; then
        nodes_to_process=("${NODES[@]}")
    else
        nodes_to_process=("$target_node")
    fi
    
    # 处理配置类型
    local types_to_process=()
    if [ "$target_type" = "all" ]; then
        types_to_process=("${CONFIG_TYPES[@]}")
    else
        types_to_process=("$target_type")
    fi
    
    # 遍历节点和配置类型
    for node in "${nodes_to_process[@]}"; do
        if ! check_node "$node"; then
            warning "节点 $node 不可达，跳过"
            continue
        fi
        
        log "处理节点: $node"
        
        for config_type in "${types_to_process[@]}"; do
            local config_file="$SECRETS_DIR/${node}-${config_type}.hcl"
            
            if [ ! -f "$config_file" ]; then
                config_file="$SECRETS_DIR/${node}-${config_type}.yml"
            fi
            
            if [ ! -f "$config_file" ]; then
                config_file="$SECRETS_DIR/${node}-${config_type}.json"
            fi
            
            if [ -f "$config_file" ]; then
                log "找到配置文件: $config_file"
                deploy_config "$node" "$config_type" "$config_file"
                restart_service "$node" "$config_type"
                verify_deployment "$node" "$config_type"
            else
                warning "未找到配置文件: $node-$config_type"
            fi
        done
    done
    
    # 清理临时文件
    rm -rf "$TEMP_DIR"
    
    success "批量部署完成！"
    log "日志文件: $LOGS_DIR"
    log "备份文件: $BACKUP_DIR"
}

# 显示帮助信息
show_help() {
    echo "使用方法: $0 [节点名] [配置类型]"
    echo ""
    echo "参数:"
    echo "  节点名     - 目标节点名称 (默认: all)"
    echo "  配置类型   - 配置类型 (默认: all)"
    echo ""
    echo "示例:"
    echo "  $0                    # 部署所有节点的所有配置"
    echo "  $0 ch4               # 部署 ch4 节点的所有配置"
    echo "  $0 all nomad         # 部署所有节点的 nomad 配置"
    echo "  $0 ch4 consul        # 部署 ch4 节点的 consul 配置"
    echo ""
    echo "支持的节点: ${NODES[*]}"
    echo "支持的配置类型: ${CONFIG_TYPES[*]}"
}

# 检查参数
if [ "$1" = "-h" ] || [ "$1" = "--help" ]; then
    show_help
    exit 0
fi

# 运行主函数
main "$@"