"""Dataplane-local integer enumerations for hot-path array indexing.

These mirror the control-plane enums but use IntEnum exclusively so they
can be used directly as array indices with zero overhead.
"""

from enum import IntEnum


class ModeId(IntEnum):
    AUTO = 0
    FAST = 1
    EXPERT = 2
    HEAVY = 3
    GROK_4_3 = 4


class PoolId(IntEnum):
    BASIC = 0
    SUPER = 1
    HEAVY = 2


class StatusId(IntEnum):
    ACTIVE = 0
    COOLING = 1
    EXPIRED = 2
    DISABLED = 3
    DELETED = 4


# Map pool string → PoolId integer (used during sync from control plane).
POOL_STR_TO_ID: dict[str, int] = {
    "basic": int(PoolId.BASIC),
    "super": int(PoolId.SUPER),
    "heavy": int(PoolId.HEAVY),
}

POOL_ID_TO_STR: dict[int, str] = {v: k for k, v in POOL_STR_TO_ID.items()}

STATUS_STR_TO_ID: dict[str, int] = {
    "active": int(StatusId.ACTIVE),
    "cooling": int(StatusId.COOLING),
    "expired": int(StatusId.EXPIRED),
    "disabled": int(StatusId.DISABLED),
}

ALL_MODE_IDS: tuple[int, ...] = (
    int(ModeId.AUTO),
    int(ModeId.FAST),
    int(ModeId.EXPERT),
    int(ModeId.HEAVY),
    int(ModeId.GROK_4_3),
)

__all__ = [
    "ModeId",
    "PoolId",
    "StatusId",
    "POOL_STR_TO_ID",
    "POOL_ID_TO_STR",
    "STATUS_STR_TO_ID",
    "ALL_MODE_IDS",
]