fix

medusajs · Jan 15, 2024 · b7290e5 · b7290e5
1 parent 3eab5e0
commit b7290e5
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 8 deletions.
diff --git a/packages/cart/src/services/cart-module.ts b/packages/cart/src/services/cart-module.ts
@@ -254,7 +254,7 @@ export default class CartModuleService implements ICartModuleService {
   addLineItems(
     data: CartTypes.CreateLineItemForCartDTO,
     sharedContext?: Context
-  ): Promise<CartTypes.CartLineItemDTO[]>
+  ): Promise<CartTypes.CartLineItemDTO>
   addLineItems(
     data: CartTypes.CreateLineItemForCartDTO[],
     sharedContext?: Context
@@ -264,21 +264,31 @@ export default class CartModuleService implements ICartModuleService {
     items: CartTypes.CreateLineItemDTO[],
     sharedContext?: Context
   ): Promise<CartTypes.CartLineItemDTO[]>
+  addLineItems(
+    cartId: string,
+    items: CartTypes.CreateLineItemDTO,
+    sharedContext?: Context
+  ): Promise<CartTypes.CartLineItemDTO>
 
   @InjectManager("baseRepository_")
   async addLineItems(
     cartIdOrData:
       | string
       | CartTypes.CreateLineItemForCartDTO[]
       | CartTypes.CreateLineItemForCartDTO,
-    dataOrSharedContext?: CartTypes.CreateLineItemDTO[] | Context,
+    dataOrSharedContext?:
+      | CartTypes.CreateLineItemDTO[]
+      | CartTypes.CreateLineItemDTO
+      | Context,
     @MedusaContext() sharedContext: Context = {}
-  ): Promise<CartTypes.CartLineItemDTO[]> {
+  ): Promise<CartTypes.CartLineItemDTO[] | CartTypes.CartLineItemDTO> {
     let items: LineItem[] = []
     if (isString(cartIdOrData)) {
       items = await this.addLineItems_(
         cartIdOrData,
-        dataOrSharedContext as CartTypes.CreateLineItemDTO[],
+        dataOrSharedContext as
+          | CartTypes.CreateLineItemDTO[]
+          | CartTypes.CreateLineItemDTO,
         sharedContext
       )
     } else if (Array.isArray(cartIdOrData)) {
@@ -304,19 +314,21 @@ export default class CartModuleService implements ICartModuleService {
   @InjectTransactionManager("baseRepository_")
   protected async addLineItems_(
     cartId: string,
-    data: CartTypes.CreateLineItemDTO[],
+    data: CartTypes.CreateLineItemDTO[] | CartTypes.CreateLineItemDTO,
     @MedusaContext() sharedContext: Context = {}
   ): Promise<LineItem[]> {
     const cart = await this.retrieve(cartId, { select: ["id"] }, sharedContext)
 
-    const items = data.map((item) => {
+    const items = Array.isArray(data) ? data : [data]
+
+    const toUpdate = items.map((item) => {
       return {
         ...item,
         cart_id: cart.id,
       }
     })
 
-    return await this.addLineItemsBulk_(items, sharedContext)
+    return await this.addLineItemsBulk_(toUpdate, sharedContext)
   }
 
   @InjectTransactionManager("baseRepository_")

diff --git a/packages/types/src/cart/service.ts b/packages/types/src/cart/service.ts
@@ -76,7 +76,7 @@ export interface ICartModuleService extends IModuleService {
   addLineItems(
     data: CreateLineItemForCartDTO,
     sharedContext?: Context
-  ): Promise<CartLineItemDTO[]>
+  ): Promise<CartLineItemDTO>
   addLineItems(
     data: CreateLineItemForCartDTO[],
     sharedContext?: Context
@@ -86,6 +86,11 @@ export interface ICartModuleService extends IModuleService {
     items: CreateLineItemDTO[],
     sharedContext?: Context
   ): Promise<CartLineItemDTO[]>
+  addLineItems(
+    cartId: string,
+    items: CreateLineItemDTO,
+    sharedContext?: Context
+  ): Promise<CartLineItemDTO>
 
   updateLineItems(
     data: UpdateLineItemDTO[],