// @flow import { difference } from 'lodash'; import type { DocumentEvent } from '../events'; import { Document, Revision, Backlink } from '../models'; import parseDocumentIds from '../../shared/utils/parseDocumentIds'; import slugify from '../utils/slugify'; export default class Backlinks { async on(event: DocumentEvent) { switch (event.name) { case 'documents.publish': { const document = await Document.findByPk(event.documentId); const linkIds = parseDocumentIds(document.text); await Promise.all( linkIds.map(async linkId => { const linkedDocument = await Document.findByPk(linkId); if (linkedDocument.id === event.documentId) return; await Backlink.findOrCreate({ where: { documentId: linkedDocument.id, reverseDocumentId: event.documentId, }, defaults: { userId: document.lastModifiedById, }, }); }) ); break; } case 'documents.update': { // no-op for now if (event.data.autosave) return; // no-op for drafts const document = await Document.findByPk(event.documentId); if (!document.publishedAt) return; const [currentRevision, previousRevision] = await Revision.findAll({ where: { documentId: event.documentId }, order: [['createdAt', 'desc']], limit: 2, }); const previousLinkIds = previousRevision ? parseDocumentIds(previousRevision.text) : []; const currentLinkIds = parseDocumentIds(currentRevision.text); const addedLinkIds = difference(currentLinkIds, previousLinkIds); const removedLinkIds = difference(previousLinkIds, currentLinkIds); // add any new backlinks that were created await Promise.all( addedLinkIds.map(async linkId => { const linkedDocument = await Document.findByPk(linkId); if (linkedDocument.id === event.documentId) return; await Backlink.findOrCreate({ where: { documentId: linkedDocument.id, reverseDocumentId: event.documentId, }, defaults: { userId: currentRevision.userId, }, }); }) ); // delete any backlinks that were removed await Promise.all( removedLinkIds.map(async linkId => { const document = await Document.findByPk(linkId, { paranoid: false, }); if (document) { await Backlink.destroy({ where: { documentId: document.id, reverseDocumentId: event.documentId, }, }); } }) ); if ( !previousRevision || currentRevision.title === previousRevision.title ) { break; } // update any link titles in documents that lead to this one const backlinks = await Backlink.findAll({ where: { documentId: event.documentId, }, include: [{ model: Document, as: 'reverseDocument' }], }); await Promise.all( backlinks.map(async backlink => { const previousUrl = `/doc/${slugify(previousRevision.title)}-${ document.urlId }`; // find links in the other document that lead to this one and have // the old title as anchor text. Go ahead and update those to the // new title automatically backlink.reverseDocument.text = backlink.reverseDocument.text.replace( `[${previousRevision.title}](${previousUrl})`, `[${document.title}](${document.url})` ); await backlink.reverseDocument.save({ silent: true, hooks: false, }); }) ); break; } case 'documents.delete': { await Backlink.destroy({ where: { reverseDocumentId: event.documentId, }, }); await Backlink.destroy({ where: { documentId: event.documentId, }, }); break; } default: } } }