const Model = require('objection').Model const path = require('path') const fs = require('fs-extra') const _ = require('lodash') const yaml = require('js-yaml') const commonHelper = require('../../helpers/common') /* global WIKI */ /** * Storage model */ module.exports = class Storage extends Model { static get tableName() { return 'storage' } static get jsonSchema () { return { type: 'object', required: ['key', 'title', 'isEnabled'], properties: { id: {type: 'integer'}, key: {type: 'string'}, title: {type: 'string'}, isEnabled: {type: 'boolean'}, mode: {type: 'string'}, config: {type: 'object'} } } } static async getTargets() { return WIKI.db.storage.query() } static async refreshTargetsFromDisk() { try { const dbTargets = await WIKI.db.storage.query() // -> Fetch definitions from disk const storageDirs = await fs.readdir(path.join(WIKI.SERVERPATH, 'modules/storage')) let diskTargets = [] for (let dir of storageDirs) { const def = await fs.readFile(path.join(WIKI.SERVERPATH, 'modules/storage', dir, 'definition.yml'), 'utf8') diskTargets.push(yaml.safeLoad(def)) } // -> Insert new targets let newTargets = [] _.forEach(diskTargets, target => { if (!_.some(dbTargets, ['key', target.key])) { newTargets.push({ key: target.key, title: target.title, isEnabled: false, mode: 'push', config: _.transform(target.props, (result, value, key) => { if (_.isPlainObject(value)) { let cfgValue = { type: value.type.toLowerCase(), value: !_.isNil(value.default) ? value.default : commonHelper.getTypeDefaultValue(value.type) } if (_.isArray(value.enum)) { cfgValue.enum = value.enum } _.set(result, key, cfgValue) } else { _.set(result, key, { type: value.toLowerCase(), value: commonHelper.getTypeDefaultValue(value) }) } return result }, {}) }) } }) if (newTargets.length > 0) { await WIKI.db.storage.query().insert(newTargets) WIKI.logger.info(`Loaded ${newTargets.length} new storage targets: [ OK ]`) } else { WIKI.logger.info(`No new storage targets found: [ SKIPPED ]`) } } catch (err) { WIKI.logger.error(`Failed to scan or load new storage providers: [ FAILED ]`) WIKI.logger.error(err) } } static async createPage(page) { const targets = await WIKI.db.storage.query().where('isEnabled', true) if (targets && targets.length > 0) { _.forEach(targets, target => { WIKI.queue.job.syncStorage.add({ event: 'created', target, page }, { removeOnComplete: true }) }) } } }