medusa-store/packages/modules/notification/src/services/notification-module-service.ts

import {
  Context,
  DAL,
  InferEntityType,
  INotificationModuleService,
  InternalModuleDeclaration,
  Logger,
  ModulesSdkTypes,
  NotificationTypes,
} from "@medusajs/framework/types"
import {
  EmitEvents,
  generateEntityId,
  InjectManager,
  MedusaContext,
  MedusaError,
  MedusaService,
  NotificationStatus,
  promiseAll,
} from "@medusajs/framework/utils"
import { Notification } from "@models"
import NotificationProviderService from "./notification-provider"

type InjectedDependencies = {
  logger?: Logger
  baseRepository: DAL.RepositoryService
  notificationService: ModulesSdkTypes.IMedusaInternalService<
    typeof Notification
  >
  notificationProviderService: NotificationProviderService
}

export default class NotificationModuleService
  extends MedusaService<{
    Notification: { dto: NotificationTypes.NotificationDTO }
  }>({ Notification })
  implements INotificationModuleService
{
  protected baseRepository_: DAL.RepositoryService
  protected readonly notificationService_: ModulesSdkTypes.IMedusaInternalService<
    typeof Notification
  >
  protected readonly notificationProviderService_: NotificationProviderService

  constructor(
    {
      baseRepository,
      notificationService,
      notificationProviderService,
    }: InjectedDependencies,
    protected readonly moduleDeclaration: InternalModuleDeclaration
  ) {
    // @ts-ignore
    super(...arguments)
    this.baseRepository_ = baseRepository
    this.notificationService_ = notificationService
    this.notificationProviderService_ = notificationProviderService
  }

  // @ts-expect-error
  createNotifications(
    data: NotificationTypes.CreateNotificationDTO[],
    sharedContext?: Context
  ): Promise<NotificationTypes.NotificationDTO[]>
  // @ts-expect-error
  createNotifications(
    data: NotificationTypes.CreateNotificationDTO,
    sharedContext?: Context
  ): Promise<NotificationTypes.NotificationDTO>

  @InjectManager()
  @EmitEvents()
  // @ts-expect-error
  async createNotifications(
    data:
      | NotificationTypes.CreateNotificationDTO
      | NotificationTypes.CreateNotificationDTO[],
    @MedusaContext() sharedContext: Context = {}
  ): Promise<
    NotificationTypes.NotificationDTO | NotificationTypes.NotificationDTO[]
  > {
    const normalized = Array.isArray(data) ? data : [data]

    const createdNotifications = await this.createNotifications_(
      normalized,
      sharedContext
    )

    const serialized = await this.baseRepository_.serialize<
      NotificationTypes.NotificationDTO[]
    >(createdNotifications)

    return Array.isArray(data) ? serialized : serialized[0]
  }

  @InjectManager()
  protected async createNotifications_(
    data: NotificationTypes.CreateNotificationDTO[],
    @MedusaContext() sharedContext: Context = {}
  ): Promise<InferEntityType<typeof Notification>[]> {
    if (!data.length) {
      return []
    }

    // TODO: At this point we should probably take a lock with the idempotency keys so we don't have race conditions.
    // Also, we should probably rely on Redis for this instead of the database.
    const idempotencyKeys = data
      .map((entry) => entry.idempotency_key)
      .filter(Boolean)

    let { notificationsToProcess, createdNotifications } =
      await this.baseRepository_.transaction(async (txManager) => {
        const context = {
          ...sharedContext,
          transactionManager: txManager,
        }

        const alreadySentNotifications = await this.notificationService_.list(
          {
            idempotency_key: idempotencyKeys,
          },
          {},
          context
        )

        const existsMap = new Map(
          alreadySentNotifications.map((n) => [n.idempotency_key as string, n])
        )

        const notificationsToProcess = data.filter(
          (entry) =>
            !entry.idempotency_key ||
            !existsMap.has(entry.idempotency_key) ||
            (existsMap.has(entry.idempotency_key) &&
              existsMap.get(entry.idempotency_key)!.status ===
                NotificationStatus.FAILURE)
        )

        const channels = notificationsToProcess.map((not) => not.channel)
        const providers =
          await this.notificationProviderService_.getProviderForChannels(
            channels
          )

        // Create the notifications to be sent to prevent concurrent actions listing the same notifications
        const normalizedNotificationsToProcess = notificationsToProcess.map(
          (entry) => {
            const provider = providers.find((provider) =>
              provider?.channels.includes(entry.channel)
            )

            return {
              provider,
              data: {
                id: generateEntityId(undefined, "noti"),
                ...entry,
                provider_id: provider?.id,
              },
            }
          }
        )

        const toCreate = normalizedNotificationsToProcess
          .filter(
            (e) =>
              !e.data.idempotency_key || !existsMap.has(e.data.idempotency_key)
          )
          .map((e) => e.data)

        const createdNotifications = toCreate.length
          ? await this.notificationService_.create(toCreate, context)
          : []

        return {
          notificationsToProcess: normalizedNotificationsToProcess,
          createdNotifications,
        }
      })

    const notificationToUpdate: { id: string; external_id?: string }[] = []

    try {
      await promiseAll(
        notificationsToProcess.map(async (entry) => {
          const provider = entry.provider

          if (!provider?.is_enabled) {
            entry.data.status = NotificationStatus.FAILURE
            notificationToUpdate.push(entry.data)

            const errorMessage = !provider
              ? `Could not find a notification provider for channel: ${entry.data.channel} for notification id ${entry.data.id}`
              : `Notification provider ${provider.id} is not enabled. To enable it, configure it as a provider in the notification module options.`

            throw new MedusaError(MedusaError.Types.NOT_FOUND, errorMessage)
          }

          const res = await this.notificationProviderService_
            .send(provider, entry.data)
            .catch((e) => {
              entry.data.status = NotificationStatus.FAILURE
              notificationToUpdate.push(entry.data)
              throw new MedusaError(
                MedusaError.Types.UNEXPECTED_STATE,
                `Failed to send notification with id ${entry.data.id}:\n${e.message}`
              )
            })

          entry.data.external_id = res.id
          entry.data.status = NotificationStatus.SUCCESS

          notificationToUpdate.push(entry.data)
        }),
        {
          aggregateErrors: true,
        }
      )
    } finally {
      const updatedNotifications = await this.notificationService_.update(
        notificationToUpdate,
        sharedContext
      )
      const updatedNotificationsMap = new Map(
        updatedNotifications.map((n) => [n.id, n])
      )

      // Maintain the order of the notifications
      createdNotifications = createdNotifications.map((notification) => {
        return updatedNotificationsMap.get(notification.id) || notification
      })
    }

    return createdNotifications
  }
}