#!/bin/bash

# Test functions for machine setup scripts
# Safe testing without system modifications

set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# Source shared library functions
source "$SCRIPT_DIR/lib.sh"

test_sheet_numbers() {
    echo "Testing sheet number generation..."
    
    # Test cases with expected sheet numbers
    local tests=(
        "topsheet:5"         # Default tinsnip sheet
    )
    
    for test_case in "${tests[@]}"; do
        IFS=':' read -r sheet expected <<< "$test_case"
        actual=$(get_sheet_number "$sheet")
        
        if [[ "$actual" == "$expected" ]]; then
            echo "✓ $sheet = $actual"
        else
            echo "✗ $sheet = $actual (expected $expected)"
        fi
    done
}

test_uid_calculation() {
    echo "Testing UID calculation..."
    
    # Test cases for default sheet (topsheet) - only station which doesn't require registry
    local tests=(
        "station:prod:50000"    # topsheet sheet (S=5), S=00
        "station:test:50010"    # topsheet sheet (S=5), S=00
    )
    
    for test_case in "${tests[@]}"; do
        IFS=':' read -r service env expected <<< "$test_case"
        actual=$(calculate_service_uid "$service" "$env")
        
        if [[ "$actual" == "$expected" ]]; then
            echo "✓ $service:$env = $actual"
        else
            echo "✗ $service:$env = $actual (expected $expected)"
        fi
    done
    unset TIN_SHEET
}

test_nfs_path_generation() {
    echo "Testing NFS path generation..."
    
    local sheet="topsheet"  # Default for testing
    local tests=(
        "station:prod:/volume1/tinsnip/${sheet}/station/prod"
        "gazette:test:/volume1/tinsnip/${sheet}/gazette/test"
    )
    
    for test_case in "${tests[@]}"; do
        IFS=':' read -r service env expected <<< "$test_case"
        actual="/volume1/tinsnip/${sheet}/${service}/${env}"
        
        if [[ "$actual" == "$expected" ]]; then
            echo "✓ $service:$env = $actual"
        else
            echo "✗ $service:$env = $actual (expected $expected)"
        fi
    done
}

test_xdg_paths() {
    echo "Testing XDG path generation..."
    
    export XDG_STATE_HOME="$HOME/.local/state"
    export XDG_DATA_HOME="$HOME/.local/share"
    export XDG_CONFIG_HOME="$HOME/.config"
    
    local service="gazette"
    local sheet="${TIN_SHEET:-topsheet}"  # Use env var or default
    
    echo "XDG paths for $service:"
    echo "  State:  ${XDG_STATE_HOME}/${sheet}/@${service}"
    echo "  Data:   ${XDG_DATA_HOME}/${sheet}/@${service}"
    echo "  Config: ${XDG_CONFIG_HOME}/${sheet}/@${service}"
}

calculate_service_ports() {
    local service_uid="$1"
    local port_count="${2:-3}"
    
    for ((i=0; i<port_count; i++)); do
        echo $((service_uid + i))
    done
}

test_port_allocation() {
    echo "Testing port allocation..."
    
    # Test cases: service:env:expected_ports (topsheet sheet = 5)
    local tests=(
        "station:prod:50000,50001,50002"
        "station:test:50010,50011,50012"
    )
    
    for test_case in "${tests[@]}"; do
        IFS=':' read -r service env expected_ports <<< "$test_case"
        
        # Calculate UID and ports
        local service_uid
        service_uid=$(calculate_service_uid "$service" "$env")
        
        local actual_ports
        actual_ports=$(calculate_service_ports "$service_uid" 3 | tr '\n' ',' | sed 's/,$//')
        
        if [[ "$actual_ports" == "$expected_ports" ]]; then
            echo "✓ $service:$env ports = $actual_ports"
        else
            echo "✗ $service:$env ports = $actual_ports (expected $expected_ports)"
        fi
    done
}

test_port_conflicts() {
    echo "Testing port conflict detection..."
    
    # Generate all service port ranges
    local services=("station:prod" "station:test" "gazette:prod" "gazette:test")
    local all_ports=()
    
    for service_env in "${services[@]}"; do
        IFS=':' read -r service env <<< "$service_env"
        local service_uid
        service_uid=$(calculate_service_uid "$service" "$env")
        
        # Get first 3 ports for this service
        local ports
        ports=$(calculate_service_ports "$service_uid" 3)
        all_ports+=($ports)
    done
    
    # Check for duplicates
    local unique_ports
    unique_ports=$(printf '%s\n' "${all_ports[@]}" | sort -n | uniq)
    local total_ports=${#all_ports[@]}
    local unique_count
    unique_count=$(echo "$unique_ports" | wc -l)
    
    if [[ $total_ports -eq $unique_count ]]; then
        echo "✓ No port conflicts detected ($total_ports unique ports)"
    else
        echo "✗ Port conflicts found! $total_ports ports, $unique_count unique"
        echo "Duplicate ports:"
        printf '%s\n' "${all_ports[@]}" | sort -n | uniq -d
    fi
}

main() {
    echo "Machine Setup Function Tests"
    echo "============================"
    echo
    
    test_sheet_numbers
    echo
    
    test_uid_calculation
    echo
    
    test_nfs_path_generation  
    echo
    
    test_xdg_paths
    echo
    
    test_port_allocation
    echo
    
    test_port_conflicts
    echo
    
    echo "All function tests completed!"
}

main "$@"