sizing of the link map

src/routes/api/note_map.js

@@ -0,0 +1,122 @@
+"use strict";
+
+const becca = require("../../becca/becca");
+
+function buildDescendantCountMap() {
+    const noteIdToCountMap = {};
+
+    function getCount(noteId) {
+        if (!(noteId in noteIdToCountMap)) {
+            const note = becca.getNote(noteId);
+
+            noteIdToCountMap[noteId] = note.children.length;
+
+            for (const child of note.children) {
+                noteIdToCountMap[noteId] += getCount(child.noteId);
+            }
+        }
+
+        return noteIdToCountMap[noteId];
+    }
+
+    getCount('root');
+
+    return noteIdToCountMap;
+}
+
+function getLinkMap(req) {
+    const mapRootNote = becca.getNote(req.params.noteId);
+
+    const noteIds = new Set();
+
+    const notes = mapRootNote.getSubtreeNotes(false)
+        .map(note => [
+            note.noteId,
+            note.isContentAvailable() ? note.title : '[protected]',
+            note.type
+        ]);
+
+    notes.forEach(([noteId]) => noteIds.add(noteId));
+
+    const links = Object.values(becca.attributes).filter(rel => {
+        if (rel.type !== 'relation' || rel.name === 'relationMapLink' || rel.name === 'template') {
+            return false;
+        }
+        else if (!noteIds.has(rel.noteId) || !noteIds.has(rel.value)) {
+            return false;
+        }
+        else if (rel.name === 'imageLink') {
+            const parentNote = becca.getNote(rel.noteId);
+
+            return !parentNote.getChildNotes().find(childNote => childNote.noteId === rel.value);
+        }
+        else {
+            return true;
+        }
+    })
+        .map(rel => ({
+        id: rel.noteId + "-" + rel.name + "-" + rel.value,
+        sourceNoteId: rel.noteId,
+        targetNoteId: rel.value,
+        name: rel.name
+    }));
+
+    return {
+        notes: notes,
+        noteIdToDescendantCountMap: buildDescendantCountMap(),
+        links: links
+    };
+}
+
+function getTreeMap(req) {
+    const mapRootNote = becca.getNote(req.params.noteId);
+    const noteIds = new Set();
+
+    const notes = mapRootNote.getSubtreeNotes(false)
+        .filter(note => !note.hasLabel('excludeFromTreeMap'))
+        .filter(note => {
+            if (note.type !== 'image' || note.getChildNotes().length > 0) {
+                return true;
+            }
+
+            const imageLinkRelation = note.getTargetRelations().find(rel => rel.name === 'imageLink');
+
+            if (!imageLinkRelation) {
+                return true;
+            }
+
+            return !note.getParentNotes().find(parentNote => parentNote.noteId === imageLinkRelation.noteId);
+        })
+        .map(note => [
+            note.noteId,
+            note.isContentAvailable() ? note.title : '[protected]',
+            note.type
+        ]);
+
+    notes.forEach(([noteId]) => noteIds.add(noteId));
+
+    const links = [];
+
+    for (const branch of Object.values(becca.branches)) {
+        if (!noteIds.has(branch.parentNoteId) || !noteIds.has(branch.noteId)) {
+            continue;
+        }
+
+        links.push({
+            id: branch.branchId,
+            sourceNoteId: branch.parentNoteId,
+            targetNoteId: branch.noteId
+        });
+    }
+
+    return {
+        notes: notes,
+        noteIdToDescendantCountMap: buildDescendantCountMap(),
+        links: links
+    };
+}
+
+module.exports = {
+    getLinkMap,
+    getTreeMap
+};