trilium/src/public/javascripts/services/protected_session.js

142 lines
3.6 KiB
JavaScript
Raw Normal View History

import treeService from './tree.js';
2018-04-08 21:25:35 +08:00
import noteDetailService from './note_detail.js';
import utils from './utils.js';
import server from './server.js';
import protectedSessionHolder from './protected_session_holder.js';
2018-03-26 09:29:35 +08:00
import infoService from "./info.js";
const $dialog = $("#protected-session-password-dialog");
const $passwordForm = $("#protected-session-password-form");
const $password = $("#protected-session-password");
const $noteDetailWrapper = $("#note-detail-wrapper");
const $protectButton = $("#protect-button");
const $unprotectButton = $("#unprotect-button");
let protectedSessionDeferred = null;
function ensureProtectedSession(requireProtectedSession, modal) {
const dfd = $.Deferred();
if (requireProtectedSession && !protectedSessionHolder.isProtectedSessionAvailable()) {
protectedSessionDeferred = dfd;
if (treeService.getCurrentNode().data.isProtected) {
$noteDetailWrapper.hide();
}
$dialog.dialog({
modal: modal,
width: 400,
open: () => {
if (!modal) {
// dialog steals focus for itself, which is not what we want for non-modal (viewing)
treeService.getCurrentNode().setFocus();
}
}
});
}
else {
dfd.resolve();
}
return dfd.promise();
}
async function setupProtectedSession() {
const password = $password.val();
$password.val("");
const response = await enterProtectedSession(password);
if (!response.success) {
2018-03-26 09:29:35 +08:00
infoService.showError("Wrong password.");
return;
2017-11-05 06:18:55 +08:00
}
protectedSessionHolder.setProtectedSessionId(response.protectedSessionId);
2017-11-11 11:55:19 +08:00
$dialog.dialog("close");
2018-04-08 21:25:35 +08:00
noteDetailService.reload();
treeService.reload();
if (protectedSessionDeferred !== null) {
ensureDialogIsClosed($dialog, $password);
$noteDetailWrapper.show();
2017-11-15 12:01:23 +08:00
protectedSessionDeferred.resolve();
protectedSessionDeferred = null;
2017-11-09 11:33:08 +08:00
}
}
function ensureDialogIsClosed() {
// this may fal if the dialog has not been previously opened
try {
$dialog.dialog('close');
}
catch (e) {}
$password.val('');
}
2017-11-11 11:55:19 +08:00
async function enterProtectedSession(password) {
return await server.post('login/protected', {
password: password
});
}
2017-11-11 11:55:19 +08:00
async function protectNoteAndSendToServer() {
await ensureProtectedSession(true, true);
2018-04-08 21:25:35 +08:00
const note = noteDetailService.getCurrentNote();
2018-03-26 11:25:17 +08:00
note.isProtected = true;
2018-04-08 21:25:35 +08:00
await noteDetailService.saveNote(note);
2018-03-26 11:25:17 +08:00
treeService.setProtected(note.noteId, note.isProtected);
2018-04-08 21:25:35 +08:00
noteDetailService.setNoteBackgroundIfProtected(note);
}
async function unprotectNoteAndSendToServer() {
await ensureProtectedSession(true, true);
2018-04-08 21:25:35 +08:00
const note = noteDetailService.getCurrentNote();
2018-03-26 11:25:17 +08:00
note.isProtected = false;
2018-04-08 21:25:35 +08:00
await noteDetailService.saveNote(note);
2018-03-26 11:25:17 +08:00
treeService.setProtected(note.noteId, note.isProtected);
2018-04-08 21:25:35 +08:00
noteDetailService.setNoteBackgroundIfProtected(note);
}
2018-04-02 08:33:10 +08:00
async function protectBranch(noteId, protect) {
await ensureProtectedSession(true, true);
2017-11-23 09:46:42 +08:00
2018-04-02 08:33:10 +08:00
await server.put('notes/' + noteId + "/protect/" + (protect ? 1 : 0));
2017-11-15 13:04:26 +08:00
2018-03-26 09:29:35 +08:00
infoService.showMessage("Request to un/protect sub tree has finished successfully");
2017-11-15 13:04:26 +08:00
treeService.reload();
2018-04-08 21:25:35 +08:00
noteDetailService.reload();
}
2017-11-15 13:04:26 +08:00
$passwordForm.submit(() => {
setupProtectedSession();
2017-11-09 11:33:08 +08:00
return false;
});
2017-11-09 11:33:08 +08:00
$protectButton.click(protectNoteAndSendToServer);
$unprotectButton.click(unprotectNoteAndSendToServer);
export default {
ensureProtectedSession,
protectNoteAndSendToServer,
unprotectNoteAndSendToServer,
2018-04-02 08:33:10 +08:00
protectBranch,
ensureDialogIsClosed
};