zshsetup.sh

#!/bin/bash

# Function to display usage
usage() {
    echo "Usage: $0 [-h] [-v] [-n] [-t THEME] [-p PLUGINS...] [-f PLUGINS_FILE]"
    echo "Options:"
    echo "  -h          Show this help message"
    echo "  -v          Verbose output"
    echo "  -n          No backup (skip backing up existing configurations)"
    echo "  -t THEME    Specify oh-my-zsh theme (default: 'powerlevel10k/powerlevel10k')"
    echo "  -p PLUGINS  Specify additional plugins (space-separated list)"
    echo "  -f FILE     Read additional plugins from file (one plugin per line)"
    echo
    echo "Example: $0 -v -t agnoster -p 'git docker kubectl' -f ~/.zsh_plugins"
    exit 1
}

# Function to log messages
log() {
    local level=$1
    shift
    local msg="$@"
    local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
    
    case "$level" in
        INFO)
            [ "$VERBOSE" = true ] && echo "[$timestamp] INFO: $msg"
            ;;
        WARN)
            echo "[$timestamp] WARNING: $msg" >&2
            ;;
        ERROR)
            echo "[$timestamp] ERROR: $msg" >&2
            ;;
        *)
            echo "[$timestamp] $msg"
            ;;
    esac
}

# Function to backup existing configuration
backup_config() {
    local backup_dir="$HOME/.zsh_backup_$(date +%Y%m%d_%H%M%S)"
    
    if [ -f "$HOME/.zshrc" ]; then
        log INFO "Backing up existing .zshrc to $backup_dir"
        mkdir -p "$backup_dir"
        cp "$HOME/.zshrc" "$backup_dir/"
    fi
    
    if [ -d "$HOME/.oh-my-zsh" ]; then
        log INFO "Backing up existing oh-my-zsh to $backup_dir"
        cp -r "$HOME/.oh-my-zsh" "$backup_dir/"
    fi
}

# Function to install dependencies
install_dependencies() {
    log INFO "Installing required packages..."
    
    if command -v apt-get >/dev/null 2>&1; then
        sudo -S apt-get update
        sudo -S apt-get install -y zsh git curl wget nano
    elif command -v yum >/dev/null 2>&1; then
        sudo -S yum -y install zsh git curl wget nano
    elif command -v pacman >/dev/null 2>&1; then
        sudo -S pacman -Sy --noconfirm zsh git curl wget nano
    else
        log ERROR "Unsupported package manager. Please install zsh, git, curl, wget, and nano manually."
        exit 1
    fi
}

# Function to install oh-my-zsh
install_oh_my_zsh() {
    log INFO "Installing oh-my-zsh..."
    
    if [ -d "$HOME/.oh-my-zsh" ]; then
        log WARN "oh-my-zsh is already installed"
        return
    fi
    
    # Install oh-my-zsh
    sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" "" --unattended
    
    if [ $? -ne 0 ]; then
        log ERROR "Failed to install oh-my-zsh"
        exit 1
    fi

    # Enable auto-updates
    log INFO "Enabling oh-my-zsh auto-updates..."
    sed -i 's/# zstyle '\'':omz:update'\'' mode auto/zstyle '\'':omz:update'\'' mode auto/' "$HOME/.zshrc"
    
    # Enable command correction
    log INFO "Enabling command correction..."
    sed -i 's/# ENABLE_CORRECTION="true"/ENABLE_CORRECTION="true"/' "$HOME/.zshrc"
    
    # Set history timestamps
    log INFO "Setting history timestamps format..."
    sed -i 's/# HIST_STAMPS="mm\/dd\/yyyy"/HIST_STAMPS="mm\/dd\/yyyy"/' "$HOME/.zshrc"
}

# Function to install plugins
install_plugins() {
    local plugins=()
    
    # First argument might be a file
    if [ -n "$1" ] && [ -f "$1" ]; then
        local plugins_file="$1"
        shift
        # Read plugins from file (one per line)
        while IFS= read -r plugin || [ -n "$plugin" ]; do
            # Skip empty lines and comments
            [[ -z "$plugin" || "$plugin" =~ ^[[:space:]]*# ]] && continue
            plugins+=("$plugin")
        done < "$plugins_file"
    fi
    
    # Add any remaining plugins from command line
    plugins+=("$@")
    
    local custom_plugins_dir="$HOME/.oh-my-zsh/custom/plugins"
    
    # Install zsh-autosuggestions
    if [ ! -d "$custom_plugins_dir/zsh-autosuggestions" ]; then
        log INFO "Installing zsh-autosuggestions..."
        git clone https://github.com/zsh-users/zsh-autosuggestions "$custom_plugins_dir/zsh-autosuggestions"
    fi
    
    # Install zsh-syntax-highlighting
    if [ ! -d "$custom_plugins_dir/zsh-syntax-highlighting" ]; then
        log INFO "Installing zsh-syntax-highlighting..."
        git clone https://github.com/zsh-users/zsh-syntax-highlighting "$custom_plugins_dir/zsh-syntax-highlighting"
    fi
    
    # Install zsh-completions
    if [ ! -d "$custom_plugins_dir/zsh-completions" ]; then
        log INFO "Installing zsh-completions..."
        git clone https://github.com/zsh-users/zsh-completions "$custom_plugins_dir/zsh-completions"
    fi
    
    # Install zsh-history-substring-search
    if [ ! -d "$custom_plugins_dir/zsh-history-substring-search" ]; then
        log INFO "Installing zsh-history-substring-search..."
        git clone https://github.com/zsh-users/zsh-history-substring-search "$custom_plugins_dir/zsh-history-substring-search"
    fi
    
    # Install apt plugin
    if [ ! -d "$custom_plugins_dir/apt-zsh-plugin" ]; then
        log INFO "Installing apt plugin..."
        git clone https://github.com/GeoLMg/apt-zsh-plugin "$custom_plugins_dir/apt-zsh-plugin"
    fi
    
    # Add custom plugins to .zshrc
    local plugin_list="git zsh-autosuggestions zsh-syntax-highlighting zsh-completions zsh-history-substring-search history-substring-search apt-zsh-plugin ${plugins[@]}"
    sed -i "s|plugins=(git)|plugins=($plugin_list)|" "$HOME/.zshrc"
}

# Function to configure theme
configure_theme() {
    local theme=$1
    
    # Install powerlevel10k theme
    if [ ! -d "${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/powerlevel10k" ]; then
        log INFO "Installing powerlevel10k theme..."
        git clone --depth=1 https://github.com/romkatv/powerlevel10k.git "${ZSH_CUSTOM:-$HOME/.oh-my-zsh/custom}/themes/powerlevel10k"
    fi
    
    log INFO "Setting theme to $theme..."
    sed -i "s|ZSH_THEME=\"[^\"]*\"|ZSH_THEME=\"$theme\"|" "$HOME/.zshrc"
}

# Default values
VERBOSE=false
NO_BACKUP=false
THEME="powerlevel10k/powerlevel10k"
PLUGINS=()
PLUGINS_FILE=""

# Parse arguments
while getopts "hvnt:p:f:" opt; do
    case $opt in
        h)
            usage
            ;;
        v)
            VERBOSE=true
            ;;
        n)
            NO_BACKUP=true
            ;;
        t)
            THEME="$OPTARG"
            ;;
        p)
            IFS=' ' read -r -a PLUGINS <<< "$OPTARG"
            ;;
        f)
            PLUGINS_FILE="$OPTARG"
            if [ ! -f "$PLUGINS_FILE" ]; then
                log ERROR "Plugins file not found: $PLUGINS_FILE"
                exit 1
            fi
            ;;
        \?)
            usage
            ;;
    esac
done

# Main installation process
log INFO "Starting zsh setup..."

# Backup existing configuration
if [ "$NO_BACKUP" = false ]; then
    backup_config
fi

# Install dependencies
install_dependencies

# Install oh-my-zsh
install_oh_my_zsh

# Install and configure plugins
if [ -n "$PLUGINS_FILE" ]; then
    install_plugins "$PLUGINS_FILE" "${PLUGINS[@]}"
else
    install_plugins "${PLUGINS[@]}"
fi

# Configure theme
configure_theme "$THEME"

# Set zsh as default shell
if [ "$SHELL" != "$(which zsh)" ]; then
    log INFO "Setting zsh as default shell..."
    chsh -s "$(which zsh)"
fi

# Add custom configurations
cat >> "$HOME/.zshrc" << 'EOL'

# Custom aliases
alias ll='ls -lah'
alias la='ls -A'
alias l='ls -CF'
alias ..='cd ..'
alias ...='cd ../..'

# History configuration
HISTSIZE=10000
SAVEHIST=10000
setopt SHARE_HISTORY
setopt HIST_IGNORE_DUPS
setopt HIST_IGNORE_SPACE

# Key bindings
bindkey '^[[A' history-substring-search-up
bindkey '^[[B' history-substring-search-down
EOL

# Set zsh as default shell
if [ "$SHELL" != "$(which zsh)" ]; then
    log INFO "Setting zsh as default shell..."
    chsh -s "$(which zsh)"
fi

log INFO "zsh setup completed successfully!"
log INFO "Please restart your terminal or run 'zsh' to start using your new shell"

# echo "Press any key to proceed..."

# # Loop until a key is pressed
# while true; do
#     read -rsn1 key  # Read a single character silently
#     if [[ -n "$key" ]]; then
#         nano .zshrc
#         break
#     fi
# done

exit 0