wikijs-fork/server/core/config.js

89 lines
2.2 KiB
JavaScript
Raw Normal View History

2017-04-02 23:56:47 +00:00
const _ = require('lodash')
2017-05-15 01:20:40 +00:00
const cfgHelper = require('../helpers/config')
const fs = require('fs')
const path = require('path')
const yaml = require('js-yaml')
/* global WIKI */
2017-07-29 21:33:08 +00:00
module.exports = {
/**
* Load root config from disk
*/
init() {
let confPaths = {
config: path.join(WIKI.ROOTPATH, 'config.yml'),
data: path.join(WIKI.SERVERPATH, 'app/data.yml'),
dataRegex: path.join(WIKI.SERVERPATH, 'app/regex.js')
2017-07-29 21:33:08 +00:00
}
let appconfig = {}
let appdata = {}
try {
appconfig = yaml.safeLoad(
cfgHelper.parseConfigValue(
fs.readFileSync(confPaths.config, 'utf8')
)
2017-05-13 18:44:04 +00:00
)
2017-07-29 21:33:08 +00:00
appdata = yaml.safeLoad(fs.readFileSync(confPaths.data, 'utf8'))
appdata.regex = require(confPaths.dataRegex)
} catch (ex) {
console.error(ex)
process.exit(1)
}
2017-04-02 23:56:47 +00:00
2017-07-29 21:33:08 +00:00
// Merge with defaults
2017-04-02 23:56:47 +00:00
2017-07-29 21:33:08 +00:00
appconfig = _.defaultsDeep(appconfig, appdata.defaults.config)
2017-04-02 23:56:47 +00:00
2017-07-29 21:33:08 +00:00
if (appconfig.port < 1) {
appconfig.port = process.env.PORT || 80
}
2017-04-19 00:31:07 +00:00
2017-09-10 05:41:22 +00:00
appconfig.public = (appconfig.public === true || _.toLower(appconfig.public) === 'true')
WIKI.config = appconfig
WIKI.data = appdata
WIKI.version = require(path.join(WIKI.ROOTPATH, 'package.json')).version
2017-07-29 21:33:08 +00:00
},
/**
* Load config from DB
*/
2018-05-28 18:46:55 +00:00
async loadFromDb() {
let conf = await WIKI.db.settings.getConfig()
if (conf) {
WIKI.config = _.defaultsDeep(conf, WIKI.config)
2017-12-17 04:41:16 +00:00
} else {
2018-05-28 18:46:55 +00:00
WIKI.logger.warn('DB Configuration is empty or incomplete. Switching to Setup mode...')
WIKI.config.setup = true
2017-12-17 04:41:16 +00:00
}
},
/**
* Save config to DB
*
2018-05-28 18:46:55 +00:00
* @param {Array} keys Array of keys to save
2017-12-17 04:41:16 +00:00
* @returns Promise
*/
2018-05-28 18:46:55 +00:00
async saveToDb(keys) {
2017-12-17 04:41:16 +00:00
try {
2018-05-28 18:46:55 +00:00
for (let key of keys) {
let value = _.get(WIKI.config, key, null)
if (!_.isPlainObject(value)) {
value = { v: value }
}
let affectedRows = await WIKI.db.settings.query().patch({ value }).where('key', key)
2018-05-28 18:46:55 +00:00
if (affectedRows === 0 && value) {
await WIKI.db.settings.query().insert({ key, value })
2018-05-28 18:46:55 +00:00
}
2017-07-29 21:33:08 +00:00
}
2017-12-17 04:41:16 +00:00
} catch (err) {
WIKI.logger.error(`Failed to save configuration to DB: ${err.message}`)
2017-12-17 04:41:16 +00:00
return false
}
return true
2017-04-02 23:56:47 +00:00
}
}