100 lines
2.6 KiB
TypeScript
100 lines
2.6 KiB
TypeScript
import { LineItem, LineItemAdjustment, LineItemTaxLine } from "@medusajs/medusa"
|
|
import faker from "faker"
|
|
import { DataSource } from "typeorm"
|
|
|
|
type TaxLineFactoryData = {
|
|
rate: number
|
|
code: string
|
|
name: string
|
|
}
|
|
|
|
type LineItemAdjustmentFactoryData = Omit<LineItemAdjustment, "discount_id"> & {
|
|
discount_id: string
|
|
discount_code: string
|
|
}
|
|
|
|
export type LineItemFactoryData = {
|
|
id?: string
|
|
cart_id?: string
|
|
order_id?: string
|
|
variant_id: string | null
|
|
product_id: string | null
|
|
title?: string
|
|
description?: string
|
|
thumbnail?: string
|
|
should_merge?: boolean
|
|
allow_discounts?: boolean
|
|
is_giftcard?: boolean
|
|
unit_price?: number
|
|
quantity?: number
|
|
fulfilled_quantity?: boolean
|
|
shipped_quantity?: boolean
|
|
returned_quantity?: boolean
|
|
tax_lines?: TaxLineFactoryData[]
|
|
adjustments: LineItemAdjustmentFactoryData[]
|
|
includes_tax?: boolean
|
|
order_edit_id?: string
|
|
}
|
|
|
|
export const simpleLineItemFactory = async (
|
|
dataSource: DataSource,
|
|
data: LineItemFactoryData,
|
|
seed?: number
|
|
): Promise<LineItem> => {
|
|
if (
|
|
typeof data.cart_id === "undefined" &&
|
|
typeof data.order_id === "undefined"
|
|
) {
|
|
throw Error()
|
|
}
|
|
|
|
if (typeof seed !== "undefined") {
|
|
faker.seed(seed)
|
|
Math
|
|
}
|
|
|
|
const manager = dataSource.manager
|
|
|
|
const id = data.id || `simple-line-${Math.random() * 1000}`
|
|
const toSave = manager.create(LineItem, {
|
|
id,
|
|
cart_id: data.cart_id,
|
|
order_id: data.order_id,
|
|
title: data.title || faker.commerce.productName(),
|
|
description: data.description || "",
|
|
thumbnail: data.thumbnail || "",
|
|
should_merge:
|
|
typeof data.should_merge !== "undefined" ? data.should_merge : true,
|
|
allow_discounts:
|
|
typeof data.allow_discounts !== "undefined" ? data.allow_discounts : true,
|
|
unit_price: typeof data.unit_price !== "undefined" ? data.unit_price : 100,
|
|
variant_id: data.variant_id,
|
|
quantity: data.quantity || 1,
|
|
fulfilled_quantity: data.fulfilled_quantity || null,
|
|
shipped_quantity: data.shipped_quantity || null,
|
|
returned_quantity: data.returned_quantity || null,
|
|
adjustments: data.adjustments,
|
|
includes_tax: data.includes_tax,
|
|
order_edit_id: data.order_edit_id,
|
|
is_giftcard: data.is_giftcard || false,
|
|
product_id: data.product_id,
|
|
})
|
|
|
|
const line = await manager.save(toSave)
|
|
|
|
if (typeof data.tax_lines !== "undefined") {
|
|
const taxLinesToSave = data.tax_lines.map((tl) =>
|
|
manager.create(LineItemTaxLine, {
|
|
item_id: id,
|
|
rate: tl.rate,
|
|
code: tl.code,
|
|
name: tl.name,
|
|
})
|
|
)
|
|
|
|
await manager.save(taxLinesToSave)
|
|
}
|
|
|
|
return line
|
|
}
|