#!/bin/bash

set -e

echo "=== ORLY Policy Test Script ==="
echo ""

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

# Get the directory where this script is located
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
# Get the repository root (two levels up from scripts/docker-policy)
REPO_ROOT="$( cd "$SCRIPT_DIR/../.." && pwd )"

echo "Script directory: $SCRIPT_DIR"
echo "Repository root: $REPO_ROOT"
echo ""

echo -e "${YELLOW}Step 1: Building ORLY binary on host...${NC}"
cd "$REPO_ROOT" && CGO_ENABLED=0 go build -o orly

echo ""
echo -e "${YELLOW}Step 2: Copying files to test directory...${NC}"
cp "$REPO_ROOT/orly" "$SCRIPT_DIR/"
cp "$REPO_ROOT/pkg/crypto/p8k/libsecp256k1.so" "$SCRIPT_DIR/"

echo ""
echo -e "${YELLOW}Step 3: Cleaning up old containers...${NC}"
cd "$SCRIPT_DIR" && docker-compose down -v 2>/dev/null || true

echo ""
echo -e "${YELLOW}Step 4: Building Docker image...${NC}"
cd "$SCRIPT_DIR" && docker-compose build

echo ""
echo -e "${YELLOW}Step 5: Starting ORLY relay container...${NC}"
cd "$SCRIPT_DIR" && docker-compose up -d

echo ""
echo -e "${YELLOW}Step 6: Waiting for relay to start (15 seconds)...${NC}"
sleep 15

echo ""
echo -e "${YELLOW}Step 7: Checking relay logs...${NC}"
docker logs orly-policy-test 2>&1 | tail -20

echo ""
echo -e "${YELLOW}Step 8: Sending test event to relay...${NC}"

# Install websocat if not available
if ! command -v websocat &> /dev/null; then
    echo "websocat not found. Installing..."
    wget -qO- https://github.com/vi/websocat/releases/download/v1.12.0/websocat.x86_64-unknown-linux-musl -O /tmp/websocat
    chmod +x /tmp/websocat
    WEBSOCAT="/tmp/websocat"
else
    WEBSOCAT="websocat"
fi

# Check which port the relay is listening on
RELAY_PORT=$(docker logs orly-policy-test 2>&1 | grep "starting listener" | grep -oP ':\K[0-9]+' | head -1)
if [ -z "$RELAY_PORT" ]; then
    RELAY_PORT="8777"
fi
echo "Relay is listening on port: $RELAY_PORT"

# Generate a test event with a properly formatted (but invalid) signature
# The policy script should still receive this event even if validation fails
TIMESTAMP=$(date +%s)
TEST_EVENT='["EVENT",{"id":"aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa","pubkey":"4db2c42f3c02079dd6feae3f88f6c8693940a00ade3cc8e5d72050bd6e577cd5","created_at":'$TIMESTAMP',"kind":1,"tags":[],"content":"Test event for policy validation","sig":"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"}]'

echo "Sending test event..."
echo "$TEST_EVENT" | timeout 5 $WEBSOCAT ws://localhost:$RELAY_PORT 2>&1 || echo "Connection attempt completed"

echo ""
echo -e "${YELLOW}Step 9: Waiting for policy script to execute (5 seconds)...${NC}"
sleep 5

echo ""
echo -e "${YELLOW}Step 10: Checking if cs-policy.js created output file...${NC}"

# Check if the output file exists in the container
if docker exec orly-policy-test test -f /home/orly/cs-policy-output.txt; then
    echo -e "${GREEN}✓ SUCCESS: cs-policy-output.txt file exists!${NC}"
    echo ""
    echo "Output file contents:"
    docker exec orly-policy-test cat /home/orly/cs-policy-output.txt
    echo ""
    echo -e "${GREEN}✓ Policy script is working correctly!${NC}"
    EXIT_CODE=0
else
    echo -e "${RED}✗ FAILURE: cs-policy-output.txt file not found!${NC}"
    echo ""
    echo "Checking relay logs for errors:"
    docker logs orly-policy-test 2>&1 | grep -i policy || echo "No policy-related logs found"
    EXIT_CODE=1
fi

echo ""
echo -e "${YELLOW}Step 11: Additional debugging info...${NC}"
echo "Files in /home/orly directory:"
docker exec orly-policy-test ls -la /home/orly/

echo ""
echo "Policy configuration:"
docker exec orly-policy-test cat /home/orly/.config/orly/policy.json || echo "Policy config not found"

echo ""
echo "=== Test Complete ==="
echo ""
echo "To view logs: docker logs orly-policy-test"
echo "To stop container: cd scripts/docker-policy && docker-compose down"
echo "To clean up: cd scripts/docker-policy && docker-compose down -v"

exit $EXIT_CODE