"""Websocket signaling server for chat, presence, and item interactions.""" from __future__ import annotations import argparse import asyncio from datetime import datetime from importlib.metadata import PackageNotFoundError, version as package_version import json import logging import os import re import ssl import time import uuid from pathlib import Path from typing import Literal from zoneinfo import ZoneInfo from pydantic import ValidationError, TypeAdapter from websockets.asyncio.server import ServerConnection, serve from .client import ClientConnection from .config import load_config from .item_catalog import ( CLOCK_DEFAULT_TIME_ZONE, CLOCK_TIME_ZONE_OPTIONS, ITEM_TYPE_EDITABLE_PROPERTIES, ITEM_TYPE_LABELS, ITEM_TYPE_PROPERTY_METADATA, ITEM_TYPE_SEQUENCE, ITEM_TYPE_TOOLTIPS, get_item_definition, get_item_global_properties, get_item_use_cooldown_ms, is_known_item_type, ) from .item_type_handlers import get_item_type_handler from .item_service import ItemService from .models import ( BroadcastChatMessagePacket, BroadcastNicknamePacket, BroadcastPositionPacket, ChatMessagePacket, ClientPacket, ForwardSignalPacket, ItemActionResultPacket, ItemAddPacket, ItemDeletePacket, ItemDropPacket, ItemPianoNoteBroadcastPacket, ItemPianoNotePacket, ItemPianoRecordingPacket, ItemPickupPacket, ItemRemovePacket, ItemUpdatePacket, ItemUpsertPacket, ItemUsePacket, ItemUseSoundPacket, NicknameResultPacket, PingPacket, PongPacket, RemoteUser, UpdateNicknamePacket, UpdatePositionPacket, UserLeftPacket, WelcomePacket, WorldItem, ) LOGGER = logging.getLogger("chgrid.server") PACKET_LOGGER = logging.getLogger("chgrid.server.packet") CLIENT_PACKET_ADAPTER = TypeAdapter(ClientPacket) MAX_ACTIVE_PIANO_KEYS_PER_CLIENT = 12 PIANO_RECORDING_MAX_MS = 30_000 PIANO_RECORDING_MAX_EVENTS = 4096 class SignalingServer: """Coordinates websocket clients, signaling, and authoritative item actions.""" def __init__( self, host: str, port: int, ssl_cert: str | None, ssl_key: str | None, max_message_size: int = 2_000_000, state_file: Path | None = None, grid_size: int = 41, state_save_debounce_ms: int = 200, state_save_max_delay_ms: int = 1000, ): """Initialize runtime state, TLS context, and item service.""" self.host = host self.port = port self.max_message_size = max_message_size self._ssl_context = self._build_ssl_context(ssl_cert, ssl_key) self.clients: dict[ServerConnection, ClientConnection] = {} self.item_service = ItemService(state_file=state_file) self.item_last_use_ms: dict[str, int] = {} self.active_piano_keys_by_client: dict[str, set[str]] = {} self.piano_recording_state_by_item: dict[str, dict] = {} self.piano_playback_tasks_by_item: dict[str, asyncio.Task[None]] = {} self.grid_size = max(1, grid_size) self.instance_id = str(uuid.uuid4()) self.server_version = self._resolve_server_version() self.state_save_debounce_ms = max(1, int(state_save_debounce_ms)) self.state_save_max_delay_ms = max(self.state_save_debounce_ms, int(state_save_max_delay_ms)) self._pending_state_save_handle: asyncio.TimerHandle | None = None self._pending_state_save_started_at: float | None = None @staticmethod def _resolve_server_version() -> str: """Resolve serverInfo version, preferring synced web version when available.""" env_override = os.getenv("CHGRID_SERVER_VERSION", "").strip() if env_override: return env_override try: version_file = Path(__file__).resolve().parents[2] / "client" / "public" / "version.js" text = version_file.read_text(encoding="utf-8") match = re.search(r'CHGRID_WEB_VERSION\s*=\s*"([^"]+)"', text) if match: token = match.group(1).strip() if token: return token except OSError: pass try: return package_version("chgrid-server") except PackageNotFoundError: return "unknown" @property def items(self) -> dict[str, WorldItem]: """Expose current item map owned by the item service.""" return self.item_service.items def _nickname_key(self, nickname: str) -> str: """Normalize nickname for case-insensitive comparisons.""" return nickname.casefold() def _flush_state_save(self) -> None: """Immediately flush pending state persistence and clear debounce state.""" if self._pending_state_save_handle is not None: self._pending_state_save_handle.cancel() self._pending_state_save_handle = None self._pending_state_save_started_at = None self.item_service.save_state() def _request_state_save(self) -> None: """Debounce/coalesce item-state persistence to reduce write churn.""" loop = asyncio.get_running_loop() now = loop.time() if self._pending_state_save_started_at is None: self._pending_state_save_started_at = now elapsed_ms = int((now - self._pending_state_save_started_at) * 1000) if elapsed_ms >= self.state_save_max_delay_ms: self._flush_state_save() return if self._pending_state_save_handle is not None: self._pending_state_save_handle.cancel() remaining_ms = max(0, self.state_save_max_delay_ms - elapsed_ms) delay_ms = min(self.state_save_debounce_ms, remaining_ms) self._pending_state_save_handle = loop.call_later(delay_ms / 1000, self._flush_state_save) def _is_nickname_taken(self, nickname: str, exclude_client_id: str | None = None) -> bool: """Check whether nickname is already used by another active client.""" wanted = self._nickname_key(nickname) for other in self.clients.values(): if exclude_client_id is not None and other.id == exclude_client_id: continue if self._nickname_key(other.nickname) == wanted: return True return False @staticmethod def _item_type_label(item: WorldItem) -> str: """Return user-facing item type wording for chat/status strings.""" return "radio" if item.type == "radio_station" else item.type @staticmethod def _resolve_item_use_sound(item: WorldItem) -> str | None: """Resolve one-shot use sound, preferring per-item param override.""" param_sound = item.params.get("useSound") if isinstance(param_sound, str): token = param_sound.strip() if token: return token return None if isinstance(item.useSound, str) and item.useSound.strip(): return item.useSound.strip() return None def _get_client_by_id(self, client_id: str) -> ClientConnection | None: """Resolve one connected client by id.""" for connected in self.clients.values(): if connected.id == client_id: return connected return None def _get_piano_source_position(self, item: WorldItem) -> tuple[int, int]: """Resolve world position used for piano note spatial broadcasts.""" if item.carrierId: carrier = self._get_client_by_id(item.carrierId) if carrier is not None: return carrier.x, carrier.y return item.x, item.y async def _broadcast_item_piano_note( self, item: WorldItem, *, sender_id: str, key_id: str, midi: int, on: bool, instrument_override: str | None = None, voice_mode_override: str | None = None, attack_override: int | None = None, decay_override: int | None = None, release_override: int | None = None, brightness_override: int | None = None, emit_range_override: int | None = None, exclude: ServerConnection | None = None, ) -> None: """Broadcast one piano note event using current item synth settings.""" instrument = (instrument_override if isinstance(instrument_override, str) else str(item.params.get("instrument", "piano"))).strip().lower() voice_mode = (voice_mode_override if isinstance(voice_mode_override, str) else str(item.params.get("voiceMode", "poly"))).strip().lower() if voice_mode not in {"poly", "mono"}: voice_mode = "poly" octave = int(item.params.get("octave", 0)) if isinstance(item.params.get("octave", 0), (int, float)) else 0 attack = ( int(attack_override) if isinstance(attack_override, int) else int(item.params.get("attack", 15)) if isinstance(item.params.get("attack", 15), (int, float)) else 15 ) decay = ( int(decay_override) if isinstance(decay_override, int) else int(item.params.get("decay", 45)) if isinstance(item.params.get("decay", 45), (int, float)) else 45 ) release = ( int(release_override) if isinstance(release_override, int) else int(item.params.get("release", 35)) if isinstance(item.params.get("release", 35), (int, float)) else 35 ) brightness = ( int(brightness_override) if isinstance(brightness_override, int) else int(item.params.get("brightness", 55)) if isinstance(item.params.get("brightness", 55), (int, float)) else 55 ) emit_range = ( int(emit_range_override) if isinstance(emit_range_override, int) else int(item.params.get("emitRange", 15)) if isinstance(item.params.get("emitRange", 15), (int, float)) else 15 ) source_x, source_y = self._get_piano_source_position(item) await self._broadcast( ItemPianoNoteBroadcastPacket( type="item_piano_note", itemId=item.id, senderId=sender_id, keyId=key_id, midi=max(0, min(127, int(midi))), on=on, instrument=instrument, voiceMode=voice_mode, octave=max(-2, min(2, octave)), attack=max(0, min(100, attack)), decay=max(0, min(100, decay)), release=max(0, min(100, release)), brightness=max(0, min(100, brightness)), x=source_x, y=source_y, emitRange=max(5, min(20, emit_range)), ), exclude=exclude, ) def _cancel_piano_playback(self, item_id: str) -> None: """Cancel active playback task for one piano item, if any.""" task = self.piano_playback_tasks_by_item.pop(item_id, None) if task is not None and not task.done(): task.cancel() @staticmethod def _recording_elapsed_ms(session: dict, now_monotonic: float | None = None) -> int: """Compute effective recorded duration, including currently active segment.""" elapsed_ms = int(session.get("elapsedMs", 0)) if isinstance(session.get("elapsedMs"), (int, float)) else 0 paused = session.get("paused") is True if paused: return max(0, elapsed_ms) last_resume = session.get("lastResumeMonotonic") if isinstance(last_resume, (int, float)): now_value = now_monotonic if isinstance(now_monotonic, (int, float)) else time.monotonic() elapsed_ms += max(0, int((now_value - float(last_resume)) * 1000)) return max(0, elapsed_ms) async def _finalize_piano_recording(self, item_id: str, *, status_message: str | None = None) -> None: """Persist and broadcast one active recording session, then clear runtime state.""" session = self.piano_recording_state_by_item.pop(item_id, None) if not session: return auto_stop_task = session.get("autoStopTask") if isinstance(auto_stop_task, asyncio.Task) and not auto_stop_task.done(): auto_stop_task.cancel() item = self.items.get(item_id) if not item or item.type != "piano": return elapsed_ms = max(0, min(PIANO_RECORDING_MAX_MS, self._recording_elapsed_ms(session))) recorded_events = session.get("events") events = list(recorded_events) if isinstance(recorded_events, list) else [] song_id = f"item:{item.id}:recording" keys: list[str] = [] key_to_index: dict[str, int] = {} states: list[list[object]] = [] state_to_index: dict[tuple[object, ...], int] = {} compact_events: list[list[int]] = [] for event in events: if not isinstance(event, dict): continue t = int(event.get("t", 0)) if isinstance(event.get("t"), (int, float)) else 0 key_id = str(event.get("keyId", "")).strip() midi = int(event.get("midi", 0)) if isinstance(event.get("midi"), (int, float)) else 0 on = 1 if event.get("on") is True else 0 instrument = str(event.get("instrument", "piano")).strip().lower() or "piano" voice_mode = str(event.get("voiceMode", "poly")).strip().lower() if voice_mode not in {"mono", "poly"}: voice_mode = "poly" attack = int(event.get("attack", 15)) if isinstance(event.get("attack"), (int, float)) else 15 decay = int(event.get("decay", 45)) if isinstance(event.get("decay"), (int, float)) else 45 release = int(event.get("release", 35)) if isinstance(event.get("release"), (int, float)) else 35 brightness = int(event.get("brightness", 55)) if isinstance(event.get("brightness"), (int, float)) else 55 emit_range = int(event.get("emitRange", 15)) if isinstance(event.get("emitRange"), (int, float)) else 15 state_key = ( instrument, voice_mode, max(0, min(100, attack)), max(0, min(100, decay)), max(0, min(100, release)), max(0, min(100, brightness)), max(5, min(20, emit_range)), ) if not key_id: continue index = key_to_index.get(key_id) if index is None: index = len(keys) keys.append(key_id) key_to_index[key_id] = index state_index = state_to_index.get(state_key) if state_index is None: state_index = len(states) states.append(list(state_key)) state_to_index[state_key] = state_index compact_events.append([max(0, min(PIANO_RECORDING_MAX_MS, t)), index, max(0, min(127, midi)), on, state_index]) compact_events.sort(key=lambda row: row[0]) first_state = states[0] if states else ["piano", "poly", 15, 45, 35, 55, 15] self.item_service.piano_songs[song_id] = { "meta": { "instrument": first_state[0], "voiceMode": first_state[1], "attack": first_state[2], "decay": first_state[3], "release": first_state[4], "brightness": first_state[5], "emitRange": first_state[6], "recordingLengthMs": elapsed_ms, }, "keys": keys, "states": states, "events": compact_events, } self.item_service.save_piano_songs() item.params["songId"] = song_id item.params.pop("recording", None) item.params.pop("recordingLengthMs", None) item.updatedAt = self.item_service.now_ms() item.version += 1 self._request_state_save() await self._broadcast_item(item) owner_id = str(session.get("ownerClientId", "")) owner = self._get_client_by_id(owner_id) if owner_id else None if owner and status_message: await self._send_item_result(owner, True, "use", status_message, item.id) async def _auto_stop_piano_recording(self, item_id: str) -> None: """Stop a recording automatically at the max recording duration.""" try: while True: session = self.piano_recording_state_by_item.get(item_id) if not isinstance(session, dict): return if self._recording_elapsed_ms(session) >= PIANO_RECORDING_MAX_MS: await self._finalize_piano_recording(item_id, status_message="stop") return await asyncio.sleep(0.25) except asyncio.CancelledError: return async def _start_piano_playback(self, item: WorldItem) -> None: """Run one piano recording playback task and broadcast note events.""" sender_id = f"item:{item.id}:playback" events: list[dict[str, object]] = [] song_id = str(item.params.get("songId", "")).strip() song_payload = self.item_service.piano_songs.get(song_id) if song_id else None if isinstance(song_payload, dict): keys = song_payload.get("keys") states = song_payload.get("states") compact_events = song_payload.get("events") meta = song_payload.get("meta") if isinstance(keys, list) and isinstance(compact_events, list): base_state = None if isinstance(meta, dict): instrument = str(meta.get("instrument", "")).strip().lower() or "piano" raw_voice_mode = str(meta.get("voiceMode", "")).strip().lower() voice_mode = raw_voice_mode if raw_voice_mode in {"mono", "poly"} else "poly" attack = int(meta.get("attack", 15)) if isinstance(meta.get("attack"), (int, float)) else 15 decay = int(meta.get("decay", 45)) if isinstance(meta.get("decay"), (int, float)) else 45 release = int(meta.get("release", 35)) if isinstance(meta.get("release"), (int, float)) else 35 brightness = int(meta.get("brightness", 55)) if isinstance(meta.get("brightness"), (int, float)) else 55 emit_range = int(meta.get("emitRange", 15)) if isinstance(meta.get("emitRange"), (int, float)) else 15 base_state = ( instrument, voice_mode, max(0, min(100, attack)), max(0, min(100, decay)), max(0, min(100, release)), max(0, min(100, brightness)), max(5, min(20, emit_range)), ) for row in compact_events: if not isinstance(row, list) or len(row) < 4: continue raw_time, raw_key_idx, raw_midi, raw_on = row[:4] if not isinstance(raw_time, (int, float)) or not isinstance(raw_key_idx, (int, float)) or not isinstance(raw_midi, (int, float)): continue key_idx = int(raw_key_idx) if key_idx < 0 or key_idx >= len(keys): continue raw_key = keys[key_idx] if not isinstance(raw_key, str) or not raw_key.strip(): continue state = base_state if len(row) >= 5 and isinstance(states, list) and isinstance(row[4], (int, float)): state_idx = int(row[4]) if 0 <= state_idx < len(states): state_row = states[state_idx] if isinstance(state_row, list) and len(state_row) >= 7: candidate_instrument = str(state_row[0]).strip().lower() or "piano" candidate_voice_mode = str(state_row[1]).strip().lower() state = ( candidate_instrument, candidate_voice_mode if candidate_voice_mode in {"mono", "poly"} else "poly", max(0, min(100, int(state_row[2]) if isinstance(state_row[2], (int, float)) else 15)), max(0, min(100, int(state_row[3]) if isinstance(state_row[3], (int, float)) else 45)), max(0, min(100, int(state_row[4]) if isinstance(state_row[4], (int, float)) else 35)), max(0, min(100, int(state_row[5]) if isinstance(state_row[5], (int, float)) else 55)), max(5, min(20, int(state_row[6]) if isinstance(state_row[6], (int, float)) else 15)), ) if state is None: continue events.append( { "t": max(0, min(PIANO_RECORDING_MAX_MS, int(raw_time))), "keyId": raw_key[:32], "midi": max(0, min(127, int(raw_midi))), "on": bool(raw_on), "instrument": state[0], "voiceMode": state[1], "attack": state[2], "decay": state[3], "release": state[4], "brightness": state[5], "emitRange": state[6], } ) events.sort(key=lambda entry: int(entry["t"])) if not events: return active_keys: dict[str, int] = {} previous_at_ms = 0 try: for event in events: current_at_ms = int(event["t"]) delay_ms = max(0, current_at_ms - previous_at_ms) if delay_ms > 0: await asyncio.sleep(delay_ms / 1000) current_item = self.items.get(item.id) if not current_item or current_item.type != "piano": break key_id = str(event["keyId"]) midi = int(event["midi"]) on = bool(event["on"]) if on: active_keys[key_id] = midi else: active_keys.pop(key_id, None) await self._broadcast_item_piano_note( current_item, sender_id=sender_id, key_id=key_id, midi=midi, on=on, instrument_override=event.get("instrument") if isinstance(event.get("instrument"), str) else None, voice_mode_override=event.get("voiceMode") if isinstance(event.get("voiceMode"), str) else None, attack_override=event.get("attack") if isinstance(event.get("attack"), int) else None, decay_override=event.get("decay") if isinstance(event.get("decay"), int) else None, release_override=event.get("release") if isinstance(event.get("release"), int) else None, brightness_override=event.get("brightness") if isinstance(event.get("brightness"), int) else None, emit_range_override=event.get("emitRange") if isinstance(event.get("emitRange"), int) else None, ) previous_at_ms = current_at_ms except asyncio.CancelledError: pass finally: current_item = self.items.get(item.id) if current_item and current_item.type == "piano": for key_id, midi in list(active_keys.items()): await self._broadcast_item_piano_note( current_item, sender_id=sender_id, key_id=key_id, midi=midi, on=False, ) current_task = self.piano_playback_tasks_by_item.get(item.id) if current_task is asyncio.current_task(): self.piano_playback_tasks_by_item.pop(item.id, None) def _is_in_bounds(self, x: int, y: int) -> bool: """Check whether a coordinate is inside server-authoritative world bounds.""" return 0 <= x < self.grid_size and 0 <= y < self.grid_size @staticmethod def _normalize_clock_timezone(value: object) -> str: """Normalize timezone input to one of supported clock zones.""" token = str(value or "").strip() if token in CLOCK_TIME_ZONE_OPTIONS: return token return CLOCK_DEFAULT_TIME_ZONE @staticmethod def _parse_clock_use_24_hour(value: object) -> bool | None: """Parse bool-like clock format values (`on/off`, `true/false`, etc.).""" if isinstance(value, bool): return value if isinstance(value, (int, float)): return bool(value) if isinstance(value, str): token = value.strip().lower() if token in {"on", "true", "1", "yes"}: return True if token in {"off", "false", "0", "no"}: return False return None @classmethod def _format_clock_display_time(cls, params: dict) -> str: """Render current clock text based on item timezone/format params.""" tz_name = cls._normalize_clock_timezone(params.get("timeZone")) use_24_hour = cls._parse_clock_use_24_hour(params.get("use24Hour")) if use_24_hour is None: use_24_hour = False now = datetime.now(ZoneInfo(tz_name)) if use_24_hour: return now.strftime("%H:%M") hour_12 = now.hour % 12 or 12 return f"{hour_12}:{now.minute:02d} {'AM' if now.hour < 12 else 'PM'}" async def _send_item_result( self, client: ClientConnection, ok: bool, action: Literal["add", "pickup", "drop", "delete", "use", "update"], message: str, item_id: str | None = None, ) -> None: """Send a structured item action result to one client.""" await self._send( client.websocket, ItemActionResultPacket( type="item_action_result", ok=ok, action=action, message=message, itemId=item_id, ), ) async def _broadcast_item(self, item: WorldItem) -> None: """Broadcast a full item snapshot update to all connected clients.""" await self._broadcast(ItemUpsertPacket(type="item_upsert", item=item)) async def start(self) -> None: """Start websocket serving and run until cancelled.""" protocol = "wss" if self._ssl_context else "ws" LOGGER.info("starting signaling server on %s://%s:%d", protocol, self.host, self.port) try: async with serve( self._handle_client, self.host, self.port, ssl=self._ssl_context, max_size=self.max_message_size, ): await asyncio.Future() finally: self._flush_state_save() async def _handle_client(self, websocket: ServerConnection) -> None: """Handle one websocket client's connect/message/disconnect lifecycle.""" client = ClientConnection(websocket=websocket, id=str(uuid.uuid4())) self.clients[websocket] = client LOGGER.info("client connected id=%s total=%d", client.id, len(self.clients)) try: await self._send_welcome(client) async for raw_message in websocket: await self._handle_message(client, raw_message) finally: if websocket in self.clients: disconnected = self.clients.pop(websocket) self.active_piano_keys_by_client.pop(disconnected.id, None) for item_id, session in list(self.piano_recording_state_by_item.items()): if session.get("ownerClientId") != disconnected.id: continue await self._finalize_piano_recording(item_id) for item in self.item_service.drop_carried_items_for_disconnect(disconnected): await self._broadcast_item(item) self._request_state_save() LOGGER.info( "client disconnected id=%s nickname=%s total=%d", disconnected.id, disconnected.nickname, len(self.clients), ) await self._broadcast(UserLeftPacket(type="user_left", id=disconnected.id), exclude=websocket) await self._broadcast( BroadcastChatMessagePacket( type="chat_message", message=f"{disconnected.nickname} has logged out.", system=True, ), exclude=websocket, ) async def _send_welcome(self, client: ClientConnection) -> None: """Send initial world snapshot to a newly connected client.""" users = [ RemoteUser(id=other.id, nickname=other.nickname, x=other.x, y=other.y) for ws, other in self.clients.items() if ws is not client.websocket ] packet = WelcomePacket( type="welcome", id=client.id, users=users, items=[item.model_dump(exclude_none=True) for item in self.items.values()], worldConfig={"gridSize": self.grid_size}, uiDefinitions=self._build_ui_definitions(), serverInfo={"instanceId": self.instance_id, "version": self.server_version}, ) await self._send(client.websocket, packet) def _build_ui_definitions(self) -> dict: """Build server-owned UI definitions for item/menu rendering.""" item_types: list[dict] = [] for item_type in ITEM_TYPE_SEQUENCE: editable = list(ITEM_TYPE_EDITABLE_PROPERTIES.get(item_type, ("title",))) item_types.append( { "type": item_type, "label": ITEM_TYPE_LABELS.get(item_type, item_type), "tooltip": ITEM_TYPE_TOOLTIPS.get(item_type), "capabilities": list(get_item_definition(item_type).capabilities), "editableProperties": editable, "propertyMetadata": ITEM_TYPE_PROPERTY_METADATA.get(item_type, {}), "globalProperties": get_item_global_properties(item_type), } ) return { "itemTypeOrder": list(ITEM_TYPE_SEQUENCE), "itemTypes": item_types, } async def _broadcast_wheel_result_after_delay( self, client: ClientConnection, self_message: str, others_message: str, delay_seconds: float = 3.0, ) -> None: """Delay then publish wheel result text to self and other users.""" await asyncio.sleep(delay_seconds) await self._broadcast( BroadcastChatMessagePacket(type="chat_message", message=others_message, system=True), exclude=client.websocket, ) if client.websocket in self.clients: await self._send( client.websocket, BroadcastChatMessagePacket(type="chat_message", message=self_message, system=True), ) async def _handle_message(self, client: ClientConnection, raw_message: str) -> None: """Decode, validate, and route one inbound client packet.""" try: payload = json.loads(raw_message) except json.JSONDecodeError: PACKET_LOGGER.warning("non-json packet from id=%s", client.id) return try: packet = CLIENT_PACKET_ADAPTER.validate_python(payload) except ValidationError as exc: PACKET_LOGGER.warning("invalid packet from id=%s: %s", client.id, exc) return if isinstance(packet, UpdatePositionPacket): if not self._is_in_bounds(packet.x, packet.y): PACKET_LOGGER.warning( "out-of-bounds position ignored id=%s x=%d y=%d grid_size=%d", client.id, packet.x, packet.y, self.grid_size, ) return client.x = packet.x client.y = packet.y await self._broadcast( BroadcastPositionPacket(type="update_position", id=client.id, x=client.x, y=client.y), exclude=client.websocket, ) carried = self.item_service.find_carried_item(client.id) if carried: carried.x = client.x carried.y = client.y carried.updatedAt = self.item_service.now_ms() await self._broadcast_item(carried) return if isinstance(packet, UpdateNicknamePacket): requested_nickname = packet.nickname.strip() if not requested_nickname: await self._send( client.websocket, NicknameResultPacket( type="nickname_result", accepted=False, requestedNickname=packet.nickname, effectiveNickname=client.nickname, reason="Nickname is required.", ), ) return old_nickname = client.nickname if self._is_nickname_taken(requested_nickname, exclude_client_id=client.id): await self._send( client.websocket, NicknameResultPacket( type="nickname_result", accepted=False, requestedNickname=requested_nickname, effectiveNickname=client.nickname, reason="Nickname already in use.", ), ) return if requested_nickname == old_nickname: await self._send( client.websocket, NicknameResultPacket( type="nickname_result", accepted=True, requestedNickname=requested_nickname, effectiveNickname=client.nickname, ), ) return client.nickname = requested_nickname if old_nickname == "user...": LOGGER.info("user login id=%s nickname=%s", client.id, client.nickname) else: LOGGER.info("nickname change id=%s old=%s new=%s", client.id, old_nickname, client.nickname) await self._send( client.websocket, NicknameResultPacket( type="nickname_result", accepted=True, requestedNickname=requested_nickname, effectiveNickname=client.nickname, ), ) await self._broadcast( BroadcastNicknamePacket(type="update_nickname", id=client.id, nickname=client.nickname), exclude=client.websocket, ) if old_nickname == "user...": await self._broadcast( BroadcastChatMessagePacket( type="chat_message", message=f"{client.nickname} has logged in.", system=True, ), exclude=client.websocket, ) else: await self._broadcast( BroadcastChatMessagePacket( type="chat_message", message=f"{old_nickname} is now known as {client.nickname}.", system=True, ), exclude=client.websocket, ) self_message = ( f"Welcome. Logged in as {client.nickname}." if old_nickname == "user..." else f"You are now known as {client.nickname}." ) await self._send( client.websocket, BroadcastChatMessagePacket( type="chat_message", message=self_message, system=True, ), ) return if isinstance(packet, ChatMessagePacket): await self._broadcast( BroadcastChatMessagePacket( type="chat_message", message=packet.message, senderId=client.id, senderNickname=client.nickname, system=False, ) ) return if isinstance(packet, PingPacket): await self._send( client.websocket, PongPacket(type="pong", clientSentAt=packet.clientSentAt), ) return if isinstance(packet, ItemAddPacket): if not is_known_item_type(packet.itemType): await self._send_item_result(client, False, "add", "Unknown item type.") return item = self.item_service.default_item(client, packet.itemType) self.item_service.add_item(item) await self._broadcast_item(item) self._request_state_save() LOGGER.info( "item created by=%s item_id=%s type=%s title=%s x=%d y=%d", client.nickname, item.id, item.type, item.title, item.x, item.y, ) item_text = f"{item.title} ({self._item_type_label(item)})" await self._broadcast( BroadcastChatMessagePacket( type="chat_message", message=f"{client.nickname} placed {item_text} at {item.x}, {item.y}.", system=True, ), exclude=client.websocket, ) await self._send_item_result( client, True, "add", f"You placed {item_text} at {item.x}, {item.y}.", item.id, ) return if isinstance(packet, ItemPickupPacket): item = self.items.get(packet.itemId) if not item: await self._send_item_result(client, False, "pickup", "Item not found.") return if item.carrierId and item.carrierId != client.id: await self._send_item_result(client, False, "pickup", "Item is already being carried.", item.id) return carried = self.item_service.find_carried_item(client.id) if carried and carried.id != item.id: await self._send_item_result(client, False, "pickup", "You are already carrying an item.", item.id) return if item.carrierId is None and (item.x != client.x or item.y != client.y): await self._send_item_result(client, False, "pickup", "Item is not on your square.", item.id) return item.carrierId = client.id item.x = client.x item.y = client.y item.updatedAt = self.item_service.now_ms() await self._broadcast_item(item) self._request_state_save() await self._send_item_result(client, True, "pickup", f"Picked up {item.title}.", item.id) return if isinstance(packet, ItemDropPacket): item = self.items.get(packet.itemId) if not item: await self._send_item_result(client, False, "drop", "Item not found.") return if item.carrierId != client.id: await self._send_item_result(client, False, "drop", "You are not carrying that item.", item.id) return if not self._is_in_bounds(packet.x, packet.y): await self._send_item_result(client, False, "drop", "Drop position is out of bounds.", item.id) return item.carrierId = None item.x = packet.x item.y = packet.y item.updatedAt = self.item_service.now_ms() await self._broadcast_item(item) self._request_state_save() await self._send_item_result(client, True, "drop", f"Dropped {item.title}.", item.id) return if isinstance(packet, ItemDeletePacket): item = self.items.get(packet.itemId) if not item: await self._send_item_result(client, False, "delete", "Item not found.") return if item.carrierId and item.carrierId != client.id: await self._send_item_result(client, False, "delete", "Item is being carried by another user.", item.id) return if item.carrierId is None and (item.x != client.x or item.y != client.y): await self._send_item_result(client, False, "delete", "Item is not on your square.", item.id) return LOGGER.info( "item deleted by=%s item_id=%s type=%s title=%s", client.nickname, item.id, item.type, item.title, ) self._cancel_piano_playback(item.id) recording_state = self.piano_recording_state_by_item.pop(item.id, None) if recording_state is not None: auto_stop_task = recording_state.get("autoStopTask") if isinstance(auto_stop_task, asyncio.Task) and not auto_stop_task.done(): auto_stop_task.cancel() song_id = str(item.params.get("songId", "")).strip() if song_id and song_id in self.item_service.piano_songs: self.item_service.piano_songs.pop(song_id, None) self.item_service.save_piano_songs() self.item_service.remove_item(item.id) self.item_last_use_ms.pop(item.id, None) await self._broadcast(ItemRemovePacket(type="item_remove", itemId=item.id)) self._request_state_save() await self._send_item_result(client, True, "delete", f"Deleted {item.title}.", item.id) return if isinstance(packet, ItemUsePacket): item = self.items.get(packet.itemId) if not item: await self._send_item_result(client, False, "use", "Item not found.") return if item.carrierId not in (None, client.id): await self._send_item_result(client, False, "use", "Item is not available.", item.id) return if item.carrierId is None and (item.x != client.x or item.y != client.y): await self._send_item_result(client, False, "use", "Item is not on your square.", item.id) return handler = get_item_type_handler(item.type) now_ms = self.item_service.now_ms() cooldown_ms = get_item_use_cooldown_ms(item.type) last_use_ms = self.item_last_use_ms.get(item.id) if last_use_ms is not None and now_ms - last_use_ms < cooldown_ms: remaining_ms = cooldown_ms - (now_ms - last_use_ms) remaining_seconds = max(0.1, round(remaining_ms / 1000, 1)) await self._send_item_result( client, False, "use", f"{item.title} is on cooldown for {remaining_seconds:.1f} s.", item.id, ) return try: use_result = handler.use(item, client.nickname, self._format_clock_display_time) except ValueError as exc: await self._send_item_result(client, False, "use", str(exc), item.id) return if use_result.updated_params is not None: try: item.params = handler.validate_update(item, {**item.params, **use_result.updated_params}) except ValueError as exc: await self._send_item_result(client, False, "use", str(exc), item.id) return item.updatedAt = now_ms self._request_state_save() await self._broadcast_item(item) self.item_last_use_ms[item.id] = now_ms await self._broadcast( BroadcastChatMessagePacket(type="chat_message", message=use_result.others_message, system=True), exclude=client.websocket, ) use_sound = self._resolve_item_use_sound(item) if use_sound: await self._broadcast( ItemUseSoundPacket( type="item_use_sound", itemId=item.id, sound=use_sound, x=item.x, y=item.y, ) ) await self._send_item_result(client, True, "use", use_result.self_message, item.id) if use_result.delayed_self_message is not None and use_result.delayed_others_message is not None: asyncio.create_task( self._broadcast_wheel_result_after_delay( client=client, self_message=use_result.delayed_self_message, others_message=use_result.delayed_others_message, ) ) return if isinstance(packet, ItemPianoNotePacket): item = self.items.get(packet.itemId) if not item or item.type != "piano": return if item.carrierId not in (None, client.id): return if item.carrierId is None and (item.x != client.x or item.y != client.y): return active_keys = self.active_piano_keys_by_client.setdefault(client.id, set()) if packet.on: if packet.keyId not in active_keys and len(active_keys) >= MAX_ACTIVE_PIANO_KEYS_PER_CLIENT: return active_keys.add(packet.keyId) else: active_keys.discard(packet.keyId) recording_state = self.piano_recording_state_by_item.get(item.id) if recording_state and recording_state.get("ownerClientId") == client.id and recording_state.get("paused") is not True: elapsed_ms = max(0, min(PIANO_RECORDING_MAX_MS, self._recording_elapsed_ms(recording_state))) events = recording_state.get("events") if isinstance(events, list) and len(events) < PIANO_RECORDING_MAX_EVENTS: instrument = str(item.params.get("instrument", "piano")).strip().lower() voice_mode = str(item.params.get("voiceMode", "poly")).strip().lower() if voice_mode not in {"poly", "mono"}: voice_mode = "poly" attack = int(item.params.get("attack", 15)) if isinstance(item.params.get("attack", 15), (int, float)) else 15 decay = int(item.params.get("decay", 45)) if isinstance(item.params.get("decay", 45), (int, float)) else 45 release = int(item.params.get("release", 35)) if isinstance(item.params.get("release", 35), (int, float)) else 35 brightness = int(item.params.get("brightness", 55)) if isinstance(item.params.get("brightness", 55), (int, float)) else 55 emit_range = int(item.params.get("emitRange", 15)) if isinstance(item.params.get("emitRange", 15), (int, float)) else 15 events.append( { "t": elapsed_ms, "keyId": packet.keyId[:32], "midi": packet.midi, "on": packet.on, "instrument": instrument, "voiceMode": voice_mode, "attack": max(0, min(100, attack)), "decay": max(0, min(100, decay)), "release": max(0, min(100, release)), "brightness": max(0, min(100, brightness)), "emitRange": max(5, min(20, emit_range)), } ) if elapsed_ms >= PIANO_RECORDING_MAX_MS: await self._finalize_piano_recording(item.id, status_message="stop") await self._broadcast_item_piano_note( item, sender_id=client.id, key_id=packet.keyId, midi=packet.midi, on=packet.on, exclude=client.websocket, ) return if isinstance(packet, ItemPianoRecordingPacket): item = self.items.get(packet.itemId) if not item or item.type != "piano": await self._send_item_result(client, False, "use", "Piano not found.") return if item.carrierId not in (None, client.id): await self._send_item_result(client, False, "use", "Piano is not available.", item.id) return if item.carrierId is None and (item.x != client.x or item.y != client.y): await self._send_item_result(client, False, "use", "Piano is not on your square.", item.id) return if packet.action == "toggle_record": existing = self.piano_recording_state_by_item.get(item.id) if existing and existing.get("ownerClientId") != client.id: await self._send_item_result(client, False, "use", "This piano is already recording.", item.id) return if existing and existing.get("ownerClientId") == client.id: if existing.get("paused") is True: existing["paused"] = False existing["lastResumeMonotonic"] = time.monotonic() await self._send_item_result(client, True, "use", "resume", item.id) else: existing["elapsedMs"] = self._recording_elapsed_ms(existing) existing["paused"] = True existing.pop("lastResumeMonotonic", None) await self._send_item_result(client, True, "use", "pause", item.id) return self._cancel_piano_playback(item.id) recording_state = { "ownerClientId": client.id, "elapsedMs": 0, "paused": False, "lastResumeMonotonic": time.monotonic(), "events": [], } self.piano_recording_state_by_item[item.id] = recording_state auto_stop_task = asyncio.create_task(self._auto_stop_piano_recording(item.id)) recording_state["autoStopTask"] = auto_stop_task await self._send_item_result(client, True, "use", "record", item.id) return if packet.action == "stop_record": existing = self.piano_recording_state_by_item.get(item.id) if existing and existing.get("ownerClientId") != client.id: await self._send_item_result(client, False, "use", "This piano is already recording.", item.id) return if existing and existing.get("ownerClientId") == client.id: await self._finalize_piano_recording(item.id, status_message="stop") return await self._send_item_result(client, True, "use", "stop", item.id) return if packet.action == "playback": if item.id in self.piano_recording_state_by_item: await self._send_item_result(client, False, "use", "Stop recording before playback.", item.id) return song_id = str(item.params.get("songId", "")).strip() has_song = isinstance(self.item_service.piano_songs.get(song_id), dict) if song_id else False if not has_song: await self._send_item_result(client, False, "use", "No recording saved on this piano.", item.id) return self._cancel_piano_playback(item.id) playback_task = asyncio.create_task(self._start_piano_playback(item)) self.piano_playback_tasks_by_item[item.id] = playback_task await self._send_item_result(client, True, "use", "play", item.id) return if packet.action == "stop_playback": self._cancel_piano_playback(item.id) await self._send_item_result(client, True, "use", "stop", item.id) return return if isinstance(packet, ItemUpdatePacket): item = self.items.get(packet.itemId) if not item: await self._send_item_result(client, False, "update", "Item not found.") return if item.carrierId not in (None, client.id): await self._send_item_result(client, False, "update", "Item is not available for editing.", item.id) return if item.carrierId is None and (item.x != client.x or item.y != client.y): await self._send_item_result(client, False, "update", "Item is not on your square.", item.id) return if packet.title is not None: title = packet.title.strip() if not title: await self._send_item_result(client, False, "update", "Title cannot be empty.", item.id) return item.title = title[:80] if packet.params: next_params = {**item.params, **packet.params} handler = get_item_type_handler(item.type) try: next_params = handler.validate_update(item, next_params) except ValueError as exc: await self._send_item_result(client, False, "update", str(exc), item.id) return item.params = next_params item.updatedAt = self.item_service.now_ms() item.version += 1 await self._broadcast_item(item) self._request_state_save() await self._send_item_result(client, True, "update", f"Updated {item.title}.", item.id) return target = self._find_by_id(packet.targetId) if not target: PACKET_LOGGER.info("signal target not found sender=%s target=%s", client.id, packet.targetId) return await self._send( target.websocket, ForwardSignalPacket( type="signal", senderId=client.id, senderNickname=client.nickname, x=client.x, y=client.y, sdp=packet.sdp, ice=packet.ice, ), ) async def _broadcast(self, packet: object, exclude: ServerConnection | None = None) -> None: """Broadcast one packet to all clients except an optional websocket.""" recipients = [websocket for websocket in self.clients if websocket is not exclude] if not recipients: return await asyncio.gather(*(self._send(websocket, packet) for websocket in recipients)) async def _send(self, websocket: ServerConnection, packet: object) -> None: """Send one packet to one websocket, swallowing per-client send failures.""" try: if hasattr(packet, "model_dump"): data = packet.model_dump(exclude_none=True) else: data = packet await websocket.send(json.dumps(data)) except Exception as exc: # intentionally broad to keep server alive per client error LOGGER.debug("send failure: %s", exc) def _find_by_id(self, client_id: str) -> ClientConnection | None: """Resolve a client id to an active connection.""" for client in self.clients.values(): if client.id == client_id: return client return None @staticmethod def _build_ssl_context(cert: str | None, key: str | None) -> ssl.SSLContext | None: """Create TLS server context when cert/key are configured.""" if not cert or not key: return None context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER) context.load_cert_chain(certfile=Path(cert), keyfile=Path(key)) return context def run() -> None: """CLI entrypoint for running the signaling server process.""" parser = argparse.ArgumentParser(description="chgrid signaling server") parser.add_argument("--config", default="config.toml") parser.add_argument("--host", default=None) parser.add_argument("--port", type=int, default=None) parser.add_argument("--ssl-cert", default=None) parser.add_argument("--ssl-key", default=None) parser.add_argument("--allow-insecure-ws", action="store_true", default=None) args = parser.parse_args() config_path = Path(args.config) if args.config else None if config_path and not config_path.exists() and args.config == "config.toml": config_path = None config = load_config(config_path) host = args.host or config.server.bind_ip port = args.port or config.server.port allow_insecure_ws = config.network.allow_insecure_ws if args.allow_insecure_ws is True: allow_insecure_ws = True ssl_cert = args.ssl_cert if args.ssl_cert is not None else config.tls.cert_file or None ssl_key = args.ssl_key if args.ssl_key is not None else config.tls.key_file or None state_file_value = config.storage.state_file.strip() state_file: Path | None = None if state_file_value: base_dir = config_path.parent if config_path is not None else Path.cwd() state_file = Path(state_file_value) if not state_file.is_absolute(): state_file = base_dir / state_file if not allow_insecure_ws and (not ssl_cert or not ssl_key): raise SystemExit( "TLS is required when insecure ws is disabled. Set tls.cert_file/tls.key_file in config.toml." ) logging.basicConfig( level=getattr(logging, config.logging.level.upper(), logging.INFO), format="%(asctime)s %(levelname)s %(name)s %(message)s", ) server = SignalingServer( host, port, ssl_cert, ssl_key, max_message_size=config.network.max_message_bytes, state_file=state_file, grid_size=config.world.grid_size, state_save_debounce_ms=config.storage.state_save_debounce_ms, state_save_max_delay_ms=config.storage.state_save_max_delay_ms, ) asyncio.run(server.start())