105 lines
2.9 KiB
Python
105 lines
2.9 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/Anchorage",
|
|
"America/Argentina/Buenos_Aires",
|
|
"America/Chicago",
|
|
"America/Detroit",
|
|
"America/Halifax",
|
|
"America/Indiana/Indianapolis",
|
|
"America/Kentucky/Louisville",
|
|
"America/Los_Angeles",
|
|
"America/St_Johns",
|
|
"Asia/Bangkok",
|
|
"Asia/Dhaka",
|
|
"Asia/Dubai",
|
|
"Asia/Hong_Kong",
|
|
"Asia/Kabul",
|
|
"Asia/Karachi",
|
|
"Asia/Kathmandu",
|
|
"Asia/Kolkata",
|
|
"Asia/Seoul",
|
|
"Asia/Singapore",
|
|
"Asia/Tehran",
|
|
"Asia/Tokyo",
|
|
"Asia/Yangon",
|
|
"Atlantic/Azores",
|
|
"Atlantic/South_Georgia",
|
|
"Australia/Brisbane",
|
|
"Australia/Darwin",
|
|
"Australia/Eucla",
|
|
"Australia/Lord_Howe",
|
|
"Europe/Berlin",
|
|
"Europe/Helsinki",
|
|
"Europe/London",
|
|
"Europe/Moscow",
|
|
"Pacific/Auckland",
|
|
"Pacific/Chatham",
|
|
"Pacific/Honolulu",
|
|
"Pacific/Kiritimati",
|
|
"Pacific/Noumea",
|
|
"Pacific/Pago_Pago",
|
|
"Pacific/Apia",
|
|
"UTC",
|
|
)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ItemDefinition:
|
|
default_title: str
|
|
capabilities: tuple[str, ...]
|
|
use_sound: str | None
|
|
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"),
|
|
use_sound=None,
|
|
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"),
|
|
use_sound="sounds/roll.ogg",
|
|
emit_sound=None,
|
|
default_params={"sides": 6, "number": 2},
|
|
),
|
|
"wheel": ItemDefinition(
|
|
default_title="wheel",
|
|
capabilities=("editable", "carryable", "deletable", "usable"),
|
|
use_sound="sounds/spin.ogg",
|
|
emit_sound=None,
|
|
default_params={"spaces": "yes, no"},
|
|
use_cooldown_ms=4000,
|
|
),
|
|
"clock": ItemDefinition(
|
|
default_title="clock",
|
|
capabilities=("editable", "carryable", "deletable", "usable"),
|
|
use_sound=None,
|
|
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
|