trilium/src/public/app/services/froca_updater.js

import LoadResults from "./load_results.js";
import froca from "./froca.js";
import utils from "./utils.js";
import options from "./options.js";
import noteAttributeCache from "./note_attribute_cache.js";
import FBranch from "../entities/fbranch.js";
import FAttribute from "../entities/fattribute.js";
import FAttachment from "../entities/fattachment.js";

async function processEntityChanges(entityChanges) {
    const loadResults = new LoadResults(entityChanges);

    for (const ec of entityChanges) {
        try {
            if (ec.entityName === 'notes') {
                processNoteChange(loadResults, ec);
            } else if (ec.entityName === 'branches') {
                await processBranchChange(loadResults, ec);
            } else if (ec.entityName === 'attributes') {
                processAttributeChange(loadResults, ec);
            } else if (ec.entityName === 'note_reordering') {
                processNoteReordering(loadResults, ec);
            } else if (ec.entityName === 'blobs') {
                for (const affectedNoteId of ec.noteIds) {
                    for (const key of Object.keys(froca.blobPromises)) {
                        if (key.includes(affectedNoteId)) {
                            delete froca.blobPromises[key];
                        }
                    }
                }

                loadResults.addNoteContent(ec.noteIds, ec.componentId);
            } else if (ec.entityName === 'revisions') {
                loadResults.addRevision(ec.entityId, ec.noteId, ec.componentId);
            } else if (ec.entityName === 'options') {
                if (ec.entity.name === 'openNoteContexts') {
                    continue; // only noise
                }

                options.set(ec.entity.name, ec.entity.value);

                loadResults.addOption(ec.entity.name);
            } else if (ec.entityName === 'attachments') {
                processAttachment(loadResults, ec);
            } else if (ec.entityName === 'etapi_tokens') {
                // NOOP
            }
            else {
                throw new Error(`Unknown entityName '${ec.entityName}'`);
            }
        }
        catch (e) {
            throw new Error(`Can't process entity ${JSON.stringify(ec)} with error ${e.message} ${e.stack}`);
        }
    }

    const missingNoteIds = [];

    for (const {entityName, entity} of entityChanges) {
        if (!entity) { // if erased
            continue;
        }

        if (entityName === 'branches' && !(entity.parentNoteId in froca.notes)) {
            missingNoteIds.push(entity.parentNoteId);
        }
        else if (entityName === 'attributes'
            && entity.type === 'relation'
            && (entity.name === 'template' || entity.name === 'inherit')
            && !(entity.value in froca.notes)) {

            missingNoteIds.push(entity.value);
        }
    }

    if (missingNoteIds.length > 0) {
        await froca.reloadNotes(missingNoteIds);
    }

    if (!loadResults.isEmpty()) {
        if (loadResults.hasAttributeRelatedChanges()) {
            noteAttributeCache.invalidate();
        }

        const appContext = (await import("../components/app_context.js")).default;
        await appContext.triggerEvent('entitiesReloaded', {loadResults});
    }
}

function processNoteChange(loadResults, ec) {
    const note = froca.notes[ec.entityId];

    if (!note) {
        // if this note has not been requested before then it's not part of froca's cached subset, and
        // we're not interested in it
        return;
    }

    loadResults.addNote(ec.entityId, ec.componentId);

    if (ec.isErased && ec.entityId in froca.notes) {
        utils.reloadFrontendApp(`${ec.entityName} '${ec.entityId}' is erased, need to do complete reload.`);
        return;
    }

    if (ec.isErased || ec.entity?.isDeleted) {
        delete froca.notes[ec.entityId];
    }
    else {
        note.update(ec.entity);
    }
}

async function processBranchChange(loadResults, ec) {
    if (ec.isErased && ec.entityId in froca.branches) {
        utils.reloadFrontendApp(`${ec.entityName} '${ec.entityId}' is erased, need to do complete reload.`);
        return;
    }

    let branch = froca.branches[ec.entityId];

    if (ec.isErased || ec.entity?.isDeleted) {
        if (branch) {
            const childNote = froca.notes[branch.noteId];
            const parentNote = froca.notes[branch.parentNoteId];

            if (childNote) {
                childNote.parents = childNote.parents.filter(parentNoteId => parentNoteId !== branch.parentNoteId);
                delete childNote.parentToBranch[branch.parentNoteId];
            }

            if (parentNote) {
                parentNote.children = parentNote.children.filter(childNoteId => childNoteId !== branch.noteId);
                delete parentNote.childToBranch[branch.noteId];
            }

            loadResults.addBranch(ec.entityId, ec.componentId);

            delete froca.branches[ec.entityId];
        }

        return;
    }

    loadResults.addBranch(ec.entityId, ec.componentId);

    const childNote = froca.notes[ec.entity.noteId];
    let parentNote = froca.notes[ec.entity.parentNoteId];

    if (childNote && !childNote.isRoot() && !parentNote) {
        // a branch cannot exist without the parent
        // a note loaded into froca has to also contain all its ancestors,
        // this problem happened e.g., in sharing where _share was hidden and thus not loaded
        // sharing meant cloning into _share, which crashed because _share was not loaded
        parentNote = await froca.getNote(ec.entity.parentNoteId);
    }

    if (branch) {
        branch.update(ec.entity);
    }
    else if (childNote || parentNote) {
        froca.branches[ec.entityId] = branch = new FBranch(froca, ec.entity);
    }

    if (childNote) {
        childNote.addParent(branch.parentNoteId, branch.branchId);
    }

    if (parentNote) {
        parentNote.addChild(branch.noteId, branch.branchId);
    }
}

function processNoteReordering(loadResults, ec) {
    const parentNoteIdsToSort = new Set();

    for (const branchId in ec.positions) {
        const branch = froca.branches[branchId];

        if (branch) {
            branch.notePosition = ec.positions[branchId];

            parentNoteIdsToSort.add(branch.parentNoteId);
        }
    }

    for (const parentNoteId of parentNoteIdsToSort) {
        const parentNote = froca.notes[parentNoteId];

        if (parentNote) {
            parentNote.sortChildren();
        }
    }

    loadResults.addNoteReordering(ec.entityId, ec.componentId);
}

function processAttributeChange(loadResults, ec) {
    let attribute = froca.attributes[ec.entityId];

    if (ec.isErased && ec.entityId in froca.attributes) {
        utils.reloadFrontendApp(`${ec.entityName} '${ec.entityId}' is erased, need to do complete reload.`);
        return;
    }

    if (ec.isErased || ec.entity?.isDeleted) {
        if (attribute) {
            const sourceNote = froca.notes[attribute.noteId];
            const targetNote = attribute.type === 'relation' && froca.notes[attribute.value];

            if (sourceNote) {
                sourceNote.attributes = sourceNote.attributes.filter(attributeId => attributeId !== attribute.attributeId);
            }

            if (targetNote) {
                targetNote.targetRelations = targetNote.targetRelations.filter(attributeId => attributeId !== attribute.attributeId);
            }

            loadResults.addAttribute(ec.entityId, ec.componentId);

            delete froca.attributes[ec.entityId];
        }

        return;
    }

    loadResults.addAttribute(ec.entityId, ec.componentId);

    const sourceNote = froca.notes[ec.entity.noteId];
    const targetNote = ec.entity.type === 'relation' && froca.notes[ec.entity.value];

    if (attribute) {
        attribute.update(ec.entity);
    } else if (sourceNote || targetNote) {
        attribute = new FAttribute(froca, ec.entity);

        froca.attributes[attribute.attributeId] = attribute;

        if (sourceNote && !sourceNote.attributes.includes(attribute.attributeId)) {
            sourceNote.attributes.push(attribute.attributeId);
        }

        if (targetNote && !targetNote.targetRelations.includes(attribute.attributeId)) {
            targetNote.targetRelations.push(attribute.attributeId);
        }
    }
}

function processAttachment(loadResults, ec) {
    if (ec.isErased && ec.entityId in froca.attachments) {
        utils.reloadFrontendApp(`${ec.entityName} '${ec.entityId}' is erased, need to do complete reload.`);
        return;
    }

    const attachment = froca.attachments[ec.entityId];

    if (ec.isErased || ec.entity?.isDeleted) {
        if (attachment) {
            const note = attachment.getNote();

            if (note && note.attachments) {
                note.attachments = note.attachments.filter(att => att.attachmentId !== attachment.attachmentId);
            }

            loadResults.addAttachment(ec.entity);

            delete froca.attachments[ec.entityId];
        }

        return;
    }

    if (attachment) {
        attachment.update(ec.entity);
    } else {
        const note = froca.notes[ec.entity.parentId];

        if (note?.attachments) {
            note.attachments.push(new FAttachment(froca, ec.entity));
        }
    }

    loadResults.addAttachment(ec.entity);
}

export default {
    processEntityChanges
}