Kristaps Fabians Geikins
101 lines
3.0 KiB
TypeScript
101 lines
3.0 KiB
TypeScript
import { CommentRecord } from '@/modules/comments/helpers/types'
|
|
import { CommentsEmitter, CommentsEvents } from '@/modules/comments/events/emitter'
|
|
import { ensureCommentSchema } from '@/modules/comments/services/commentTextService'
|
|
import type { JSONContent } from '@tiptap/core'
|
|
import { iterateContentNodes } from '@/modules/core/services/richTextEditorService'
|
|
import { publishNotification } from '@/modules/notifications/services/publication'
|
|
import { difference, flatten } from 'lodash'
|
|
import { NotificationType } from '@/modules/notifications/helpers/types'
|
|
import { addStreamCommentMentionActivity } from '@/modules/activitystream/services/streamActivity'
|
|
|
|
function findMentionedUserIds(doc: JSONContent) {
|
|
const mentionedUserIds = new Set<string>()
|
|
|
|
for (const node of iterateContentNodes(doc)) {
|
|
if (node.type === 'mention') {
|
|
const uid = node.attrs?.id
|
|
if (uid) {
|
|
mentionedUserIds.add(uid)
|
|
}
|
|
}
|
|
}
|
|
|
|
return [...mentionedUserIds]
|
|
}
|
|
|
|
function collectMentionedUserIds(comment: CommentRecord): string[] {
|
|
if (!comment.text) return []
|
|
|
|
const { doc } = ensureCommentSchema(comment.text)
|
|
if (!doc) return []
|
|
|
|
return findMentionedUserIds(doc)
|
|
}
|
|
|
|
async function sendNotificationsForUsers(userIds: string[], comment: CommentRecord) {
|
|
const { id, streamId, authorId, parentComment } = comment
|
|
const threadId = parentComment || id
|
|
|
|
await Promise.all(
|
|
flatten(
|
|
userIds.map((uid) => {
|
|
return [
|
|
// Actually send out notification
|
|
publishNotification(NotificationType.MentionedInComment, {
|
|
targetUserId: uid,
|
|
data: {
|
|
threadId,
|
|
streamId,
|
|
authorId,
|
|
commentId: id
|
|
}
|
|
}),
|
|
// Create activity item
|
|
addStreamCommentMentionActivity({
|
|
streamId,
|
|
mentionAuthorId: authorId,
|
|
mentionTargetId: uid,
|
|
commentId: id,
|
|
threadId
|
|
})
|
|
]
|
|
})
|
|
)
|
|
)
|
|
}
|
|
|
|
async function processCommentMentions(
|
|
newComment: CommentRecord,
|
|
previousComment?: CommentRecord
|
|
) {
|
|
const newMentionedUserIds = collectMentionedUserIds(newComment)
|
|
const previouslyMentionedUserIds = previousComment
|
|
? collectMentionedUserIds(previousComment)
|
|
: []
|
|
|
|
const newMentions = difference(newMentionedUserIds, previouslyMentionedUserIds)
|
|
if (!newMentions.length) return
|
|
|
|
await sendNotificationsForUsers(newMentions, newComment)
|
|
}
|
|
|
|
/**
|
|
* Hook into the comments lifecycle to generate notifications accordingly
|
|
* @returns Callback to invoke when you wish to stop listening for comments events
|
|
*/
|
|
export async function notifyUsersOnCommentEvents() {
|
|
const exitCbs = [
|
|
CommentsEmitter.listen(CommentsEvents.Created, async ({ comment }) => {
|
|
await processCommentMentions(comment)
|
|
}),
|
|
CommentsEmitter.listen(
|
|
CommentsEvents.Updated,
|
|
async ({ newComment, previousComment }) => {
|
|
await processCommentMentions(newComment, previousComment)
|
|
}
|
|
)
|
|
]
|
|
|
|
return () => exitCbs.forEach((cb) => cb())
|
|
}
|