2017-10-22 09:10:33 +08:00
|
|
|
"use strict";
|
|
|
|
|
2017-10-16 07:47:05 +08:00
|
|
|
const sql = require('../../services/sql');
|
2018-04-02 09:27:46 +08:00
|
|
|
const optionService = require('../../services/options');
|
|
|
|
const protectedSessionService = require('../../services/protected_session');
|
2018-04-17 04:26:47 +08:00
|
|
|
const utils = require('../../services/utils');
|
2017-10-15 11:31:44 +08:00
|
|
|
|
2018-04-01 23:05:09 +08:00
|
|
|
async function getTree() {
|
2018-03-25 09:39:15 +08:00
|
|
|
const branches = await sql.getRows(`
|
2018-04-17 04:26:47 +08:00
|
|
|
WITH RECURSIVE
|
|
|
|
tree(branchId, noteId, isExpanded) AS (
|
|
|
|
SELECT branchId, noteId, isExpanded FROM branches WHERE branchId = 'root'
|
|
|
|
UNION ALL
|
|
|
|
SELECT branches.branchId, branches.noteId, branches.isExpanded FROM branches
|
|
|
|
JOIN tree ON branches.parentNoteId = tree.noteId
|
|
|
|
WHERE tree.isExpanded = 1 AND branches.isDeleted = 0
|
|
|
|
)
|
|
|
|
SELECT branches.* FROM tree JOIN branches USING(branchId);`);
|
|
|
|
|
|
|
|
const noteIds = branches.map(b => b.noteId);
|
|
|
|
const questionMarks = branches.map(() => "?").join(",");
|
|
|
|
|
|
|
|
const notes = await sql.getRows(`
|
|
|
|
SELECT noteId, title, isProtected, type, mime
|
|
|
|
FROM notes WHERE isDeleted = 0 AND noteId IN (${questionMarks})`, noteIds);
|
2018-03-25 08:41:27 +08:00
|
|
|
|
2018-04-02 09:27:46 +08:00
|
|
|
protectedSessionService.decryptNotes(notes);
|
2018-03-25 08:41:27 +08:00
|
|
|
|
2018-04-17 04:26:47 +08:00
|
|
|
notes.forEach(note => note.isProtected = !!note.isProtected);
|
|
|
|
|
|
|
|
const relationships = await sql.getRows(`SELECT noteId, parentNoteId FROM branches WHERE isDeleted = 0
|
|
|
|
AND parentNoteId IN (${questionMarks})`, noteIds);
|
|
|
|
|
|
|
|
const parentToChild = {};
|
|
|
|
|
|
|
|
for (const rel of relationships) {
|
|
|
|
parentToChild[rel.parentNoteId] = parentToChild[rel.parentNoteId] || [];
|
|
|
|
parentToChild[rel.parentNoteId].push(rel.noteId);
|
|
|
|
}
|
2018-02-14 11:30:33 +08:00
|
|
|
|
2018-03-31 00:57:22 +08:00
|
|
|
return {
|
2018-04-03 09:47:46 +08:00
|
|
|
startNotePath: await optionService.getOption('startNotePath'),
|
2018-03-25 09:39:15 +08:00
|
|
|
branches: branches,
|
2018-04-17 04:26:47 +08:00
|
|
|
notes: notes,
|
|
|
|
parentToChild
|
2018-03-31 00:57:22 +08:00
|
|
|
};
|
|
|
|
}
|
2017-10-15 11:31:44 +08:00
|
|
|
|
2018-03-31 00:57:22 +08:00
|
|
|
module.exports = {
|
2018-04-02 08:33:10 +08:00
|
|
|
getTree
|
2018-03-31 00:57:22 +08:00
|
|
|
};
|