install.sh

#!/bin/bash

# nwt Installation Script
# This script installs the nwt (New Worktree) function to your shell configuration

set -e

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Detect shell and config file
detect_shell_config() {
    # Check user's default shell from SHELL environment variable
    if [[ "$SHELL" == *"zsh"* ]]; then
        echo "$HOME/.zshrc"
    elif [[ "$SHELL" == *"bash"* ]]; then
        echo "$HOME/.bashrc"
    # On macOS, default to zsh (it's been the default since macOS Catalina)
    elif [[ "$OSTYPE" == "darwin"* ]]; then
        echo "$HOME/.zshrc"
    else
        # Fallback detection based on which config files exist
        if [[ -f "$HOME/.zshrc" ]]; then
            echo "$HOME/.zshrc"
        elif [[ -f "$HOME/.bashrc" ]]; then
            echo "$HOME/.bashrc"
        else
            echo ""
        fi
    fi
}

main() {
    echo -e "${BLUE}🚀 Installing nwt (New Worktree) for Warp Terminal${NC}"
    echo

    # Detect config file
    CONFIG_FILE=$(detect_shell_config)
    
    if [[ -z "$CONFIG_FILE" ]]; then
        echo -e "${RED}❌ Could not detect shell configuration file${NC}"
        echo "Please manually add the contents of nwt.zsh to your shell configuration"
        exit 1
    fi

    echo -e "${YELLOW}📝 Detected config file: $CONFIG_FILE${NC}"
    
    # Check if already installed
    if grep -q "new_worktree()" "$CONFIG_FILE" 2>/dev/null; then
        echo -e "${YELLOW}⚠️  nwt appears to already be installed${NC}"
        read -p "Do you want to reinstall? (y/N): " -r
        if [[ ! $REPLY =~ ^[Yy]$ ]]; then
            echo "Installation cancelled"
            exit 0
        fi
        
        # Remove existing installation
        echo -e "${YELLOW}🔄 Removing existing installation...${NC}"
        # Create a temp file without the nwt function
        grep -v -A 200 "# Git worktree + Warp function" "$CONFIG_FILE" | \
        grep -v -B 200 "alias nwt='new_worktree'" > "$CONFIG_FILE.tmp" || true
        mv "$CONFIG_FILE.tmp" "$CONFIG_FILE"
    fi

    # Add the function to config file
    echo -e "${YELLOW}📦 Installing nwt function...${NC}"
    
    echo "" >> "$CONFIG_FILE"
    echo "# Git worktree + Warp function (installed by nwt installer)" >> "$CONFIG_FILE"
    cat nwt.zsh >> "$CONFIG_FILE"
    echo "" >> "$CONFIG_FILE"

    echo -e "${GREEN}✅ nwt installed successfully!${NC}"
    echo
    echo -e "${BLUE}📖 Usage:${NC}"
    echo "  nwt <branch-name>              # Create worktree with auto-detected base branch"
    echo "  nwt <branch-name> <base-branch> # Create worktree from specific base branch"
    echo
    echo -e "${BLUE}Examples:${NC}"
    echo "  nwt feature/user-auth         # Creates from main/master/develop"
    echo "  nwt hotfix/critical-bug main  # Creates from main branch"
    echo
    echo -e "${YELLOW}🔄 To start using nwt, run: source $CONFIG_FILE${NC}"
    echo -e "${YELLOW}   Or restart your terminal${NC}"
}

# Check if script is being sourced or executed directly
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
    main "$@"
fi