#!/bin/bash

# Common functionality shared across all AI thinking mechanisms
# This file contains utilities and initialization code used by multiple scripts

# --- Script Directory Setup ---
# Get the directory where this script is located
get_script_dir() {
    echo "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
}

# --- Initialization Functions ---

# Initialize a thinking mechanism script with common dependencies
init_thinking_mechanism() {
    local script_path="$1"

    # Set up script directory
    SCRIPT_DIR="$(cd "$(dirname "${script_path}")" && pwd)"

    # Source configuration
    source "${SCRIPT_DIR}/config.sh"

    # Source logging system
    source "${SCRIPT_DIR}/logging.sh"

    # Source quality guard for output quality protection
    source "${SCRIPT_DIR}/quality_guard.sh"

    # Get mechanism name automatically
    MECHANISM_NAME=$(get_mechanism_name "$script_path")

    # Set up resource management
    setup_cleanup_trap

    export SCRIPT_DIR MECHANISM_NAME
}

# Initialize the main dispatcher with common dependencies
init_dispatcher() {
    local script_path="$1"

    # Set up script directory
    SCRIPT_DIR="$(cd "$(dirname "${script_path}")" && pwd)"

    # Source logging system (dispatcher sources this later)
    # source "${SCRIPT_DIR}/logging.sh"

    export SCRIPT_DIR
}

# --- Model Validation Functions ---

# Validate and set a model with fallback
validate_and_set_model() {
    local model_var="$1"
    local model_name="$2"
    local fallback_model="$3"

    local validated_model
    validated_model=$(validate_model "$model_name" "$fallback_model")

    if [ $? -ne 0 ]; then
        log_error "No valid model available for $model_var"
        return 1
    fi

    eval "$model_var=\"$validated_model\""
    echo "Set $model_var to: $validated_model"
}

# --- Argument Processing Functions ---

# Common file path validation
validate_file_arg() {
    local file_path="$1"

    if [ -n "$file_path" ]; then
        validate_file_path "$file_path"
        if [ $? -ne 0 ]; then
            return 1
        fi
    fi

    echo "$file_path"
}

# --- Cleanup Functions ---

# Global array to track resources for cleanup
declare -a CLEANUP_RESOURCES=()

# Register a resource for cleanup
register_cleanup_resource() {
    local resource="$1"
    CLEANUP_RESOURCES+=("$resource")
}

# Clean up temporary resources
cleanup_resources() {
    local exit_code=$?

    # Clean up registered resources
    for resource in "${CLEANUP_RESOURCES[@]}"; do
        if [ -d "$resource" ]; then
            rm -rf "$resource" 2>/dev/null || true
        elif [ -f "$resource" ]; then
            rm -f "$resource" 2>/dev/null || true
        fi
    done

    # Clean up any additional temp directories
    if [ -n "$TEMP_DIR" ] && [ -d "$TEMP_DIR" ]; then
        rm -rf "$TEMP_DIR" 2>/dev/null || true
    fi

    exit $exit_code
}

# Set up trap for cleanup on script exit
setup_cleanup_trap() {
    trap cleanup_resources EXIT INT TERM
}

# Create a temporary directory with automatic cleanup
create_managed_temp_dir() {
    local prefix="${1:-ai_thinking}"
    local temp_dir

    temp_dir=$(mktemp -d -t "${prefix}_XXXXXX")
    register_cleanup_resource "$temp_dir"

    echo "$temp_dir"
}

# Create a temporary file with automatic cleanup
create_managed_temp_file() {
    local prefix="${1:-ai_thinking}"
    local suffix="${2:-tmp}"
    local temp_file

    temp_file=$(mktemp -t "${prefix}_XXXXXX.${suffix}")
    register_cleanup_resource "$temp_file"

    echo "$temp_file"
}

# --- Standardized Error Handling ---

# Standardized error codes
ERROR_INVALID_ARGUMENT=1
ERROR_FILE_NOT_FOUND=2
ERROR_MODEL_UNAVAILABLE=3
ERROR_VALIDATION_FAILED=4
ERROR_PROCESSING_FAILED=5
ERROR_RESOURCE_ERROR=6

# Standardized error handling function
handle_error() {
    local error_code="$1"
    local error_message="$2"
    local script_name="${3:-$(basename "${BASH_SOURCE[1]}")}"
    local line_number="${4:-${BASH_LINENO[0]}}"

    # Log the error
    log_error "[$script_name:$line_number] $error_message"

    # Print user-friendly error message
    echo "Error: $error_message" >&2

    # Exit with appropriate code
    exit "$error_code"
}

# Validation error handler
handle_validation_error() {
    local error_message="$1"
    local script_name="${2:-$(basename "${BASH_SOURCE[1]}")}"

    handle_error "$ERROR_VALIDATION_FAILED" "$error_message" "$script_name" "${BASH_LINENO[0]}"
}

# Model error handler
handle_model_error() {
    local model_name="$1"
    local script_name="${2:-$(basename "${BASH_SOURCE[1]}")}"

    handle_error "$ERROR_MODEL_UNAVAILABLE" "Model '$model_name' is not available" "$script_name" "${BASH_LINENO[0]}"
}

# File error handler
handle_file_error() {
    local file_path="$1"
    local operation="$2"
    local script_name="${3:-$(basename "${BASH_SOURCE[1]}")}"

    handle_error "$ERROR_FILE_NOT_FOUND" "Cannot $operation file: $file_path" "$script_name" "${BASH_LINENO[0]}"
}

# Processing error handler
handle_processing_error() {
    local operation="$1"
    local details="$2"
    local script_name="${3:-$(basename "${BASH_SOURCE[1]}")}"

    handle_error "$ERROR_PROCESSING_FAILED" "Failed to $operation: $details" "$script_name" "${BASH_LINENO[0]}"
}

# --- Utility Functions ---

# Check if a command exists
command_exists() {
    command -v "$1" >/dev/null 2>&1
}

# Create a temporary directory with cleanup
create_temp_dir() {
    local prefix="${1:-ai_thinking}"
    local temp_dir

    temp_dir=$(mktemp -d -t "${prefix}_XXXXXX")
    echo "$temp_dir"
}

# --- Common Constants ---

# Default values
DEFAULT_ROUNDS=2
DEFAULT_MODEL="gemma3n:e2b"

# File paths
LOG_DIR=~/tmp/ai_thinking

export DEFAULT_ROUNDS DEFAULT_MODEL LOG_DIR