#!/bin/bash

# CentOS 自动安装 wrk 脚本
# 使用方法: 
# chmod +x install_wrk.sh
# sudo ./install_wrk.sh

set -e  # 遇到错误立即停止执行

echo "开始安装 wrk..."

# 1. 检查是否为 root 用户
if [ "$EUID" -ne 0 ]; then 
  echo "请使用 root 权限运行此脚本 (使用 sudo)"
  exit 1
fi

# 2. 安装必要依赖
echo "正在安装依赖环境 (gcc, make, openssl-devel, git)..."
yum update -y
yum groupinstall "Development Tools" -y
yum install openssl-devel git -y

# 3. 下载源码到临时目录
BUILD_DIR="/tmp/wrk_build"
if [ -d "$BUILD_DIR" ]; then
    rm -rf "$BUILD_DIR"
fi

echo "正在从 GitHub 克隆 wrk 源码..."
git clone https://github.com/wg/wrk.git "$BUILD_DIR"
cd "$BUILD_DIR"

# 4. 编译
echo "开始编译 (这可能需要一分钟)..."
make

# 5. 移动到系统路径
echo "正在配置全局命令..."
cp wrk /usr/local/bin/

# 6. 清理临时文件
cd ~
rm -rf "$BUILD_DIR"

# 7. 验证
echo "--------------------------------------"
if command -v wrk >/dev/null 2>&1; then
    echo "安装成功！"
    wrk --version
else
    echo "安装似乎失败了，请检查上述错误日志。"
    exit 1
fi
echo "--------------------------------------"