@labdigital/commercetools-mock 2.17.1 → 2.18.1

package/src/repositories/business-unit.ts

@@ -1,4 +1,9 @@
 import {
+	BusinessUnitChangeAssociateModeAction,
+	BusinessUnitChangeStatusAction,
+	BusinessUnitUpdateAction,
+	CompanyDraft,
+	DivisionDraft,
 	type Associate,
 	type BusinessUnit,
 	type BusinessUnitAddAddressAction,
@@ -13,35 +18,28 @@ import {
 	type BusinessUnitSetStoreModeAction,
 	type Company,
 	type Division,
-	BusinessUnitChangeStatusAction,
-	CompanyDraft,
-	DivisionDraft,
-} from '@commercetools/platform-sdk'
+} from "@commercetools/platform-sdk";
+import { getBaseResourceProperties } from "../helpers";
+import { AbstractStorage } from "../storage/abstract";
+import { Writable } from "../types";
 import {
 	AbstractResourceRepository,
+	AbstractUpdateHandler,
+	UpdateHandlerInterface,
 	type RepositoryContext,
-} from './abstract.js'
-import { getBaseResourceProperties } from '../helpers.js'
+} from "./abstract";
 import {
 	createAddress,
 	createAssociate,
 	createCustomFields,
 	getBusinessUnitKeyReference,
 	getStoreKeyReference,
-} from './helpers.js'
-import { Writable } from '../types.js'
-
-export class BusinessUnitRepository extends AbstractResourceRepository<'business-unit'> {
-	getTypeId() {
-		return 'business-unit' as const
-	}
+} from "./helpers";
 
-	private _isCompanyDraft(draft: BusinessUnitDraft): draft is CompanyDraft {
-		return draft.unitType === 'Company'
-	}
-
-	private _isDivisionDraft(draft: BusinessUnitDraft): draft is DivisionDraft {
-		return draft.unitType === 'Division'
+export class BusinessUnitRepository extends AbstractResourceRepository<"business-unit"> {
+	constructor(storage: AbstractStorage) {
+		super("business-unit", storage);
+		this.actions = new BusinessUnitUpdateHandler(this._storage);
 	}
 
 	create(context: RepositoryContext, draft: BusinessUnitDraft): BusinessUnit {
@@ -50,27 +48,27 @@ export class BusinessUnitRepository extends AbstractResourceRepository<'business
 			key: draft.key,
 			status: draft.status,
 			stores: draft.stores?.map((s) =>
-				getStoreKeyReference(s, context.projectKey, this._storage)
+				getStoreKeyReference(s, context.projectKey, this._storage),
 			),
 			storeMode: draft.storeMode,
 			name: draft.name,
 			contactEmail: draft.contactEmail,
 			addresses: draft.addresses?.map((a) =>
-				createAddress(a, context.projectKey, this._storage)
+				createAddress(a, context.projectKey, this._storage),
 			),
 			custom: createCustomFields(
 				draft.custom,
 				context.projectKey,
-				this._storage
+				this._storage,
 			),
 			shippingAddressIds: draft.shippingAddresses,
 			defaultShippingAddressId: draft.defaultShippingAddress,
 			billingAddressIds: draft.billingAddresses,
 			associateMode: draft.associateMode,
 			associates: draft.associates?.map((a) =>
-				createAssociate(a, context.projectKey, this._storage)
+				createAssociate(a, context.projectKey, this._storage),
 			),
-		}
+		};
 
 		if (this._isDivisionDraft(draft)) {
 			const division = {
@@ -78,138 +76,160 @@ export class BusinessUnitRepository extends AbstractResourceRepository<'business
 				parentUnit: getBusinessUnitKeyReference(
 					draft.parentUnit,
 					context.projectKey,
-					this._storage
+					this._storage,
 				),
-			} as Division
+			} as Division;
 
-			this.saveNew(context, division)
-			return division
+			this.saveNew(context, division);
+			return division;
 		} else if (this._isCompanyDraft(draft)) {
-			const company = resource as Company
+			const company = resource as Company;
 
-			this.saveNew(context, company)
-			return company
+			this.saveNew(context, company);
+			return company;
 		}
 
-		throw new Error('Invalid business unit type')
+		throw new Error("Invalid business unit type");
 	}
 
-	actions = {
-		addAddress: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ address }: BusinessUnitAddAddressAction
-		) => {
-			const newAddress = createAddress(
-				address,
-				context.projectKey,
-				this._storage
-			)
-			if (newAddress) {
-				resource.addresses.push(newAddress)
-			}
-		},
-		addAssociate: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ associate }: BusinessUnitAddAssociateAction
-		) => {
-			const newAssociate = createAssociate(
-				associate,
-				context.projectKey,
-				this._storage
-			)
-			if (newAssociate) {
-				resource.associates.push(newAssociate)
-			}
-		},
-		setAssociates: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ associates }: BusinessUnitSetAssociatesAction
-		) => {
-			const newAssociates = associates
-				.map((a) => createAssociate(a, context.projectKey, this._storage))
-				.filter((a): a is Writable<Associate> => a !== undefined)
-			resource.associates = newAssociates || undefined
-		},
-		setContactEmail: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ contactEmail }: BusinessUnitSetContactEmailAction
-		) => {
-			resource.contactEmail = contactEmail
-		},
-		setStoreMode: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ storeMode }: BusinessUnitSetStoreModeAction
-		) => {
-			resource.storeMode = storeMode
-		},
-		changeAssociateMode: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ storeMode }: BusinessUnitSetStoreModeAction
-		) => {
-			resource.associateMode = storeMode
-		},
-		changeName: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ name }: BusinessUnitChangeNameAction
-		) => {
-			resource.name = name
-		},
-		changeAddress: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ address }: BusinessUnitChangeAddressAction
-		) => {
-			const newAddress = createAddress(
-				address,
-				context.projectKey,
-				this._storage
-			)
-			if (newAddress) {
-				resource.addresses.push(newAddress)
-			}
-		},
-		addStore: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ store }: BusinessUnitAddStoreAction
-		) => {
-			const newStore = getStoreKeyReference(
-				store,
-				context.projectKey,
-				this._storage
-			)
-			if (newStore) {
-				if (!resource.stores) {
-					resource.stores = []
-				}
-
-				resource.stores.push(newStore)
+	private _isCompanyDraft(draft: BusinessUnitDraft): draft is CompanyDraft {
+		return draft.unitType === "Company";
+	}
+
+	private _isDivisionDraft(draft: BusinessUnitDraft): draft is DivisionDraft {
+		return draft.unitType === "Division";
+	}
+}
+
+class BusinessUnitUpdateHandler
+	extends AbstractUpdateHandler
+	implements
+		Partial<UpdateHandlerInterface<BusinessUnit, BusinessUnitUpdateAction>>
+{
+	addAddress(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ address }: BusinessUnitAddAddressAction,
+	) {
+		const newAddress = createAddress(
+			address,
+			context.projectKey,
+			this._storage,
+		);
+		if (newAddress) {
+			resource.addresses.push(newAddress);
+		}
+	}
+
+	addAssociate(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ associate }: BusinessUnitAddAssociateAction,
+	) {
+		const newAssociate = createAssociate(
+			associate,
+			context.projectKey,
+			this._storage,
+		);
+		if (newAssociate) {
+			resource.associates.push(newAssociate);
+		}
+	}
+
+	addStore(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ store }: BusinessUnitAddStoreAction,
+	) {
+		const newStore = getStoreKeyReference(
+			store,
+			context.projectKey,
+			this._storage,
+		);
+		if (newStore) {
+			if (!resource.stores) {
+				resource.stores = [];
 			}
-		},
-		changeParentUnit: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ parentUnit }: BusinessUnitChangeParentUnitAction
-		) => {
-			resource.parentUnit = getBusinessUnitKeyReference(
-				parentUnit,
-				context.projectKey,
-				this._storage
-			)
-		},
-		changeStatus: (
-			context: RepositoryContext,
-			resource: Writable<BusinessUnit>,
-			{ status }: BusinessUnitChangeStatusAction
-		) => {
-			resource.status = status
-		},
+
+			resource.stores.push(newStore);
+		}
+	}
+
+	changeAddress(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ address }: BusinessUnitChangeAddressAction,
+	) {
+		const newAddress = createAddress(
+			address,
+			context.projectKey,
+			this._storage,
+		);
+		if (newAddress) {
+			resource.addresses.push(newAddress);
+		}
+	}
+
+	changeAssociateMode(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ associateMode }: BusinessUnitChangeAssociateModeAction,
+	) {
+		resource.associateMode = associateMode;
+	}
+
+	changeName(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ name }: BusinessUnitChangeNameAction,
+	) {
+		resource.name = name;
+	}
+
+	changeParentUnit(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ parentUnit }: BusinessUnitChangeParentUnitAction,
+	) {
+		resource.parentUnit = getBusinessUnitKeyReference(
+			parentUnit,
+			context.projectKey,
+			this._storage,
+		);
+	}
+
+	changeStatus(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ status }: BusinessUnitChangeStatusAction,
+	) {
+		resource.status = status;
+	}
+
+	setAssociates(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ associates }: BusinessUnitSetAssociatesAction,
+	) {
+		const newAssociates = associates
+			.map((a) => createAssociate(a, context.projectKey, this._storage))
+			.filter((a): a is Writable<Associate> => a !== undefined);
+		resource.associates = newAssociates || undefined;
+	}
+
+	setContactEmail(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ contactEmail }: BusinessUnitSetContactEmailAction,
+	) {
+		resource.contactEmail = contactEmail;
+	}
+
+	setStoreMode(
+		context: RepositoryContext,
+		resource: Writable<BusinessUnit>,
+		{ storeMode }: BusinessUnitSetStoreModeAction,
+	) {
+		resource.storeMode = storeMode;
 	}
 }