#!/bin/bash

deploy_wazuh() {
    # 创建目录（如果不存在）
    mkdir -p /data/wazuh

    # 下载脚本
    echo "正在下载setup_wazuh_agent.sh..."
    wget -q -O /data/wazuh/setup_wazuh_agent.sh http://121.41.78.13:8017/setup_wazuh_agent.sh

    # 检查下载是否成功
    if [ ! -f "/data/wazuh/setup_wazuh_agent.sh" ]; then
        echo "错误：脚本下载失败"
        exit 1
    fi

    # 设置脚本执行权限
    chmod +x /data/wazuh/setup_wazuh_agent.sh

    # 定义要添加的crontab任务
    CRON_JOB="*/10 * * * * sudo /bin/bash /data/wazuh/setup_wazuh_agent.sh 1>/dev/null 2>&1"

    # 检查任务是否已存在于crontab中
    if ! crontab -l 2>/dev/null | grep -qF "$CRON_JOB"; then
        echo "添加定时任务到crontab..."
        # 现有任务 + 新任务 重新写入crontab
        (crontab -l 2>/dev/null; echo "$CRON_JOB") | crontab -
        echo "定时任务已添加"
    else
        echo "定时任务已存在，无需重复添加"
    fi

    echo "操作完成"
} 

deploy_wazuh
