import type { WrpcRoom } from "./wrpcRoom";
import type { Maybe } from "./wrpcSchema";

export type WrpcData<WrpcRooms extends Maybe<Record<string, WrpcRoom>> = unknown> =
  WrpcRooms extends Record<string, WrpcRoom>
    ? {
        rooms: WrpcRooms;
      }
    : {
        rooms?: WrpcRooms;
      };

export type WrpcDefinition<WrpcRooms extends Maybe<Record<string, WrpcRoom>> = unknown> = {
  rooms: WrpcRooms extends Record<string, WrpcRoom> ? WrpcRooms : Record<string, WrpcRoom>;
};

export type Wrpc<WrpcRooms extends Maybe<Record<string, WrpcRoom>> = unknown> = WrpcDefinition<WrpcRooms>;

export const defineWrpc = <WrpcRooms extends Maybe<Record<string, WrpcRoom>> = unknown>(
  wrpcData: WrpcData<WrpcRooms>,
): Wrpc<WrpcRooms> => ({
  rooms: (wrpcData.rooms ?? {}) as WrpcRooms extends Record<string, WrpcRoom> ? WrpcRooms : Record<string, WrpcRoom>,
});