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

export type ExtractErrorNames<Errors extends Maybe<Record<string, Maybe<Shape>>>> =
  Errors extends Record<string, Maybe<Shape>> ? keyof Errors : never;

export type ExtractErrorData<
  Errors extends Maybe<Record<string, Maybe<Shape>>>,
  Name extends ExtractErrorNames<Errors>,
  Fallback = unknown,
> =
  Errors extends Record<string, Maybe<Shape>>
    ? Name extends keyof Errors
      ? Errors[Name] extends Shape
        ? Errors[Name]
        : Fallback
      : Fallback
    : Fallback;

export type ExtractRoomErrors<Errors extends Maybe<Record<string, Maybe<Shape>>>> =
  Errors extends Record<string, Maybe<Shape>>
    ? {
        [K in keyof Errors]: WrpcError<K & string, Errors[K] extends Shape ? InferShapeOutput<Errors[K]> : undefined>;
      }[keyof Errors]
    : never;