"""
Stealth Browser Stack - FastAPI on Port 8888
=============================================
This handles healthchecks and API tasks.
noVNC moved to 6080 to avoid conflict.
"""

import os
import asyncio
import logging
from datetime import datetime
from typing import Dict, Any, Optional

from fastapi import FastAPI, HTTPException
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

from agent import BrowserSession, validate_order

app = FastAPI(title="Stealth Browser Worker", version="2.5.0")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

browser_session: Optional[BrowserSession] = None


class TaskRequest(BaseModel):
    task_id: str
    task_type: str
    data: Dict[str, Any]


class TaskResponse(BaseModel):
    task_id: str
    status: str
    result: Optional[Dict[str, Any]] = None
    error: Optional[str] = None
    logs: Optional[list] = None
    execution_time_ms: int = 0


@app.get("/")
async def root():
    return {
        "service": "AI System Bridge",
        "status": "active",
        "timestamp": datetime.now().isoformat()
    }


@app.get("/health")
async def health():
    return {
        "status": "healthy",
        "display": os.environ.get("DISPLAY", ":99"),
        "browser": browser_session is not None and browser_session.is_active
    }


@app.post("/run-task", response_model=TaskResponse)
async def run_task(request: TaskRequest):
    global browser_session
    start_time = datetime.now()
    
    try:
        if request.task_type in ["validate_order", "validate_email"]:
            order_data = {**request.data, "task_id": request.task_id}
            result = await validate_order(order_data, browser_session)
            execution_time = int((datetime.now() - start_time).total_seconds() * 1000)
            
            return TaskResponse(
                task_id=request.task_id,
                status="success",
                result=result,
                logs=result.get("logs", []),
                execution_time_ms=execution_time
            )
        else:
            raise HTTPException(status_code=400, detail="Invalid task_type")
            
    except Exception as e:
        return TaskResponse(task_id=request.task_id, status="error", error=str(e))


@app.on_event("startup")
async def startup():
    global browser_session
    try:
        browser_session = await BrowserSession.create()
        logger.info("✅ Persistent browser session created!")
    except Exception as e:
        logger.error(f"⚠️ Browser fail: {e}")


if __name__ == "__main__":
    import uvicorn
    # Port 8888 for HF healthcheck
    uvicorn.run(app, host="0.0.0.0", port=8888)