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

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

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

export type WrpcRoomEvent<Input extends Maybe<Shape> = unknown> = WrpcRoomEventDefinition<Input>;

export type WrpcRoomEvents<Events extends Maybe<Record<string, WrpcRoomEvent>> = unknown> =
  Events extends Record<string, WrpcRoomEvent>
    ? {
        events: Events;
      }
    : {
        events?: Events;
      };

export const defineWrpcRoomEvent = <Input extends Maybe<Shape> = unknown>(
  wrpcRoomEventData: WrpcRoomEventData<Input>,
): WrpcRoomEvent<Input> => ({
  input: wrpcRoomEventData.input as Input extends Shape ? Input : undefined,
});