import type { WrpcRoomEvent } from "./wrpcRoomEvent";
import type { Maybe, Shape } from "./wrpcSchema";

export type ExtractEventNames<Events extends Maybe<Record<string, WrpcRoomEvent>>> =
  Events extends Record<string, WrpcRoomEvent> ? keyof Events : never;

export type ExtractEventInput<
  Events extends Maybe<Record<string, WrpcRoomEvent>>,
  Name extends ExtractEventNames<Events>,
  Fallback = unknown,
> =
  Events extends Record<string, WrpcRoomEvent>
    ? Events[Name] extends WrpcRoomEvent<infer Input>
      ? Input extends Shape
        ? Input
        : Fallback
      : Fallback
    : Fallback;