@unchainedshop/core 4.0.0-rc.2 → 4.0.0-rc.21

package/src/directors/WorkerAdapter.ts

@@ -1,8 +1,8 @@
 import { BaseAdapter, IBaseAdapter } from '@unchainedshop/utils';
 import { WorkResult } from '@unchainedshop/core-worker';
-import { Modules } from '../modules.js';
+import { ModuleOptions, Modules } from '../modules.js';
 import { Services } from '../services/index.js';
-import { BulkImporter } from '../core-index.js';
+import { BulkImporter } from '../bulk-importer/index.js';
 
 export type IWorkerAdapter<Input, Output> = IBaseAdapter & {
   type: string;
@@ -11,7 +11,12 @@ export type IWorkerAdapter<Input, Output> = IBaseAdapter & {
 
   doWork: (
     input: Input,
-    unchainedAPI: { modules: Modules; services: Services; bulkImporter: BulkImporter },
+    unchainedAPI: {
+      modules: Modules;
+      services: Services;
+      bulkImporter: BulkImporter;
+      options: ModuleOptions;
+    },
     workId: string,
   ) => Promise<WorkResult<Output>>;
 };
@@ -21,6 +26,6 @@ export const WorkerAdapter: Omit<IWorkerAdapter<any, void>, 'key' | 'label' | 't
   external: false,
 
   async doWork() {
-    return { success: false, result: null };
+    return { success: false };
   },
 };

package/src/directors/WorkerDirector.ts

@@ -13,7 +13,7 @@ export type WorkScheduleConfiguration = Pick<
 > & {
   type: string;
   input?: (workData: Omit<WorkData, 'input'>) => Promise<Record<string, any> | null>;
-  schedule: ScheduleData;
+  schedule?: ScheduleData;
   scheduleId?: string;
 };
 
@@ -24,7 +24,7 @@ export type IWorkerDirector = IBaseDirector<IWorkerAdapter<any, any>> & {
   configureAutoscheduling: (workScheduleConfiguration: WorkScheduleConfiguration) => void;
   getAutoSchedules: () => [string, WorkScheduleConfiguration][];
   doWork: (work: Work, unchainedAPI) => Promise<WorkResult>;
-  processNextWork: (unchainedAPI: { modules: Modules }, workerId?: string) => Promise<Work>;
+  processNextWork: (unchainedAPI: { modules: Modules }, workerId?: string) => Promise<Work | null>;
 };
 
 const AutoScheduleMap = new Map<string, WorkScheduleConfiguration>();
@@ -36,7 +36,7 @@ const baseDirector = BaseDirector<IWorkerAdapter<any, any>>('WorkerDirector', {
 export const WorkerDirector: IWorkerDirector = {
   ...baseDirector,
 
-  getActivePluginTypes: ({ external } = { external: null }) => {
+  getActivePluginTypes: ({ external } = {}) => {
     return WorkerDirector.getAdapters()
       .filter((adapter) => {
         if (external === null || external === undefined) return true;
@@ -55,8 +55,7 @@ export const WorkerDirector: IWorkerDirector = {
   },
 
   disableAutoscheduling: (type) => {
-    const config = AutoScheduleMap.get(type);
-    AutoScheduleMap.set(type, { ...config, schedule: null });
+    AutoScheduleMap.delete(type);
   },
 
   unregisterAdapter: (key) => {
@@ -110,7 +109,7 @@ export const WorkerDirector: IWorkerDirector = {
     }
   },
 
-  processNextWork: async (unchainedAPI: { modules: Modules }, workerId?: string): Promise<Work> => {
+  processNextWork: async (unchainedAPI: { modules: Modules }, workerId?: string) => {
     const adapters = WorkerDirector.getAdapters();
 
     const allocationMap = await unchainedAPI.modules.worker.allocationMap();

package/src/factory/index.ts

@@ -0,0 +1,3 @@
+import registerProductDiscoverabilityFilter from './registerProductDiscoverabilityFilter.js';
+
+export { registerProductDiscoverabilityFilter };

package/src/factory/registerProductDiscoverabilityFilter.ts

@@ -0,0 +1,36 @@
+import { FilterAdapter, FilterDirector } from '../core-index.js';
+
+export default function registerProductDiscoverabilityFilter({
+  orderIndex = 0,
+  hiddenTagValue = 'hidden',
+}: {
+  orderIndex?: number;
+  hiddenTagValue?: string;
+}) {
+  FilterDirector.registerAdapter({
+    ...FilterAdapter,
+
+    key: `shop.unchained.filters.product-discoverability-${crypto.randomUUID()}`,
+    label: 'Product Discoverability Filter',
+    version: '1.0.0',
+    orderIndex,
+
+    actions: (params) => {
+      return {
+        ...FilterAdapter.actions(params),
+
+        transformProductSelector: async (lastSelector, options) => {
+          const { key } = options || {};
+
+          if (!key) {
+            const newSelector = { ...lastSelector };
+            if (!newSelector.$and) newSelector.$and = [];
+            newSelector.$and.push({ tags: { $ne: hiddenTagValue } });
+            return newSelector;
+          }
+          return lastSelector;
+        },
+      };
+    },
+  });
+}

package/src/modules.ts

@@ -69,17 +69,17 @@ export interface ModuleOptions {
   assortments?: AssortmentsSettingsOptions;
   products?: ProductsSettingsOptions;
   delivery?: DeliverySettingsOptions;
+  payment?: PaymentSettingsOptions;
   filters?: FiltersSettingsOptions;
   enrollments?: EnrollmentsSettingsOptions;
   orders?: OrdersSettingsOptions;
   quotations?: QuotationsSettingsOptions;
   files?: FilesSettingsOptions;
-  payment?: PaymentSettingsOptions;
   worker?: WorkerSettingsOptions;
   users?: UserSettingsOptions;
 }
 
-const initModules = async (
+export default async function initModules(
   {
     db,
     migrationRepository,
@@ -87,7 +87,7 @@ const initModules = async (
   }: {
     db: mongodb.Db;
     migrationRepository: MigrationRepository<unknown>;
-    options?: ModuleOptions;
+    options: ModuleOptions;
   },
   customModules: Record<
     string,
@@ -95,7 +95,7 @@ const initModules = async (
       configure: (params: ModuleInput<any>) => any;
     }
   >,
-): Promise<Modules> => {
+): Promise<Modules> {
   const assortments = await configureAssortmentsModule({
     db,
     options: options.assortments,
@@ -154,6 +154,7 @@ const initModules = async (
   const products = await configureProductsModule({
     db,
     migrationRepository,
+    options: options.products,
   });
   const quotations = await configureQuotationsModule({
     db,
@@ -203,6 +204,4 @@ const initModules = async (
   }
 
   return modules;
-};
-
-export default initModules;
+}

package/src/services/activateEnrollment.ts

@@ -1,30 +1,26 @@
 import { Enrollment, EnrollmentStatus } from '@unchainedshop/core-enrollments';
 import { Modules } from '../modules.js';
 import { processEnrollmentService } from './processEnrollment.js';
+import { addMessageService } from './addMessage.js';
 
 export async function activateEnrollmentService(this: Modules, enrollment: Enrollment) {
   if (enrollment.status === EnrollmentStatus.TERMINATED) return enrollment;
 
-  let updatedEnrollment = await this.enrollments.updateStatus(enrollment._id, {
+  let updatedEnrollment = (await this.enrollments.updateStatus(enrollment._id, {
     status: EnrollmentStatus.ACTIVE,
     info: 'activated manually',
-  });
+  })) as Enrollment;
 
-  updatedEnrollment = await processEnrollmentService.bind(this)(updatedEnrollment);
+  updatedEnrollment = (await processEnrollmentService.bind(this)(updatedEnrollment)) as Enrollment;
 
   const user = await this.users.findUserById(enrollment.userId);
   const locale = this.users.userLocale(user);
 
-  await this.worker.addWork({
-    type: 'MESSAGE',
-    retries: 0,
-    input: {
-      reason: 'status_change',
-      locale: locale.baseName,
-      template: 'ENROLLMENT_STATUS',
-      enrollmentId: updatedEnrollment._id,
-    },
+  await addMessageService.bind(this)('ENROLLMENT_STATUS', {
+    reason: 'status_change',
+    locale: locale.baseName,
+    enrollmentId: updatedEnrollment!._id,
   });
 
-  return updatedEnrollment;
+  return updatedEnrollment!;
 }

package/src/services/addMessage.ts

@@ -0,0 +1,20 @@
+import { Work } from '@unchainedshop/core-worker';
+import { MessagingDirector } from '../core-index.js';
+import { Modules } from '../modules.js';
+
+export async function addMessageService<T = Record<string, any>>(
+  this: Modules,
+  template: string,
+  input: T,
+): Promise<Work | null> {
+  const templateResolver = MessagingDirector.getTemplate(template); // Validate template exists
+  if (!templateResolver) return null;
+  return this.worker.addWork({
+    type: 'MESSAGE',
+    input: {
+      template,
+      ...input,
+    },
+    retries: 0,
+  });
+}

package/src/services/calculateDiscountTotal.ts

@@ -20,19 +20,19 @@ export async function calculateDiscountTotalService(
 
   // Delivery discounts
   const orderDelivery = await this.orders.deliveries.findDelivery({
-    orderDeliveryId: order.deliveryId,
+    orderDeliveryId: order.deliveryId!,
   });
   const orderDeliveryDiscountSum = DeliveryPricingSheet({
-    calculation: orderDelivery.calculation || [],
+    calculation: orderDelivery?.calculation || [],
     currencyCode: order.currencyCode,
   }).total({ category: DeliveryPricingRowCategory.Discount, discountId: orderDiscountId });
 
   // Payment discounts
   const orderPayment = await this.orders.payments.findOrderPayment({
-    orderPaymentId: order.paymentId,
+    orderPaymentId: order.paymentId!,
   });
   const orderPaymentDiscountSum = PaymentPricingSheet({
-    calculation: orderPayment.calculation || [],
+    calculation: orderPayment?.calculation || [],
     currencyCode: order.currencyCode,
   }).total({ category: PaymentPricingRowCategory.Discount, discountId: orderDiscountId });
 

package/src/services/checkoutOrder.ts

@@ -3,6 +3,7 @@ import { Modules } from '../modules.js';
 import { nextUserCartService } from './nextUserCart.js';
 import { validateOrderService } from './validateOrder.js';
 import { processOrderService } from './processOrder.js';
+import { User } from '@unchainedshop/core-users';
 
 export async function checkoutOrderService(
   this: Modules,
@@ -15,6 +16,7 @@ export async function checkoutOrderService(
   },
 ) {
   const order = await this.orders.findOrder({ orderId });
+  if (!order) return null;
   if (order.status !== null) return order;
 
   await validateOrderService.bind(this)(order);
@@ -25,15 +27,18 @@ export async function checkoutOrderService(
     const processedOrder = await processOrderService.bind(this)(order, transactionContext);
 
     // After checkout, store last checkout information on user
-    await this.users.updateLastBillingAddress(processedOrder.userId, processedOrder.billingAddress);
-    await this.users.updateLastContact(processedOrder.userId, processedOrder.contact);
-
+    if (processedOrder.billingAddress) {
+      await this.users.updateLastBillingAddress(processedOrder.userId, processedOrder.billingAddress);
+    }
+    if (processedOrder.contact) {
+      await this.users.updateLastContact(processedOrder.userId, processedOrder.contact);
+    }
     // Then eventually build next cart
-    const user = await this.users.findUserById(processedOrder.userId);
+    const user = (await this.users.findUserById(processedOrder.userId)) as User;
     const locale = this.users.userLocale(user);
     await nextUserCartService.bind(this)({
       user,
-      countryCode: locale.region,
+      countryCode: locale?.region || processedOrder.countryCode,
     });
 
     return processedOrder;

package/src/services/createDownloadStream.ts

@@ -11,6 +11,7 @@ export async function createDownloadStreamService(
 ) {
   const fileAdapter = getFileAdapter();
   const file = await this.files.findFile({ fileId });
+  if (!file) return null;
   const stream = fileAdapter.createDownloadStream(file, { modules: this });
   return stream;
 }

package/src/services/createEnrollmentFromCheckout.ts

@@ -25,42 +25,44 @@ export async function createEnrollmentFromCheckoutService(
   const orderId = order._id;
 
   const payment = await this.orders.payments.findOrderPayment({
-    orderPaymentId: order.paymentId,
+    orderPaymentId: order.paymentId!,
   });
   const delivery = await this.orders.deliveries.findDelivery({
-    orderDeliveryId: order.deliveryId,
+    orderDeliveryId: order.deliveryId!,
   });
 
   const template = {
-    billingAddress: order.billingAddress,
-    contact: order.contact,
+    billingAddress: order.billingAddress!,
+    contact: order.contact!,
     countryCode: order.countryCode,
     currencyCode: order.currencyCode,
     delivery: {
-      deliveryProviderId: delivery.deliveryProviderId,
-      context: delivery.context,
+      deliveryProviderId: delivery!.deliveryProviderId,
+      context: delivery!.context,
     },
     orderIdForFirstPeriod: orderId,
     payment: {
-      paymentProviderId: payment.paymentProviderId,
-      context: payment.context,
+      paymentProviderId: payment!.paymentProviderId,
+      context: payment!.context,
     },
     userId: order.userId,
     meta: order.context,
     ...context,
   };
 
-  return Promise.all(
-    items.map(async (item) => {
-      const enrollmentData = await EnrollmentDirector.transformOrderItemToEnrollment(item, template, {
-        modules: this,
-      });
+  return (
+    await Promise.all(
+      items.map(async (item) => {
+        const enrollmentData = await EnrollmentDirector.transformOrderItemToEnrollment(item, template, {
+          modules: this,
+        });
 
-      const enrollment = await this.enrollments.create(enrollmentData);
-      return await initializeEnrollmentService.bind(this)(enrollment, {
-        orderIdForFirstPeriod: enrollment.orderIdForFirstPeriod,
-        reason: 'new_enrollment',
-      });
-    }),
-  );
+        const enrollment = await this.enrollments.create(enrollmentData);
+        return await initializeEnrollmentService.bind(this)(enrollment, {
+          orderIdForFirstPeriod: enrollment.orderIdForFirstPeriod,
+          reason: 'new_enrollment',
+        });
+      }),
+    )
+  ).filter(Boolean) as Enrollment[];
 }

package/src/services/createManualOrderDiscount.ts

@@ -1,11 +1,11 @@
-import { Order, OrderDiscount } from '@unchainedshop/core-orders';
+import { Order } from '@unchainedshop/core-orders';
 import { OrderDiscountDirector } from '../directors/OrderDiscountDirector.js';
 import { Modules } from '../modules.js';
 
 export async function createManualOrderDiscountService(
   this: Modules,
   { order, code }: { code: string; order: Order },
-): Promise<OrderDiscount> {
+) {
   // Use an already existing discount if available!
   const spareDiscount = await this.orders.discounts.findSpareDiscount({ code });
   if (spareDiscount) {

package/src/services/createSignedURL.ts

@@ -9,6 +9,7 @@ export async function createSignedURLService(
   const preparedFileData = await fileUploadAdapter.createSignedURL(directoryName, fileName, {
     modules: this,
   });
+  if (!preparedFileData) throw new Error('Could not prepare signed URL');
   const fileData = getFileFromFileData(preparedFileData, meta);
   const fileId = await this.files.create(fileData);
   const file = await this.files.findFile({ fileId });

package/src/services/deleteUser.ts

@@ -9,9 +9,14 @@ export async function deleteUserService(this: Modules, { userId }: { userId: str
   await this.quotations.deleteRequestedUserQuotations(userId);
   await this.enrollments.deleteInactiveUserEnrollments(userId);
   await this.payment.paymentCredentials.deleteUserPaymentCredentials(userId);
-  await this.users.webAuthn.deleteUserWebAuthnCredentials(user.username);
 
-  const carts = await this.orders.findOrders({ userId, status: null });
+  if (user.username) {
+    await this.users.webAuthn.deleteUserWebAuthnCredentials(user.username);
+  }
+
+  const carts = (await this.orders.findOrders({ userId, includeCarts: true })).filter(
+    (c) => c.status === null,
+  );
 
   await Array.fromAsync(carts, async (userCart) => {
     await this.orders.positions.deleteOrderPositions(userCart?._id);

package/src/services/discountedEntities.ts

@@ -15,32 +15,32 @@ export async function discountedEntitiesService(
 ): Promise<OrderPricingDiscount[]> {
   // Delivery discounts
   const orderDelivery = await this.orders.deliveries.findDelivery({
-    orderDeliveryId: order.deliveryId,
+    orderDeliveryId: order.deliveryId!,
   });
 
   const deliveryPricingSheet = DeliveryPricingSheet({
-    calculation: orderDelivery.calculation || [],
+    calculation: orderDelivery!.calculation || [],
     currencyCode: order.currencyCode,
   });
   const orderDeliveryDiscounts = deliveryPricingSheet
     .discountPrices(orderDiscount._id)
     .map((discount) => ({
-      delivery: orderDelivery,
+      delivery: orderDelivery!,
       ...discount,
     }));
 
   // Payment discounts
   const orderPayment = await this.orders.payments.findOrderPayment({
-    orderPaymentId: order.paymentId,
+    orderPaymentId: order.paymentId!,
   });
   const paymentPricingSheet = PaymentPricingSheet({
-    calculation: orderPayment.calculation || [],
+    calculation: orderPayment!.calculation || [],
     currencyCode: order.currencyCode,
   });
   const orderPaymentDiscounts = paymentPricingSheet
     .discountPrices(orderDiscount._id)
     .map((discount) => ({
-      payment: orderPayment,
+      payment: orderPayment!,
       ...discount,
     }));
 

package/src/services/findOrInitCart.ts

@@ -1,5 +1,4 @@
 import { User } from '@unchainedshop/core-users';
-import { Order } from '@unchainedshop/core-orders';
 import { Modules } from '../modules.js';
 import { nextUserCartService } from './nextUserCart.js';
 
@@ -25,5 +24,5 @@ export async function findOrInitCartService(
     user,
     countryCode,
     forceCartCreation: true,
-  }) as Order;
+  });
 }

package/src/services/fullfillQuotation.ts

@@ -10,5 +10,5 @@ export async function fullfillQuotationService(this: Modules, quotation: Quotati
     info: JSON.stringify(info),
   });
 
-  return processQuotationService.bind(this)(updatedQuotation, {});
+  return processQuotationService.bind(this)(updatedQuotation!, {});
 }

package/src/services/generateOrderFromEnrollment.ts

@@ -0,0 +1,54 @@
+import { Modules } from '../modules.js';
+import { Enrollment, EnrollmentOrderPositionTemplate } from '@unchainedshop/core-enrollments';
+import { updateCalculationService } from './updateCalculation.js';
+import { checkoutOrderService } from './checkoutOrder.js';
+
+export async function generateOrderFromEnrollmentService(
+  this: Modules,
+  enrollment: Enrollment,
+  {
+    orderPositionTemplates,
+    orderContext,
+  }: {
+    orderPositionTemplates: EnrollmentOrderPositionTemplate[];
+    orderContext?: any;
+  },
+) {
+  if (!enrollment.payment || !enrollment.delivery) return null;
+
+  const order = await this.orders.create({
+    userId: enrollment.userId,
+    currencyCode: enrollment.currencyCode,
+    countryCode: enrollment.countryCode,
+    contact: enrollment.contact,
+    billingAddress: enrollment.billingAddress,
+    originEnrollmentId: enrollment._id,
+    context: orderContext,
+  });
+  const orderId = order._id;
+
+  await Promise.all(
+    orderPositionTemplates.map((orderPositionTemplate) =>
+      this.orders.positions.addProductItem({
+        ...orderPositionTemplate,
+        orderId: order._id,
+      }),
+    ),
+  );
+
+  const { paymentProviderId, context: paymentContext } = enrollment.payment;
+  if (paymentProviderId) {
+    await this.orders.setPaymentProvider(orderId, paymentProviderId);
+  }
+
+  const { deliveryProviderId, context: deliveryContext } = enrollment.delivery;
+  if (deliveryProviderId) {
+    await this.orders.setDeliveryProvider(orderId, deliveryProviderId);
+  }
+
+  await updateCalculationService.bind(this)(orderId);
+  return checkoutOrderService.bind(this)(order._id, {
+    deliveryContext,
+    paymentContext,
+  });
+}

package/src/services/index.ts

@@ -44,6 +44,8 @@ import { simulateProductPricingService } from './simulateProductPricing.js';
 import { simulateProductDispatchingService } from './simulateProductDispatching.js';
 import { simulateProductInventoryService } from './simulateProductInventory.js';
 import { findOrInitCartService } from './findOrInitCart.js';
+import { addMessageService } from './addMessage.js';
+import { generateOrderFromEnrollmentService } from './generateOrderFromEnrollment.js';
 
 // Auto-Inject Unchained API as last parameter
 // https://developer.mozilla.org/en-US/docs/Web/JavaScript/Reference/Global_Objects/Proxy
@@ -134,6 +136,9 @@ export default function initServices(modules: Modules, customServices: CustomSer
       createEnrollmentFromCheckout: createEnrollmentFromCheckoutService as Bound<
         typeof createEnrollmentFromCheckoutService
       >,
+      generateOrderFromEnrollment: generateOrderFromEnrollmentService as Bound<
+        typeof generateOrderFromEnrollmentService
+      >,
       processEnrollment: processOrderService as Bound<typeof processOrderService>,
       initializeEnrollment: initializeEnrollmentService as Bound<typeof initializeEnrollmentService>,
       activateEnrollment: activateEnrollmentService as Bound<typeof activateEnrollmentService>,
@@ -156,6 +161,9 @@ export default function initServices(modules: Modules, customServices: CustomSer
     warehousing: {
       ercMetadata: ercMetadataService as Bound<typeof ercMetadataService>,
     },
+    worker: {
+      addMessage: addMessageService as Bound<typeof addMessageService>,
+    },
   };
 
   // This Proxy Binds all services to the Modules Object

package/src/services/initCartProviders.ts

@@ -14,7 +14,7 @@ export async function initCartProvidersService(this: Modules, order: Order) {
   });
 
   const orderDelivery = await this.orders.deliveries.findDelivery({
-    orderDeliveryId: updatedOrder.deliveryId,
+    orderDeliveryId: updatedOrder.deliveryId!,
   });
   const deliveryProviderId = orderDelivery?.deliveryProviderId;
 
@@ -33,10 +33,9 @@ export async function initCartProvidersService(this: Modules, order: Order) {
       { modules: this },
     );
     if (defaultOrderDeliveryProvider) {
-      updatedOrder = await this.orders.setDeliveryProvider(
-        updatedOrder._id,
-        defaultOrderDeliveryProvider._id,
-      );
+      updatedOrder =
+        (await this.orders.setDeliveryProvider(updatedOrder._id, defaultOrderDeliveryProvider._id)) ||
+        updatedOrder;
     }
   }
 
@@ -46,7 +45,7 @@ export async function initCartProvidersService(this: Modules, order: Order) {
   });
 
   const orderPayment = await this.orders.payments.findOrderPayment({
-    orderPaymentId: updatedOrder.paymentId,
+    orderPaymentId: updatedOrder.paymentId!,
   });
   const paymentProviderId = orderPayment?.paymentProviderId;
 
@@ -76,10 +75,9 @@ export async function initCartProvidersService(this: Modules, order: Order) {
     );
 
     if (defaultOrderPaymentProvider) {
-      updatedOrder = await this.orders.setPaymentProvider(
-        updatedOrder._id,
-        defaultOrderPaymentProvider._id,
-      );
+      updatedOrder =
+        (await this.orders.setPaymentProvider(updatedOrder._id, defaultOrderPaymentProvider._id)) ||
+        updatedOrder;
     }
   }
   return updatedOrder;

package/src/services/initializeEnrollment.ts

@@ -2,36 +2,39 @@ import { Enrollment } from '@unchainedshop/core-enrollments';
 import { EnrollmentDirector } from '../core-index.js';
 import { processEnrollmentService } from './processEnrollment.js';
 import { Modules } from '../modules.js';
+import { addMessageService } from './addMessage.js';
 
 export async function initializeEnrollmentService(
   this: Modules,
   enrollment: Enrollment,
   params: { orderIdForFirstPeriod?: string; reason: string },
 ) {
-  const director = await EnrollmentDirector.actions({ enrollment }, { modules: this });
+  const product = await this.products.findProduct({
+    productId: enrollment.productId,
+  });
+
+  const director = await EnrollmentDirector.actions(
+    { enrollment, product: product! },
+    { modules: this },
+  );
   const period = await director.nextPeriod();
 
   let updatedEnrollment = enrollment;
   if (period && (params.orderIdForFirstPeriod || period.isTrial)) {
-    updatedEnrollment = await this.enrollments.addEnrollmentPeriod(enrollment._id, {
+    updatedEnrollment = (await this.enrollments.addEnrollmentPeriod(enrollment._id, {
       ...period,
       orderId: params.orderIdForFirstPeriod,
-    });
+    })) as Enrollment;
   }
 
   const processedEnrollment = await processEnrollmentService.bind(this)(updatedEnrollment);
   const user = await this.users.findUserById(enrollment.userId);
-  const locale = this.users.userLocale(user);
+  const locale = this.users.userLocale(user!);
 
-  await this.worker.addWork({
-    type: 'MESSAGE',
-    retries: 0,
-    input: {
-      reason: params.reason || 'status_change',
-      locale: locale.baseName,
-      template: 'ENROLLMENT_STATUS',
-      enrollmentId: processedEnrollment._id,
-    },
+  await addMessageService.bind(this)('ENROLLMENT_STATUS', {
+    reason: 'status_change',
+    locale: locale.baseName,
+    enrollmentId: processedEnrollment._id,
   });
 
   return processedEnrollment;