import type { z } from "@zod/zod";
import type { Document, IndexDescription, ObjectId } from "mongodb";

/**
 * Type alias for Zod schema objects
 */
export type Schema = z.ZodObject<z.ZodRawShape>;

/**
 * Infer the TypeScript type from a Zod schema, including MongoDB Document
 */
export type Infer<T extends Schema> = z.infer<T> & Document;

/**
 * Infer the model type from a Zod schema, including MongoDB Document and ObjectId
 */
export type InferModel<T extends Schema> = Infer<T> & {
  _id?: ObjectId;
};

/**
 * Infer the input type for a Zod schema (handles defaults)
 */
export type Input<T extends Schema> = z.input<T>;

/**
 * Type for indexes that can be passed to the Model constructor
 */
export type Indexes = IndexDescription[];

/**
 * Complete definition of a model, including schema and indexes
 *
 * @example
 * ```ts
 * const userDef: ModelDef<typeof userSchema> = {
 *   schema: userSchema,
 *   indexes: [
 *     { key: { email: 1 }, unique: true },
 *     { key: { name: 1 } }
 *   ]
 * };
 * ```
 */
export type ModelDef<T extends Schema> = {
  schema: T;
  indexes?: Indexes;
};