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

type WrpcRoomCommandData<Input extends Maybe<Shape> = unknown, Output extends Maybe<Shape> = unknown> = {
  input?: Input;
  output?: Output;
};

type WrpcRoomCommandDefinition<Input extends Maybe<Shape> = unknown, Output extends Maybe<Shape> = unknown> = {
  input: Input extends Shape ? Input : undefined;
  output: Output extends Shape ? Output : undefined;
};

export type WrpcRoomCommand<
  Input extends Maybe<Shape> = unknown,
  Output extends Maybe<Shape> = unknown,
> = WrpcRoomCommandDefinition<Input, Output>;

export type WrpcRoomCommands<Commands extends Maybe<Record<string, WrpcRoomCommand>> = unknown> =
  Commands extends Record<string, WrpcRoomCommand>
    ? {
        commands: Commands;
      }
    : {
        commands?: Commands;
      };

export const defineWrpcRoomCommand = <Input extends Maybe<Shape> = unknown, Output extends Maybe<Shape> = unknown>(
  wrpcRoomCommandData: WrpcRoomCommandData<Input, Output>,
): WrpcRoomCommand<Input, Output> => ({
  input: wrpcRoomCommandData.input as Input extends Shape ? Input : undefined,
  output: wrpcRoomCommandData.output as Output extends Shape ? Output : undefined,
});