#!/bin/bash

# 检查是否以 root 用户运行
if [[ $EUID -ne 0 ]]; then
  echo "请以 root 用户或使用 sudo 运行此脚本。"
  exit 1
fi


# 检查系统架构
ARCH=$(uname -m)
case $ARCH in
  x86_64)
    DOCKER_COMPOSE_ARCH="x86_64"
    ;;
  aarch64)
    DOCKER_COMPOSE_ARCH="aarch64"
    ;;
  armv7l)
    DOCKER_COMPOSE_ARCH="armv7"
    ;;
  *)
    echo "不支持的系统架构: $ARCH"
    exit 1
    ;;
esac

# 定义安装路径和版本
INSTALL_PATH="/usr/local/bin/docker-compose"
LATEST_VERSION=$(curl -s https://api.github.com/repos/docker/compose/releases/latest | grep "tag_name" | cut -d '"' -f 4)

echo "正在安装 docker-compose 版本: $LATEST_VERSION"

# 下载 docker-compose
DOWNLOAD_URL="https://gh-proxy.com/https://github.com/docker/compose/releases/download/$LATEST_VERSION/docker-compose-Linux-$DOCKER_COMPOSE_ARCH"
curl -L "$DOWNLOAD_URL" -o "$INSTALL_PATH"

# 检查下载是否成功
if [[ $? -ne 0 ]]; then
  echo "下载 docker-compose 失败，请检查网络连接。"
  exit 1
fi

# 设置可执行权限
chmod +x "$INSTALL_PATH"

# 验证安装
if docker-compose --version; then
  echo "docker-compose 安装成功！"
else
  echo "docker-compose 安装失败，请检查安装路径和权限。"
  exit 1
fi