"""Operator handler for BackgroundJobs"""

from uuid import UUID
import traceback

from btrixcloud.utils import (
    str_to_date,
    dt_now,
)

from .models import MCDecoratorSyncData
from .baseoperator import BaseOperator


# ============================================================================
class BgJobOperator(BaseOperator):
    """BgJobOperator"""

    def init_routes(self, app):
        """init routes for this operator"""

        # nop, but needed for metacontroller
        @app.post("/op/backgroundjob/sync")
        async def mc_sync_background_jobs():
            return {"attachments": []}

        @app.post("/op/backgroundjob/finalize")
        async def mc_finalize_background_jobs(data: MCDecoratorSyncData):
            return await self.finalize_background_job(data)

    async def finalize_background_job(self, data: MCDecoratorSyncData) -> dict:
        """handle finished background job"""

        metadata = data.object["metadata"]
        labels: dict[str, str] = metadata.get("labels", {})
        oid: str = labels.get("btrix.org") or ""
        job_type: str = labels.get("job_type") or ""
        job_id: str = labels.get("job_id") or metadata.get("name")

        status = data.object["status"]
        spec = data.object["spec"]
        success = status.get("succeeded") == spec.get("parallelism")
        if not success:
            print(
                "Succeeded: {status.get('succeeded')}, Num Pods: {spec.get('parallelism')}"
            )
        completion_time = status.get("completionTime")

        finalized = True

        finished = None
        if completion_time:
            finished = str_to_date(completion_time)
        if not finished:
            finished = dt_now()

        try:
            org_id = UUID(oid)
        # pylint: disable=broad-except
        except Exception:
            org_id = None

        try:
            await self.background_job_ops.job_finished(
                job_id, job_type, success=success, finished=finished, oid=org_id
            )
            # print(
            #    f"{job_type} background job completed: success: {success}, {job_id}",
            #    flush=True,
            # )

        # pylint: disable=broad-except
        except Exception:
            print("Update Background Job Error", flush=True)
            traceback.print_exc()

        return {"attachments": [], "finalized": finalized}