Files
cfw-autumn/shared/models/attachModels/attachBody.ts
2025-09-09 10:36:57 -07:00

127 lines
3.1 KiB
TypeScript

import { z } from "zod";
import { FeatureOptionsSchema } from "../cusProductModels/cusProductModels.js";
import { ProductItemSchema } from "../productV2Models/productItemModels/productItemModels.js";
import { CreateFreeTrialSchema } from "../productModels/freeTrialModels/freeTrialModels.js";
import { notNullish } from "../../utils/utils.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>;