Class ZodSocketMessageSender

packages/core/src/v3/zodSocket.ts:221–273

}[keyof TMessageCatalog];

export class ZodSocketMessageSender<TMessageCatalog extends ZodSocketMessageCatalogSchema> {
  #schema: TMessageCatalog;
  #socket: ZodSocket<any, TMessageCatalog>;

  constructor(options: ZodSocketMessageSenderOptions<TMessageCatalog>) {
    this.#schema = options.schema;
    this.#socket = options.socket;
  }

  public send<K extends GetSocketMessagesWithoutCallback<TMessageCatalog>>(
    type: K,
    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>
  ): void {
    const schema = this.#schema[type]["message"];

    if (!schema) {
      throw new Error(`Unknown message type: ${type as string}`);
    }

    const parsedPayload = schema.safeParse(payload);

    if (!parsedPayload.success) {
      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
    }

    // @ts-expect-error
    this.#socket.emit(type, { payload, version: "v1" });

    return;
  }

  public async sendWithAck<K extends GetSocketMessagesWithCallback<TMessageCatalog>>(
    type: K,
    payload: z.input<GetSocketMessageSchema<TMessageCatalog, K>>
  ): Promise<z.infer<GetSocketCallbackSchema<TMessageCatalog, K>>> {
    const schema = this.#schema[type]["message"];

    if (!schema) {
      throw new Error(`Unknown message type: ${type as string}`);
    }

    const parsedPayload = schema.safeParse(payload);

    if (!parsedPayload.success) {
      throw new Error(`Failed to parse message payload: ${JSON.stringify(parsedPayload.error)}`);
    }

    // @ts-expect-error
    const callbackResult = await this.#socket.emitWithAck(type, { payload, version: "v1" });

    return callbackResult;
  }
}

export type ZodSocket<
  TListenEvents extends ZodSocketMessageCatalogSchema,