161 lines
5.4 KiB
JavaScript
161 lines
5.4 KiB
JavaScript
"use strict";
|
|
|
|
var _path = _interopRequireDefault(require("path"));
|
|
var _fs = _interopRequireDefault(require("fs"));
|
|
var _url = _interopRequireDefault(require("url"));
|
|
var _lodash = _interopRequireDefault(require("lodash"));
|
|
var _util = require("util");
|
|
var _index = _interopRequireDefault(require("./index"));
|
|
var _yargs = _interopRequireDefault(require("../core/yargs"));
|
|
var _importHelper = _interopRequireDefault(require("./import-helper"));
|
|
var _process = _interopRequireDefault(require("process"));
|
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
|
const args = (0, _yargs.default)().argv;
|
|
const api = {
|
|
config: undefined,
|
|
rawConfig: undefined,
|
|
error: undefined,
|
|
async init() {
|
|
let config;
|
|
try {
|
|
if (args.url) {
|
|
config = api.parseDbUrl(args.url);
|
|
} else {
|
|
const module = await _importHelper.default.importModule(api.getConfigFile());
|
|
config = await module.default;
|
|
}
|
|
} catch (e) {
|
|
api.error = e;
|
|
}
|
|
if (typeof config === 'function') {
|
|
// accepts callback parameter
|
|
if (config.length === 1) {
|
|
config = await (0, _util.promisify)(config)();
|
|
} else {
|
|
// returns a promise.
|
|
config = await config();
|
|
}
|
|
}
|
|
api.rawConfig = config;
|
|
return api;
|
|
},
|
|
getConfigFile() {
|
|
if (args.config) {
|
|
return _path.default.resolve(_process.default.cwd(), args.config);
|
|
}
|
|
const defaultPath = _path.default.resolve(_process.default.cwd(), 'config', 'config.json');
|
|
const alternativePath = defaultPath.replace('.json', '.js');
|
|
return _index.default.path.existsSync(alternativePath) ? alternativePath : defaultPath;
|
|
},
|
|
relativeConfigFile() {
|
|
return _path.default.relative(_process.default.cwd(), api.getConfigFile());
|
|
},
|
|
configFileExists() {
|
|
return _index.default.path.existsSync(api.getConfigFile());
|
|
},
|
|
getDefaultConfig() {
|
|
return JSON.stringify({
|
|
development: {
|
|
username: 'root',
|
|
password: null,
|
|
database: 'database_development',
|
|
host: '127.0.0.1',
|
|
dialect: 'mysql'
|
|
},
|
|
test: {
|
|
username: 'root',
|
|
password: null,
|
|
database: 'database_test',
|
|
host: '127.0.0.1',
|
|
dialect: 'mysql'
|
|
},
|
|
production: {
|
|
username: 'root',
|
|
password: null,
|
|
database: 'database_production',
|
|
host: '127.0.0.1',
|
|
dialect: 'mysql'
|
|
}
|
|
}, undefined, 2) + '\n';
|
|
},
|
|
writeDefaultConfig() {
|
|
const configPath = _path.default.dirname(api.getConfigFile());
|
|
if (!_index.default.path.existsSync(configPath)) {
|
|
_index.default.asset.mkdirp(configPath);
|
|
}
|
|
_fs.default.writeFileSync(api.getConfigFile(), api.getDefaultConfig());
|
|
},
|
|
readConfig() {
|
|
if (!api.config) {
|
|
const env = _index.default.generic.getEnvironment();
|
|
if (api.rawConfig === undefined) {
|
|
throw new Error('Error reading "' + api.relativeConfigFile() + '". Error: ' + api.error);
|
|
}
|
|
if (typeof api.rawConfig !== 'object') {
|
|
throw new Error('Config must be an object or a promise for an object: ' + api.relativeConfigFile());
|
|
}
|
|
if (args.url) {
|
|
_index.default.view.log('Parsed url ' + api.filteredUrl(args.url, api.rawConfig));
|
|
} else {
|
|
_index.default.view.log('Loaded configuration file "' + api.relativeConfigFile() + '".');
|
|
}
|
|
if (api.rawConfig[env]) {
|
|
_index.default.view.log('Using environment "' + env + '".');
|
|
api.rawConfig = api.rawConfig[env];
|
|
}
|
|
|
|
// The Sequelize library needs a function passed in to its logging option
|
|
if (api.rawConfig.logging && !_lodash.default.isFunction(api.rawConfig.logging)) {
|
|
api.rawConfig.logging = console.log;
|
|
}
|
|
|
|
// in case url is present - we overwrite the configuration
|
|
if (api.rawConfig.url) {
|
|
api.rawConfig = _lodash.default.merge(api.rawConfig, api.parseDbUrl(api.rawConfig.url));
|
|
} else if (api.rawConfig.use_env_variable) {
|
|
api.rawConfig = _lodash.default.merge(api.rawConfig, api.parseDbUrl(_process.default.env[api.rawConfig.use_env_variable]));
|
|
}
|
|
api.config = api.rawConfig;
|
|
}
|
|
return api.config;
|
|
},
|
|
filteredUrl(uri, config) {
|
|
const regExp = new RegExp(':?' + _lodash.default.escapeRegExp(config.password) + '@');
|
|
return uri.replace(regExp, ':*****@');
|
|
},
|
|
urlStringToConfigHash(urlString) {
|
|
try {
|
|
const urlParts = _url.default.parse(urlString);
|
|
let result = {
|
|
database: urlParts.pathname.replace(/^\//, ''),
|
|
host: urlParts.hostname,
|
|
port: urlParts.port,
|
|
protocol: urlParts.protocol.replace(/:$/, ''),
|
|
ssl: urlParts.query ? urlParts.query.indexOf('ssl=true') >= 0 : false
|
|
};
|
|
if (urlParts.auth) {
|
|
const authParts = urlParts.auth.split(':');
|
|
result.username = authParts[0];
|
|
if (authParts.length > 1) {
|
|
result.password = authParts.slice(1).join(':');
|
|
}
|
|
}
|
|
return result;
|
|
} catch (e) {
|
|
throw new Error('Error parsing url: ' + urlString);
|
|
}
|
|
},
|
|
parseDbUrl(urlString) {
|
|
let config = api.urlStringToConfigHash(urlString);
|
|
config = _lodash.default.assign(config, {
|
|
dialect: config.protocol
|
|
});
|
|
if (config.dialect === 'sqlite' && config.database.indexOf(':memory') !== 0) {
|
|
config = _lodash.default.assign(config, {
|
|
storage: '/' + config.database
|
|
});
|
|
}
|
|
return config;
|
|
}
|
|
};
|
|
module.exports = api; |