from fastapi import APIRouter
from backend.services.data_service import data_service
from backend.utils.common import make_response
from backend.schemas.schemas import RegisterRequest, AddUrlsRequest, PendingRequest, SearchRequest

router = APIRouter(prefix="/api/v1", tags=["V1 Manual"])

@router.post("/register")
async def register(req: RegisterRequest):
    try:
        res = data_service.register_task(req.url)
        return make_response(1, res.pop("msg", "Success"), res)
    except Exception as e:
        return make_response(0, str(e))

@router.post("/add_urls")
async def add_urls(req: AddUrlsRequest):
    try:
        res = data_service.add_urls(req.task_id, req.urls_obj["urls"])
        return make_response(1, res.pop("msg", "Success"), res)
    except Exception as e:
        return make_response(0, str(e))

@router.post("/search")
async def search_manual(req: SearchRequest):
    try:
        res = data_service.search(req.query_embedding['vector'], req.task_id, req.limit)
        return make_response(1, res.pop("msg", "Success"), res)
    except Exception as e:
        return make_response(0, str(e))