outline/server/queues/tasks/CommentCreatedNotificationTask.ts

import { Node } from "prosemirror-model";
import subscriptionCreator from "@server/commands/subscriptionCreator";
import { sequelize } from "@server/database/sequelize";
import { schema } from "@server/editor";
import CommentCreatedEmail from "@server/emails/templates/CommentCreatedEmail";
import { Comment, Document, Notification, Team } from "@server/models";
import DocumentHelper from "@server/models/helpers/DocumentHelper";
import NotificationHelper from "@server/models/helpers/NotificationHelper";
import ProsemirrorHelper from "@server/models/helpers/ProsemirrorHelper";
import { CommentEvent } from "@server/types";
import BaseTask, { TaskPriority } from "./BaseTask";

export default class CommentCreatedNotificationTask extends BaseTask<
  CommentEvent
> {
  public async perform(event: CommentEvent) {
    const [document, comment, team] = await Promise.all([
      Document.scope("withCollection").findOne({
        where: {
          id: event.documentId,
        },
      }),
      Comment.findByPk(event.modelId),
      Team.findByPk(event.teamId),
    ]);
    if (!document || !comment || !team) {
      return;
    }

    // Commenting on a doc automatically creates a subscription to the doc
    // if they haven't previously had one.
    await sequelize.transaction(async (transaction) => {
      await subscriptionCreator({
        user: comment.createdBy,
        documentId: document.id,
        event: "documents.update",
        resubscribe: false,
        transaction,
        ip: event.ip,
      });
    });

    const recipients = await NotificationHelper.getCommentNotificationRecipients(
      document,
      comment,
      comment.createdById
    );
    if (!recipients.length) {
      return;
    }

    let content = ProsemirrorHelper.toHTML(
      Node.fromJSON(schema, comment.data),
      {
        centered: false,
      }
    );
    if (!content) {
      return;
    }

    content = await DocumentHelper.attachmentsToSignedUrls(
      content,
      event.teamId,
      86400 * 4
    );

    for (const recipient of recipients) {
      const notification = await Notification.create({
        event: event.name,
        userId: recipient.user.id,
        actorId: comment.createdById,
        teamId: team.id,
        documentId: document.id,
      });
      await CommentCreatedEmail.schedule(
        {
          to: recipient.user.email,
          documentId: document.id,
          teamUrl: team.url,
          isReply: !!comment.parentCommentId,
          actorName: comment.createdBy.name,
          commentId: comment.id,
          content,
          collectionName: document.collection?.name,
          unsubscribeUrl: recipient.unsubscribeUrl,
        },
        { notificationId: notification.id }
      );
    }
  }

  public get options() {
    return {
      attempts: 1,
      priority: TaskPriority.Background,
    };
  }
}