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

type WrpcRoomBroadcastData<Input extends Maybe<Shape> = unknown> = {
  input?: Input;
};

type WrpcRoomBroadcastDefinition<Input extends Maybe<Shape> = unknown> = {
  input: Input extends Shape ? Input : undefined;
};

export type WrpcRoomBroadcast<Input extends Maybe<Shape> = unknown> = WrpcRoomBroadcastDefinition<Input>;

export type WrpcRoomBroadcasts<Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown> =
  Broadcasts extends Record<string, WrpcRoomBroadcast>
    ? {
        broadcasts: Broadcasts;
      }
    : {
        broadcasts?: Broadcasts;
      };

export const defineWrpcRoomBroadcast = <Input extends Maybe<Shape> = unknown>(
  wrpcRoomBroadcastData: WrpcRoomBroadcastData<Input>,
): WrpcRoomBroadcast<Input> => ({
  input: wrpcRoomBroadcastData.input as Input extends Shape ? Input : undefined,
});