#!/bin/bash

# Don't use set -e because we want to continue even if tests fail
set -uo pipefail

PROJECT="getpkg"
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
GETPKG="${SCRIPT_DIR}/output/${PROJECT}"
TEST_DIR="${SCRIPT_DIR}/test_temp"
TEST_TOOL_NAME="test-tool-$RANDOM"
TEST_ARCH=$(uname -m | sed 's/x86_64/x86_64/;s/arm64/aarch64/;s/aarch64/aarch64/')

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

# Test counters
TESTS_PASSED=0
TESTS_FAILED=0

# Function to print test results
print_test_result() {
    local test_name="$1"
    local result="$2"
    if [ "$result" -eq 0 ]; then
        echo -e "${GREEN}✓${NC} $test_name"
        ((TESTS_PASSED++))
    else
        echo -e "${RED}✗${NC} $test_name"
        ((TESTS_FAILED++))
    fi
}

# Function to cleanup test artifacts
CLEANED=0
cleanup() {
    if [ "$CLEANED" -eq 1 ]; then
        return
    fi
    CLEANED=1
    echo -e "\n${YELLOW}Cleaning up test artifacts...${NC}"
    
    # Remove local test directories
    rm -rf "$TEST_DIR"
    rm -rf ~/.config/getpkg/"${TEST_TOOL_NAME}.json" 2>/dev/null || true
    rm -rf ~/.config/getpkg/"${TEST_TOOL_NAME}-noarch.json" 2>/dev/null || true
    rm -rf ~/.getpkg/"${TEST_TOOL_NAME}" 2>/dev/null || true
    rm -rf ~/.getpkg/"${TEST_TOOL_NAME}-noarch" 2>/dev/null || true
    rm -rf ~/.local/bin/getpkg/"${TEST_TOOL_NAME}" 2>/dev/null || true
    rm -rf ~/.local/bin/getpkg/"${TEST_TOOL_NAME}-noarch" 2>/dev/null || true
    
    # Remove test tool from bashrc_getpkg if it exists
    # if [ -f ~/.bashrc_getpkg ]; then
    #     sed -i "/${TEST_TOOL_NAME}/d" ~/.bashrc_getpkg 2>/dev/null || true
    # fi
    $GETPKG uninstall "$TEST_TOOL_NAME" 2>/dev/null || true
    $GETPKG uninstall "${TEST_TOOL_NAME}-noarch" 2>/dev/null || true
    
    # Clean up from getpkg.xyz if we have write access
    if [ -n "${SOS_WRITE_TOKEN:-}" ]; then
        echo "Attempting to clean up test objects from getpkg.xyz..."
        
        # Clean up main architecture variant
        $GETPKG unpublish "${TEST_TOOL_NAME}:${TEST_ARCH}" 2>/dev/null || true
        
        # Clean up noarch variant
        $GETPKG unpublish "${TEST_TOOL_NAME}-noarch:universal" 2>/dev/null || true
        
        echo "Cleaned up test tools from getpkg.xyz"
    else
        echo "Note: SOS_WRITE_TOKEN not set, cannot clean up remote test objects"
    fi
}

# Set up trap to ensure cleanup runs
trap cleanup EXIT

# Create test directory
mkdir -p "$TEST_DIR"

echo -e "${YELLOW}Running getpkg tests...${NC}\n"

# Check if getpkg binary exists
if [ ! -f "$GETPKG" ]; then
    echo -e "${RED}Error: getpkg binary not found at $GETPKG${NC}"
    echo "Please run ./build.sh first to build getpkg"
    exit 1
fi

if [ ! -x "$GETPKG" ]; then
    echo -e "${RED}Error: getpkg binary is not executable${NC}"
    exit 1
fi

echo "Using getpkg binary: $GETPKG"

# Test 1: Version command
echo "Test 1: Version command"
VERSION=$(timeout 3 "$GETPKG" version 2>&1) || VERSION=""
if [[ "$VERSION" =~ ^[0-9]{4}\.[0-9]{4}\.[0-9]{4}$ ]]; then
    print_test_result "Version format (YYYY.MMDD.HHMM)" 0
else
    print_test_result "Version format (YYYY.MMDD.HHMM)" 1
fi

# Test 2: Help command
echo -e "\nTest 2: Help command"
HELP_OUTPUT=$(timeout 3 "$GETPKG" help 2>&1) || HELP_OUTPUT=""
if [[ "$HELP_OUTPUT" =~ getpkg\ -\ Package\ manager\ for\ dropshell\ tools ]] || [[ "$HELP_OUTPUT" =~ USAGE: ]]; then
    print_test_result "Help command output" 0
else
    print_test_result "Help command output" 1
fi

# Test 3: Autocomplete command
echo -e "\nTest 3: Autocomplete command"
AUTOCOMPLETE_OUTPUT=$(timeout 3 "$GETPKG" autocomplete 2>&1) || AUTOCOMPLETE_OUTPUT=""
if [[ "$AUTOCOMPLETE_OUTPUT" =~ install ]] && [[ "$AUTOCOMPLETE_OUTPUT" =~ publish ]]; then
    print_test_result "Autocomplete command output" 0
else
    print_test_result "Autocomplete command output" 1
fi

# Test 4: Hash command
echo -e "\nTest 4: Hash command"

# Test 4a: Hash command with missing argument
echo "Test 4a: Hash command error handling"
HASH_ERROR_OUTPUT=$("$GETPKG" hash 2>&1) || true
if [[ "$HASH_ERROR_OUTPUT" =~ "Usage: getpkg hash" ]]; then
    print_test_result "Hash command error handling (missing argument)" 0
else
    print_test_result "Hash command error handling (missing argument)" 1
fi

# Test 4b: Hash command with nonexistent file
echo "Test 4b: Hash command with nonexistent file"
HASH_NONEXISTENT_OUTPUT=$("$GETPKG" hash nonexistent_file.txt 2>&1) || true
if [[ "$HASH_NONEXISTENT_OUTPUT" =~ "Error: Path does not exist" ]]; then
    print_test_result "Hash command error handling (nonexistent file)" 0
else
    print_test_result "Hash command error handling (nonexistent file)" 1
fi

# Test 4c: Hash command with test file
echo "Test 4c: Hash command with test file"
TEST_FILE="${TEST_DIR}/test_hash_file.txt"
echo "test content for hashing" > "$TEST_FILE"
HASH_FILE_OUTPUT=$("$GETPKG" hash "$TEST_FILE" 2>&1) || HASH_FILE_OUTPUT=""
if [[ "$HASH_FILE_OUTPUT" =~ ^[0-9]+$ ]]; then
    print_test_result "Hash command with file (numeric output)" 0
    
    # Test that the hash is consistent
    HASH_FILE_OUTPUT2=$("$GETPKG" hash "$TEST_FILE" 2>&1) || HASH_FILE_OUTPUT2=""
    if [ "$HASH_FILE_OUTPUT" = "$HASH_FILE_OUTPUT2" ]; then
        print_test_result "Hash command consistency (same file, same hash)" 0
    else
        print_test_result "Hash command consistency (same file, same hash)" 1
    fi
else
    print_test_result "Hash command with file (numeric output)" 1
    print_test_result "Hash command consistency (same file, same hash)" 1
fi

# Test 4d: Hash command with directory
echo "Test 4d: Hash command with directory"
TEST_HASH_DIR="${TEST_DIR}/test_hash_dir"
mkdir -p "$TEST_HASH_DIR"
echo "dir content 1" > "$TEST_HASH_DIR/file1.txt"
echo "dir content 2" > "$TEST_HASH_DIR/file2.txt"
HASH_DIR_OUTPUT=$("$GETPKG" hash "$TEST_HASH_DIR" 2>&1) || HASH_DIR_OUTPUT=""
if [[ "$HASH_DIR_OUTPUT" =~ ^[0-9]+$ ]]; then
    print_test_result "Hash command with directory (numeric output)" 0
    
    # Test that directory hash is different from file hash
    if [ "$HASH_DIR_OUTPUT" != "$HASH_FILE_OUTPUT" ]; then
        print_test_result "Hash command different for file vs directory" 0
    else
        print_test_result "Hash command different for file vs directory" 1
    fi
else
    print_test_result "Hash command with directory (numeric output)" 1
    print_test_result "Hash command different for file vs directory" 1
fi

# Test 4e: Hash command compatibility with tr (sos script format)
echo "Test 4e: Hash command compatibility with tr command"
LOCALHASH=$("$GETPKG" hash "$TEST_FILE" | tr -d '[:space:]') || LOCALHASH=""
if [[ "$LOCALHASH" =~ ^[0-9]+$ ]] && [ ${#LOCALHASH} -gt 0 ]; then
    print_test_result "Hash command compatible with tr -d '[:space:]'" 0
else
    print_test_result "Hash command compatible with tr -d '[:space:]'" 1
fi

# Test 4f: Hash command appears in autocomplete
echo "Test 4f: Hash command in autocomplete"
if [[ "$AUTOCOMPLETE_OUTPUT" =~ hash ]]; then
    print_test_result "Hash command appears in autocomplete" 0
else
    print_test_result "Hash command appears in autocomplete" 1
fi

# Test 4g: Hash command appears in help
echo "Test 4g: Hash command in help"
if [[ "$HELP_OUTPUT" =~ hash.*file_or_directory ]] && [[ "$HELP_OUTPUT" =~ "Outputs raw hash value to stdout" ]]; then
    print_test_result "Hash command appears in help" 0
else
    print_test_result "Hash command appears in help" 1
fi

# Test 5: Create command
echo -e "\nTest 5: Create command"
CREATE_DIR="${TEST_DIR}/${TEST_TOOL_NAME}"
timeout 3 "$GETPKG" create "$TEST_TOOL_NAME" "$CREATE_DIR" >/dev/null 2>&1 || true
if [ -d "$CREATE_DIR" ] && [ -f "$CREATE_DIR/setup_script.sh" ]; then
    print_test_result "Create tool directory" 0
else
    print_test_result "Create tool directory" 1
fi

# Test 6: Create test tool binary
echo -e "\nTest 6: Create test tool"
cat > "$CREATE_DIR/${TEST_TOOL_NAME}" << 'EOF'
#!/bin/bash
case "$1" in
    version)
        echo "1.0.0"
        ;;
    autocomplete)
        echo "subcommand1"
        echo "subcommand2"
        ;;
    *)
        echo "Test tool - $1"
        ;;
esac
EOF
chmod +x "$CREATE_DIR/${TEST_TOOL_NAME}"


if [ -x "$CREATE_DIR/${TEST_TOOL_NAME}" ]; then
    print_test_result "Create test tool binary" 0
else
    print_test_result "Create test tool binary" 1
fi

# Test 6b: Test basic publish functionality
echo -e "\nTest 6b: Testing basic publish command"
# Test with a dummy token to see error message
DUMMY_OUTPUT=$(timeout 3 "$GETPKG" publish "dummy:x86_64" "$CREATE_DIR" 2>&1 <<< "") || true
echo "Dummy publish output: $DUMMY_OUTPUT"

# Test 7: Publish command (requires SOS_WRITE_TOKEN)
if [ -n "${SOS_WRITE_TOKEN:-}" ]; then
    echo -e "\nTest 7: Publish command with ARCH"
    echo "SOS_WRITE_TOKEN is set: ${#SOS_WRITE_TOKEN} characters"
    echo "Publishing ${TEST_TOOL_NAME}:${TEST_ARCH} from $CREATE_DIR"
    echo "Directory contents:"
    ls -la "$CREATE_DIR"
    echo "Running: $GETPKG publish ${TEST_TOOL_NAME}:${TEST_ARCH} $CREATE_DIR"
    # Make sure the token is exported for the subprocess
    export SOS_WRITE_TOKEN
    # Run publish command and capture both stdout and stderr
    PUBLISH_OUTPUT=$(timeout 3 "$GETPKG" publish "${TEST_TOOL_NAME}:${TEST_ARCH}" "$CREATE_DIR" 2>&1)
    PUBLISH_EXIT_CODE=$?
    echo "Publish exit code: $PUBLISH_EXIT_CODE"
    echo "Publish output: $PUBLISH_OUTPUT"
    
    # Check if the output indicates a crash or missing dependencies
    if [ -z "$PUBLISH_OUTPUT" ] && [ $PUBLISH_EXIT_CODE -ne 0 ]; then
        echo "Publish command failed with no output. Checking for missing dependencies..."
        ldd "$GETPKG" 2>&1 | grep "not found" || echo "All dependencies found"
    fi
    if [[ "$PUBLISH_OUTPUT" =~ Published! ]] && [[ "$PUBLISH_OUTPUT" =~ URL: ]] && [[ "$PUBLISH_OUTPUT" =~ Hash: ]]; then
        print_test_result "Publish tool with ARCH to getpkg.xyz" 0
        
        # Extract hash for later cleanup
        PUBLISHED_HASH=$(echo "$PUBLISH_OUTPUT" | grep "Hash:" | cut -d' ' -f2)
        echo "Published hash: $PUBLISHED_HASH"
        
        # Test 8: Check if published tool exists
        echo -e "\nTest 8: Check published tool exists"
        EXISTS_CHECK=$(curl -s "https://getpkg.xyz/exists/${TEST_TOOL_NAME}:${TEST_ARCH}" 2>/dev/null || echo "error")
        if [[ "$EXISTS_CHECK" != "error" ]] && [[ "$EXISTS_CHECK" != "false" ]]; then
            print_test_result "Published tool exists on server" 0
        else
            print_test_result "Published tool exists on server" 1
        fi
        
        # Test 9: Install command
        echo -e "\nTest 9: Install command"
        INSTALL_OUTPUT=$(timeout 3 "$GETPKG" install "$TEST_TOOL_NAME" 2>&1) || INSTALL_OUTPUT=""
        if [[ "$INSTALL_OUTPUT" =~ Installed\ ${TEST_TOOL_NAME}\ successfully ]] || [[ "$INSTALL_OUTPUT" =~ ${TEST_TOOL_NAME}\ is\ already\ up\ to\ date ]]; then
            print_test_result "Install tool from getpkg.xyz" 0
            
            # Test 10: Check installed files
            echo -e "\nTest 10: Check installed files"
            if [ -f ~/.config/getpkg/"${TEST_TOOL_NAME}.json" ] && [ -d ~/.getpkg/"${TEST_TOOL_NAME}" ] && [ -L ~/.local/bin/getpkg/"${TEST_TOOL_NAME}" ]; then
                print_test_result "Tool files installed correctly" 0
            else
                print_test_result "Tool files installed correctly" 1
            fi
            
            # Test 11: Check bashrc_getpkg modifications
            echo -e "\nTest 11: Check bashrc modifications"
            if [ -f ~/.bashrc_getpkg ] && grep -q "${TEST_TOOL_NAME}" ~/.bashrc_getpkg; then
                print_test_result "Bashrc modifications for PATH and autocomplete" 0
            else
                print_test_result "Bashrc modifications for PATH and autocomplete" 1
            fi
            
            # Test 12: Reinstall after manual removal
            echo -e "\nTest 12: Reinstall after manual removal"
            # First remove the tool
            rm -rf ~/.getpkg/"${TEST_TOOL_NAME}"
            rm -rf ~/.local/bin/getpkg/"${TEST_TOOL_NAME}"  
            rm -f ~/.config/getpkg/"${TEST_TOOL_NAME}.json"
            
            REINSTALL_OUTPUT=$(timeout 3 "$GETPKG" install "$TEST_TOOL_NAME" 2>&1) || REINSTALL_OUTPUT=""
            if [[ "$REINSTALL_OUTPUT" =~ Installed\ ${TEST_TOOL_NAME}\ successfully ]] || [[ "$REINSTALL_OUTPUT" =~ ${TEST_TOOL_NAME}\ is\ already\ up\ to\ date ]]; then
                print_test_result "Reinstall after manual removal" 0
            else
                print_test_result "Reinstall after manual removal" 1
            fi
            
            # Test 13: Update already installed tool (should say up to date)
            echo -e "\nTest 13: Update check for installed tool"
            UPDATE_OUTPUT=$(timeout 3 "$GETPKG" install "$TEST_TOOL_NAME" 2>&1) || UPDATE_OUTPUT=""
            if [[ "$UPDATE_OUTPUT" =~ ${TEST_TOOL_NAME}\ is\ already\ up\ to\ date ]]; then
                print_test_result "Update check recognizes up-to-date tool" 0
            else
                print_test_result "Update check recognizes up-to-date tool" 1
            fi
            
        else
            print_test_result "Install tool from getpkg.xyz" 1
            # Skip dependent tests
            print_test_result "Tool files installed correctly" 1
            print_test_result "Bashrc modifications for PATH and autocomplete" 1
            print_test_result "Reinstall after manual removal" 1
            print_test_result "Update check recognizes up-to-date tool" 1
        fi
        
    else
        print_test_result "Publish tool with ARCH to getpkg.xyz" 1
        # Skip dependent tests
        print_test_result "Published tool exists on server" 1
        print_test_result "Install tool from getpkg.xyz" 1
        print_test_result "Tool files installed correctly" 1
        print_test_result "Bashrc modifications for PATH and autocomplete" 1
        print_test_result "Reinstall after manual removal" 1
        print_test_result "Update check recognizes up-to-date tool" 1
    fi
    
    # Test 13b: Publish without ARCH (for cross-platform tools)
    echo -e "\nTest 13b: Publish without ARCH"
    TEST_TOOL_NOARCH="${TEST_TOOL_NAME}-noarch"
    mkdir -p "${TEST_DIR}/${TEST_TOOL_NOARCH}"
    cat > "${TEST_DIR}/${TEST_TOOL_NOARCH}/${TEST_TOOL_NOARCH}" << 'EOF'
#!/bin/bash
case "$1" in
    version)
        echo "1.0.0"
        ;;
    *)
        echo "Cross-platform test tool"
        ;;
esac
EOF
    chmod +x "${TEST_DIR}/${TEST_TOOL_NOARCH}/${TEST_TOOL_NOARCH}"
    
    PUBLISH_NOARCH_OUTPUT=$(timeout 3 "$GETPKG" publish "${TEST_TOOL_NOARCH}" "${TEST_DIR}/${TEST_TOOL_NOARCH}" 2>&1) || PUBLISH_NOARCH_OUTPUT=""
    if [[ "$PUBLISH_NOARCH_OUTPUT" =~ Published! ]] && [[ "$PUBLISH_NOARCH_OUTPUT" =~ URL: ]] && [[ "$PUBLISH_NOARCH_OUTPUT" =~ Hash: ]]; then
        print_test_result "Publish tool without ARCH" 0
        # Clean up the noarch tool
        NOARCH_HASH=$(curl -s "https://getpkg.xyz/hash/${TEST_TOOL_NOARCH}" 2>/dev/null || echo "")
        if [ -n "$NOARCH_HASH" ] && [ "$NOARCH_HASH" != "null" ] && [ "$NOARCH_HASH" != "Not found" ]; then
            curl -s -H "Authorization: Bearer ${SOS_WRITE_TOKEN}" \
                "https://getpkg.xyz/deleteobject?hash=${NOARCH_HASH}" >/dev/null 2>&1 || true
        fi
    else
        print_test_result "Publish tool without ARCH" 1
    fi
    
    # Test 13c: Install universal tool (arch fallback)
    echo -e "\nTest 13c: Install universal tool (arch fallback)"
    rm -rf ~/.config/getpkg/"${TEST_TOOL_NOARCH}.json" ~/.getpkg/"${TEST_TOOL_NOARCH}" ~/.local/bin/getpkg/"${TEST_TOOL_NOARCH}" 2>/dev/null || true
    FALLBACK_INSTALL_OUTPUT=$(timeout 3 "$GETPKG" install "${TEST_TOOL_NOARCH}" 2>&1) || FALLBACK_INSTALL_OUTPUT=""
    if [[ "$FALLBACK_INSTALL_OUTPUT" =~ Arch-specific\ version\ not\ found,\ trying\ universal\ version ]] && [[ "$FALLBACK_INSTALL_OUTPUT" =~ Installed\ ${TEST_TOOL_NOARCH}\ successfully ]]; then
        print_test_result "Install universal tool with arch fallback" 0
        
        # Test update check for universal tool
        UPDATE_UNIVERSAL_OUTPUT=$(timeout 3 "$GETPKG" install "${TEST_TOOL_NOARCH}" 2>&1) || UPDATE_UNIVERSAL_OUTPUT=""
        if [[ "$UPDATE_UNIVERSAL_OUTPUT" =~ ${TEST_TOOL_NOARCH}\ is\ already\ up\ to\ date ]]; then
            print_test_result "Update check for universal tool" 0
        else
            print_test_result "Update check for universal tool" 1
        fi
    else
        print_test_result "Install universal tool with arch fallback" 1
        print_test_result "Update check for universal tool" 1
    fi
else
    echo -e "\n${YELLOW}Skipping publish/install tests (SOS_WRITE_TOKEN not set)${NC}"
    echo "To run full tests, set SOS_WRITE_TOKEN environment variable"
fi

# Test 14: Invalid tool name validation
echo -e "\nTest 14: Invalid tool name validation"
INVALID_OUTPUT=$(timeout 3 "$GETPKG" install "../evil-tool" 2>&1)
INVALID_EXIT_CODE=$?
echo "Invalid tool output: $INVALID_OUTPUT"
echo "Invalid tool exit code: $INVALID_EXIT_CODE"
if [[ "$INVALID_OUTPUT" =~ Invalid\ tool\ name ]]; then
    print_test_result "Invalid tool name rejection" 0
else
    print_test_result "Invalid tool name rejection" 1
fi

# Test 15: Update command (if we have tools installed)
if [ -d ~/.config/getpkg ] && [ "$(find ~/.config/getpkg -name "*.json" -type f 2>/dev/null | wc -l)" -gt 0 ]; then
    echo -e "\nTest 15: Update command"
    UPDATE_ALL_OUTPUT=$(timeout 30 "$GETPKG" update 2>&1) || UPDATE_ALL_OUTPUT=""
    if [[ "$UPDATE_ALL_OUTPUT" =~ Update\ complete ]]; then
        print_test_result "Update all tools command" 0
    else
        print_test_result "Update all tools command" 1
    fi
else
    echo -e "\n${YELLOW}Skipping update all test (no tools installed)${NC}"
fi

cleanup

# Print summary
echo -e "\n${YELLOW}Test Summary:${NC}"
echo -e "Tests passed: ${GREEN}${TESTS_PASSED}${NC}"
echo -e "Tests failed: ${RED}${TESTS_FAILED}${NC}"

if [ "$TESTS_FAILED" -eq 0 ]; then
    echo -e "\n${GREEN}All tests passed!${NC}"
    exit 0
else
    echo -e "\n${RED}Some tests failed!${NC}"
    exit 1
fi