mirror of
https://github.com/zadam/trilium.git
synced 2024-12-26 09:12:08 +08:00
334 lines
No EOL
9.8 KiB
JavaScript
334 lines
No EOL
9.8 KiB
JavaScript
"use strict";
|
|
|
|
const log = require('./log');
|
|
const rp = require('request-promise');
|
|
const sql = require('./sql');
|
|
const options = require('./options');
|
|
const utils = require('./utils');
|
|
const source_id = require('./source_id');
|
|
const notes = require('./notes');
|
|
const syncUpdate = require('./sync_update');
|
|
const content_hash = require('./content_hash');
|
|
const event_log = require('./event_log');
|
|
const fs = require('fs');
|
|
const app_info = require('./app_info');
|
|
const messaging = require('./messaging');
|
|
const sync_setup = require('./sync_setup');
|
|
|
|
let syncInProgress = false;
|
|
let proxyToggle = true;
|
|
let syncServerCertificate = null;
|
|
|
|
async function sync() {
|
|
if (syncInProgress) {
|
|
log.info("Sync already in progress");
|
|
|
|
return {
|
|
success: false,
|
|
message: "Sync already in progress"
|
|
};
|
|
}
|
|
|
|
syncInProgress = true;
|
|
|
|
try {
|
|
if (!await sql.isDbUpToDate()) {
|
|
return {
|
|
success: false,
|
|
message: "DB not up to date"
|
|
};
|
|
}
|
|
|
|
const syncContext = await login();
|
|
|
|
await pushSync(syncContext);
|
|
|
|
await pullSync(syncContext);
|
|
|
|
await pushSync(syncContext);
|
|
|
|
await checkContentHash(syncContext);
|
|
|
|
return {
|
|
success: true
|
|
};
|
|
}
|
|
catch (e) {
|
|
proxyToggle = !proxyToggle;
|
|
|
|
if (e.message.indexOf('ECONNREFUSED') !== -1) {
|
|
log.info("No connection to sync server.");
|
|
|
|
return {
|
|
success: false,
|
|
message: "No connection to sync server."
|
|
};
|
|
}
|
|
else {
|
|
log.info("sync failed: " + e.stack);
|
|
|
|
return {
|
|
success: false,
|
|
message: e.message
|
|
}
|
|
}
|
|
}
|
|
finally {
|
|
syncInProgress = false;
|
|
}
|
|
}
|
|
|
|
async function login() {
|
|
const timestamp = utils.nowDate();
|
|
|
|
const documentSecret = await options.getOption('document_secret');
|
|
const hash = utils.hmac(documentSecret, timestamp);
|
|
|
|
const syncContext = { cookieJar: rp.jar() };
|
|
|
|
const resp = await syncRequest(syncContext, 'POST', '/api/login/sync', {
|
|
timestamp: timestamp,
|
|
dbVersion: app_info.db_version,
|
|
hash: hash
|
|
});
|
|
|
|
syncContext.sourceId = resp.sourceId;
|
|
|
|
return syncContext;
|
|
}
|
|
|
|
async function getLastSyncedPull() {
|
|
return parseInt(await options.getOption('last_synced_pull'));
|
|
}
|
|
|
|
async function setLastSyncedPull(syncId) {
|
|
await sql.doInTransaction(async () => {
|
|
await options.setOption('last_synced_pull', syncId);
|
|
});
|
|
}
|
|
|
|
async function pullSync(syncContext) {
|
|
const lastSyncedPull = await getLastSyncedPull();
|
|
|
|
const changesUri = '/api/sync/changed?lastSyncId=' + lastSyncedPull;
|
|
|
|
const syncRows = await syncRequest(syncContext, 'GET', changesUri);
|
|
|
|
log.info("Pulled " + syncRows.length + " changes from " + changesUri);
|
|
|
|
for (const sync of syncRows) {
|
|
if (source_id.isLocalSourceId(sync.source_id)) {
|
|
log.info(`Skipping pull #${sync.id} ${sync.entity_name} ${sync.entity_id} because ${sync.source_id} is a local source id.`);
|
|
|
|
await setLastSyncedPull(sync.id);
|
|
|
|
continue;
|
|
}
|
|
|
|
const resp = await syncRequest(syncContext, 'GET', "/api/sync/" + sync.entity_name + "/" + encodeURIComponent(sync.entity_id));
|
|
|
|
if (!resp || (sync.entity_name === 'notes' && !resp.entity)) {
|
|
log.error(`Empty response to pull for sync #${sync.id} ${sync.entity_name}, id=${sync.entity_id}`);
|
|
}
|
|
else if (sync.entity_name === 'notes') {
|
|
await syncUpdate.updateNote(resp.entity, syncContext.sourceId);
|
|
}
|
|
else if (sync.entity_name === 'notes_tree') {
|
|
await syncUpdate.updateNoteTree(resp, syncContext.sourceId);
|
|
}
|
|
else if (sync.entity_name === 'notes_history') {
|
|
await syncUpdate.updateNoteHistory(resp, syncContext.sourceId);
|
|
}
|
|
else if (sync.entity_name === 'notes_reordering') {
|
|
await syncUpdate.updateNoteReordering(resp, syncContext.sourceId);
|
|
}
|
|
else if (sync.entity_name === 'options') {
|
|
await syncUpdate.updateOptions(resp, syncContext.sourceId);
|
|
}
|
|
else if (sync.entity_name === 'recent_notes') {
|
|
await syncUpdate.updateRecentNotes(resp, syncContext.sourceId);
|
|
}
|
|
else {
|
|
throw new Error(`Unrecognized entity type ${sync.entity_name} in sync #${sync.id}`);
|
|
}
|
|
|
|
await setLastSyncedPull(sync.id);
|
|
}
|
|
|
|
log.info("Finished pull");
|
|
}
|
|
|
|
async function getLastSyncedPush() {
|
|
return parseInt(await options.getOption('last_synced_push'));
|
|
}
|
|
|
|
async function setLastSyncedPush(lastSyncedPush) {
|
|
await sql.doInTransaction(async () => {
|
|
await options.setOption('last_synced_push', lastSyncedPush);
|
|
});
|
|
}
|
|
|
|
async function pushSync(syncContext) {
|
|
let lastSyncedPush = await getLastSyncedPush();
|
|
|
|
while (true) {
|
|
const sync = await sql.getSingleResultOrNull('SELECT * FROM sync WHERE id > ? LIMIT 1', [lastSyncedPush]);
|
|
|
|
if (sync === null) {
|
|
// nothing to sync
|
|
|
|
log.info("Nothing to push");
|
|
|
|
break;
|
|
}
|
|
|
|
if (sync.source_id === syncContext.sourceId) {
|
|
log.info(`Skipping push #${sync.id} ${sync.entity_name} ${sync.entity_id} because it originates from sync target`);
|
|
}
|
|
else {
|
|
await pushEntity(sync, syncContext);
|
|
}
|
|
|
|
lastSyncedPush = sync.id;
|
|
|
|
await setLastSyncedPush(lastSyncedPush);
|
|
}
|
|
}
|
|
|
|
async function pushEntity(sync, syncContext) {
|
|
let entity;
|
|
|
|
if (sync.entity_name === 'notes') {
|
|
entity = await sql.getSingleResult('SELECT * FROM notes WHERE note_id = ?', [sync.entity_id]);
|
|
}
|
|
else if (sync.entity_name === 'notes_tree') {
|
|
entity = await sql.getSingleResult('SELECT * FROM notes_tree WHERE note_tree_id = ?', [sync.entity_id]);
|
|
}
|
|
else if (sync.entity_name === 'notes_history') {
|
|
entity = await sql.getSingleResult('SELECT * FROM notes_history WHERE note_history_id = ?', [sync.entity_id]);
|
|
}
|
|
else if (sync.entity_name === 'notes_reordering') {
|
|
entity = {
|
|
note_pid: sync.entity_id,
|
|
ordering: await sql.getMap('SELECT note_tree_id, note_pos FROM notes_tree WHERE note_pid = ?', [sync.entity_id])
|
|
};
|
|
}
|
|
else if (sync.entity_name === 'options') {
|
|
entity = await sql.getSingleResult('SELECT * FROM options WHERE opt_name = ?', [sync.entity_id]);
|
|
}
|
|
else if (sync.entity_name === 'recent_notes') {
|
|
entity = await sql.getSingleResult('SELECT * FROM recent_notes WHERE note_tree_id = ?', [sync.entity_id]);
|
|
}
|
|
else {
|
|
throw new Error(`Unrecognized entity type ${sync.entity_name} in sync #${sync.id}`);
|
|
}
|
|
|
|
if (!entity) {
|
|
log.info(`Sync #${sync.id} entity for ${sync.entity_name} ${sync.entity_id} doesn't exist. Skipping.`);
|
|
return;
|
|
}
|
|
|
|
log.info(`Pushing changes in sync #${sync.id} ${sync.entity_name} ${sync.entity_id}`);
|
|
|
|
const payload = {
|
|
sourceId: await source_id.getCurrentSourceId(),
|
|
entity: entity
|
|
};
|
|
|
|
await syncRequest(syncContext, 'PUT', '/api/sync/' + sync.entity_name, payload);
|
|
}
|
|
|
|
async function checkContentHash(syncContext) {
|
|
const resp = await syncRequest(syncContext, 'GET', '/api/sync/check');
|
|
|
|
if (await getLastSyncedPull() < resp.max_sync_id) {
|
|
log.info("There are some outstanding pulls, skipping content check.");
|
|
|
|
return;
|
|
}
|
|
|
|
const lastSyncedPush = await getLastSyncedPush();
|
|
const notPushedSyncs = await sql.getSingleValue("SELECT COUNT(*) FROM sync WHERE id > ?", [lastSyncedPush]);
|
|
|
|
if (notPushedSyncs > 0) {
|
|
log.info("There's " + notPushedSyncs + " outstanding pushes, skipping content check.");
|
|
|
|
return;
|
|
}
|
|
|
|
const hashes = await content_hash.getHashes();
|
|
let allChecksPassed = true;
|
|
|
|
for (const key in hashes) {
|
|
if (hashes[key] !== resp.hashes[key]) {
|
|
allChecksPassed = false;
|
|
|
|
await event_log.addEvent(`Content hash check for ${key} FAILED. Local is ${hashes[key]}, remote is ${resp.hashes[key]}`);
|
|
|
|
if (key !== 'recent_notes') {
|
|
// let's not get alarmed about recent notes which get updated often and can cause failures in race conditions
|
|
await messaging.sendMessage({type: 'sync-hash-check-failed'});
|
|
}
|
|
}
|
|
}
|
|
|
|
if (allChecksPassed) {
|
|
log.info("Content hash checks PASSED");
|
|
}
|
|
}
|
|
|
|
async function syncRequest(syncContext, method, uri, body) {
|
|
const fullUri = sync_setup.SYNC_SERVER + uri;
|
|
|
|
try {
|
|
const options = {
|
|
method: method,
|
|
uri: fullUri,
|
|
jar: syncContext.cookieJar,
|
|
json: true,
|
|
body: body,
|
|
timeout: sync_setup.SYNC_TIMEOUT
|
|
};
|
|
|
|
if (syncServerCertificate) {
|
|
options.ca = syncServerCertificate;
|
|
}
|
|
|
|
if (sync_setup.SYNC_PROXY && proxyToggle) {
|
|
options.proxy = sync_setup.SYNC_PROXY;
|
|
}
|
|
|
|
return await rp(options);
|
|
}
|
|
catch (e) {
|
|
throw new Error(`Request to ${method} ${fullUri} failed, inner exception: ${e.stack}`);
|
|
}
|
|
}
|
|
|
|
sql.dbReady.then(() => {
|
|
if (sync_setup.isSyncSetup) {
|
|
log.info("Setting up sync to " + sync_setup.SYNC_SERVER + " with timeout " + sync_setup.SYNC_TIMEOUT);
|
|
|
|
if (sync_setup.SYNC_PROXY) {
|
|
log.info("Sync proxy: " + sync_setup.SYNC_PROXY);
|
|
}
|
|
|
|
if (sync_setup.SYNC_CERT_PATH) {
|
|
log.info('Sync certificate: ' + sync_setup.SYNC_CERT_PATH);
|
|
|
|
syncServerCertificate = fs.readFileSync(sync_setup.SYNC_CERT_PATH);
|
|
}
|
|
|
|
setInterval(sync, 60000);
|
|
|
|
// kickoff initial sync immediately
|
|
setTimeout(sync, 1000);
|
|
}
|
|
else {
|
|
log.info("Sync server not configured, sync timer not running.")
|
|
}
|
|
});
|
|
|
|
module.exports = {
|
|
sync
|
|
}; |