You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
116 lines
3.4 KiB
JavaScript
116 lines
3.4 KiB
JavaScript
/*jshint esversion: 6 */
|
|
/* globals Buffer*/
|
|
const Quota = module.exports;
|
|
|
|
//const Util = require("../common-util");
|
|
const Keys = require("../keys");
|
|
const Package = require('../../package.json');
|
|
const Https = require("https");
|
|
|
|
Quota.applyCustomLimits = function (Env) {
|
|
var isLimit = function (o) {
|
|
var valid = o && typeof(o) === 'object' &&
|
|
typeof(o.limit) === 'number' &&
|
|
typeof(o.plan) === 'string' &&
|
|
typeof(o.note) === 'string';
|
|
return valid;
|
|
};
|
|
|
|
// read custom limits from the Environment (taken from config)
|
|
var customLimits = (function (custom) {
|
|
var limits = {};
|
|
Object.keys(custom).forEach(function (k) {
|
|
var user;
|
|
try {
|
|
user = Keys.parseUser(k);
|
|
} catch (err) {
|
|
return void Env.Log.error("PARSE_CUSTOM_LIMIT_BLOCK", {
|
|
user: k,
|
|
error: err.message,
|
|
});
|
|
}
|
|
|
|
var unsafeKey = user.pubkey;
|
|
limits[unsafeKey] = custom[k];
|
|
});
|
|
return limits;
|
|
}(Env.customLimits || {}));
|
|
|
|
Object.keys(customLimits).forEach(function (k) {
|
|
if (!isLimit(customLimits[k])) { return; }
|
|
Env.limits[k] = customLimits[k];
|
|
});
|
|
};
|
|
|
|
Quota.updateCachedLimits = function (Env, cb) {
|
|
Quota.applyCustomLimits(Env);
|
|
if (Env.blockDailyCheck === true ||
|
|
(typeof(Env.blockDailyCheck) === 'undefined' && Env.adminEmail === false && Env.allowSubscriptions === false)) {
|
|
return void cb();
|
|
}
|
|
|
|
var body = JSON.stringify({
|
|
domain: Env.myDomain,
|
|
subdomain: Env.mySubdomain || null,
|
|
adminEmail: Env.adminEmail,
|
|
version: Package.version
|
|
});
|
|
var options = {
|
|
host: 'accounts.cryptpad.fr',
|
|
path: '/api/getauthorized',
|
|
method: 'POST',
|
|
headers: {
|
|
"Content-Type": "application/json",
|
|
"Content-Length": Buffer.byteLength(body)
|
|
}
|
|
};
|
|
|
|
var req = Https.request(options, function (response) {
|
|
if (!('' + response.statusCode).match(/^2\d\d$/)) {
|
|
return void cb('SERVER ERROR ' + response.statusCode);
|
|
}
|
|
var str = '';
|
|
|
|
response.on('data', function (chunk) {
|
|
str += chunk;
|
|
});
|
|
|
|
response.on('end', function () {
|
|
try {
|
|
var json = JSON.parse(str);
|
|
Env.limits = json;
|
|
Quota.applyCustomLimits(Env);
|
|
cb(void 0);
|
|
} catch (e) {
|
|
cb(e);
|
|
}
|
|
});
|
|
});
|
|
|
|
req.on('error', function (e) {
|
|
Quota.applyCustomLimits(Env);
|
|
if (!Env.myDomain) { return cb(); }
|
|
// only return an error if your server allows subscriptions
|
|
cb(e);
|
|
});
|
|
|
|
req.end(body);
|
|
};
|
|
|
|
// The limits object contains storage limits for all the publicKey that have paid
|
|
// To each key is associated an object containing the 'limit' value and a 'note' explaining that limit
|
|
Quota.getUpdatedLimit = function (Env, safeKey, cb) { // FIXME BATCH?S
|
|
Quota.updateCachedLimits(Env, function (err) {
|
|
if (err) { return void cb(err); }
|
|
|
|
var limit = Env.limits[safeKey];
|
|
|
|
if (limit && typeof(limit.limit) === 'number') {
|
|
return void cb(void 0, [limit.limit, limit.plan, limit.note]);
|
|
}
|
|
|
|
return void cb(void 0, [Env.defaultStorageLimit, '', '']);
|
|
});
|
|
};
|
|
|