#!/bin/bash
# classify_question.sh - Question classifier for smart routing
#
# Classifies questions by type to optimize consultant selection.
#
# Usage: ./classify_question.sh "question"
# Output: category (CODE_REVIEW, ARCHITECTURE, BUG_DEBUG, etc.)

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "$SCRIPT_DIR/lib/common.sh"

QUESTION="${1:-}"

if [[ -z "$QUESTION" ]]; then
    log_error "Usage: $0 \"question\""
    exit 1
fi

# =============================================================================
# PATTERN-BASED CLASSIFICATION (fast, no API call)
# =============================================================================

classify_by_patterns() {
    local question="$1"
    local q_lower=$(echo "$question" | tr '[:upper:]' '[:lower:]')

    # CODE_REVIEW patterns
    if echo "$q_lower" | grep -qE 'review|check|code review|analyze.*code|quality|qualit'; then
        echo "CODE_REVIEW"
        return 0
    fi

    # BUG_DEBUG patterns
    if echo "$q_lower" | grep -qE 'bug|error|crash|fix|debug|problem|issue|not working|fails|exception|traceback'; then
        echo "BUG_DEBUG"
        return 0
    fi

    # ARCHITECTURE patterns
    if echo "$q_lower" | grep -qE 'architect|design|pattern|structure|microservic|monolit|scalabil|refactor|organiz|system design'; then
        echo "ARCHITECTURE"
        return 0
    fi

    # ALGORITHM patterns
    if echo "$q_lower" | grep -qE 'algorithm|optimi|performance|complexit|O\(|big-o|efficien|sort|search|data structure'; then
        echo "ALGORITHM"
        return 0
    fi

    # SECURITY patterns
    if echo "$q_lower" | grep -qE 'security|vulnerabil|injection|xss|csrf|auth|authentication|password|encrypt|decrypt|token|jwt'; then
        echo "SECURITY"
        return 0
    fi

    # QUICK_SYNTAX patterns
    if echo "$q_lower" | grep -qE 'syntax|how to write|example of|snippet|one-liner'; then
        echo "QUICK_SYNTAX"
        return 0
    fi

    # DATABASE patterns
    if echo "$q_lower" | grep -qE 'database|sql|query|mongodb|postgres|mysql|redis|index|migration|schema'; then
        echo "DATABASE"
        return 0
    fi

    # API patterns
    if echo "$q_lower" | grep -qE 'api|rest|graphql|endpoint|request|response|http|webhook'; then
        echo "API_DESIGN"
        return 0
    fi

    # TESTING patterns
    if echo "$q_lower" | grep -qE 'test|testing|unit test|integration|mock|stub|coverage|tdd|bdd'; then
        echo "TESTING"
        return 0
    fi

    # Default
    echo "GENERAL"
    return 0
}

# =============================================================================
# LLM-BASED CLASSIFICATION (more accurate, requires API)
# =============================================================================

classify_with_llm() {
    local question="$1"

    local prompt="Classify this programming question into ONE category only:

CATEGORIES:
- CODE_REVIEW: Code review, quality analysis, best practices
- BUG_DEBUG: Debugging, error fixing, troubleshooting
- ARCHITECTURE: System design, patterns, project structure
- ALGORITHM: Algorithms, data structures, complexity, optimization
- SECURITY: Security, vulnerabilities, authentication
- QUICK_SYNTAX: Quick syntax questions, snippets, one-liners
- DATABASE: SQL queries, schema design, database operations
- API_DESIGN: API design, REST, GraphQL, endpoints
- TESTING: Unit testing, integration testing, TDD
- GENERAL: Other

QUESTION: $question

Reply ONLY with the category name (e.g.: ARCHITECTURE), nothing else."

    if command -v claude &> /dev/null; then
        local result=$(echo "$prompt" | claude --print 2>/dev/null | tr -d '\n' | tr '[:lower:]' '[:upper:]')
        # Validate that it's a valid category
        case "$result" in
            CODE_REVIEW|BUG_DEBUG|ARCHITECTURE|ALGORITHM|SECURITY|QUICK_SYNTAX|DATABASE|API_DESIGN|TESTING|GENERAL)
                echo "$result"
                return 0
                ;;
            *)
                # Fallback to pattern matching
                classify_by_patterns "$question"
                return 0
                ;;
        esac
    else
        # No LLM available, use pattern matching
        classify_by_patterns "$question"
        return 0
    fi
}

# =============================================================================
# MAIN
# =============================================================================

# Use classification mode from config or default to pattern
CLASSIFICATION_MODE="${CLASSIFICATION_MODE:-pattern}"

case "$CLASSIFICATION_MODE" in
    llm)
        classify_with_llm "$QUESTION"
        ;;
    pattern|*)
        classify_by_patterns "$QUESTION"
        ;;
esac