"use strict"; const log = require('./log'); const rp = require('request-promise'); const sql = require('./sql'); const options = require('./options'); const migration = require('./migration'); const utils = require('./utils'); const config = require('./config'); 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 SYNC_SERVER = config['Sync']['syncServerHost']; const isSyncSetup = !!SYNC_SERVER; const SYNC_TIMEOUT = config['Sync']['syncServerTimeout'] || 5000; const SYNC_PROXY = config['Sync']['syncProxy']; let syncInProgress = false; let proxyToggle = true; async function sync() { if (syncInProgress) { log.info("Sync already in progress"); return { success: false, message: "Sync already in progress" }; } syncInProgress = true; try { if (!await migration.isDbUpToDate()) { log.info("DB not up to date"); 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.nowTimestamp(); 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: migration.APP_DB_VERSION, hash: hash }); syncContext.sourceId = resp.sourceId; return syncContext; } async function getLastSyncedPull() { return parseInt(await options.getOption('last_synced_pull')); } 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 " + sync.entity_name + " " + sync.entity_id + " because it has local source id."); continue; } console.log("Pulling ", sync); const resp = await syncRequest(syncContext, 'GET', "/api/sync/" + sync.entity_name + "/" + encodeURIComponent(sync.entity_id)); if (sync.entity_name === 'notes') { await syncUpdate.updateNote(resp.entity, resp.links, 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); } await options.setOption('last_synced_pull', sync.id); } log.info("Finished pull"); } async function getLastSyncedPush() { return parseInt(await options.getOption('last_synced_push')); } 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 sync " + sync.entity_name + " " + sync.entity_id + " because it originates from sync target"); } else { await readAndPushEntity(sync, syncContext); } lastSyncedPush = sync.id; await options.setOption('last_synced_push', lastSyncedPush); } } async function readAndPushEntity(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_path = ?', [sync.entity_id]); } else { throw new Error("Unrecognized entity type " + sync.entity_name); } if (!entity) { log.info("Sync entity for " + sync.entity_name + " " + sync.entity_id + " doesn't exist. Skipping."); return; } log.info("Pushing changes in " + sync.entity_name + " " + sync.entity_id); await sendEntity(syncContext, entity, sync.entity_name); } async function sendEntity(syncContext, entity, entityName) { const payload = { sourceId: source_id.currentSourceId, entity: entity }; if (entityName === 'notes') { payload.links = await sql.getResults('SELECT * FROM links WHERE note_id = ?', [entity.note_id]); } await syncRequest(syncContext, 'PUT', '/api/sync/' + entityName, payload); } async function checkContentHash(syncContext) { 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 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 localContentHash = await content_hash.getContentHash(); if (resp.content_hash === localContentHash) { log.info("Content hash check passed with value: " + localContentHash); } else { await event_log.addEvent("Content hash check failed. Local is " + localContentHash + ", remote is " + resp.content_hash); } } async function syncRequest(syncContext, method, uri, body) { const fullUri = SYNC_SERVER + uri; try { const options = { method: method, uri: fullUri, jar: syncContext.cookieJar, json: true, body: body, timeout: SYNC_TIMEOUT }; if (SYNC_PROXY && proxyToggle) { options.proxy = SYNC_PROXY; } return await rp(options); } catch (e) { throw new Error("Request to " + method + " " + fullUri + " failed, inner exception: " + e.stack); } } if (isSyncSetup) { log.info("Setting up sync to " + SYNC_SERVER + " with timeout " + SYNC_TIMEOUT); if (SYNC_PROXY) { log.info("Sync proxy: " + SYNC_PROXY); } setInterval(sync, 60000); // kickoff initial sync immediately setTimeout(sync, 1000); } else { log.info("Sync server not configured, sync timer not running.") } module.exports = { sync, isSyncSetup };