from typing import Any

from backend.app.config import get_settings
from backend.magic_plan.magic_plan_client import MagicPlanClient
from backend.magic_plan.magic_plan_service import MagicPlanService
from backend.magic_plan.magic_plan_trigger_request import MagicPlanTriggerRequest
from backend.utils.subtasks import task_handler
from utils.logger import setup_logger

logger = setup_logger()


@task_handler()
def handler(body: dict[str, Any], context: Any) -> str:
    settings = get_settings()
    payload = MagicPlanTriggerRequest.model_validate(body)
    client = MagicPlanClient(
        customer_id=settings.MAGICPLAN_CUSTOMER_ID,
        api_key=settings.MAGICPLAN_API_KEY,
    )
    plan = MagicPlanService(client).run(payload.address, payload.uprn)
    logger.info("Saved MagicPlan plan uid=%s", plan.uid)
    return plan.uid


if __name__ == "__main__":
    event = {"Records": [{"body": '{"address": "2 Laburnum Way Bromley BR2 8BZ"}', "messageId": "local-test"}]}
    handler(event, None)