import type { WrpcRoomBroadcast, WrpcRoomBroadcasts } from "./wrpcRoomBroadcast";
import type { WrpcRoomCommand, WrpcRoomCommands } from "./wrpcRoomCommand";
import type { WrpcRoomConnectionMetadata } from "./wrpcRoomConnectionMetadata";
import type { WrpcRoomErrors } from "./wrpcRoomError";
import type { WrpcRoomEvent, WrpcRoomEvents } from "./wrpcRoomEvent";
import type { WrpcRoomLifecycles } from "./wrpcRoomLifecycle";
import type { WrpcRoomMetadata } from "./wrpcRoomMetadata";
import type { WrpcRoomConfig, WrpcRoomSettings } from "./wrpcRoomSettings";
import type { WrpcRoomState } from "./wrpcRoomState";
import type { InferShapeOutput, Maybe, Shape } from "./wrpcSchema";

export type WrpcRoomData<
  State extends Maybe<Shape> = unknown,
  Metadata extends Maybe<Shape> = unknown,
  ConnectionMetadata extends Maybe<Shape> = unknown,
  Errors extends Maybe<Record<string, Maybe<Shape>>> = unknown,
  Create extends Maybe<Shape> = unknown,
  Join extends Maybe<Shape> = unknown,
  Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown,
  Events extends Maybe<Record<string, WrpcRoomEvent>> = unknown,
  Commands extends Maybe<Record<string, WrpcRoomCommand>> = unknown,
> = WrpcRoomSettings &
  WrpcRoomState<State> &
  WrpcRoomMetadata<Metadata> &
  WrpcRoomConnectionMetadata<ConnectionMetadata> &
  WrpcRoomErrors<Errors> &
  WrpcRoomLifecycles<Create, Join> &
  WrpcRoomBroadcasts<Broadcasts> &
  WrpcRoomEvents<Events> &
  WrpcRoomCommands<Commands>;

export type WrpcRoomDefinition<
  State extends Maybe<Shape> = unknown,
  Metadata extends Maybe<Shape> = unknown,
  ConnectionMetadata extends Maybe<Shape> = unknown,
  Errors extends Maybe<Record<string, Maybe<Shape>>> = unknown,
  Create extends Maybe<Shape> = unknown,
  Join extends Maybe<Shape> = unknown,
  Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown,
  Events extends Maybe<Record<string, WrpcRoomEvent>> = unknown,
  Commands extends Maybe<Record<string, WrpcRoomCommand>> = unknown,
> = {
  config?: WrpcRoomConfig;
  state: State extends Shape ? State : undefined;
  initialState: InferShapeOutput<State, undefined>;
  metadata: Metadata extends Shape ? Metadata : undefined;
  initialMetadata: InferShapeOutput<Metadata, undefined>;
  connectionMetadata: ConnectionMetadata extends Shape ? ConnectionMetadata : undefined;
  initialConnectionMetadata: InferShapeOutput<ConnectionMetadata, undefined>;
  errors: Errors extends Record<string, Maybe<Shape>> ? Errors : Record<string, Maybe<Shape>>;
  create: Create extends Shape ? Create : undefined;
  join: Join extends Shape ? Join : undefined;
  broadcasts: Broadcasts extends Record<string, WrpcRoomBroadcast> ? Broadcasts : Record<string, WrpcRoomBroadcast>;
  events: Events extends Record<string, WrpcRoomEvent> ? Events : Record<string, WrpcRoomEvent>;
  commands: Commands extends Record<string, WrpcRoomCommand> ? Commands : Record<string, WrpcRoomCommand>;
};

export type WrpcRoom<
  State extends Maybe<Shape> = unknown,
  Metadata extends Maybe<Shape> = unknown,
  ConnectionMetadata extends Maybe<Shape> = unknown,
  Errors extends Maybe<Record<string, Maybe<Shape>>> = unknown,
  Create extends Maybe<Shape> = unknown,
  Join extends Maybe<Shape> = unknown,
  Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown,
  Events extends Maybe<Record<string, WrpcRoomEvent>> = unknown,
  Commands extends Maybe<Record<string, WrpcRoomCommand>> = unknown,
  WrpcRoomBase extends Maybe<WrpcRoom> = unknown,
> = WrpcRoomDefinition<State, Metadata, ConnectionMetadata, Errors, Create, Join, Broadcasts, Events, Commands> & {
  wrpcRoomBase?: WrpcRoomBase;
  extend: <
    NewState extends Maybe<Shape> = State,
    NewRoomMetadata extends Maybe<Shape> = unknown,
    NewConnectionMetadata extends Maybe<Shape> = unknown,
    NewErrors extends Maybe<Record<string, Maybe<Shape>>> = unknown,
    NewCreate extends Maybe<Shape> = unknown,
    NewJoin extends Maybe<Shape> = unknown,
    NewBroadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown,
    NewEvents extends Maybe<Record<string, WrpcRoomEvent>> = unknown,
    NewCommands extends Maybe<Record<string, WrpcRoomCommand>> = unknown,
  >(
    extender: (
      wrpcRoomBase: WrpcRoomDefinition<
        State,
        Metadata,
        ConnectionMetadata,
        Errors,
        Create,
        Join,
        Broadcasts,
        Events,
        Commands
      >,
    ) => WrpcRoomData<
      NewState,
      NewRoomMetadata,
      NewConnectionMetadata,
      NewErrors,
      NewCreate,
      NewJoin,
      NewBroadcasts,
      NewEvents,
      NewCommands
    >,
  ) => WrpcRoom<
    NewState,
    NewRoomMetadata,
    NewConnectionMetadata,
    NewErrors,
    NewCreate,
    NewJoin,
    NewBroadcasts,
    NewEvents,
    NewCommands,
    WrpcRoom<State, Metadata, ConnectionMetadata, Errors, Create, Join, Broadcasts, Events, Commands>
  >;
};

export const defineWrpcRoom = <
  State extends Maybe<Shape> = unknown,
  Metadata extends Maybe<Shape> = unknown,
  ConnectionMetadata extends Maybe<Shape> = unknown,
  Errors extends Maybe<Record<string, Maybe<Shape>>> = unknown,
  Create extends Maybe<Shape> = unknown,
  Join extends Maybe<Shape> = unknown,
  Broadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown,
  Events extends Maybe<Record<string, WrpcRoomEvent>> = unknown,
  Commands extends Maybe<Record<string, WrpcRoomCommand>> = unknown,
  WrpcRoomBase extends Maybe<WrpcRoom> = unknown,
>(
  wrpcRoomData: WrpcRoomData<State, Metadata, ConnectionMetadata, Errors, Create, Join, Broadcasts, Events, Commands>,
  wrpcRoomBase?: WrpcRoomBase,
): WrpcRoom<State, Metadata, ConnectionMetadata, Errors, Create, Join, Broadcasts, Events, Commands, WrpcRoomBase> => ({
  config: wrpcRoomData.config,
  state: wrpcRoomData.state as State extends Shape ? State : undefined,
  initialState: wrpcRoomData.initialState as InferShapeOutput<State, undefined>,
  metadata: wrpcRoomData.metadata as Metadata extends Shape ? Metadata : undefined,
  initialMetadata: wrpcRoomData.initialMetadata as InferShapeOutput<Metadata, undefined>,
  connectionMetadata: wrpcRoomData.connectionMetadata as ConnectionMetadata extends Shape
    ? ConnectionMetadata
    : undefined,
  initialConnectionMetadata: wrpcRoomData.initialConnectionMetadata as InferShapeOutput<ConnectionMetadata, undefined>,
  errors: (wrpcRoomData.errors ?? {}) as Errors extends Record<string, Maybe<Shape>>
    ? Errors
    : Record<string, Maybe<Shape>>,
  create: wrpcRoomData.create as Create extends Shape ? Create : undefined,
  join: wrpcRoomData.join as Join extends Shape ? Join : undefined,
  broadcasts: (wrpcRoomData.broadcasts ?? {}) as Broadcasts extends Record<string, WrpcRoomBroadcast>
    ? Broadcasts
    : Record<string, WrpcRoomBroadcast>,
  events: (wrpcRoomData.events ?? {}) as Events extends Record<string, WrpcRoomEvent>
    ? Events
    : Record<string, WrpcRoomEvent>,
  commands: (wrpcRoomData.commands ?? {}) as Commands extends Record<string, WrpcRoomCommand>
    ? Commands
    : Record<string, WrpcRoomCommand>,

  wrpcRoomBase,
  extend: <
    NewState extends Maybe<Shape> = State,
    NewMetadata extends Maybe<Shape> = unknown,
    NewConnectionMetadata extends Maybe<Shape> = unknown,
    NewErrors extends Maybe<Record<string, Maybe<Shape>>> = unknown,
    NewCreate extends Maybe<Shape> = unknown,
    NewJoin extends Maybe<Shape> = unknown,
    NewBroadcasts extends Maybe<Record<string, WrpcRoomBroadcast>> = unknown,
    NewEvents extends Maybe<Record<string, WrpcRoomEvent>> = unknown,
    NewCommands extends Maybe<Record<string, WrpcRoomCommand>> = unknown,
  >(
    extender: (
      wrpcRoomBase: WrpcRoomDefinition<
        State,
        Metadata,
        ConnectionMetadata,
        Errors,
        Create,
        Join,
        Broadcasts,
        Events,
        Commands
      >,
    ) => WrpcRoomData<
      NewState,
      NewMetadata,
      NewConnectionMetadata,
      NewErrors,
      NewCreate,
      NewJoin,
      NewBroadcasts,
      NewEvents,
      NewCommands
    >,
  ) => {
    const nextWrpcRoomData = extender({
      config: wrpcRoomData.config,
      state: wrpcRoomData.state as State extends Shape ? State : undefined,
      initialState: wrpcRoomData.initialState as InferShapeOutput<State, undefined>,
      metadata: wrpcRoomData.metadata as Metadata extends Shape ? Metadata : undefined,
      initialMetadata: wrpcRoomData.initialMetadata as InferShapeOutput<Metadata, undefined>,
      connectionMetadata: wrpcRoomData.connectionMetadata as ConnectionMetadata extends Shape
        ? ConnectionMetadata
        : undefined,
      initialConnectionMetadata: wrpcRoomData.initialConnectionMetadata as InferShapeOutput<
        ConnectionMetadata,
        undefined
      >,
      errors: (wrpcRoomData.errors ?? {}) as Errors extends Record<string, Maybe<Shape>>
        ? Errors
        : Record<string, Maybe<Shape>>,
      create: wrpcRoomData.create as Create extends Shape ? Create : undefined,
      join: wrpcRoomData.join as Join extends Shape ? Join : undefined,
      broadcasts: (wrpcRoomData.broadcasts ?? {}) as Broadcasts extends Record<string, WrpcRoomBroadcast>
        ? Broadcasts
        : Record<string, WrpcRoomBroadcast>,
      events: (wrpcRoomData.events ?? {}) as Events extends Record<string, WrpcRoomEvent>
        ? Events
        : Record<string, WrpcRoomEvent>,
      commands: (wrpcRoomData.commands ?? {}) as Commands extends Record<string, WrpcRoomCommand>
        ? Commands
        : Record<string, WrpcRoomCommand>,
    });

    return defineWrpcRoom(nextWrpcRoomData, wrpcRoomBase) as WrpcRoom<
      NewState,
      NewMetadata,
      NewConnectionMetadata,
      NewErrors,
      NewCreate,
      NewJoin,
      NewBroadcasts,
      NewEvents,
      NewCommands,
      WrpcRoom<State, Metadata, ConnectionMetadata, Errors, Create, Join, Broadcasts, Events, Commands, WrpcRoomBase>
    >;
  },
});