166 lines
3.2 KiB
JavaScript
166 lines
3.2 KiB
JavaScript
var fs = require('fs')
|
|
var _ = require('lodash')
|
|
|
|
var re_comment = /^\s*;/
|
|
var re_setion = /^\s*\[([^\]]+)\]\s*$/
|
|
var re_kv = /^([^=]+)=(.*)$/
|
|
var re_boolean = /^(true|false)$/i
|
|
|
|
/**
|
|
* Namespaces of socket.io
|
|
* @type {{SYS: string, LOG: string, PROC: string}}
|
|
*/
|
|
exports.NSP = {
|
|
SYS: '/sys',
|
|
LOG: '/log',
|
|
PROC: '/proc'
|
|
}
|
|
|
|
/**
|
|
* Configurations
|
|
* @type {[type]}
|
|
*/
|
|
exports.File = File
|
|
|
|
/**
|
|
* Configurations store in a disk file.
|
|
* @param {Object} options
|
|
* @constructor
|
|
*/
|
|
function File (options) {
|
|
if (!(this instanceof File)) {
|
|
return new File(options)
|
|
}
|
|
|
|
if (typeof options === 'string') {
|
|
options = {
|
|
file: options
|
|
}
|
|
}
|
|
options = _.assign({}, options || {})
|
|
if (!options.file) {
|
|
throw new Error('`file` is required.')
|
|
}
|
|
Object.freeze(options)
|
|
this.options = options
|
|
}
|
|
|
|
/**
|
|
* Load data from file (sync).
|
|
*/
|
|
File.prototype.loadSync = function () {
|
|
if (!fs.existsSync(this.options.file)) {
|
|
this._data = {}
|
|
return this
|
|
}
|
|
|
|
var json = {}
|
|
var sec
|
|
fs.readFileSync(this.options.file, {
|
|
encoding: 'utf8'
|
|
}).split(/[\r\n]/).forEach(function (line) {
|
|
// Empty line.
|
|
if (!line) {
|
|
sec = null
|
|
return
|
|
}
|
|
// Remove comments.
|
|
if (re_comment.test(line)) {
|
|
return
|
|
}
|
|
var ms
|
|
// Sections.
|
|
if ((ms = line.match(re_setion)) && ms.length === 2) {
|
|
json[sec = ms[1].trim()] = {}
|
|
return
|
|
}
|
|
|
|
// Key-value pairs.
|
|
if ((ms = line.match(re_kv)) && ms.length === 3) {
|
|
var key = ms[1].trim()
|
|
var value = ms[2].trim()
|
|
// Parse boolean and number.
|
|
if (!isNaN(value)) {
|
|
value = parseFloat(value)
|
|
} else if (re_boolean.test(value)) {
|
|
value = value.toLowerCase() === 'true'
|
|
}
|
|
if (sec) {
|
|
json[sec][key] = value
|
|
} else {
|
|
json[key] = value
|
|
}
|
|
}
|
|
})
|
|
|
|
this._data = json
|
|
return this
|
|
}
|
|
|
|
/**
|
|
* Save data to a disk file (sync).
|
|
* @returns {File}
|
|
*/
|
|
File.prototype.saveSync = function () {
|
|
function wrapValue (key, value) {
|
|
return key + ' = ' + (typeof value === 'string' ? value : JSON.stringify(value)) + '\n'
|
|
}
|
|
var ini = ''
|
|
for (var key in this._data) {
|
|
var value = this._data[key]
|
|
// TODO: Array type.
|
|
if (typeof value === 'object') {
|
|
ini += '[ ' + key + ' ]\n'
|
|
for (var subKey in value) {
|
|
ini += wrapValue(subKey, value[subKey])
|
|
}
|
|
ini += '\n'
|
|
}
|
|
ini += wrapValue(key, value)
|
|
}
|
|
fs.writeFileSync(this.options.file, ini)
|
|
return this
|
|
}
|
|
|
|
/**
|
|
* Get data.
|
|
* @returns {{}|*}
|
|
*/
|
|
File.prototype.valueOf = function () {
|
|
return this._data
|
|
}
|
|
|
|
/**
|
|
* Get/set/remove key-value pairs.
|
|
* @param {String} key
|
|
* @param {Mixed} value
|
|
* @param {Mixed} def
|
|
* @returns {*}
|
|
*/
|
|
File.prototype.val = function (key, value, def) {
|
|
if (!key) {
|
|
return
|
|
}
|
|
// Load config from File.
|
|
this.loadSync()
|
|
|
|
if (typeof value === 'undefined') {
|
|
// Get config.
|
|
return this._data[key]
|
|
} else if (value == null) {
|
|
// Clear config.
|
|
delete this._data[key]
|
|
// Reset to default if necessary.
|
|
if (typeof def !== 'undefined') {
|
|
this._data[key] = def
|
|
}
|
|
return this.saveSync()
|
|
}
|
|
|
|
this._data[key] = value
|
|
|
|
// Save it.
|
|
this.saveSync()
|
|
return this
|
|
}
|