setup.sh

#!/bin/bash

# 定义当前脚本目录
SCRIPT_DIR=$(cd "$(dirname "$0")" && pwd)
echo "Script directory: $SCRIPT_DIR"

# 检查操作系统类型
OS_TYPE=$(uname)
echo "Operating System: $OS_TYPE"

# 安装 Homebrew（如果尚未安装），并在 Linux 上添加 Linuxbrew 路径
setup_brew() {
    source "${SCRIPT_DIR}/setup_brew.sh"
}

# 安装 Bash 和 Zsh
setup_shells() {
    source "${SCRIPT_DIR}/setup_shells.sh"
}

# 安装必要的软件
install_software() {
    echo "Installing brew apps from brew-apps.txt..."
    bash "${SCRIPT_DIR}/install_brew_apps.sh"

    echo "Installing vim-plug for Vim..."
    curl -fLo "$HOME/.vim/autoload/plug.vim" --create-dirs \
        https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim

    echo "Installing vim-plug for Neovim..."
    curl -fLo "$HOME/.local/share/nvim/site/autoload/plug.vim" --create-dirs \
        https://raw.githubusercontent.com/junegunn/vim-plug/master/plug.vim

    if ! command -v zoxide &> /dev/null 2>&1; then
        echo "Installing zoxide..."
        curl -sS https://raw.githubusercontent.com/ajeetdsouza/zoxide/main/install.sh | bash
    fi

    if [ ! -d "$HOME/.nvm" ]; then
        echo "Installing NVM..."
        curl -o install.sh https://raw.githubusercontent.com/nvm-sh/nvm/v0.40.1/install.sh
        chmod +x install.sh
        bash install.sh
        rm install.sh
    fi

    if ! command -v rustup &> /dev/null 2>&1; then
        echo "Installing Rust..."
        curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
        source "$HOME/.cargo/env"
    fi

    if [ ! -d "$HOME/miniconda3" ] || [ -z "$(ls -A "$HOME/miniconda3")" ]; then
        echo "Installing Miniconda..."
        mkdir -p $HOME/miniconda3
        if [ "$OS_TYPE" = "Darwin" ]; then
            ARCH=$(uname -m)  # arm64 or x86_64
            curl -o "$HOME/miniconda.sh" "https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-${ARCH}.sh"
        elif [ "$OS_TYPE" = "Linux" ]; then
            curl -o "$HOME/miniconda.sh" -sS https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
        fi
        bash "$HOME/miniconda.sh" -b -u -p "$HOME/miniconda3"
        if [ $? -ne 0 ]; then
            echo "Miniconda installation failed."
        else
            echo "Miniconda installation succeeded."
        fi
        rm "$HOME/miniconda.sh"
    else
        echo "Miniconda is already installed and not empty."
    fi

    # 安装 AWS CLI
    if ! command -v aws &> /dev/null 2>&1; then
        echo "Installing AWS CLI..."
        if [ "$OS_TYPE" = "Darwin" ]; then
            curl "https://awscli.amazonaws.com/AWSCLIV2.pkg" -o "AWSCLIV2.pkg"
            sudo installer -pkg "AWSCLIV2.pkg" -target /
            rm "AWSCLIV2.pkg"
        elif [ "$OS_TYPE" = "Linux" ]; then
            curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o "awscliv2.zip"
            unzip awscliv2.zip
            sudo ./aws/install
            rm -rf awscliv2.zip aws
        fi
    else
        echo "AWS CLI is already installed."
    fi

    echo "Creating necessary directories..."
    mkdir -p "$HOME/.config/bat"
    mkdir -p "$HOME/.config/ghostty"
    mkdir -p "$HOME/.config/zellij"
    mkdir -p "$HOME/.config"
    mkdir -p "$HOME/.claude"

    # Neovim uses the LazyVim config in this repo. Back up any existing config
    # directory so we do not leave both init.lua and init.vim in place.
    if [ -e "$HOME/.config/nvim" ] && [ ! -L "$HOME/.config/nvim" ]; then
        backup_path="$HOME/.config/nvim.backup.$(date +%Y%m%d%H%M%S)"
        echo "Backing up existing Neovim config to $backup_path"
        mv "$HOME/.config/nvim" "$backup_path"
    fi

    echo "Creating symbolic links..."
    ln -sf "$SCRIPT_DIR/.vimrc" "$HOME/.vimrc"
    ln -sfn "$SCRIPT_DIR/lazyvim" "$HOME/.config/nvim"
    ln -sf "$SCRIPT_DIR/.zshrc" "$HOME/.zshrc"
    ln -sf "$SCRIPT_DIR/.gitconfig" "$HOME/.gitconfig"
    ln -sf "$SCRIPT_DIR/.gitignore" "$HOME/.gitignore"
    ln -sf "$SCRIPT_DIR/.bash_profile" "$HOME/.bash_profile"
    ln -sf "$SCRIPT_DIR/.bashrc" "$HOME/.bashrc"
    ln -sf "$SCRIPT_DIR/.commonrc" "$HOME/.commonrc"
    ln -sf "$SCRIPT_DIR/.common_vimrc" "$HOME/.common_vimrc"
    ln -sf "$SCRIPT_DIR/zellij/config.kdl" "$HOME/.config/zellij/config.kdl"
    ln -sf "$SCRIPT_DIR/.editorconfig" "$HOME/.editorconfig"
    ln -sf "$SCRIPT_DIR/.ideavimrc" "$HOME/.ideavimrc"
    ln -sf "$SCRIPT_DIR/.aliases" "$HOME/.aliases"
    ln -sf "$SCRIPT_DIR/.functions" "$HOME/.functions"
    ln -sf "$SCRIPT_DIR/.tools" "$HOME/.tools"
    ln -sf "$SCRIPT_DIR/.ripgreprc" "$HOME/.ripgreprc"
    ln -sf "$SCRIPT_DIR/bat/config" "$HOME/.config/bat/config"
    ln -sf "$SCRIPT_DIR/starship.toml" "$HOME/.config/starship.toml"
    ln -sf "$SCRIPT_DIR/ghostty/config" "$HOME/.config/ghostty/config"
    ln -sf "$SCRIPT_DIR/claude/CLAUDE.md" "$HOME/.claude/CLAUDE.md"
    ln -sf "$SCRIPT_DIR/claude/RTK.md" "$HOME/.claude/RTK.md"

    echo "Installing Vim and Neovim plugins..."
    vim +PlugInstall +qall
    nvim --headless "+Lazy! sync" +qa

    echo "=================================================="
    echo "Dotfiles setup completed successfully."
    echo "Restart your shell (bash or zsh) to apply the settings"
    echo "=================================================="
}


extra_macos_setup() {
    # turn off accent chars by holding a key
    defaults write -g ApplePressAndHoldEnabled -bool false
}


if [ "$OS_TYPE" = "Darwin" ]; then
    setup_brew
    setup_shells
    install_software
    extra_macos_setup
elif [ "$OS_TYPE" = "Linux" ]; then
    setup_brew
    setup_shells
    install_software
else
    echo "Unsupported operating system: $OS_TYPE"
    exit 1
fi

echo "Done. Run './test.sh' to verify the setup."