netlify-hi/src/helpers/filetreeUtils.js

115 lines
2.7 KiB
JavaScript

const sortTree = (unsorted) => {
//Sort by folder before file, then by name
const orderedTree = Object.keys(unsorted)
.sort((a, b) => {
let a_pinned = unsorted[a].pinned || false;
let b_pinned = unsorted[b].pinned || false;
if (a_pinned != b_pinned) {
if (a_pinned) {
return -1;
} else {
return 1;
}
}
const a_is_note = a.indexOf(".md") > -1;
const b_is_note = b.indexOf(".md") > -1;
if (a_is_note && !b_is_note) {
return 1;
}
if (!a_is_note && b_is_note) {
return -1;
}
if (a.toLowerCase() > b.toLowerCase()) {
return 1;
}
return -1;
})
.reduce((obj, key) => {
obj[key] = unsorted[key];
return obj;
}, {});
for (const key of Object.keys(orderedTree)) {
if (orderedTree[key].isFolder) {
orderedTree[key] = sortTree(orderedTree[key]);
}
}
return orderedTree;
};
function getPermalinkMeta(note, key) {
let permalink = "/";
let parts = note.filePathStem.split("/");
let name = parts[parts.length - 1];
let noteIcon = process.env.NOTE_ICON_DEFAULT;
let hide = false;
let pinned = false;
let folders = null;
try {
if (note.data.permalink) {
permalink = note.data.permalink;
}
if (note.data.tags && note.data.tags.indexOf("gardenEntry") != -1) {
permalink = "/";
}
if (note.data.title) {
name = note.data.title;
}
if (note.data.noteIcon) {
noteIcon = note.data.noteIcon;
}
// Reason for adding the hide flag instead of removing completely from file tree is to
// allow users to use the filetree data elsewhere without the fear of losing any data.
if (note.data.hide) {
hide = note.data.hide;
}
if (note.data.pinned) {
pinned = note.data.pinned;
}
if (note.data["dg-path"]) {
folders = note.data["dg-path"].split("/");
} else {
folders = note.filePathStem
.split("notes/")[1]
.split("/");
}
folders[folders.length - 1]+= ".md";
} catch {
//ignore
}
return [{ permalink, name, noteIcon, hide, pinned }, folders];
}
function assignNested(obj, keyPath, value) {
lastKeyIndex = keyPath.length - 1;
for (var i = 0; i < lastKeyIndex; ++i) {
key = keyPath[i];
if (!(key in obj)) {
obj[key] = { isFolder: true };
}
obj = obj[key];
}
obj[keyPath[lastKeyIndex]] = value;
}
function getFileTree(data) {
const tree = {};
(data.collections.note || []).forEach((note) => {
const [meta, folders] = getPermalinkMeta(note);
assignNested(tree, folders, { isNote: true, ...meta });
});
const fileTree = sortTree(tree);
return fileTree;
}
exports.getFileTree = getFileTree;