from __future__ import annotations

from enum import StrEnum


class CeleryRuntimeDomain(StrEnum):
    GENERATION_CREATE = "generation_create"
    GENERATION_POLL = "generation_poll"
    GENERATION_DOWNLOAD = "generation_download"
    MODULE_ASYNC = "module_async"
    SHOT_ANALYSIS = "shot_analysis"
    SHOT_SPLIT = "shot_split"
    VIDEO_UPSCALE = "video_upscale"
    PRIVATE_PORTRAIT_POLL = "private_portrait_poll"
    PRIVATE_PORTRAIT_DELETE = "private_portrait_delete"


class CeleryRuntimeState(StrEnum):
    ACTIVE = "active"
    WAITING_LOCK_EXPIRE = "waiting_lock_expire"
    WAITING_DB_LEASE = "waiting_db_lease"
    RECOVERY_CANDIDATE = "recovery_candidate"


class WorkerIdentityQuality(StrEnum):
    FULL = "full"
    INSTANCE_TOKEN_ONLY = "instance_token_only"
    SINGLE_PROCESS = "single_process"
    FALLBACK = "fallback"


class CeleryRuntimeEvent(StrEnum):
    WORKER_IDENTITY_INITIALIZED = "CELERY_WORKER_IDENTITY_INITIALIZED"
    WORKER_IDENTITY_FALLBACK = "CELERY_WORKER_IDENTITY_FALLBACK"
    WORKER_REGISTERED = "CELERY_WORKER_REGISTERED"
    WORKER_HEARTBEAT_LOST = "CELERY_WORKER_HEARTBEAT_LOST"
    WORKER_DUPLICATE_NAME = "CELERY_WORKER_DUPLICATE_NAME"
    WORKER_STALE_INSTANCE_FOUND = "CELERY_WORKER_STALE_INSTANCE_FOUND"
    WORKER_INSTANCE_RECOVERY_START = "CELERY_WORKER_INSTANCE_RECOVERY_START"
    WORKER_INSTANCE_RECOVERY_DONE = "CELERY_WORKER_INSTANCE_RECOVERY_DONE"
    WORKER_SHUTDOWN = "CELERY_WORKER_SHUTDOWN"
    RUNTIME_ACQUIRE_START = "CELERY_RUNTIME_ACQUIRE_START"
    RUNTIME_ACQUIRED = "CELERY_RUNTIME_ACQUIRED"
    RUNTIME_LOCK_HELD = "CELERY_RUNTIME_LOCK_HELD"
    RUNTIME_REDIS_UNAVAILABLE = "CELERY_RUNTIME_REDIS_UNAVAILABLE"
    RUNTIME_HEARTBEAT_LOST = "CELERY_RUNTIME_HEARTBEAT_LOST"
    RUNTIME_DB_LEASE_LOST = "CELERY_RUNTIME_DB_LEASE_LOST"
    RUNTIME_RESULT_DISCARDED = "CELERY_RUNTIME_RESULT_DISCARDED"
    RUNTIME_COMPLETED = "CELERY_RUNTIME_COMPLETED"
    RECOVERY_BARRIER_SKIPPED = "CELERY_RECOVERY_BARRIER_SKIPPED"
    RECOVERY_LIVE_LOCK_SKIPPED = "CELERY_RECOVERY_LIVE_LOCK_SKIPPED"
    RECOVERY_WAIT_DB_LEASE = "CELERY_RECOVERY_WAIT_DB_LEASE"
    RECOVERY_CAS_ACQUIRED = "CELERY_RECOVERY_CAS_ACQUIRED"
    RECOVERY_REQUEUED = "CELERY_RECOVERY_REQUEUED"
    REGISTRY_RECONCILE_DONE = "CELERY_REGISTRY_RECONCILE_DONE"
    REGISTRY_GC_DONE = "CELERY_REGISTRY_GC_DONE"