import { z } from "zod/v4";
import { notNullish } from "../../utils/utils.js";
import { FeatureOptionsSchema } from "../cusProductModels/cusProductModels.js";
import { CreateFreeTrialSchema } from "../productModels/freeTrialModels/freeTrialModels.js";
import { ProductItemSchema } from "../productV2Models/productItemModels/productItemModels.js";

export const ProductOptions = z.object({
	product_id: z.string(),
	quantity: z.number().nullish(),
	entity_id: z.string().nullish(),
	options: z.array(FeatureOptionsSchema).nullish(),
});

export const AttachBodySchema = z
	.object({
		// Customer Info
		customer_id: z
			.string()
			.describe("ID of the customer to attach the product to"),

		customer_data: z
			.any()
			.nullish()
			.describe("Customer data if using attach to auto create customer"),

		// Entity Info
		entity_id: z.string().nullish(),
		entity_data: z.any().nullish(),

		// Product Info
		product_id: z.string().nullish(),
		product_ids: z.array(z.string()).min(1).nullish(),
		options: z.array(FeatureOptionsSchema).nullish(),

		products: z.array(ProductOptions).nullish(),

		// Custom Product
		is_custom: z.boolean().optional(),
		items: z.array(ProductItemSchema).optional(),
		free_trial: CreateFreeTrialSchema.or(z.boolean()).optional(),

		// New Version
		version: z.number().optional(),

		// Others
		success_url: z.string().optional(),
		force_checkout: z.boolean().optional(),
		invoice_only: z.boolean().optional(),
		metadata: z.any().optional(),
		billing_cycle_anchor: z.number().optional(),
		checkout_session_params: z.any().optional(),
		reward: z.string().or(z.array(z.string())).optional(),
		invoice: z.boolean().optional(),
		enable_product_immediately: z.boolean().optional(),
		finalize_invoice: z.boolean().optional(),

		// Checkout params
		skip_checkout: z.boolean().optional(),
		setup_payment: z.boolean().optional(),
	})
	.refine(
		(data) => {
			if (!data.product_id && !data.product_ids && !data.products) {
				return false;
			}

			return true;
		},
		{
			message: "`product_id` is not provided",
		},
	)
	.refine(
		(data) => {
			if (data.product_id && data.product_ids) {
				return false;
			}

			return true;
		},
		{
			message: "provide either one `product_id` or `product_ids`",
		},
	)
	.refine(
		(data) => {
			if (
				notNullish(data.product_ids) &&
				new Set(data.product_ids).size !== data.product_ids!.length
			) {
				return false;
			}

			return true;
		},
		{
			message: "Can't pass in duplicate product_ids",
		},
	)
	.refine(
		(data) => {
			if (data.product_ids && data.is_custom) {
				return false;
			}

			return true;
		},
		{
			message: "Can't pass in product_ids if is_custom is true",
		},
	)
	.refine(
		(data) => {
			if (data.items && !data.is_custom) {
				return false;
			}

			return true;
		},
		{
			message: "Can't pass in items if is_custom is false",
		},
	);

export type AttachBody = z.infer<typeof AttachBodySchema>;
export type ProductOptions = z.infer<typeof ProductOptions>;