64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Literal
|
|
|
|
ItemType = Literal["radio_station", "dice", "wheel", "clock"]
|
|
CLOCK_DEFAULT_TIME_ZONE = "America/Detroit"
|
|
CLOCK_TIME_ZONE_OPTIONS: tuple[str, ...] = (
|
|
"America/Detroit",
|
|
"America/New_York",
|
|
"America/Indiana/Indianapolis",
|
|
"America/Kentucky/Louisville",
|
|
)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ItemDefinition:
|
|
default_title: str
|
|
capabilities: tuple[str, ...]
|
|
emit_sound: str | None
|
|
default_params: dict
|
|
use_cooldown_ms: int = 1000
|
|
|
|
|
|
ITEM_DEFINITIONS: dict[ItemType, ItemDefinition] = {
|
|
"radio_station": ItemDefinition(
|
|
default_title="radio",
|
|
capabilities=("editable", "carryable", "deletable", "usable"),
|
|
emit_sound=None,
|
|
default_params={"streamUrl": "", "enabled": True, "channel": "stereo", "volume": 50, "effect": "off", "effectValue": 50},
|
|
),
|
|
"dice": ItemDefinition(
|
|
default_title="Dice",
|
|
capabilities=("editable", "carryable", "deletable", "usable"),
|
|
emit_sound="sounds/roll.ogg",
|
|
default_params={"sides": 6, "number": 2},
|
|
),
|
|
"wheel": ItemDefinition(
|
|
default_title="wheel",
|
|
capabilities=("editable", "carryable", "deletable", "usable"),
|
|
emit_sound="sounds/spin.ogg",
|
|
default_params={"spaces": "yes, no"},
|
|
use_cooldown_ms=4000,
|
|
),
|
|
"clock": ItemDefinition(
|
|
default_title="clock",
|
|
capabilities=("editable", "carryable", "deletable", "usable"),
|
|
emit_sound="sounds/clock.ogg",
|
|
default_params={"timeZone": CLOCK_DEFAULT_TIME_ZONE, "use24Hour": False},
|
|
),
|
|
}
|
|
|
|
|
|
def get_item_definition(item_type: ItemType) -> ItemDefinition:
|
|
return ITEM_DEFINITIONS[item_type]
|
|
|
|
|
|
def get_item_use_cooldown_ms(item_type: ItemType) -> int:
|
|
definition = get_item_definition(item_type)
|
|
cooldown_ms = definition.use_cooldown_ms
|
|
if isinstance(cooldown_ms, int) and cooldown_ms > 0:
|
|
return cooldown_ms
|
|
return 1000
|