import os
import socket
import json
import logging
import httpx
from tenacity import retry, stop_after_attempt, wait_exponential

from src.schemas.search_request import SearchRequest
from src.config import config

logger = logging.getLogger(__name__)

SPACE_ID = config.app.id
API_ENDPOINT = config.tracking_db.api_endpoint
ANON_KEY = config.tracking_db.anon_key
JWT_TOKEN = config.tracking_db.jwt_token

@retry(stop=stop_after_attempt(3), wait=wait_exponential(multiplier=1, min=1, max=5))
async def send_tracking_event_to_db(search_data: SearchRequest):

    headers = {
        "apikey": ANON_KEY.get_secret_value(),
        "Authorization": f"Bearer {JWT_TOKEN.get_secret_value()}",
        "Content-Type": "application/json",
        "Prefer": "return=minimal"
    }

    data = search_data.model_dump()
    session_id = data.pop('session_id')

    payload = {
        "space_id": SPACE_ID,
        "session_id": session_id,
        "environment": "dev" if os.getenv("DEV_MODE", "").lower() == "true" else "prod",
        "server_host": socket.gethostname(),        
        "data": data
    }

    try:
        async with httpx.AsyncClient(timeout=10.0) as client:
            response = await client.post(API_ENDPOINT, headers=headers, json=payload)
            response.raise_for_status()
            logger.info("Successfully sent tracking event to DB!")
    except httpx.TimeoutException:
        logger.warning("Tracking request timed out - continuing anyway")
    except httpx.HTTPStatusError as e:
        logger.error(f"HTTP {e.response.status_code} error tracking event")
    except Exception as e:
        logger.error(f"Unexpected error in tracking: {e} with payload: {json.dumps(payload)}")