import asyncio
import json
from datetime import datetime, timezone

from oom_analyzer import oom_analyzer


def lambda_handler(event, context):
    oom_event_id = event["kwargs"]["activity_id"]
    cluster_id = event["kwargs"]["cluster_id"]
    instance = event["kwargs"]["instance"]
    tenant_id = event["kwargs"]["tenant_id"]
    project_id = event["kwargs"]["project_id"]
    occurrence_ts = event["kwargs"]["occurrence_ts"]
    print(
        datetime.fromtimestamp(occurrence_ts, tz=timezone.utc).strftime("%Y-%m-%d %H:%M:%S"),
        "UTC : OOM event ",
        oom_event_id,
    )

    asyncio.run(
        oom_analyzer(
            oom_event_id,
            cluster_id,
            instance,
            occurrence_ts,
            tenant_id,
            project_id,
        )
    )


if __name__ == "__main__":
    event = {
        "kwargs": {
            "activity_id": "3b61bcad-968d-4f8e-9dc5-150d288353a3",
            "cluster_id": "1379661944629574082",
            "instance": "db-tidb-0",
            "occurrence_ts": 1666791974,
            "tenant_id": "1372813089198981282",
            "project_id": "1372813089196881283",
        }
    }
    lambda_handler(event, None)