#!/bin/bash

echo "=== 安装 SSH 代理工具 ==="
echo ""

echo "选择要安装的工具："
echo "1. connect-proxy (HTTP 代理，推荐)"
echo "2. socat (SOCKS5 代理)"
echo "3. proxychains4 (通用代理工具)"
echo "4. 全部安装"
echo ""
read -p "请选择 (1-4): " choice

case $choice in
    1)
        echo "安装 connect-proxy..."
        sudo apt-get update
        sudo apt-get install -y connect-proxy
        ;;
    2)
        echo "安装 socat..."
        sudo apt-get update
        sudo apt-get install -y socat
        ;;
    3)
        echo "安装 proxychains4..."
        sudo apt-get update
        sudo apt-get install -y proxychains4
        ;;
    4)
        echo "安装所有工具..."
        sudo apt-get update
        sudo apt-get install -y connect-proxy socat proxychains4
        ;;
    *)
        echo "无效选择"
        exit 1
        ;;
esac

echo ""
echo "✓ 安装完成"
echo ""
echo "现在运行诊断脚本："
echo "  ./diagnose_and_fix.sh"