from enum import Enum


class CeleryQueue(str, Enum):
    GEN_CHATAPI_CREATE = "gen_chatapi_create"
    GEN_PROVIDER_POLL = "gen_provider_poll"
    GEN_RESULT_DOWNLOAD = "gen_result_download"
    GEN_VIDEO_UPSCALE_LOCAL = "gen_video_upscale_local"
    GEN_VIDEO_UPSCALE_REMOTE = "gen_video_upscale_remote"
    GEN_RECOVERY = "gen_recovery"
    GEN_PRIVATE_PORTRAIT = "gen_private_portrait"
    GEN_SHOT_ANALYSIS = "gen_shot_analysis"
    GEN_SHOT_SPLIT = "gen_shot_split"
    DEFAULT = "default"


class CeleryTaskName(str, Enum):
    CHATAPI_CREATE = "generation.chatapi_create_generation_task"
    POLL_GENERATION = "generation.poll_generation_task"
    DOWNLOAD_GENERATION_RESULT = "generation.download_generation_result_task"
    RECOVER_DOWNLOAD = "generation.recover_download_tasks_once"
    RECOVER_GENERATION = "generation.recover_generation_tasks_once"
    RECOVER_CREATE = "generation.recover_create_tasks_once"
    VIDEO_UPSCALE_EXECUTE_LOCAL = "video_upscale.execute_local"
    VIDEO_UPSCALE_SUBMIT_REMOTE = "video_upscale.submit_remote"
    VIDEO_UPSCALE_POLL_REMOTE = "video_upscale.poll_remote"
    VIDEO_UPSCALE_DOWNLOAD_REMOTE_RESULT = "video_upscale.download_remote_result"
    VIDEO_UPSCALE_FINALIZE = "video_upscale.finalize"
    VIDEO_UPSCALE_RECOVER = "video_upscale.recover_once"
    DISPATCH_DUE_POLL = "generation.dispatch_due_poll_tasks"
    STARTUP_RECOVERY = "recovery.startup_recovery_once"
    MODULE_ASYNC_RECOVERY = "module_async.recover_module_async_tasks_once"
    SHOT_SPLIT_RECOVERY = "shot_replicate.recover_split_tasks_once"
    SHOT_ANALYSIS_RECOVERY = "shot_replicate.recover_analysis_tasks_once"
    CELERY_RUNTIME_RECONCILE = "celery_runtime.reconcile_once"
    CELERY_RUNTIME_GC = "celery_runtime.registry_gc_once"
    SHOT_ANALYZE_ORIGINAL = "shot_replicate.analyze_original_video"
    SHOT_ANALYZE_CUSTOM_SEGMENT = "shot_replicate.analyze_custom_segment_video"
    SHOT_SPLIT_ONE = "shot_replicate.split_one_segment"
    PRIVATE_PORTRAIT_POLL_ASSET = "private_portrait.poll_asset_status"
    PRIVATE_PORTRAIT_SYNC_DUE_ASSETS = "private_portrait.sync_due_assets"
    PRIVATE_PORTRAIT_DELETE_ASSET = "private_portrait.delete_asset_remote"
    PRIVATE_PORTRAIT_DELETE_GROUP = "private_portrait.delete_group_remote"
    PRIVATE_PORTRAIT_DELETE_PROJECT = "private_portrait.delete_project_remote"
    PRIVATE_PORTRAIT_RECOVER_REMOTE_DELETES = "private_portrait.recover_remote_deletes"