#!/bin/bash

# 오류 발생 시에 스크립트 종료 및 미정의 변수 사용 방지
set -e
set -u

# 변수 정의
# OWNER: GitHub 저장소 소유자 이름
# REPO: GitHub 저장소 이름
# KEYWORD: 다운로드할 릴리스 자산 필터링 키워드
# DEFAULT_EXTRACT_BASE: 기본 설치 디렉토리 경로
OWNER="pyhub-kr"
REPO="pyhub-mcptools"
DEFAULT_EXTRACT_BASE="$HOME/mcptools"

FORCE_OVERWRITE=0
INSTALL_NAME=""
USE_ALPHA=0
SPECIFIC_VERSION=""

while [[ $# -gt 0 ]]; do
  case "$1" in
    -f|--force)
      FORCE_OVERWRITE=1
      shift
      ;;
    --install-name)
      INSTALL_NAME="$2"
      shift 2
      ;;
    --alpha)
      USE_ALPHA=1
      shift
      ;;
    --version)
      SPECIFIC_VERSION="$2"
      shift 2
      ;;
    *)
      shift
      ;;
  esac
done

# 프로그레스바 표시 함수
show_progress() {
  local width=50
  local percent=$1
  local completed=$((width * percent / 100))
  local remaining=$((width - completed))
  
  printf "\r[%s%s] %d%%" "$(printf '#%.0s' $(seq 1 $completed))" "$(printf ' %.0s' $(seq 1 $remaining))" "$percent"
}

# OS 및 CPU 아키텍처 검사 함수
check_os() {
  if [[ "$(uname)" != "Darwin" ]]; then
    echo "Unsupported OS. This script only supports macOS."
    exit 1
  fi
  echo "✅ OS check passed: macOS detected"

  # CPU 아키텍처 확인
  CPU_ARCH=$(uname -m)
  case "$CPU_ARCH" in
    "x86_64")
      echo "✅ CPU Architecture: Intel (x86_64)"
      KEYWORD="macOS-x86_64"
      ;;
    "arm64")
      echo "✅ CPU Architecture: Apple Silicon (ARM64)"
      KEYWORD="macOS-arm64"
      ;;
    *)
      echo "❌ Unsupported CPU architecture: $CPU_ARCH"
      exit 1
      ;;
  esac
}

# 설치 경로 설정 함수
setup_path() {
  if [ -n "$INSTALL_NAME" ]; then
    case "$(uname)" in
      "Darwin")
        EXTRACT_PATH="$HOME/Library/Application Support/$INSTALL_NAME"
        ;;
      "Linux")
        EXTRACT_PATH="$HOME/.local/share/$INSTALL_NAME"
        ;;
      "MINGW"*|"MSYS"*|"CYGWIN"*|"Windows_NT")
        if [ -n "$APPDATA" ]; then
          EXTRACT_PATH="$APPDATA/$INSTALL_NAME"
        else
          EXTRACT_PATH="$HOME/.config/$INSTALL_NAME"
        fi
        ;;
      *)
        echo "❌ Unsupported OS for install-name option."
        exit 1
        ;;
    esac
    echo "Install name provided. Using path: $EXTRACT_PATH"
  else
    echo "Default extraction path is: $DEFAULT_EXTRACT_BASE"
    # 파이프 실행 환경을 감지하여, 사용자 입력 가능 여부 확인
    if [ -t 0 ]; then
      read -p "Use this path? (Press Enter to accept or type a new path): " EXTRACT_BASE
    else
      EXTRACT_BASE=""
    fi
    if [ -z "$EXTRACT_BASE" ]; then
      EXTRACT_BASE="$DEFAULT_EXTRACT_BASE"
    fi
    EXTRACT_PATH="$EXTRACT_BASE/pyhub.mcptools"
  fi

  # 기존 폴더가 있으면 삭제 여부 확인 또는 강제 삭제
  if [ -d "$EXTRACT_PATH" ]; then
    if [ "$FORCE_OVERWRITE" -eq 1 ]; then
      echo "The path '$EXTRACT_PATH' already exists. Deleting (force mode)..."
      rm -rf "$EXTRACT_PATH"
    else
      if [ -t 0 ]; then
        read -p "The path '$EXTRACT_PATH' already exists. Delete and continue? (y/N): " CONFIRM
      else
        CONFIRM="n"
      fi
      if [[ "$CONFIRM" == "y" || "$CONFIRM" == "Y" ]]; then
        rm -rf "$EXTRACT_PATH"
      else
        echo "Installation aborted."
        exit 1
      fi
    fi
  fi

  echo "📁 Installation path set to: $EXTRACT_PATH"
  return 0
}

# 릴리스 정보 가져오기 함수
get_release_info() {
  local api_url
  local release_json
  
  # URL 결정: 특정 버전 > 알파 > 최신 stable
  if [ -n "$SPECIFIC_VERSION" ]; then
    echo "🔍 Fetching release information for version: $SPECIFIC_VERSION..."
    api_url="https://api.github.com/repos/$OWNER/$REPO/releases/tags/$SPECIFIC_VERSION"
  elif [ "$USE_ALPHA" -eq 1 ]; then
    echo "🔍 Fetching latest alpha release information..."
    api_url="https://api.github.com/repos/$OWNER/$REPO/releases"
  else
    echo "🔍 Fetching latest stable release information..."
    api_url="https://api.github.com/repos/$OWNER/$REPO/releases/latest"
  fi
  
  release_json=$(curl -s "$api_url")
  
  # 알파 버전인 경우, prerelease=true 중 최신 선택
  if [ "$USE_ALPHA" -eq 1 ] && [ -z "$SPECIFIC_VERSION" ]; then
    # jq가 설치되어 있는지 확인
    if command -v jq >/dev/null 2>&1; then
      release_json=$(echo "$release_json" | jq '[.[] | select(.prerelease == true)] | .[0]')
      local tag_name=$(echo "$release_json" | jq -r '.tag_name')
      echo "🔬 Selected alpha version: $tag_name"
    else
      # jq가 없으면 첫 번째 prerelease 찾기 (간단한 방법)
      echo "⚠️  jq not found. Using basic parsing..."
      # 첫 번째 prerelease 릴리스 추출 (완벽하지 않지만 대부분 작동)
      release_json=$(echo "$release_json" | awk 'BEGIN{RS="},{";ORS="},{"} /"prerelease":true/{print; exit}' | sed 's/^\[{/{/' | sed 's/}$/}/')
    fi
    
    if [ -z "$release_json" ] || [ "$release_json" = "null" ]; then
      echo "❌ No alpha releases found"
      exit 1
    fi
  fi
  
  # 특정 버전 요청 시 404 체크
  if [ -n "$SPECIFIC_VERSION" ]; then
    if echo "$release_json" | grep -q '"message".*"Not Found"'; then
      echo "❌ Version $SPECIFIC_VERSION not found"
      exit 1
    fi
  fi
  
  RELEASE_JSON="$release_json"
  
  # 키워드에 맞는 zip 파일 링크 찾기
  DOWNLOAD_URL=$(echo "$RELEASE_JSON" | grep "browser_download_url" | grep "pyhub.mcptools-$KEYWORD" | cut -d '"' -f 4 | grep "\.zip$" )
  SHA256_URL=$(echo "$RELEASE_JSON" | grep "browser_download_url" | grep "pyhub.mcptools-$KEYWORD" | cut -d '"' -f 4 | grep "\.sha256$" )

  if [ -z "$DOWNLOAD_URL" ]; then
    echo "❌ No release asset found matching the architecture: $KEYWORD"
    echo "Available assets:"
    echo "$RELEASE_JSON" | grep -o '"name"[[:space:]]*:[[:space:]]*"[^"]*"' | cut -d'"' -f4 | sed 's/^/  - /'
    exit 1
  fi
  
  echo "🔗 Found download URL: $DOWNLOAD_URL"
  echo "🔐 Found checksum URL: $SHA256_URL"
}

# 파일 다운로드 함수
download_file() {
  local url=$1
  local output=$2
  local size_in_bytes=0
  local percent=0

  # 파일 크기 가져오기
  size_in_bytes=$(curl -sI "$url" | grep -i content-length | awk '{print $2}' | tr -d '\r')

  # 크기 정보를 가져올 수 없으면 기본값 사용
  if [ -z "$size_in_bytes" ] || [ "$size_in_bytes" -eq 0 ]; then
    echo "⚠️ Could not determine file size, downloading without progress bar"
    curl -L "$url" -o "$output"
    return
  fi

  # 프로그레스바와 함께 다운로드
  echo "📥 Downloading: $url"
  curl -L "$url" -o "$output" -#
  echo ""
  echo "✅ Download complete: $output"
}

# 체크섬 검증 함수
verify_checksum() {
  local file=$1
  local checksum_file=$2
  
  echo "🔍 Verifying file integrity..."
  
  # SHA256 해시값만 추출
  EXPECTED_HASH=$(grep -Eo '^[a-fA-F0-9]{64}' "$checksum_file")
  if [ -z "$EXPECTED_HASH" ]; then
    echo "❌ Could not extract SHA256 hash from checksum file."
    exit 1
  fi

  # zip 파일의 실제 해시 계산 (Linux/macOS 호환 처리)
  if command -v sha256sum &> /dev/null; then
    ACTUAL_HASH=$(sha256sum "$file" | awk '{print $1}')
  elif command -v shasum &> /dev/null; then
    ACTUAL_HASH=$(shasum -a 256 "$file" | awk '{print $1}')
  else
    echo "❌ No SHA256 tool found (sha256sum or shasum required)."
    exit 1
  fi

  # 해시 비교
  if [ "$ACTUAL_HASH" != "$EXPECTED_HASH" ]; then
    echo "❌ Checksum verification failed!"
    echo "Expected: $EXPECTED_HASH"
    echo "Actual  : $ACTUAL_HASH"
    rm -f "$file"
    exit 1
  else
    echo "✅ Checksum verified successfully."
  fi
}

# 압축 해제 및 설치 함수
extract_and_install() {
  local file=$1
  local tmp_dir=$2
  local extract_path=$3
  
  echo "📦 Extracting files..."
  # 임시 디렉토리에 압축 해제
  unzip -q "$file" -d "$tmp_dir"
  show_progress 50
  
  # 최종 목적지 디렉토리 생성
  mkdir -p "$extract_path"
  
  # 압축 해제된 내용 중 첫 번째 디렉토리 찾기 (압축 파일이 하나의 루트 디렉토리를 포함한다고 가정)
  first_dir=$(find "$tmp_dir" -mindepth 1 -maxdepth 1 -type d | head -n 1)
  
  if [ -n "$first_dir" ]; then
    # 첫 번째 디렉토리 내의 내용물만 이동
    mv "$first_dir"/* "$extract_path"
  else
    # 압축 해제된 모든 파일을 직접 이동
    mv "$tmp_dir"/* "$extract_path"
  fi
  
  show_progress 100
  echo ""
  
  # 정리
  rm -rf "$tmp_dir"
  echo "✅ Installation complete at: $extract_path"
}

# PATH 환경변수 설정 함수
setup_path_env() {
  local extract_path=$1
  
  cd "$extract_path"
  echo ""
  echo "Changed directory to: $extract_path"

  # Check if path is already in PATH
  if [[ ":$PATH:" != *":$extract_path:"* ]]; then
    SHELL_PROFILE=""
    case "$SHELL" in
      */zsh)
        SHELL_PROFILE="$HOME/.zshrc"
        ;;
      */bash)
        if [ -f "$HOME/.bash_profile" ]; then
          SHELL_PROFILE="$HOME/.bash_profile"
        else
          SHELL_PROFILE="$HOME/.bashrc"
        fi
        ;;
    esac

    if [ -n "$SHELL_PROFILE" ]; then
      if ! grep -q "$extract_path" "$SHELL_PROFILE"; then
        echo "export PATH=\"\$PATH:$extract_path\"" >> "$SHELL_PROFILE"
        echo "✅ Added to PATH in $SHELL_PROFILE"
      else
        echo "✅ Path already present in $SHELL_PROFILE"
      fi
    else
      echo "⚠️ No known shell profile found. Please add the following to your PATH manually:"
      echo "export PATH=\"\$PATH:$extract_path\""
    fi
  fi
}

# 설치 후 안내 함수
show_post_install_instructions() {
  local extract_path=$1
  
  echo ""
  echo "📝 To add configuration for Claude, please run the following commands:"
  echo ""
  echo "cd \"$extract_path\""
  echo "./pyhub.mcptools --version"
  echo "./pyhub.mcptools kill claude"
  echo "./pyhub.mcptools setup-add"
  echo "./pyhub.mcptools setup-print"
  echo ""
  echo "🔄 To use pyhub.mcptools in your current terminal session, run:"
  echo "export PATH=\"\$PATH:$extract_path\""
}

# 메인 함수
main() {
  check_os
  setup_path
  get_release_info
  
  # Create temporary directory
  TMP_DIR=$(mktemp -d)
  
  # Download files
  FILENAME="$TMP_DIR/$(basename "$DOWNLOAD_URL")"
  CHECKSUM_FILE="$TMP_DIR/$(basename "$SHA256_URL")"
  
  download_file "$DOWNLOAD_URL" "$FILENAME"
  download_file "$SHA256_URL" "$CHECKSUM_FILE"
  
  verify_checksum "$FILENAME" "$CHECKSUM_FILE"
  extract_and_install "$FILENAME" "$TMP_DIR" "$EXTRACT_PATH"
  setup_path_env "$EXTRACT_PATH"
  show_post_install_instructions "$EXTRACT_PATH"
}

# 스크립트 실행
main