i.sh

#!/bin/bash

# Exit immediately if any command fails
set -e

# Function to install Homebrew
install_homebrew() {
    echo "🔍 Homebrew not found. Installing now..."
    /bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

    # Ensure Homebrew is in the PATH (handles both Apple Silicon & Intel)
    eval "$(/opt/homebrew/bin/brew shellenv 2>/dev/null || /usr/local/bin/brew shellenv)"
    
    if ! command -v brew &> /dev/null; then
        echo "❌ Homebrew installation failed. Please install it manually from https://brew.sh/"
        exit 1
    fi
    echo "✅ Homebrew installed successfully!"
}

# Function to get a valid yes/no input
get_yes_no() {
    local prompt="$1"
    local response
    while true; do
        read -p "$prompt (y/n) " response
        case "$response" in
            [Yy]) return 0 ;;  # Yes
            [Nn]) return 1 ;;  # No
            *) echo "❌ Invalid choice. Please enter 'y' for Yes or 'n' for No." ;;
        esac
    done
}

# Check if Homebrew is installed, prompt user to install if missing
if ! command -v brew &> /dev/null; then
    if get_yes_no "🍺 Homebrew is not installed. Do you want to install it now?"; then
        install_homebrew
    else
        echo "❌ Homebrew is required for this script. Exiting."
        exit 1
    fi
fi

# Check if Stow is installed, install it via Homebrew if missing
if ! command -v stow &> /dev/null; then
    echo "🛠 Stow is not installed. Installing it via Homebrew..."
    brew install stow
    echo "✅ Stow installed successfully!"
fi

# Clone dotfiles repository
DOTFILES_DIR="$HOME/dotfiles"
CONFIG_DIR="$HOME/.config"

echo "🚀 Setting up dotfiles..."
cd ~
rm -rf "$DOTFILES_DIR"
git clone --depth 1 https://github.com/phucleeuwu/dotfiles.git "$DOTFILES_DIR"

# Remove existing .zshrc and .config
rm -f ~/.zshrc
rm -rf "$CONFIG_DIR"
mkdir -p "$CONFIG_DIR"

# Use GNU Stow to manage dotfiles
cd "$DOTFILES_DIR"
stow .
stow zsh git -t ~

# Ask if user wants to remove unwanted files
if get_yes_no "🗑 Do you want to remove unwanted files (e.g., .gitignore, .stowrc, raycast🔍)?"; then
    rm -rf .git .gitignore README.md raycast🔍
    echo "✅ Unwanted files removed."
else
    ln -sf "$HOME/Documents/Personal/github-copilot" "$CONFIG_DIR"
    ln -sf "$HOME/Documents/Personal/raycast" "$CONFIG_DIR"
    echo "✅ Symlink created for GitHub Copilot and Raycast."
fi

# Ask if user wants to install Brew packages
BREWFILE="$DOTFILES_DIR/Brewfile"
if [[ -f "$BREWFILE" ]]; then
    if get_yes_no "🍺 Do you want to install my Homebrew packages (Recommended)?"; then
        brew bundle --file="$BREWFILE"
    else
        echo "⏭ Skipping Homebrew package installation."
    fi
else
    echo "⚠ No Brewfile found in ~/dotfiles. Skipping Homebrew package installation."
fi

# Final notice
echo "🎉 Setup complete! All dotfiles have been symlinked and configured."
echo "🛠 If you make any changes to your dotfiles, remember to apply them using: "
echo "   $ cd ~/dotfiles && stow ."