#!/bin/bash

RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m'

BASE_URL="http://localhost:8080"

log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
log_success() { echo -e "${GREEN}[SUCCESS]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }

extract_json() {
    echo "$1" | grep -o "\"$2\":\"[^\"]*\"" | head -1 | sed "s/\"$2\":\"//;s/\"$//"
}

extract_json_number() {
    echo "$1" | grep -o "\"$2\":[0-9]*" | head -1 | sed "s/\"$2\"://"
}

http_post() {
    local url=$1; local data=$2; local token=$3
    if [ -n "$token" ]; then
        curl -s -X POST "$url" -H "Content-Type: application/json" -H "Authorization: Bearer $token" -d "$data"
    else
        curl -s -X POST "$url" -H "Content-Type: application/json" -d "$data"
    fi
}

http_get() {
    local url=$1; local token=$2
    if [ -n "$token" ]; then
        curl -s -X GET "$url" -H "Authorization: Bearer $token"
    else
        curl -s -X GET "$url"
    fi
}

http_put() {
    local url=$1; local data=$2; local token=$3
    curl -s -X PUT "$url" -H "Content-Type: application/json" -H "Authorization: Bearer $token" -d "$data"
}

echo "============================================================"
echo "          EVENTHUB TICKETS API TEST SCRIPT"
echo "============================================================"
echo ""

log_info "Checking if server is running..."
if ! curl -s "$BASE_URL/health" | grep -q "ok"; then
    log_error "Server is not running"
    exit 1
fi
log_success "Server is running"

echo ""
log_info "============================================================"
log_info "STEP 1: Create test users"
log_info "============================================================"

# Админ (первый пользователь)
ADMIN_EMAIL="ticket_admin_$(date +%s)@example.com"
ADMIN_PASSWORD="admin123"

log_info "Creating admin user..."
response=$(http_post "$BASE_URL/v1/register" "{\"email\":\"$ADMIN_EMAIL\",\"password\":\"$ADMIN_PASSWORD\"}" "")
ADMIN_TOKEN=$(extract_json "$response" "token")
ADMIN_ID=$(extract_json "$response" "id")
log_success "Admin created"

# Обычный пользователь
USER_EMAIL="ticket_user_$(date +%s)@example.com"
USER_PASSWORD="user123"

log_info "Creating regular user..."
response=$(http_post "$BASE_URL/v1/register" "{\"email\":\"$USER_EMAIL\",\"password\":\"$USER_PASSWORD\"}" "")
USER_TOKEN=$(extract_json "$response" "token")
USER_ID=$(extract_json "$response" "id")
log_success "User created"

echo ""
log_info "============================================================"
log_info "TEST 1: Report error (user)"
log_info "============================================================"

log_info "User reporting error..."
response=$(http_post "$BASE_URL/v1/tickets" \
    "{\"error_message\":\"Test error occurred\",\"stacktrace\":\"line 1\\nline 2\",\"context\":{\"user_id\":\"$USER_ID\"}}" "$USER_TOKEN")
TICKET1_ID=$(extract_json "$response" "id")

if [ -n "$TICKET1_ID" ]; then
    log_success "Ticket created: $TICKET1_ID"
else
    log_error "Failed to create ticket: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 2: Report same error again (should increment count)"
log_info "============================================================"

log_info "User reporting same error..."
response=$(http_post "$BASE_URL/v1/tickets" \
    "{\"error_message\":\"Test error occurred\",\"stacktrace\":\"line 1\\nline 2\"}" "$USER_TOKEN")
COUNT=$(extract_json_number "$response" "count")

if [ "$COUNT" -eq 2 ]; then
    log_success "Ticket count incremented to $COUNT"
else
    log_error "Count should be 2, got $COUNT"
fi

echo ""
log_info "============================================================"
log_info "TEST 3: Report different error"
log_info "============================================================"

log_info "User reporting different error..."
response=$(http_post "$BASE_URL/v1/tickets" \
    "{\"error_message\":\"Another error\"}" "$USER_TOKEN")
TICKET2_ID=$(extract_json "$response" "id")

if [ -n "$TICKET2_ID" ] && [ "$TICKET2_ID" != "$TICKET1_ID" ]; then
    log_success "New ticket created: $TICKET2_ID"
else
    log_error "Failed to create new ticket"
fi

echo ""
log_info "============================================================"
log_info "TEST 4: Admin views all tickets"
log_info "============================================================"

log_info "Admin getting all tickets..."
response=$(http_get "$BASE_URL/v1/admin/tickets" "$ADMIN_TOKEN")
TICKET_COUNT=$(echo "$response" | grep -o "\"id\"" | wc -l)

if [ "$TICKET_COUNT" -eq 2 ]; then
    log_success "Admin sees $TICKET_COUNT tickets"
else
    log_error "Admin should see 2 tickets, found $TICKET_COUNT"
fi

echo ""
log_info "============================================================"
log_info "TEST 5: User cannot view tickets"
log_info "============================================================"

log_info "User trying to view tickets..."
response=$(http_get "$BASE_URL/v1/admin/tickets" "$USER_TOKEN")

if echo "$response" | grep -q "Admin access required"; then
    log_success "User correctly denied access"
else
    log_error "User should be denied: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 6: Admin views tickets by status"
log_info "============================================================"

log_info "Admin getting open tickets..."
response=$(http_get "$BASE_URL/v1/admin/tickets?status=open" "$ADMIN_TOKEN")
OPEN_COUNT=$(echo "$response" | grep -o "\"id\"" | wc -l)

if [ "$OPEN_COUNT" -eq 2 ]; then
    log_success "Found $OPEN_COUNT open tickets"
else
    log_error "Should find 2 open tickets, found $OPEN_COUNT"
fi

echo ""
log_info "============================================================"
log_info "TEST 7: Admin updates ticket status"
log_info "============================================================"

log_info "Admin marking ticket as in_progress..."
response=$(http_put "$BASE_URL/v1/admin/tickets/$TICKET1_ID" \
    "{\"action\":\"status\",\"status\":\"in_progress\"}" "$ADMIN_TOKEN")

STATUS=$(echo "$response" | grep -o "\"status\":\"[^\"]*\"" | sed 's/"status":"//;s/"//')
if [ "$STATUS" = "in_progress" ]; then
    log_success "Ticket status updated to in_progress"
else
    log_error "Failed to update status: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 8: Admin assigns ticket"
log_info "============================================================"

log_info "Admin assigning ticket..."
response=$(http_put "$BASE_URL/v1/admin/tickets/$TICKET1_ID" \
    "{\"action\":\"assign\",\"admin_id\":\"$ADMIN_ID\"}" "$ADMIN_TOKEN")

ASSIGNED=$(echo "$response" | grep -o "\"assigned_to\":\"[^\"]*\"" | sed 's/"assigned_to":"//;s/"//')
if [ "$ASSIGNED" = "$ADMIN_ID" ]; then
    log_success "Ticket assigned to admin"
else
    log_error "Failed to assign ticket: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 9: Admin resolves ticket"
log_info "============================================================"

log_info "Admin resolving ticket..."
response=$(http_put "$BASE_URL/v1/admin/tickets/$TICKET1_ID" \
    "{\"action\":\"resolve\",\"note\":\"Fixed in version 1.0\"}" "$ADMIN_TOKEN")

STATUS=$(echo "$response" | grep -o "\"status\":\"[^\"]*\"" | sed 's/"status":"//;s/"//')
NOTE=$(echo "$response" | grep -o "\"resolution_note\":\"[^\"]*\"" | sed 's/"resolution_note":"//;s/"//')

if [ "$STATUS" = "resolved" ] && [ "$NOTE" = "Fixed in version 1.0" ]; then
    log_success "Ticket resolved with note"
else
    log_error "Failed to resolve ticket: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 10: Admin closes ticket"
log_info "============================================================"

log_info "Admin closing ticket..."
response=$(http_put "$BASE_URL/v1/admin/tickets/$TICKET1_ID" \
    "{\"action\":\"close\"}" "$ADMIN_TOKEN")

STATUS=$(echo "$response" | grep -o "\"status\":\"[^\"]*\"" | sed 's/"status":"//;s/"//')
if [ "$STATUS" = "closed" ]; then
    log_success "Ticket closed"
else
    log_error "Failed to close ticket: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 11: Admin views statistics"
log_info "============================================================"

log_info "Admin getting statistics..."
response=$(http_get "$BASE_URL/v1/admin/tickets/stats" "$ADMIN_TOKEN")

TOTAL=$(extract_json_number "$response" "total_tickets")
OPEN=$(extract_json_number "$response" "open")
CLOSED=$(extract_json_number "$response" "closed")

if [ "$TOTAL" -eq 2 ] && [ "$OPEN" -eq 1 ] && [ "$CLOSED" -eq 1 ]; then
    log_success "Statistics: total=$TOTAL, open=$OPEN, closed=$CLOSED"
else
    log_error "Statistics incorrect: $response"
fi

echo ""
log_info "============================================================"
log_info "TEST 12: Get single ticket"
log_info "============================================================"

log_info "Admin getting ticket $TICKET2_ID..."
response=$(http_get "$BASE_URL/v1/admin/tickets/$TICKET2_ID" "$ADMIN_TOKEN")

if echo "$response" | grep -q "$TICKET2_ID"; then
    log_success "Ticket retrieved"
else
    log_error "Failed to get ticket: $response"
fi

echo ""
echo "============================================================"
log_success "TICKETS API TESTS COMPLETED!"
echo "============================================================"
echo ""
echo "Summary of created resources:"
echo "  Admin: $ADMIN_EMAIL"
echo "  User: $USER_EMAIL"
echo "  Tickets: $TICKET1_ID, $TICKET2_ID"
echo ""