From 16eddcefd2bc8c20998b652896a441ca6b1dbae2 Mon Sep 17 00:00:00 2001
From: John Yeo <johnyeocx@gmail.com>
Date: Thu, 13 Mar 2025 03:01:05 +0800
Subject: [PATCH] fix: server build

---
 .../add-product/createFullCusProduct.ts       |  2 +-
 .../customers/products/cusProductUtils.ts     | 19 +++++++++++++++++++
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/server/src/internal/customers/add-product/createFullCusProduct.ts b/server/src/internal/customers/add-product/createFullCusProduct.ts
index c4dc72928..4d87f5b2e 100644
--- a/server/src/internal/customers/add-product/createFullCusProduct.ts
+++ b/server/src/internal/customers/add-product/createFullCusProduct.ts
@@ -33,7 +33,7 @@ import { getEntRelatedPrice } from "@/internal/products/entitlements/entitlement
 import { CusService } from "../CusService.js";
 import { getExistingCusProducts } from "./handleExistingProduct.js";
 import { isOneOff } from "@/internal/products/productUtils.js";
-import { searchCusProducts } from "tests/utils/genUtils.js";
+import { searchCusProducts } from "@/internal/customers/products/cusProductUtils.js";
 import { updateOneTimeCusProduct } from "./createOneTimeCusProduct.js";
 import { initCusEntitlement } from "./initCusEnt.js";
 import { createLogtailWithContext } from "@/external/logtail/logtailUtils.js";
diff --git a/server/src/internal/customers/products/cusProductUtils.ts b/server/src/internal/customers/products/cusProductUtils.ts
index a0fdc6e90..7c7792d88 100644
--- a/server/src/internal/customers/products/cusProductUtils.ts
+++ b/server/src/internal/customers/products/cusProductUtils.ts
@@ -391,3 +391,22 @@ export const fullCusProductToProduct = (cusProduct: FullCusProduct) => {
     entitlements: cusProduct.customer_entitlements.map((ce) => ce.entitlement),
   };
 };
+
+export const searchCusProducts = ({
+  productId,
+  cusProducts,
+  status,
+}: {
+  productId: string;
+  cusProducts: FullCusProduct[];
+  status?: CusProductStatus;
+}) => {
+  if (!cusProducts) {
+    return undefined;
+  }
+  return cusProducts.find(
+    (cusProduct: FullCusProduct) =>
+      cusProduct.product.id === productId &&
+      (status ? cusProduct.status === status : true)
+  );
+};