#!/bin/bash
# ClawIDE Installation Script
# Usage: curl -fsSL https://www.clawide.app/install.sh | bash
# Pin a version: VERSION=2.0.0 curl -fsSL https://www.clawide.app/install.sh | bash

set -e

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

# Configuration — binaries are published to the GitLab generic package
# registry (public pull is enabled, so no authentication is required).
GL_PROJECT="axs-dev/clawide"
GL_PROJECT_ENC="axs-dev%2Fclawide"
GL_PKG="https://gitlab.com/api/v4/projects/${GL_PROJECT_ENC}/packages/generic/clawide"
INSTALL_DIR="${INSTALL_DIR:-$HOME/.local/bin}"
BINARY_NAME="clawide"

echo -e "${GREEN}╔════════════════════════════════════════╗${NC}"
echo -e "${GREEN}║    ClawIDE Installation Script        ║${NC}"
echo -e "${GREEN}╚════════════════════════════════════════╝${NC}"
echo ""

# Detect system
OS=$(uname -s)
ARCH=$(uname -m)

case "$OS" in
  Linux*)     OS_TYPE="linux" ;;
  Darwin*)    OS_TYPE="darwin" ;;
  *)          echo -e "${RED}Error: Unsupported OS ($OS)${NC}"; exit 1 ;;
esac

case "$ARCH" in
  x86_64)     ARCH_TYPE="amd64" ;;
  aarch64)    ARCH_TYPE="arm64" ;;
  arm64)      ARCH_TYPE="arm64" ;;
  *)          echo -e "${RED}Error: Unsupported architecture ($ARCH)${NC}"; exit 1 ;;
esac

echo -e "${BLUE}Detected system: ${OS_TYPE} ${ARCH_TYPE}${NC}"

# Choose archive extension for this platform.
EXT="tar.gz"
[ "$OS_TYPE" = "windows" ] && EXT="zip"

# Resolve the download URL from the GitLab package registry. The "latest"
# channel always holds the newest build under version-less filenames, so no
# version lookup is needed; a pinned VERSION uses the versioned channel.
if [ -n "$VERSION" ]; then
  VERSION="${VERSION#v}"   # accept VERSION=2.0.0 or VERSION=v2.0.0
  VERSION_LABEL="v$VERSION"
  FILENAME="clawide-v${VERSION}-${OS_TYPE}-${ARCH_TYPE}.${EXT}"
  DOWNLOAD_URL="${GL_PKG}/v${VERSION}/${FILENAME}"
else
  VERSION_LABEL="latest"
  FILENAME="clawide-${OS_TYPE}-${ARCH_TYPE}.${EXT}"
  DOWNLOAD_URL="${GL_PKG}/latest/${FILENAME}"
fi

echo ""
echo -e "${YELLOW}Installation Plan:${NC}"
echo "  Project:     $GL_PROJECT"
echo "  Version:     $VERSION_LABEL"
echo "  Download:    $DOWNLOAD_URL"
echo "  Install to:  $INSTALL_DIR/$BINARY_NAME"
echo ""

# Create temp directory
TEMP_DIR=$(mktemp -d)
trap "rm -rf $TEMP_DIR" EXIT

# Download
echo -e "${BLUE}Downloading ClawIDE ${VERSION_LABEL}...${NC}"
if ! curl -fsSL "$DOWNLOAD_URL" -o "$TEMP_DIR/$FILENAME"; then
  echo -e "${RED}Error: Failed to download binary${NC}"
  echo -e "${RED}URL: $DOWNLOAD_URL${NC}"
  exit 1
fi

echo -e "${BLUE}Extracting binary...${NC}"
if [[ "$FILENAME" == *.tar.gz ]]; then
  tar -xzf "$TEMP_DIR/$FILENAME" -C "$TEMP_DIR"
elif [[ "$FILENAME" == *.zip ]]; then
  unzip -q "$TEMP_DIR/$FILENAME" -d "$TEMP_DIR"
fi

# Find the binary
BINARY_PATH=$(find "$TEMP_DIR" -type f \( -name "clawide" -o -name "clawide.exe" \) | head -1)

if [ -z "$BINARY_PATH" ]; then
  echo -e "${RED}Error: Binary not found in archive${NC}"
  exit 1
fi

echo -e "${BLUE}Installing to $INSTALL_DIR...${NC}"

# Create install directory if it doesn't exist
mkdir -p "$INSTALL_DIR"

if [ ! -w "$INSTALL_DIR" ]; then
  echo -e "${RED}Error: $INSTALL_DIR is not writable${NC}"
  exit 1
fi

# Install binary
mv "$BINARY_PATH" "$INSTALL_DIR/$BINARY_NAME"
chmod +x "$INSTALL_DIR/$BINARY_NAME"

# Update shell configuration if needed
if [[ "$INSTALL_DIR" == *"/.local/bin" ]]; then
  # Check if ~/.local/bin is in PATH
  if ! grep -q "\.local/bin" "$HOME/.bashrc" 2>/dev/null && ! grep -q "\.local/bin" "$HOME/.zshrc" 2>/dev/null; then
    echo ""
    echo -e "${YELLOW}Note: $INSTALL_DIR is not in your PATH${NC}"
    echo -e "${BLUE}Adding to shell configuration...${NC}"

    # Add to bashrc if it exists
    if [ -f "$HOME/.bashrc" ]; then
      echo 'export PATH="$HOME/.local/bin:$PATH"' >> "$HOME/.bashrc"
      echo "  Added to ~/.bashrc"
    fi

    # Add to zshrc if it exists
    if [ -f "$HOME/.zshrc" ]; then
      echo 'export PATH="$HOME/.local/bin:$PATH"' >> "$HOME/.zshrc"
      echo "  Added to ~/.zshrc"
    fi

    # If neither exists, create .bashrc
    if [ ! -f "$HOME/.bashrc" ] && [ ! -f "$HOME/.zshrc" ]; then
      echo 'export PATH="$HOME/.local/bin:$PATH"' > "$HOME/.bashrc"
      echo "  Created ~/.bashrc with PATH"
    fi

    echo -e "${YELLOW}Please reload your shell: exec \$SHELL${NC}"
  fi
fi

echo -e "${GREEN}✓ Installation successful!${NC}"
echo ""
echo -e "${BLUE}Getting started:${NC}"
echo "  1. Start ClawIDE:        $BINARY_NAME"
echo "  2. Open browser:         http://localhost:9800"
echo "  3. Configure if needed:  ~/.clawide/config.json"
echo ""
echo -e "${BLUE}Documentation:${NC}"
echo "  https://www.clawide.app/"