import electron from 'electron'; import NeDB from 'nedb'; import fsPath from 'path'; import crypto from 'crypto'; import * as util from '../common/misc'; import {DB_PERSIST_INTERVAL} from '../common/constants'; const TYPE_RESOURCE = 'Resource'; const TYPE_CONFIG = 'Config'; export const SYNC_MODE_OFF = 'paused'; export const SYNC_MODE_ON = 'active'; export const SYNC_MODE_NEVER = 'never'; export const SYNC_MODE_UNSET = 'unset'; let changeListeners = []; export function onChange (callback) { changeListeners.push(callback); } export function offChange (callback) { changeListeners = changeListeners.filter(l => l !== callback); } let _changeTimeout = null; function _notifyChange () { clearTimeout(_changeTimeout); _changeTimeout = setTimeout(() => { for (const fn of changeListeners) { fn(); } }, 200); } export function allActiveResources (resourceGroupId = null) { if (resourceGroupId) { return findActiveResources({resourceGroupId}); } else { return findActiveResources({}); } } export function activeResourcesForResourceGroup (resourceGroupId) { return findActiveResources({resourceGroupId}); } export function allResources () { return findResources({}); } export async function findResources (query = {}) { return _execDB(TYPE_RESOURCE, 'find', query); } export async function findActiveResources (query) { const configs = await findActiveConfigs(); const resourceGroupIds = configs.map(c => c.resourceGroupId); return findResources(Object.assign({resourceGroupId: {$in: resourceGroupIds}}, query)); } export async function findActiveDirtyResources () { return findActiveResources({dirty: true}); } export async function findActiveDirtyResourcesForResourceGroup (resourceGroupId) { return findActiveResources({dirty: true, resourceGroupId}); } export async function findDirtyResourcesForResourceGroup (resourceGroupId) { return findResources({dirty: true, resourceGroupId}); } export async function findResourcesForResourceGroup (resourceGroupId) { return findResources({resourceGroupId}); } export async function getResourceByDocId (id, resourceGroupId = null) { let query; if (resourceGroupId) { query = {id, resourceGroupId}; } else { query = {id}; } const rawDocs = await _execDB(TYPE_RESOURCE, 'find', query); return rawDocs.length >= 1 ? rawDocs[0] : null; } /** * This function is temporary and should only be called when cleaning * up duplicate ResourceGroups * @param id * @returns {*} */ export function findResourcesByDocId (id) { return _execDB(TYPE_RESOURCE, 'find', {id}); } /** * This function is temporary and should only be called when cleaning * up duplicate ResourceGroups * @param resourceGroupId * @returns {*} */ export async function removeResourceGroup (resourceGroupId) { await _execDB(TYPE_RESOURCE, 'remove', {resourceGroupId}, {multi: true}); await _execDB(TYPE_CONFIG, 'remove', {resourceGroupId}, {multi: true}); _notifyChange(); } export async function insertResource (resource) { const h = crypto.createHash('md5'); h.update(resource.resourceGroupId); h.update(resource.id); const newResource = Object.assign({}, resource, {_id: `rs_${h.digest('hex')}`}); await _execDB(TYPE_RESOURCE, 'insert', newResource); _notifyChange(); return newResource; } export async function updateResource (resource, ...patches) { const newDoc = Object.assign({}, resource, ...patches); await _execDB(TYPE_RESOURCE, 'update', {_id: resource._id}, newDoc, {multi: true}); _notifyChange(); return newDoc; } export async function removeResource (resource) { await _execDB(TYPE_RESOURCE, 'remove', {_id: resource._id}, {multi: true}); _notifyChange(); } // ~~~~~~ // // Config // // ~~~~~~ // export function findConfigs (query) { return _execDB(TYPE_CONFIG, 'find', query); } export function allConfigs () { return findConfigs({}); } export function findInactiveConfigs (excludedResourceGroupId = null) { if (excludedResourceGroupId) { return findConfigs({$not: {syncMode: SYNC_MODE_ON, excludedResourceGroupId}}); } else { return findConfigs({$not: {syncMode: SYNC_MODE_ON}}); } } export function findActiveConfigs (resourceGroupId = null) { if (resourceGroupId) { return findConfigs({syncMode: SYNC_MODE_ON, resourceGroupId}); } else { return findConfigs({syncMode: SYNC_MODE_ON}); } } export async function getConfig (resourceGroupId) { const rawDocs = await _execDB(TYPE_CONFIG, 'find', {resourceGroupId}); return rawDocs.length >= 1 ? _initConfig(rawDocs[0]) : null; } export async function updateConfig (config, ...patches) { const doc = _initConfig(Object.assign(config, ...patches)); await _execDB(TYPE_CONFIG, 'update', {_id: doc._id}, doc); return doc; } export function removeConfig (config) { return _execDB(TYPE_CONFIG, 'remove', {_id: config._id}); } export async function insertConfig (config) { const doc = _initConfig(config); await _execDB(TYPE_CONFIG, 'insert', doc); return doc; } function _initConfig (data) { return Object.assign({ _id: util.generateId('scf'), syncMode: SYNC_MODE_UNSET, resourceGroupId: null }, data); } export function initDB (config, forceReset) { if (!_database || forceReset) { const basePath = electron.remote.app.getPath('userData'); _database = {}; // NOTE: Do not EVER change this. EVER! const resourcePath = fsPath.join(basePath, 'sync/Resource.db'); const configPath = fsPath.join(basePath, 'sync/Config.db'); // Fill in the defaults _database['Resource'] = new NeDB( Object.assign({filename: resourcePath, autoload: true}, config) ); _database['Config'] = new NeDB( Object.assign({filename: configPath, autoload: true}, config) ); for (const key of Object.keys(_database)) { _database[key].persistence.setAutocompactionInterval(DB_PERSIST_INTERVAL); } // Done console.log(`-- Initialize Sync DB at ${basePath} --`); } } // ~~~~~~~ // // Helpers // // ~~~~~~~ // let _database = null; function _getDB (type, config = {}) { initDB(config); return _database[type]; } function _execDB (type, fnName, ...args) { return new Promise((resolve, reject) => { _getDB(type)[fnName](...args, (err, data) => { err ? reject(err) : resolve(data); }); }); }