#!/bin/bash
ARCH=$(uname -m)
echo "请选择要执行的脚本（可多选，用空格分隔）："
echo "1. 开启BBR"
echo "2. 设置Swap大小"
echo "3. 配置常用alias缩写"
echo "4. apt安装常用软件"
echo "5. 配置防火墙"
echo "6. 安装公钥"
echo "7. 安装caddy"
echo "8. 安装conda"
echo "9. 安装docker" 
read -p "输入你的选择: " choices

# 处理选择
for choice in $choices; do
    case $choice in
    1)
        echo "执行开启BBR的脚本..."
        echo -e "net.core.default_qdisc=fq\nnet.ipv4.tcp_congestion_control=bbr" | tee -a /etc/sysctl.conf
        sysctl -p
        ;;
    2)
        echo "输入Swap内存大小(MB) [默认: 2048]: "
        read -t 30 -p "> " swap_size
        if [ $? -ne 0 ] || [ -z "$swap_size" ]; then
            swap_size=2048
            echo -e "\n使用默认值: ${swap_size}MB"
        fi
        dd if=/dev/zero of=/swapfile bs=1M count=$swap_size
        mkswap /swapfile
        swapon /swapfile
        echo "/swapfile swap swap defaults 0 0" | tee -a /etc/fstab
        ;;
    3)
        echo "配置常用alias缩写"
        wget https://git.nvme.cc/adminroot/setup/raw/branch/main/src/alias.txt -qO ~/.alias.txt
        echo -e 'if [ -f ~/.alias.txt ]; then\n    . ~/.alias.txt\nfi' >>~/.bashrc
        ;;
    4)
        echo "即将安装htop cbm tree ufw curl wget lsof micro git git-lfs 7z zellij build-essential ffmpeg"
        apt update -y
        apt install -y htop cbm tree ufw curl wget lsof micro git git-lfs build-essential ffmpeg
        wget https://git.nvme.cc/adminroot/auto-update/raw/branch/main/app/zellij_$ARCH -O /usr/local/bin/zellij
        wget https://git.nvme.cc/adminroot/auto-update/raw/branch/main/app/7z_$ARCH -O /usr/local/bin/7z
        chmod +x /usr/local/bin/7z /usr/local/bin/zellij
        git config --global credential.helper store # git记住用户名密码
        ;;
    5)
        echo "配置防火墙"
        apt update -y
        apt install -y ufw
        ufw allow 22
        ufw allow 80
        ufw allow 443
        ufw allow 13576
        ufw allow 24687
        echo "y" | ufw enable

        # Check if Docker rules already exist
        if grep -q "BEGIN UFW AND DOCKER" /etc/ufw/after.rules; then
            echo "UFW Docker规则已存在, 跳过..."
        else
            echo "下载并追加UFW Docker配置规则..."
            if curl -fsSL https://git.nvme.cc/adminroot/setup/raw/branch/main/src/ufw-docker.txt | tee -a /etc/ufw/after.rules > /dev/null; then
                echo "UFW Docker规则已成功追加到文件末尾"
            else
                echo "错误: 无法下载UFW Docker规则"
                exit 1
            fi
        fi
        ufw reload
        ufw status
        ;;
    6)
        echo "安装公钥"
        mkdir -p ~/.ssh
        echo "ssh-ed25519 AAAAC3NzaC1lZDI1NTE5AAAAIIDtyoFZlC25rUrOyGgkxd4xO3DgcJy751y6w0d6Rbx4 eureka" >>~/.ssh/authorized_keys
        chmod 600 ~/.ssh/authorized_keys
        chmod 700 ~/.ssh
        ;;
    7)
        echo "安装caddy"
        sudo apt install -y debian-keyring debian-archive-keyring apt-transport-https curl gnupg
        curl -1sLf 'https://dl.cloudsmith.io/public/caddy/stable/gpg.key' | sudo gpg --dearmor -o /usr/share/keyrings/caddy-stable-archive-keyring.gpg
        curl -1sLf 'https://dl.cloudsmith.io/public/caddy/stable/debian.deb.txt' | sudo tee /etc/apt/sources.list.d/caddy-stable.list
        sudo apt update
        sudo apt install caddy
        mkdir /etc/caddy/sites
        sed -i '$a import /etc/caddy/sites/*.txt' /etc/caddy/Caddyfile
        ;;
    8)
        read -t 30 -p "Conda安装路径 (默认: ~/.miniconda3): " CONDA_PATH
        CONDA_PATH=${CONDA_PATH:-"$HOME/.miniconda3"}
        PROFILE="$HOME/.bashrc"
        case $ARCH in
            x86_64)
                MINICONDA_INSTALLER="Miniconda3-latest-Linux-x86_64.sh"
            ;;
            aarch64|arm64)
                MINICONDA_INSTALLER="Miniconda3-latest-Linux-aarch64.sh"
            ;;
            *)
                echo "不支持的架构: $ARCH"
                echo "支持的架构: x86_64, aarch64, armv7l"
                exit 1
            ;;
        esac
        echo "检测到系统架构: $ARCH"
        echo "将下载: $MINICONDA_INSTALLER"
        mkdir -p "$CONDA_PATH"
        wget "https://repo.anaconda.com/miniconda/$MINICONDA_INSTALLER" -O "$CONDA_PATH/miniconda.sh"
        bash "$CONDA_PATH/miniconda.sh" -b -u -p "$CONDA_PATH"
        rm -f "$CONDA_PATH/miniconda.sh"
        echo "source $CONDA_PATH/bin/activate" >> "$PROFILE"
        ;;
    9)
        echo "安装docker"
        curl -fsSL https://get.docker.com -o get-docker.sh
        sh get-docker.sh
        rm ./get-docker.sh
        ;;
    *)
        echo "无效的选择: $choice"
        ;;
    esac
done


    echo " ============= Copy and run ============"
    echo "|                                       |"
    echo "|                                       |"
    echo "|            source ~/.bashrc           |"
    echo "|                                       |"
    echo "|                                       |"
    echo " ======================================="