import type { WrpcRoomBroadcast } from "./wrpcRoomBroadcast";
import type { Maybe, Shape } from "./wrpcSchema";

export type ExtractBroadcastNames<Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>>> =
  Broadcasts extends Record<string, WrpcRoomBroadcast> ? keyof Broadcasts : never;

export type ExtractBroadcastInput<
  Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>>,
  Name extends ExtractBroadcastNames<Broadcasts>,
  Fallback = unknown,
> =
  Broadcasts extends Record<string, WrpcRoomBroadcast>
    ? Broadcasts[Name] extends WrpcRoomBroadcast<infer Input>
      ? Input extends Shape
        ? Input
        : Fallback
      : Fallback
    : Fallback;