#!/bin/bash
# End-to-end test script for SNO Quickstarts repository integration with Kanbn
# This script demonstrates a real-world example of using Kanbn with a GitHub repository

# Set environment variables for testing
export KANBN_ENV=test
export DEBUG=true

# Function to print info message
print_info() {
  echo -e "\033[0;34m ℹ️ $1\033[0m"
}

# Function to print success message
print_success() {
  echo -e "\033[0;32m ✅ $1\033[0m"
}

# Function to print warning message
print_warning() {
  echo -e "\033[0;33m ⚠️ $1\033[0m"
}

# Function to print error message
print_error() {
  echo -e "\033[0;31m ❌ $1\033[0m"
  # Don't exit immediately to allow for graceful failure
}

print_info "Starting SNO Quickstarts Integration Test"

# Create a test directory
TEST_DIR="/tmp/kanbn_sno_test_$(date +%s)"
mkdir -p "$TEST_DIR"
cd "$TEST_DIR" || { print_error "Failed to create and change to test directory"; exit 1; }

print_info "Testing in: $TEST_DIR"

# Check for Ollama connectivity
print_info "Checking for Ollama..."
OLLAMA_HOST=${OLLAMA_HOST:-"http://localhost:11434"}
print_info "Using Ollama at: $OLLAMA_HOST"

if curl -s "$OLLAMA_HOST/api/tags" >/dev/null 2>&1; then
  print_success "Ollama is reachable at $OLLAMA_HOST"

  # Check for an embedding model
  AVAILABLE_MODELS=$(curl -s "$OLLAMA_HOST/api/tags" | grep -o '"name":"[^"]*"' | cut -d'"' -f4)

  if echo "$AVAILABLE_MODELS" | grep -q "${OLLAMA_MODEL:-llama3}"; then
    print_success "Embedding model found: ${OLLAMA_MODEL:-llama3}"
  else
    print_warning "Specified model ${OLLAMA_MODEL:-llama3} not found"
    print_info "Available models: $AVAILABLE_MODELS"

    # Use the first available model
    if [ -n "$AVAILABLE_MODELS" ]; then
      FIRST_MODEL=$(echo "$AVAILABLE_MODELS" | head -n 1)
      print_info "Using available model: $FIRST_MODEL"
      export OLLAMA_MODEL="$FIRST_MODEL"
    else
      print_warning "No models available - tests will use fallback embedding method"
    fi
  fi
else
  print_warning "Ollama not reachable at $OLLAMA_HOST - tests will use fallback embedding method"
fi

# Clone the SNO Quickstarts repository
print_info "Cloning the SNO Quickstarts repository..."
if git clone https://github.com/tosin2013/sno-quickstarts.git; then
  print_success "Repository cloned successfully"
  cd sno-quickstarts || { print_error "Failed to change to repository directory"; exit 1; }
else
  print_error "Failed to clone repository"
  exit 1
fi

# Initialize kanbn board
print_info "Initializing Kanbn board..."
NODE_OPTIONS=--no-deprecation kanbn init --name "SNO Quickstarts" --description "Single Node OpenShift Quickstart Guides and Automation"

# Check if kanbn was initialized
if [ -d ".kanbn" ]; then
  print_success "Board initialized"
else
  print_error "Board initialization failed"
  exit 1
fi

# Add repository content as integrations
print_info "Adding repository content as integrations..."

# Add README.md as integration
if [ -f "README.md" ]; then
  print_info "Adding README.md as integration..."
  NODE_OPTIONS=--no-deprecation kanbn integrations --add --name "readme" --content "$(cat README.md)" || print_warning "Failed to add README.md integration"
  print_success "Added README.md as integration"
fi

# Add documentation files as integrations
for doc_file in *.md docs/*.md; do
  if [ -f "$doc_file" ] && [ "$doc_file" != "README.md" ]; then
    filename=$(basename "$doc_file")
    print_info "Adding $doc_file as integration..."
    NODE_OPTIONS=--no-deprecation kanbn integrations --add --name "$filename" --content "$(cat "$doc_file")" || print_warning "Failed to add $doc_file integration"
    print_success "Added $doc_file as integration"
  fi
done

# Add scripts as integrations
print_info "Adding scripts as integrations..."
for script_file in *.sh scripts/*.sh; do
  if [ -f "$script_file" ]; then
    filename=$(basename "$script_file")
    print_info "Adding $script_file as integration..."
    NODE_OPTIONS=--no-deprecation kanbn integrations --add --name "$filename" --content "$(cat "$script_file")" || print_warning "Failed to add $script_file integration"
    print_success "Added $script_file as integration"
  fi
done

# List integrations to verify they were added
print_info "Listing integrations after adding..."
NODE_OPTIONS=--no-deprecation kanbn integrations --list

# Create tasks based on repository content
print_info "Creating tasks based on repository content..."

# Add task for setting up SNO
NODE_OPTIONS=--no-deprecation kanbn add "setup-sno" \
    --name "Set up Single Node OpenShift" \
    --description "Configure and deploy a Single Node OpenShift instance using the quickstart guides" \
    --column "Backlog" \
    --tags "infrastructure,openshift,high-priority" || print_warning "Failed to add setup-sno task"

# Add task for automation scripts
NODE_OPTIONS=--no-deprecation kanbn add "automation-scripts" \
    --name "Implement automation scripts" \
    --description "Create and test automation scripts for SNO deployment and configuration" \
    --column "Backlog" \
    --tags "automation,scripts,medium-priority" || print_warning "Failed to add automation-scripts task"

# Add task for documentation
NODE_OPTIONS=--no-deprecation kanbn add "documentation" \
    --name "Update documentation" \
    --description "Improve and expand the documentation for SNO quickstarts" \
    --column "Backlog" \
    --tags "documentation,low-priority" || print_warning "Failed to add documentation task"

print_success "Tasks created based on repository content"

# Show the board
print_info "Displaying Kanban board..."
NODE_OPTIONS=--no-deprecation kanbn board

# Test the RAG functionality with a specific query
print_info "Testing chat with repository integrations..."
NODE_OPTIONS=--no-deprecation kanbn chat --message "What is SNO Quickstarts about?" --with-integrations --quiet || print_warning "Chat test with integrations failed"

# Test RAG functionality with specific queries using the built-in kanbn commands
print_info "Testing RAG functionality with specific queries..."

# Test query about SNO setup
print_info "Testing query: 'How to set up Single Node OpenShift'"
NODE_OPTIONS=--no-deprecation kanbn chat --message "How to set up Single Node OpenShift" --with-integrations --quiet || print_warning "Chat test for SNO setup query failed"

# Test query about automation
print_info "Testing query: 'What automation scripts are available'"
NODE_OPTIONS=--no-deprecation kanbn chat --message "What automation scripts are available" --with-integrations --quiet || print_warning "Chat test for automation scripts query failed"

# Test query about documentation
print_info "Testing query: 'Show me the documentation structure'"
NODE_OPTIONS=--no-deprecation kanbn chat --message "Show me the documentation structure" --with-integrations --quiet || print_warning "Chat test for documentation query failed"

print_info "All tests completed"
print_info "Your test environment is available at: $TEST_DIR/sno-quickstarts"
print_info "To chat with integrations: cd $TEST_DIR/sno-quickstarts && kanbn chat --with-integrations"

# No tempfiles to clean up

# Return success even if some tests failed
# This allows the pipeline to continue
exit 0