setup_proxy.sh

#!/usr/bin/env bash
# =============================================================================
# setup_proxy.sh — Ubuntu 24.04.3 Global Proxy Configuration Script
# Coverage:  shell, curl, wget, git, apt, pip, npm, yarn, pnpm,
#            docker, snap, Claude Code, .NET, Go, Rust(cargo), Java(Maven/Gradle)
# Usage:
#   Interactive mode (recommended): bash setup_proxy.sh
#   Direct mode:                    bash setup_proxy.sh http://127.0.0.1:7890 user pass
#   Clear proxy:                    bash setup_proxy.sh --clear
# =============================================================================

set -euo pipefail

# ── Colors ────────────────────────────────────────────────────────────────────
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'
BLUE='\033[0;34m'; CYAN='\033[0;36m'; BOLD='\033[1m'; NC='\033[0m'

info()    { echo -e "${CYAN}[INFO]${NC}  $*"; }
success() { echo -e "${GREEN}[OK]${NC}    $*"; }
warn()    { echo -e "${YELLOW}[WARN]${NC}  $*"; }
section() { echo -e "\n${BOLD}${BLUE}══ $* ══${NC}"; }

# ── Help ──────────────────────────────────────────────────────────────────────
usage() {
  cat <<EOF
Usage:
  $0                              Interactive mode, step-by-step prompts
  $0 <proxy_url>                  Set proxy directly (no authentication)
  $0 <proxy_url> <user> <pass>    Set proxy directly (with authentication)
  $0 --clear                      Clear all configured proxy settings
  $0 --help                       Show this help message

Examples:
  $0 http://127.0.0.1:7890
  $0 http://proxy.example.com:8080 admin secret123
  $0 --clear
EOF
}

# ── Clear Proxy ───────────────────────────────────────────────────────────────
clear_proxy() {
  section "Clearing all proxy settings"

  # 1. shell profiles
  local profiles=("$HOME/.bashrc" "$HOME/.zshrc" "$HOME/.profile" "$HOME/.bash_profile")
  for f in "${profiles[@]}"; do
    [[ -f "$f" ]] || continue
    if grep -q "# >>> proxy setup >>>" "$f" 2>/dev/null; then
      sed -i '/# >>> proxy setup >>>/,/# <<< proxy setup <<</d' "$f"
      success "Cleared proxy block from $f"
    fi
  done

  # 2. /etc/environment
  if [[ -f /etc/environment ]] && grep -q "http_proxy\|https_proxy\|no_proxy" /etc/environment 2>/dev/null; then
    sudo sed -i '/http_proxy\|https_proxy\|HTTP_PROXY\|HTTPS_PROXY\|no_proxy\|NO_PROXY\|ftp_proxy\|FTP_PROXY/d' /etc/environment
    success "Cleared /etc/environment"
  fi

  # 3. git
  git config --global --unset http.proxy  2>/dev/null && success "Cleared git http.proxy"  || true
  git config --global --unset https.proxy 2>/dev/null && success "Cleared git https.proxy" || true

  # 4. curl
  [[ -f "$HOME/.curlrc" ]] && sed -i '/^proxy=/d' "$HOME/.curlrc" && success "Cleared ~/.curlrc"

  # 5. wget
  [[ -f "$HOME/.wgetrc" ]] && sed -i '/^https_proxy\|^http_proxy\|^use_proxy/d' "$HOME/.wgetrc" && success "Cleared ~/.wgetrc"

  # 6. apt
  [[ -f /etc/apt/apt.conf.d/99proxy ]] && sudo rm -f /etc/apt/apt.conf.d/99proxy && success "Cleared apt proxy"

  # 7. pip
  if [[ -f "$HOME/.config/pip/pip.conf" ]]; then
    sed -i '/^proxy\s*=/d' "$HOME/.config/pip/pip.conf"
    success "Cleared pip proxy"
  fi

  # 8. npm
  npm config delete proxy       2>/dev/null && true
  npm config delete https-proxy 2>/dev/null && true
  success "Cleared npm proxy"

  # 9. yarn
  command -v yarn &>/dev/null && {
    yarn config delete proxy       2>/dev/null || true
    yarn config delete https-proxy 2>/dev/null || true
    success "Cleared yarn proxy"
  }

  # 10. pnpm
  command -v pnpm &>/dev/null && {
    pnpm config delete proxy       2>/dev/null || true
    pnpm config delete https-proxy 2>/dev/null || true
    success "Cleared pnpm proxy"
  }

  # 11. docker
  [[ -f /etc/systemd/system/docker.service.d/proxy.conf ]] && \
    sudo rm -f /etc/systemd/system/docker.service.d/proxy.conf && \
    sudo systemctl daemon-reload && \
    sudo systemctl restart docker 2>/dev/null || true
    success "Cleared docker proxy"

  # 12. snap
  command -v snap &>/dev/null && {
    sudo snap set system proxy.http=""  2>/dev/null || true
    sudo snap set system proxy.https="" 2>/dev/null || true
    success "Cleared snap proxy"
  }

  # 13. Claude Code
  if [[ -f "$HOME/.claude/settings.json" ]] && command -v python3 &>/dev/null; then
    python3 - <<'PYEOF'
import json, pathlib, sys
p = pathlib.Path.home() / ".claude/settings.json"
try:
    d = json.loads(p.read_text())
    for k in ("HTTPS_PROXY","HTTP_PROXY","NO_PROXY","https_proxy","http_proxy","no_proxy"):
        d.get("env",{}).pop(k, None)
    p.write_text(json.dumps(d, indent=2, ensure_ascii=False))
    print("  -> Proxy fields cleared from ~/.claude/settings.json")
except Exception as e:
    print(f"  -> Skipped ({e})")
PYEOF
  fi

  # 14. Maven settings.xml
  warn "Maven/Gradle proxies must be cleared manually: ~/.m2/settings.xml and ~/.gradle/gradle.properties"

  echo ""
  success "All proxy settings cleared. Reload your shell: source ~/.bashrc"
  exit 0
}

# ── Argument Parsing ──────────────────────────────────────────────────────────
[[ "${1:-}" == "--help" || "${1:-}" == "-h" ]] && { usage; exit 0; }
[[ "${1:-}" == "--clear" ]] && clear_proxy

PROXY_URL=""
PROXY_USER=""
PROXY_PASS=""

if [[ $# -ge 1 ]]; then
  PROXY_URL="$1"
  PROXY_USER="${2:-}"
  PROXY_PASS="${3:-}"
else
  # Interactive mode
  echo -e "\n${BOLD}Ubuntu 24.04.3 Global Proxy Setup Wizard${NC}\n"
  read -rp "Proxy address (e.g. http://127.0.0.1:7890): " PROXY_URL
  [[ -z "$PROXY_URL" ]] && { warn "No address provided. Exiting."; exit 1; }
  read -rp "Username (press Enter to skip if no auth): " PROXY_USER
  if [[ -n "$PROXY_USER" ]]; then
    read -rsp "Password: " PROXY_PASS; echo ""
  fi
fi

# Parse URL components (scheme, host, port)
SCHEME=$(echo "$PROXY_URL" | grep -oP '^https?')
HOST_PORT=$(echo "$PROXY_URL" | sed 's|^https\?://||;s|/$||')
HOST=$(echo "$HOST_PORT" | cut -d: -f1)
PORT=$(echo "$HOST_PORT" | cut -d: -f2)

# Build full URL with or without credentials
if [[ -n "$PROXY_USER" ]]; then
  ENCODED_PASS=$(python3 -c "import urllib.parse,sys; print(urllib.parse.quote('$PROXY_PASS', safe=''))" 2>/dev/null || echo "$PROXY_PASS")
  PROXY_FULL="${SCHEME}://${PROXY_USER}:${ENCODED_PASS}@${HOST_PORT}"
else
  PROXY_FULL="$PROXY_URL"
fi

NO_PROXY_DEFAULT="localhost,127.0.0.1,::1,10.0.0.0/8,172.16.0.0/12,192.168.0.0/16,.local"

echo ""
info "Proxy address : $PROXY_URL"
info "Auth user     : ${PROXY_USER:-( none )}"
info "NO_PROXY      : $NO_PROXY_DEFAULT"

# ── Helper Functions ──────────────────────────────────────────────────────────
ensure_dir()  { mkdir -p "$1"; }
append_block() {
  # append_block <file> <content-block>
  local file="$1"; shift
  # Remove existing proxy block first, then append
  if grep -q "# >>> proxy setup >>>" "$file" 2>/dev/null; then
    sed -i '/# >>> proxy setup >>>/,/# <<< proxy setup <<</d' "$file"
  fi
  cat >> "$file" <<<"$@"
}

PROFILE_BLOCK="
# >>> proxy setup >>>
export http_proxy=\"${PROXY_FULL}\"
export HTTP_PROXY=\"${PROXY_FULL}\"
export https_proxy=\"${PROXY_FULL}\"
export HTTPS_PROXY=\"${PROXY_FULL}\"
export ftp_proxy=\"${PROXY_FULL}\"
export FTP_PROXY=\"${PROXY_FULL}\"
export no_proxy=\"${NO_PROXY_DEFAULT}\"
export NO_PROXY=\"${NO_PROXY_DEFAULT}\"
# <<< proxy setup <<<"

# =============================================================================
section "1. Shell environment variables (~/.bashrc / ~/.zshrc)"
# =============================================================================
for rc in "$HOME/.bashrc" "$HOME/.zshrc"; do
  if [[ -f "$rc" ]] || [[ "$rc" == "$HOME/.bashrc" ]]; then
    touch "$rc"
    append_block "$rc" "$PROFILE_BLOCK"
    success "$rc"
  fi
done

# =============================================================================
section "2. /etc/environment (system-wide, requires sudo)"
# =============================================================================
if sudo -n true 2>/dev/null; then
  sudo sed -i '/http_proxy\|https_proxy\|HTTP_PROXY\|HTTPS_PROXY\|no_proxy\|NO_PROXY\|ftp_proxy\|FTP_PROXY/d' /etc/environment
  {
    echo "http_proxy=\"${PROXY_FULL}\""
    echo "HTTP_PROXY=\"${PROXY_FULL}\""
    echo "https_proxy=\"${PROXY_FULL}\""
    echo "HTTPS_PROXY=\"${PROXY_FULL}\""
    echo "ftp_proxy=\"${PROXY_FULL}\""
    echo "FTP_PROXY=\"${PROXY_FULL}\""
    echo "no_proxy=\"${NO_PROXY_DEFAULT}\""
    echo "NO_PROXY=\"${NO_PROXY_DEFAULT}\""
  } | sudo tee -a /etc/environment > /dev/null
  success "/etc/environment"
else
  warn "No sudo access, skipping /etc/environment (safe to ignore for non-root users)"
fi

# =============================================================================
section "3. curl (~/.curlrc)"
# =============================================================================
touch "$HOME/.curlrc"
sed -i '/^proxy=/d' "$HOME/.curlrc"
echo "proxy = ${PROXY_FULL}" >> "$HOME/.curlrc"
success "~/.curlrc"

# =============================================================================
section "4. wget (~/.wgetrc)"
# =============================================================================
touch "$HOME/.wgetrc"
sed -i '/^https_proxy\|^http_proxy\|^use_proxy\|^no_proxy/d' "$HOME/.wgetrc"
{
  echo "use_proxy = on"
  echo "http_proxy = ${PROXY_FULL}"
  echo "https_proxy = ${PROXY_FULL}"
  echo "no_proxy = ${NO_PROXY_DEFAULT}"
} >> "$HOME/.wgetrc"
success "~/.wgetrc"

# =============================================================================
section "5. git"
# =============================================================================
git config --global http.proxy  "${PROXY_FULL}"
git config --global https.proxy "${PROXY_FULL}"
success "git global config"

# =============================================================================
section "6. apt (requires sudo)"
# =============================================================================
if sudo -n true 2>/dev/null; then
  sudo tee /etc/apt/apt.conf.d/99proxy > /dev/null <<EOF
Acquire::http::Proxy "${PROXY_FULL}";
Acquire::https::Proxy "${PROXY_FULL}";
Acquire::ftp::Proxy "${PROXY_FULL}";
EOF
  success "/etc/apt/apt.conf.d/99proxy"
else
  warn "No sudo access, skipping apt proxy configuration"
fi

# =============================================================================
section "7. pip (~/.config/pip/pip.conf)"
# =============================================================================
ensure_dir "$HOME/.config/pip"
touch "$HOME/.config/pip/pip.conf"
if ! grep -q '^\[global\]' "$HOME/.config/pip/pip.conf"; then
  echo -e "\n[global]" >> "$HOME/.config/pip/pip.conf"
fi
sed -i '/^proxy\s*=/d' "$HOME/.config/pip/pip.conf"
sed -i '/^\[global\]/a proxy = '"${PROXY_FULL}" "$HOME/.config/pip/pip.conf"
success "~/.config/pip/pip.conf"

# =============================================================================
section "8. npm"
# =============================================================================
if command -v npm &>/dev/null; then
  npm config set proxy       "${PROXY_FULL}"
  npm config set https-proxy "${PROXY_FULL}"
  success "npm (~/.npmrc)"
else
  warn "npm not found, skipping"
fi

# =============================================================================
section "9. yarn"
# =============================================================================
if command -v yarn &>/dev/null; then
  yarn config set proxy       "${PROXY_FULL}" --silent 2>/dev/null || yarn config set proxy       "${PROXY_FULL}"
  yarn config set https-proxy "${PROXY_FULL}" --silent 2>/dev/null || yarn config set https-proxy "${PROXY_FULL}"
  success "yarn (~/.yarnrc / ~/.yarnrc.yml)"
else
  warn "yarn not found, skipping"
fi

# =============================================================================
section "10. pnpm"
# =============================================================================
if command -v pnpm &>/dev/null; then
  pnpm config set proxy       "${PROXY_FULL}"
  pnpm config set https-proxy "${PROXY_FULL}"
  success "pnpm (~/.npmrc)"
else
  warn "pnpm not found, skipping"
fi

# =============================================================================
section "11. docker (systemd drop-in)"
# =============================================================================
if command -v docker &>/dev/null && sudo -n true 2>/dev/null; then
  sudo mkdir -p /etc/systemd/system/docker.service.d
  sudo tee /etc/systemd/system/docker.service.d/proxy.conf > /dev/null <<EOF
[Service]
Environment="HTTP_PROXY=${PROXY_FULL}"
Environment="HTTPS_PROXY=${PROXY_FULL}"
Environment="NO_PROXY=${NO_PROXY_DEFAULT}"
EOF
  sudo systemctl daemon-reload
  sudo systemctl restart docker 2>/dev/null || warn "Failed to restart docker (may not be running)"
  success "docker systemd proxy.conf"

  # Docker daemon pull proxy (~/.docker/config.json)
  ensure_dir "$HOME/.docker"
  if command -v python3 &>/dev/null; then
    python3 - <<PYEOF
import json, pathlib
p = pathlib.Path.home() / ".docker/config.json"
d = json.loads(p.read_text()) if p.exists() else {}
d.setdefault("proxies", {})["default"] = {
    "httpProxy":  "${PROXY_FULL}",
    "httpsProxy": "${PROXY_FULL}",
    "noProxy":    "${NO_PROXY_DEFAULT}"
}
p.write_text(json.dumps(d, indent=2, ensure_ascii=False))
print("  -> ~/.docker/config.json updated")
PYEOF
  fi
else
  warn "docker not found or no sudo access, skipping"
fi

# =============================================================================
section "12. snap (requires sudo)"
# =============================================================================
if command -v snap &>/dev/null && sudo -n true 2>/dev/null; then
  sudo snap set system proxy.http="${PROXY_FULL}"
  sudo snap set system proxy.https="${PROXY_FULL}"
  success "snap system proxy"
else
  warn "snap not found or no sudo access, skipping"
fi

# =============================================================================
section "13. Claude Code (~/.claude/settings.json)"
# =============================================================================
if command -v python3 &>/dev/null; then
  ensure_dir "$HOME/.claude"
  python3 - <<PYEOF
import json, pathlib
p = pathlib.Path.home() / ".claude/settings.json"
d = json.loads(p.read_text()) if p.exists() else {}
d.setdefault("env", {}).update({
    "HTTP_PROXY":  "${PROXY_FULL}",
    "HTTPS_PROXY": "${PROXY_FULL}",
    "NO_PROXY":    "${NO_PROXY_DEFAULT}",
    "http_proxy":  "${PROXY_FULL}",
    "https_proxy": "${PROXY_FULL}",
    "no_proxy":    "${NO_PROXY_DEFAULT}",
})
p.write_text(json.dumps(d, indent=2, ensure_ascii=False))
print("  -> ~/.claude/settings.json updated")
PYEOF
  success "Claude Code settings.json"
else
  warn "python3 not found, skipping Claude Code configuration"
fi

# =============================================================================
section "14. Go (GOPATH tools; shell env vars already cover standard go commands)"
# =============================================================================
if command -v go &>/dev/null; then
  go env -w GOPROXY="https://proxy.golang.org,direct"
  go env -w GONOSUMCHECK="*"
  # HTTP_PROXY/HTTPS_PROXY are passed via shell env vars — no extra step needed
  success "go env GOPROXY"
else
  warn "go not found, skipping"
fi

# =============================================================================
section "15. Rust / cargo (~/.cargo/config.toml)"
# =============================================================================
if command -v cargo &>/dev/null; then
  ensure_dir "$HOME/.cargo"
  cat > "$HOME/.cargo/config.toml" <<EOF
[http]
proxy = "${PROXY_FULL}"

[https]
proxy = "${PROXY_FULL}"
EOF
  success "~/.cargo/config.toml"
else
  warn "cargo not found, skipping"
fi

# =============================================================================
section "16. Maven (~/.m2/settings.xml)"
# =============================================================================
if command -v mvn &>/dev/null || [[ -d "$HOME/.m2" ]]; then
  ensure_dir "$HOME/.m2"
  if command -v python3 &>/dev/null; then
    python3 - <<PYEOF
import re, pathlib
p = pathlib.Path.home() / ".m2/settings.xml"
proxy_block = """  <proxies>
    <proxy>
      <id>setup-proxy</id>
      <active>true</active>
      <protocol>http</protocol>
      <host>${HOST}</host>
      <port>${PORT}</port>
      <username>${PROXY_USER}</username>
      <password>${PROXY_PASS}</password>
      <nonProxyHosts>${NO_PROXY_DEFAULT}</nonProxyHosts>
    </proxy>
  </proxies>"""

if p.exists():
    text = p.read_text()
    text = re.sub(r'\s*<proxies>.*?</proxies>', '', text, flags=re.DOTALL)
    text = text.replace('</settings>', proxy_block + '\n</settings>')
    p.write_text(text)
else:
    p.write_text(f"""<settings xmlns="http://maven.apache.org/SETTINGS/1.0.0">
{proxy_block}
</settings>""")
print("  -> ~/.m2/settings.xml updated")
PYEOF
    success "~/.m2/settings.xml"
  else
    warn "python3 not found, skipping Maven configuration"
  fi
else
  warn "Maven not found, skipping"
fi

# =============================================================================
section "17. Gradle (~/.gradle/gradle.properties)"
# =============================================================================
if command -v gradle &>/dev/null || [[ -d "$HOME/.gradle" ]]; then
  ensure_dir "$HOME/.gradle"
  touch "$HOME/.gradle/gradle.properties"
  sed -i '/^systemProp\.http\|^systemProp\.https/d' "$HOME/.gradle/gradle.properties"
  {
    echo "systemProp.http.proxyHost=${HOST}"
    echo "systemProp.http.proxyPort=${PORT}"
    echo "systemProp.http.nonProxyHosts=${NO_PROXY_DEFAULT}"
    echo "systemProp.https.proxyHost=${HOST}"
    echo "systemProp.https.proxyPort=${PORT}"
    echo "systemProp.https.nonProxyHosts=${NO_PROXY_DEFAULT}"
    [[ -n "$PROXY_USER" ]] && {
      echo "systemProp.http.proxyUser=${PROXY_USER}"
      echo "systemProp.http.proxyPassword=${PROXY_PASS}"
      echo "systemProp.https.proxyUser=${PROXY_USER}"
      echo "systemProp.https.proxyPassword=${PROXY_PASS}"
    }
  } >> "$HOME/.gradle/gradle.properties"
  success "~/.gradle/gradle.properties"
else
  warn "Gradle not found, skipping"
fi

# =============================================================================
section "18. .NET / NuGet (~/.nuget/NuGet/NuGet.Config)"
# =============================================================================
if command -v dotnet &>/dev/null; then
  ensure_dir "$HOME/.nuget/NuGet"
  # dotnet respects HTTP_PROXY env var; write config for nuget CLI as well
  local nuget_cfg="$HOME/.nuget/NuGet/NuGet.Config"
  if [[ ! -f "$nuget_cfg" ]]; then
    echo '<?xml version="1.0" encoding="utf-8"?><configuration></configuration>' > "$nuget_cfg"
  fi
  # dotnet/nuget reads HTTP_PROXY from /etc/environment (already set above)
  success ".NET (via HTTP_PROXY environment variable)"
else
  warn ".NET not found, skipping"
fi

# =============================================================================
section "19. uv (Python package manager)"
# =============================================================================
if command -v uv &>/dev/null; then
  # uv reads standard HTTP_PROXY / HTTPS_PROXY env vars (already set via shell profile)
  success "uv (via HTTP_PROXY / HTTPS_PROXY environment variables)"
  info "For persistent config, add to ~/.config/uv/uv.toml:"
  info "  [network]\n  proxy = \"${PROXY_FULL}\""
else
  warn "uv not found, skipping"
fi

# =============================================================================
# Summary
# =============================================================================
echo ""
echo -e "${BOLD}${GREEN}════════════════════════════════════════${NC}"
echo -e "${BOLD}${GREEN}  All proxy settings configured!${NC}"
echo -e "${BOLD}${GREEN}════════════════════════════════════════${NC}"
echo ""
echo -e "  Proxy address : ${CYAN}${PROXY_URL}${NC}"
echo -e "  Auth user     : ${CYAN}${PROXY_USER:-( none )}${NC}"
echo ""
echo -e "${YELLOW}  Apply changes to the current shell:${NC}"
echo -e "     ${BOLD}source ~/.bashrc${NC}"
echo ""
echo -e "${YELLOW}  To clear all proxy settings:${NC}"
echo -e "     ${BOLD}bash $0 --clear${NC}"
echo ""