import type { z } from "zod";

export type InferRow<TTable> = TTable extends { $inferSelect: infer R }
  ? R
  : never;

/**
 * Infer the output type from either a Zod schema or a Drizzle table.
 */
export type InferOutputEntry<T> = T extends z.ZodTypeAny
  ? z.infer<T>
  : T extends { $inferSelect: any }
    ? InferRow<T>
    : never;

type FallbackTableName<Schema> = [keyof Schema & string] extends [never]
  ? string
  : never;

export type OutputAccessor<Schema> = {
  (table: FallbackTableName<Schema>): Array<any>;
  <K extends keyof Schema & string>(table: K): Array<InferOutputEntry<Schema[K]>>;
} & {
  [K in keyof Schema & string]: Array<InferOutputEntry<Schema[K]>>;
};