import { withValidation } from "next-validations";
import { z } from "zod";

import { stringOrNumber } from "@calcom/prisma/zod-utils";

import { baseApiParams } from "./baseApiParams";

// Extracted out as utility function so can be reused
// at different endpoints that require this validation.
export const schemaQueryUserId = baseApiParams
  .extend({
    userId: stringOrNumber,
  })
  .strict();

export const schemaQuerySingleOrMultipleUserIds = z.object({
  userId: z.union([stringOrNumber, z.array(stringOrNumber)]),
});

export const withValidQueryUserId = withValidation({
  schema: schemaQueryUserId,
  type: "Zod",
  mode: "query",
});