#!/bin/bash
# Docker HTTP proxy manager script
PROXY_ADDR="http://127.0.0.1:1080"
CONF_DIR="/etc/systemd/system/docker.service.d"
CONF_FILE="$CONF_DIR/http-proxy.conf"
SCRIPT_PATH="/usr/local/bin/dockerproxy"

install_proxy() {
  echo "📥 安装 dockerproxy 系统命令..."
  sudo cp "$0" "$SCRIPT_PATH"
  sudo chmod +x "$SCRIPT_PATH"
  echo "✅ 安装完成，可通过 dockerproxy [on|off] 管理代理"
}

uninstall_proxy() {
  echo "🧹 卸载 dockerproxy..."
  sudo rm -f "$SCRIPT_PATH"
  echo "✅ 卸载完成，建议先执行 dockerproxy off 关闭代理"
}

enable_proxy() {
  echo "🔧 正在开启 Docker HTTP 代理..."
  sudo mkdir -p "$CONF_DIR"
  sudo bash -c "cat > $CONF_FILE" <<EOF
[Service]
Environment="HTTP_PROXY=$PROXY_ADDR"
Environment="HTTPS_PROXY=$PROXY_ADDR"
Environment="NO_PROXY=localhost,127.0.0.1"
EOF
  echo "🔄 重启 Docker 服务以应用代理..."
  sudo systemctl daemon-reexec
  sudo systemctl daemon-reload
  sudo systemctl restart docker
  echo "✅ Docker HTTP 代理已开启 [[1,3,5]]"
}

disable_proxy() {
  echo "🧹 正在关闭 Docker HTTP 代理..."
  [ -f "$CONF_FILE" ] && sudo rm -f "$CONF_FILE"
  echo "🔄 重启 Docker 服务以移除代理..."
  sudo systemctl daemon-reexec
  sudo systemctl daemon-reload
  sudo systemctl restart docker
  echo "✅ Docker HTTP 代理已关闭 [[4]]"
}

show_usage() {
  echo "用法: $0 [on|off|install|uninstall]"
  echo "  on        - 开启 Docker HTTP 代理"
  echo "  off       - 关闭 Docker HTTP 代理"
  echo "  install   - 安装为系统命令"
  echo "  uninstall - 卸载系统命令"
}

case "$1" in
  on)    enable_proxy ;;
  off)   disable_proxy ;;
  install) install_proxy ;;
  uninstall) uninstall_proxy ;;
  *)     show_usage ;;
esac