#!/bin/bash

GITHUB_REPO=${GITHUB_REPO:-"cestef/punch"}
SKIP_VERSION_CHECK=${SKIP_VERSION_CHECK:-false}
SPECIFIC_VERSION=${SPECIFIC_VERSION:-""}
NO_COLORS=${NO_COLORS:-false}

PROJECT_NAME=$(basename "$GITHUB_REPO")
INSTALLER_NAME="${PROJECT_NAME}-installer.sh"

if [ "$NO_COLORS" = true ]; then
    C_BOLD="" C_DIM="" C_GREEN="" C_RED="" C_YELLOW="" C_BLUE="" C_RESET=""
else
    C_BOLD="\033[1m"
    C_DIM="\033[2m"
    C_GREEN="\033[0;32m"
    C_RED="\033[0;31m"
    C_YELLOW="\033[0;33m"
    C_BLUE="\033[0;34m"
    C_RESET="\033[0m"
fi

msg() {
    local color="$1"
    local message="$2"
    local level="${3:-}"
    
    if [ -n "$level" ]; then
        echo -e "${color}[${level}] ${message}${C_RESET}"
    else
        echo -e "${color}${message}${C_RESET}"
    fi
}

echo -e "${C_BOLD}Installing $PROJECT_NAME...${C_RESET}"

if [ -z "$SPECIFIC_VERSION" ] && [ "$SKIP_VERSION_CHECK" != "true" ]; then
    echo -e "${C_DIM}Fetching latest version information...${C_RESET}"
    
    LATEST_VERSION=$(curl -s "https://api.github.com/repos/$GITHUB_REPO/releases/latest" | 
                    grep '"tag_name":' | sed -E 's/.*"([^"]+)".*/\1/')
    
    if [ -z "$LATEST_VERSION" ]; then
        msg "$C_RED" "Failed to fetch latest version from GitHub API" "ERROR"
        exit 1
    fi
    
    msg "$C_GREEN" "Latest version: ${C_BOLD}$LATEST_VERSION${C_RESET}" "SUCCESS"
    VERSION_TO_INSTALL=$LATEST_VERSION
else
    VERSION_TO_INSTALL=$SPECIFIC_VERSION
    if [ -n "$SPECIFIC_VERSION" ]; then
        msg "$C_YELLOW" "Using specified version: ${C_BOLD}$VERSION_TO_INSTALL${C_RESET}" "VERSION"
    else
        echo -e "${C_DIM}Using latest available version${C_RESET}"
    fi
fi

echo -e "${C_BLUE}Downloading installer...${C_RESET}"
INSTALLER_URL="https://github.com/$GITHUB_REPO/releases/download/${VERSION_TO_INSTALL}/$INSTALLER_NAME"
echo -e "${C_DIM}From: $INSTALLER_URL${C_RESET}"

TEMP_INSTALLER=$(mktemp)
curl --proto '=https' --tlsv1.2 -LsSf "$INSTALLER_URL" -o "$TEMP_INSTALLER"
if [ $? -eq 0 ]; then
    chmod +x "$TEMP_INSTALLER"
    msg "$C_BLUE" "Running installer..."
    if command -v bash >/dev/null 2>&1; then
        bash "$TEMP_INSTALLER"
    else
        sh "$TEMP_INSTALLER"
    fi
    INSTALL_STATUS=$?
    rm "$TEMP_INSTALLER"
    exit $INSTALL_STATUS
else
    msg "$C_RED" "Failed to download installer" "ERROR"
    rm "$TEMP_INSTALLER"
    exit 1
fi