#!/usr/bin/env bash
set -euo pipefail

APP=vibebox
REPO_URL="https://github.com/robcholz/vibebox"
REQUESTED_VERSION=${VERSION:-}

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
ORANGE='\033[38;2;255;140;0m'
NC='\033[0m' # No Color

print_message() {
  local level=$1
  local message=$2
  local color=""

  case $level in
    info) color="${GREEN}" ;;
    warning) color="${YELLOW}" ;;
    error) color="${RED}" ;;
  esac

  echo -e "${color}${message}${NC}"
}

require_cmd() {
  local cmd=$1
  local hint=$2

  if ! command -v "$cmd" >/dev/null 2>&1; then
    print_message error "Missing required command: ${cmd}"
    if [[ -n "$hint" ]]; then
      print_message info "$hint"
    fi
    exit 1
  fi
}

require_cmd git "Install git and retry."

ensure_cargo() {
  if command -v cargo >/dev/null 2>&1; then
    return 0
  fi

  print_message warning "Rust (cargo) is required but not found."
  print_message info "You should review and approve the Rust installer before proceeding."
  reply=""
  if [[ -t 0 ]]; then
    read -r -p "Install Rust using rustup now? (y/N) " reply
  elif [[ -r /dev/tty ]]; then
    read -r -p "Install Rust using rustup now? (y/N) " reply </dev/tty
  else
    print_message error "Non-interactive shell: cannot prompt to install Rust."
    print_message info "Install Rust manually from https://rustup.rs and retry."
    exit 1
  fi

  case "${reply}" in
    y|Y)
      if command -v curl >/dev/null 2>&1; then
        curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
      elif command -v wget >/dev/null 2>&1; then
        wget -qO- https://sh.rustup.rs | sh -s -- -y
      else
        print_message error "Missing required command: curl or wget"
        print_message info "Install Rust manually from https://rustup.rs and retry."
        exit 1
      fi
      # shellcheck source=/dev/null
      if [[ -f "$HOME/.cargo/env" ]]; then
        # shellcheck disable=SC1090
        source "$HOME/.cargo/env"
      fi
    ;;
    *)
      print_message info "Install Rust manually from https://rustup.rs and retry."
      exit 1
    ;;
  esac

  if ! command -v cargo >/dev/null 2>&1; then
    print_message error "Cargo still not available after installation."
    print_message info "Open a new shell or run: source \"$HOME/.cargo/env\""
    exit 1
  fi
}

ensure_cargo

CARGO_HOME=${CARGO_HOME:-$HOME/.cargo}
INSTALL_DIR="$CARGO_HOME/bin"

installed_version=""
if command -v "$APP" >/dev/null 2>&1; then
  installed_version=$("$APP" --version 2>/dev/null | awk '{print $2}' | head -n1 || true)
fi

if [[ -n "$REQUESTED_VERSION" && "$installed_version" == "$REQUESTED_VERSION" ]]; then
  print_message info "Version ${YELLOW}$REQUESTED_VERSION${GREEN} already installed"
  exit 0
fi

install_args=(install "$APP" --locked --git "$REPO_URL")
if [[ -n "$REQUESTED_VERSION" ]]; then
  install_args+=(--tag "v$REQUESTED_VERSION")
fi

if command -v "$APP" >/dev/null 2>&1; then
  install_args+=(--force)
fi

print_message info "Installing ${ORANGE}${APP}${GREEN}..."
print_message info "Using cargo install from ${ORANGE}${REPO_URL}${GREEN}..."

cargo "${install_args[@]}"

if [[ ":$PATH:" != *":$INSTALL_DIR:"* ]]; then
  print_message warning "Cargo bin directory is not on PATH."
  print_message info "Add this to your shell profile if you can't run ${APP}:"
  print_message info "  export PATH=$INSTALL_DIR:\$PATH"
fi

if [ -n "${GITHUB_ACTIONS-}" ] && [ "${GITHUB_ACTIONS}" == "true" ]; then
  echo "$INSTALL_DIR" >> "$GITHUB_PATH"
  print_message info "Added $INSTALL_DIR to \$GITHUB_PATH"
fi