From 90f8fe7e538bb7e52d2558152a0390e693f39b11 Mon Sep 17 00:00:00 2001 From: Akihiko Odaki Date: Thu, 29 Mar 2018 01:20:40 +0900 Subject: Introduce processor --- src/api/limitter.ts | 83 ----------------------------------------------------- 1 file changed, 83 deletions(-) delete mode 100644 src/api/limitter.ts (limited to 'src/api/limitter.ts') diff --git a/src/api/limitter.ts b/src/api/limitter.ts deleted file mode 100644 index 9d2c42d335..0000000000 --- a/src/api/limitter.ts +++ /dev/null @@ -1,83 +0,0 @@ -import * as Limiter from 'ratelimiter'; -import * as debug from 'debug'; -import limiterDB from '../db/redis'; -import { Endpoint } from './endpoints'; -import { IAuthContext } from './authenticate'; -import getAcct from '../common/user/get-acct'; - -const log = debug('misskey:limitter'); - -export default (endpoint: Endpoint, ctx: IAuthContext) => new Promise((ok, reject) => { - const limitation = endpoint.limit; - - const key = limitation.hasOwnProperty('key') - ? limitation.key - : endpoint.name; - - const hasShortTermLimit = - limitation.hasOwnProperty('minInterval'); - - const hasLongTermLimit = - limitation.hasOwnProperty('duration') && - limitation.hasOwnProperty('max'); - - if (hasShortTermLimit) { - min(); - } else if (hasLongTermLimit) { - max(); - } else { - ok(); - } - - // Short-term limit - function min() { - const minIntervalLimiter = new Limiter({ - id: `${ctx.user._id}:${key}:min`, - duration: limitation.minInterval, - max: 1, - db: limiterDB - }); - - minIntervalLimiter.get((err, info) => { - if (err) { - return reject('ERR'); - } - - log(`@${getAcct(ctx.user)} ${endpoint.name} min remaining: ${info.remaining}`); - - if (info.remaining === 0) { - reject('BRIEF_REQUEST_INTERVAL'); - } else { - if (hasLongTermLimit) { - max(); - } else { - ok(); - } - } - }); - } - - // Long term limit - function max() { - const limiter = new Limiter({ - id: `${ctx.user._id}:${key}`, - duration: limitation.duration, - max: limitation.max, - db: limiterDB - }); - - limiter.get((err, info) => { - if (err) { - return reject('ERR'); - } - - log(`@${getAcct(ctx.user)} ${endpoint.name} max remaining: ${info.remaining}`); - - if (info.remaining === 0) { - reject('RATE_LIMIT_EXCEEDED'); - } else { - ok(); - } - }); - } -}); -- cgit v1.2.3-freya