'use strict' ;
const { Captchas , Ratelimits } = require ( _ _dirname + '/../../db/' )
, config = require ( _ _dirname + '/../../lib/misc/config.js' )
, production = process . env . NODE _ENV === 'production' ;
module . exports = async ( req , res , next ) => {
const { secureCookies , rateLimitCost , captchaOptions } = config . get ;
if ( ! [ 'grid' , 'text' ] . includes ( captchaOptions . type ) ) {
return next ( ) ; //only grid and text captcha continue
}
const generateCaptcha = require ( _ _dirname + ` /../../lib/captcha/generators/ ${ captchaOptions . type } .js ` ) ;
if ( ! production && req . cookies [ 'captchaid' ] != null ) {
return res . redirect ( ` /captcha/ ${ req . cookies [ 'captchaid' ] } .jpg ` ) ;
}
let captchaId ;
let maxAge = 5 * 60 * 1000 ;
try {
if ( ! res . locals . anonymizer ) {
const ratelimit = await Ratelimits . incrmentQuota ( res . locals . ip . cloak , 'captcha' , rateLimitCost . captcha ) ;
if ( ratelimit > 100 ) {
return res . status ( 429 ) . redirect ( '/file/ratelimit.png' ) ;
}
}
const captchaCount = await Captchas . db . estimatedDocumentCount ( ) ;
if ( captchaCount >= captchaOptions . generateLimit ) {
//TODOs: round robin sample? store in redis? only sample random with longer than x expiry?
const captchaSample = await Captchas . randomSample ( ) ;
const randomCaptcha = captchaSample [ 0 ] ;
captchaId = randomCaptcha . _id ;
maxAge = Math . abs ( ( randomCaptcha . expireAt . getTime ( ) + maxAge ) - Date . now ( ) ) ; //abs in case mongo hasn't pruned, and will not be too big since it can't be too far away from pruning anyway
} else {
( { captchaId } = await generateCaptcha ( ) ) ;
}
} catch ( err ) {
return next ( err ) ;
}
return res
. cookie ( 'captchaid' , captchaId . toString ( ) , {
'secure' : production && secureCookies && ( req . headers [ 'x-forwarded-proto' ] === 'https' ) ,
'sameSite' : 'strict' ,
maxAge ,
} )
. redirect ( ` /captcha/ ${ captchaId } .jpg ` ) ;
} ;