restore revision with attachments

src/becca/entities/bnote.js

@@ -1569,11 +1569,9 @@ class BNote extends AbstractBeccaEntity {
             noteRevision.save(); // to generate noteRevisionId which is then used to save attachments
 
             for (const noteAttachment of this.getAttachments()) {
-                const attachmentContent = noteAttachment.getContent();
-
                 const revisionAttachment = noteAttachment.copy();
                 revisionAttachment.parentId = noteRevision.noteRevisionId;
-                revisionAttachment.setContent(attachmentContent, {
+                revisionAttachment.setContent(noteAttachment.getContent(), {
                     forceSave: true,
                     forceCold: true
                 });

src/becca/entities/bnote_revision.js

@@ -5,6 +5,8 @@ const utils = require('../../services/utils');
 const dateUtils = require('../../services/date_utils');
 const becca = require('../becca');
 const AbstractBeccaEntity = require("./abstract_becca_entity");
+const sql = require("../../services/sql");
+const BAttachment = require("./battachment");
 
 /**
  * NoteRevision represents snapshot of note's title and content at some point in the past.
@@ -92,6 +94,16 @@ class BNoteRevision extends AbstractBeccaEntity {
         this._setContent(content, opts);
     }
 
+    /** @returns {BAttachment[]} */
+    getAttachments() {
+        return sql.getRows(`
+                SELECT attachments.*
+                FROM attachments 
+                WHERE parentId = ? 
+                  AND isDeleted = 0`, [this.noteRevisionId])
+            .map(row => new BAttachment(row));
+    }
+
     beforeSaving() {
         super.beforeSaving();
 

src/public/app/widgets/type_widgets/attachment_detail.js

@@ -28,9 +28,8 @@ export default class AttachmentDetailTypeWidget extends TypeWidget {
     async doRefresh(note) {
         this.$wrapper.empty();
         this.children = [];
-        this.renderedAttachmentIds = new Set();
 
-        const attachment = await server.get(`attachments/${this.noteContext.viewScope.attachmentId}/?includeContent=true`);
+        const attachment = await server.get(`attachments/${this.attachmentId}/?includeContent=true`);
 
         if (!attachment) {
             this.$wrapper.html("<strong>This attachment has been deleted.</strong>");
@@ -46,10 +45,14 @@ export default class AttachmentDetailTypeWidget extends TypeWidget {
     }
 
     async entitiesReloadedEvent({loadResults}) {
-        const attachmentChange = loadResults.getAttachments().find(att => att.attachmentId === this.attachment.attachmentId);
+        const attachmentChange = loadResults.getAttachments().find(att => att.attachmentId === this.attachmentId);
 
         if (attachmentChange?.isDeleted) {
             this.refresh(); // all other updates are handled within AttachmentDetailWidget
         }
     }
+
+    get attachmentId() {
+        return this.noteContext.viewScope.attachmentId;
+    }
 }

src/routes/api/note_revisions.js

@@ -95,11 +95,27 @@ function restoreNoteRevision(req) {
     if (noteRevision) {
         const note = noteRevision.getNote();
 
-        note.saveNoteRevision();
+        sql.transactional(() => {
+            note.saveNoteRevision();
 
-        note.title = noteRevision.title;
-        note.setContent(noteRevision.getContent());
-        note.save();
+            for (const oldNoteAttachment of note.getAttachments()) {
+                oldNoteAttachment.markAsDeleted();
+            }
+
+            let revisionContent = noteRevision.getContent();
+
+            for (const revisionAttachment of noteRevision.getAttachments()) {
+                const noteAttachment = revisionAttachment.copy();
+                noteAttachment.parentId = note.noteId;
+                noteAttachment.setContent(revisionAttachment.getContent(), { forceSave: true });
+
+                // content is rewritten to point to the restored revision attachments
+                revisionContent = revisionContent.replaceAll(`attachments/${revisionAttachment.attachmentId}`, `attachments/${noteAttachment.attachmentId}`);
+            }
+
+            note.title = noteRevision.title;
+            note.setContent(revisionContent, { forceSave: true });
+        });
     }
 }