import { z } from "zod";
import { DATA_LIMIT_METRIC } from "@marzneshin/modules/users";

export const UserSchema = z.object({
    username: z.string().min(1, { message: "Username is required" }),
    note: z.string().nullable(),
    data_limit: z
        .union([z.string().transform((str) => Number(str)), z.number()])
        .refine((val) => val >= 0, { message: "The minimum number is 0" })
        .transform((val) => Math.round(val * DATA_LIMIT_METRIC))
        .nullable()
        .optional(),
    data_limit_reset_strategy: z
        .enum(["no_reset", "day", "week", "month", "year"])
        .or(z.number())
        .or(z.string())
        .nullable()
        .optional(),
    expire_strategy: z.enum(["never", "fixed_date", "start_on_first_use"]),
    usage_duration: z.number().nullable().optional(),
    activation_deadline: z.string().or(z.date()).nullable().optional(),
    expire_date: z.string().or(z.date()).nullable().optional(),
    service_ids: z
        .array(z.number().or(z.string()))
        .nonempty({ message: "At least one service is required" })
        .transform((v) => v.map(Number)),
});

export type UserMutationType = z.infer<typeof UserSchema>;