import json
from typing import Mapping, Any
from io import BytesIO

from backend.condition.condition_trigger_request import ConditionTriggerRequest
from backend.condition.lookups.uprn_lookup_s3 import UprnLookupS3
from backend.condition.processor import process_file
from utils.logger import setup_logger
from utils.s3 import read_io_from_s3


logger = setup_logger()


def handler(event: Mapping[str, Any], context: Any) -> None:

    for record in event.get("Records", []):
        try:
            body_dict = json.loads(record["body"])
            payload = ConditionTriggerRequest.model_validate(body_dict)

            if payload.uprn_lookup_file_bucket and payload.uprn_lookup_file_key:
                uprn_lookup = UprnLookupS3(
                    bucket=payload.uprn_lookup_file_bucket,
                    key=payload.uprn_lookup_file_key,
                )  # TODO: replace with postgres implementation
            else:
                uprn_lookup = None

            file_bytes: BytesIO = read_io_from_s3(
                bucket_name=payload.trigger_file_bucket,
                file_key=payload.trigger_file_key,
            )

            process_file(
                file_stream=file_bytes,
                file_type=payload.file_type,
                uprn_lookup=uprn_lookup,
            )

        except Exception as e:
            logger.error(f"Failed to process record: {e}")