|
const Keyv = require('keyv'); |
|
const keyvMongo = require('./keyvMongo'); |
|
const keyvRedis = require('./keyvRedis'); |
|
const { math, isEnabled } = require('../server/utils'); |
|
const { logFile, violationFile } = require('./keyvFiles'); |
|
const { BAN_DURATION, USE_REDIS } = process.env ?? {}; |
|
|
|
const duration = math(BAN_DURATION, 7200000); |
|
|
|
const createViolationInstance = (namespace) => { |
|
const config = isEnabled(USE_REDIS) ? { store: keyvRedis } : { store: violationFile, namespace }; |
|
return new Keyv(config); |
|
}; |
|
|
|
|
|
const pending_req = isEnabled(USE_REDIS) |
|
? new Keyv({ store: keyvRedis }) |
|
: new Keyv({ namespace: 'pending_req' }); |
|
|
|
const namespaces = { |
|
pending_req, |
|
ban: new Keyv({ store: keyvMongo, namespace: 'bans', duration }), |
|
general: new Keyv({ store: logFile, namespace: 'violations' }), |
|
concurrent: createViolationInstance('concurrent'), |
|
non_browser: createViolationInstance('non_browser'), |
|
message_limit: createViolationInstance('message_limit'), |
|
token_balance: createViolationInstance('token_balance'), |
|
registrations: createViolationInstance('registrations'), |
|
logins: createViolationInstance('logins'), |
|
}; |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
const getLogStores = (type) => { |
|
if (!type || !namespaces[type]) { |
|
throw new Error(`Invalid store type: ${type}`); |
|
} |
|
return namespaces[type]; |
|
}; |
|
|
|
module.exports = getLogStores; |
|
|