mirror of
https://git.sr.ht/~cadence/bibliogram
synced 2025-04-04 06:32:03 +00:00
34 lines
901 B
JavaScript
34 lines
901 B
JavaScript
const constants = require("../constants")
|
|
const LimitByFrame = require("./LimitByFrame")
|
|
const {getIdentifier} = require("./get_identifier")
|
|
require("../testimports")(LimitByFrame, getIdentifier)
|
|
|
|
const limiter = new LimitByFrame()
|
|
|
|
function getIPFromReq(req) {
|
|
if (constants.quota.ip_mode === "header") {
|
|
return req.headers[constants.quota.ip_header]
|
|
} else { // constants.quota.ip_mode === "address"
|
|
return req.connection.remoteAddress
|
|
}
|
|
}
|
|
|
|
function remaining(req) {
|
|
if (!constants.quota.enabled) return Infinity // sure.
|
|
|
|
const ip = getIPFromReq(req)
|
|
const identifier = getIdentifier(ip)
|
|
return limiter.remaining(identifier)
|
|
}
|
|
|
|
function add(req, count) {
|
|
if (!constants.quota.enabled) return Infinity // why not.
|
|
|
|
const ip = getIPFromReq(req)
|
|
const identifier = getIdentifier(ip)
|
|
return limiter.add(identifier, count)
|
|
}
|
|
|
|
module.exports.remaining = remaining
|
|
module.exports.add = add
|