""" entry point for K8s crawl job which manages the stateful crawl """

import datetime
from fastapi import FastAPI

from .utils import send_signal_to_pods
from .base_job import K8SJobMixin

from ..crawl_job import CrawlJob

app = FastAPI()


# =============================================================================
class K8SCrawlJob(K8SJobMixin, CrawlJob):
    """Crawl Job State"""

    async def _do_scale(self, new_scale):
        crawl = await self._get_crawl()
        if not crawl:
            raise RuntimeError("crawl_not_found")

        # if making scale smaller, ensure existing crawlers saved their data
        pods = []
        for inx in range(new_scale, crawl.spec.replicas):
            pods.append(
                await self.core_api.read_namespaced_pod(
                    name=f"crawl-{self.job_id}-{inx}",
                    namespace=self.namespace,
                )
            )

        if pods:
            await send_signal_to_pods(self.core_api_ws, self.namespace, pods, "SIGUSR2")

        crawl.spec.replicas = new_scale

        await self.apps_api.patch_namespaced_stateful_set(
            name=crawl.metadata.name, namespace=self.namespace, body=crawl
        )

    async def _rollover_restart(self):
        """patch existing crawl statefulset with new timestamp to force restart"""
        now = datetime.datetime.utcnow()
        now = str(now.isoformat("T") + "Z")
        patch_config = {
            "spec": {
                "template": {"metadata": {"annotations": {"btrix.restartedAt": now}}}
            }
        }

        await self.apps_api.patch_namespaced_stateful_set(
            name=f"crawl-{self.job_id}", namespace=self.namespace, body=patch_config
        )

        return {"success": True}

    async def _get_crawl(self):
        try:
            return await self.apps_api.read_namespaced_stateful_set(
                name=f"crawl-{self.job_id}",
                namespace=self.namespace,
            )
        # pylint: disable=bare-except
        except:
            return None

    def _get_crawl_scale(self, crawl):
        """get scale from crawl, if any"""

        return crawl.spec.replicas if crawl else 0

    async def _send_shutdown_signal(self, signame):
        pods = await self.core_api.list_namespaced_pod(
            namespace=self.namespace,
            label_selector=f"crawl={self.job_id},role=crawler",
        )

        return await send_signal_to_pods(
            self.core_api_ws, self.namespace, pods.items, signame
        )

    # pylint: disable=line-too-long
    @property
    def redis_url(self):
        return f"redis://redis-{self.job_id}-0.redis-{self.job_id}.{self.namespace}.svc.cluster.local/0"


# ============================================================================
@app.on_event("startup")
async def startup():
    """init on startup"""
    job = K8SCrawlJob()
    job.register_handlers(app)