Make payments schema validator handle partial overrides

Author: mantrakp04

packages/stack-shared/src/config/schema.test.ts

@@ -0,0 +1,57 @@
+import { describe, expect, it } from "vitest";
+import { branchPaymentsSchema } from "./schema";
+
+describe("branchPaymentsSchema", () => {
+  it("accepts partial payments config without products", async () => {
+    await expect(branchPaymentsSchema.validate({
+      blockNewPurchases: true,
+    })).resolves.toMatchObject({
+      blockNewPurchases: true,
+    });
+  });
+
+  it("accepts product lines without products", async () => {
+    await expect(branchPaymentsSchema.validate({
+      productLines: {
+        pro: {
+          displayName: "Pro",
+          customerType: "user",
+        },
+      },
+    })).resolves.toMatchObject({
+      productLines: {
+        pro: {
+          displayName: "Pro",
+          customerType: "user",
+        },
+      },
+    });
+  });
+
+  it("rejects a product that references a missing product line", async () => {
+    await expect(branchPaymentsSchema.validate({
+      products: {
+        pro: {
+          customerType: "user",
+          productLineId: "missing-line",
+        },
+      },
+    })).rejects.toThrow('Product "pro" specifies product line ID "missing-line", but that product line does not exist');
+  });
+
+  it("rejects a product whose customer type differs from its product line", async () => {
+    await expect(branchPaymentsSchema.validate({
+      productLines: {
+        teamLine: {
+          customerType: "team",
+        },
+      },
+      products: {
+        pro: {
+          customerType: "user",
+          productLineId: "teamLine",
+        },
+      },
+    })).rejects.toThrow('Product "pro" has customer type "user" but its product line "teamLine" has customer type "team"');
+  });
+});

packages/stack-shared/src/config/schema.ts

@@ -181,9 +181,13 @@ export const branchPaymentsSchema = yupObject({
   'Product customer type must match its product line customer type',
   function(this: yup.TestContext<yup.AnyObject>, value) {
     if (!value) return true;
-    for (const [productId, product] of Object.entries(value.products)) {
-      if (!product.productLineId) continue;
-      const productLine = getOrUndefined(value.productLines, product.productLineId);
+    const products = Reflect.get(value, "products");
+    if (!isObjectLike(products)) return true;
+
+    const productLines = Reflect.get(value, "productLines");
+    for (const [productId, product] of Object.entries(products)) {
+      if (product.productLineId == null) continue;
+      const productLine = isObjectLike(productLines) ? getOrUndefined(productLines, product.productLineId) : undefined;
       if (productLine === undefined) {
         return this.createError({
           message: `Product "${productId}" specifies product line ID "${product.productLineId}", but that product line does not exist`,