initialize Env from server and deduplicate several attributes
ansuz
4 years ago
parent
6a51e4e88e
commit
369c92c01d
@ -0,0 +1,200 @@
|
|||||||
|
/* jshint esversion: 6 */
|
||||||
|
/* globals process */
|
||||||
|
|
||||||
|
const Crypto = require('crypto');
|
||||||
|
const WriteQueue = require("./write-queue");
|
||||||
|
const BatchRead = require("./batch-read");
|
||||||
|
|
||||||
|
const Keys = require("./keys");
|
||||||
|
const Core = require("./commands/core");
|
||||||
|
|
||||||
|
const Quota = require("./commands/quota");
|
||||||
|
const Util = require("./common-util");
|
||||||
|
|
||||||
|
module.exports.create = function (config) {
|
||||||
|
// mode can be FRESH (default), DEV, or PACKAGE
|
||||||
|
|
||||||
|
const Env = {
|
||||||
|
FRESH_KEY: '',
|
||||||
|
FRESH_MODE: true,
|
||||||
|
DEV_MODE: false,
|
||||||
|
configCache: {},
|
||||||
|
flushCache: function () {
|
||||||
|
Env.configCache = {};
|
||||||
|
Env.FRESH_KEY = +new Date();
|
||||||
|
if (!(Env.DEV_MODE || Env.FRESH_MODE)) { Env.FRESH_MODE = true; }
|
||||||
|
if (!Env.Log) { return; }
|
||||||
|
Env.Log.info("UPDATING_FRESH_KEY", Env.FRESH_KEY);
|
||||||
|
},
|
||||||
|
|
||||||
|
Log: undefined,
|
||||||
|
// store
|
||||||
|
id: Crypto.randomBytes(8).toString('hex'),
|
||||||
|
|
||||||
|
launchTime: +new Date(),
|
||||||
|
|
||||||
|
inactiveTime: config.inactiveTime,
|
||||||
|
archiveRetentionTime: config.archiveRetentionTime,
|
||||||
|
accountRetentionTime: config.accountRetentionTime,
|
||||||
|
|
||||||
|
// TODO implement mutability
|
||||||
|
adminEmail: config.adminEmail,
|
||||||
|
supportMailbox: config.supportMailboxPublicKey,
|
||||||
|
|
||||||
|
metadata_cache: {},
|
||||||
|
channel_cache: {},
|
||||||
|
queueStorage: WriteQueue(),
|
||||||
|
queueDeletes: WriteQueue(),
|
||||||
|
queueValidation: WriteQueue(),
|
||||||
|
|
||||||
|
batchIndexReads: BatchRead("HK_GET_INDEX"),
|
||||||
|
batchMetadata: BatchRead('GET_METADATA'),
|
||||||
|
batchRegisteredUsers: BatchRead("GET_REGISTERED_USERS"),
|
||||||
|
batchDiskUsage: BatchRead('GET_DISK_USAGE'),
|
||||||
|
batchUserPins: BatchRead('LOAD_USER_PINS'),
|
||||||
|
batchTotalSize: BatchRead('GET_TOTAL_SIZE'),
|
||||||
|
batchAccountQuery: BatchRead("QUERY_ACCOUNT_SERVER"),
|
||||||
|
|
||||||
|
intervals: {},
|
||||||
|
maxUploadSize: config.maxUploadSize || (20 * 1024 * 1024),
|
||||||
|
premiumUploadSize: false, // overridden below...
|
||||||
|
Sessions: {},
|
||||||
|
paths: {},
|
||||||
|
//msgStore: config.store,
|
||||||
|
|
||||||
|
netfluxUsers: {},
|
||||||
|
|
||||||
|
pinStore: undefined,
|
||||||
|
|
||||||
|
limits: {},
|
||||||
|
admins: [],
|
||||||
|
WARN: function (e, output) { // TODO deprecate this
|
||||||
|
if (!Env.Log) { return; }
|
||||||
|
if (e && output) {
|
||||||
|
Env.Log.warn(e, {
|
||||||
|
output: output,
|
||||||
|
message: String(e),
|
||||||
|
stack: new Error(e).stack,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
allowSubscriptions: config.allowSubscriptions === true,
|
||||||
|
blockDailyCheck: config.blockDailyCheck === true,
|
||||||
|
|
||||||
|
myDomain: config.myDomain,
|
||||||
|
mySubdomain: config.mySubdomain, // only exists for the accounts integration
|
||||||
|
customLimits: {},
|
||||||
|
// FIXME this attribute isn't in the default conf
|
||||||
|
// but it is referenced in Quota
|
||||||
|
domain: config.domain,
|
||||||
|
|
||||||
|
maxWorkers: config.maxWorkers,
|
||||||
|
disableIntegratedTasks: config.disableIntegratedTasks || false,
|
||||||
|
disableIntegratedEviction: config.disableIntegratedEviction || false,
|
||||||
|
lastEviction: +new Date(),
|
||||||
|
};
|
||||||
|
|
||||||
|
(function () {
|
||||||
|
if (process.env.PACKAGE) {
|
||||||
|
// `PACKAGE=1 node server` uses the version string from package.json as the cache string
|
||||||
|
console.log("PACKAGE MODE ENABLED");
|
||||||
|
Env.FRESH_MODE = false;
|
||||||
|
Env.DEV_MODE = false;
|
||||||
|
} else if (process.env.DEV) {
|
||||||
|
// `DEV=1 node server` will use a random cache string on every page reload
|
||||||
|
console.log("DEV MODE ENABLED");
|
||||||
|
Env.FRESH_MODE = false;
|
||||||
|
Env.DEV_MODE = true;
|
||||||
|
} else {
|
||||||
|
// `FRESH=1 node server` will set a random cache string when the server is launched
|
||||||
|
// and use it for the process lifetime or until it is reset from the admin panel
|
||||||
|
console.log("FRESH MODE ENABLED");
|
||||||
|
Env.FRESH_KEY = +new Date();
|
||||||
|
}
|
||||||
|
}());
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
(function () {
|
||||||
|
var custom = config.customLimits;
|
||||||
|
if (!custom) { return; }
|
||||||
|
|
||||||
|
var stored = Env.customLimits;
|
||||||
|
|
||||||
|
Object.keys(custom).forEach(function (k) {
|
||||||
|
var unsafeKey = Keys.canonicalize(k);
|
||||||
|
|
||||||
|
if (!unsafeKey) {
|
||||||
|
console.log("INVALID_CUSTOM_LIMIT_ID", {
|
||||||
|
message: "A custom quota upgrade was provided via your config with an invalid identifier. It will be ignored.",
|
||||||
|
key: k,
|
||||||
|
value: custom[k],
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (stored[unsafeKey]) {
|
||||||
|
console.log("INVALID_CUSTOM_LIMIT_DUPLICATED", {
|
||||||
|
message: "A duplicated custom quota upgrade was provided via your config which would have overridden an existing value. It will be ignored.",
|
||||||
|
key: k,
|
||||||
|
value: custom[k],
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!Quota.isValidLimit(custom[k])) {
|
||||||
|
console.log("INVALID_CUSTOM_LIMIT_VALUE", {
|
||||||
|
message: "A custom quota upgrade was provided via your config with an invalid value. It will be ignored.",
|
||||||
|
key: k,
|
||||||
|
value: custom[k],
|
||||||
|
});
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
var limit = stored[unsafeKey] = Util.clone(custom[k]);
|
||||||
|
limit.origin = 'config';
|
||||||
|
});
|
||||||
|
}());
|
||||||
|
|
||||||
|
(function () {
|
||||||
|
var pes = config.premiumUploadSize;
|
||||||
|
if (!isNaN(pes) && pes >= Env.maxUploadSize) {
|
||||||
|
Env.premiumUploadSize = pes;
|
||||||
|
}
|
||||||
|
}());
|
||||||
|
|
||||||
|
var paths = Env.paths;
|
||||||
|
|
||||||
|
var keyOrDefaultString = function (key, def) {
|
||||||
|
return typeof(config[key]) === 'string'? config[key]: def;
|
||||||
|
};
|
||||||
|
|
||||||
|
paths.pin = keyOrDefaultString('pinPath', './pins');
|
||||||
|
paths.block = keyOrDefaultString('blockPath', './block');
|
||||||
|
paths.data = keyOrDefaultString('filePath', './datastore');
|
||||||
|
paths.staging = keyOrDefaultString('blobStagingPath', './blobstage');
|
||||||
|
paths.blob = keyOrDefaultString('blobPath', './blob');
|
||||||
|
paths.decree = keyOrDefaultString('decreePath', './data/');
|
||||||
|
paths.archive = keyOrDefaultString('archivePath', './data/archive');
|
||||||
|
paths.task = keyOrDefaultString('taskPath', './tasks');
|
||||||
|
|
||||||
|
Env.defaultStorageLimit = typeof(config.defaultStorageLimit) === 'number' && config.defaultStorageLimit >= 0?
|
||||||
|
config.defaultStorageLimit:
|
||||||
|
Core.DEFAULT_LIMIT;
|
||||||
|
|
||||||
|
try {
|
||||||
|
Env.admins = (config.adminKeys || []).map(function (k) {
|
||||||
|
try {
|
||||||
|
return Keys.canonicalize(k);
|
||||||
|
} catch (err) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}).filter(Boolean);
|
||||||
|
} catch (e) {
|
||||||
|
console.error("Can't parse admin keys. Please update or fix your config.js file!");
|
||||||
|
}
|
||||||
|
|
||||||
|
return Env;
|
||||||
|
};
|
||||||
Loading…
Reference in New Issue