• Index
• Search by APIs
• Search by Words
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• ts-di-starter-master
  • .eslintignore
  • src
    • configLoader.ts
    • config.ts
    • server.ts
    • lib
      • services
        • limiters
          • error.ts
          • general.ts
          • limiterGroup.ts
        • events.ts
        • users.ts
        • records.ts
      • express
        • responseBuilder.ts
        • middleware
          • user.ts
          • index.ts
          • exceptionHandler.ts
          • logger.ts
          • limiter.ts
        • controllers
          • user.ts
          • index.ts
          • records.ts
          • health.ts
        • utils.ts
        • .eslintrc.js
      • models
        • record.ts
        • user.ts
        • event.ts
        • limiter.ts
      • socket
        • socket.ts
        • middleware
          • user.ts
          • index.ts
        • controllers
          • index.ts
          • userData.ts
          • .eslintrc.js
      • requests
        • createRecord.ts
    • logger.ts
    • app
      • index.ts
      • api
        • v1
          • health
            • index.ts
          • user
            • index.ts
          • index.ts
          • records
            • index.ts
        • index.ts
      • socketMiddleware.ts
      • expressMiddleware.ts
      • serviceManager.ts
    • start.ts
  • .circleci
    • config.yml
  • example.env
  • images
  • LICENSE
  • commitlint.config.js
  • webpack.config.js
  • example.env.local
  • README.md
  • package.json
  • tests
    • mockRedis.ts
    • lib
      • services
        • events.it.ts
        • users.unit.ts
        • events.unit.ts
    • getAppForTest.ts
    • firestore.ts
    • app
      • health
        • health.unit.ts
      • fixture.ts
      • user
        • user.unit.ts
      • records
        • records.unit.ts
    • .eslintrc.js
  • .eslintrc.js
  • tsconfig.json
  • .gitignore
  • .nycrc.json
  • .prettierrc.js
  • .dockerignore
  • docker-compose.yml

import assert from 'assert';
import { isNil, isObject } from 'lodash';
import { DateTime } from 'luxon';

import { LimitResultInterface } from '@/models/limiter';

/**
 * Attach limit headers
 *
 * @param {Request} request
 * @param {Response} response
 * @param {Date} curDate
 * @returns {void}
 */
export function attachLimitToRequest(request, response, curDate = DateTime.utc().toJSDate()): void {
  const { limitResult }: { limitResult: LimitResultInterface } = request?.locals || {};
  if (!limitResult) return;

  response.set('X-RateLimit-Limit', limitResult.limit);
  response.set('X-RateLimit-Remaining', limitResult.remaining);
  response.set('X-RateLimit-Reset', DateTime.fromJSDate(curDate).plus({ milliseconds: limitResult.resetMs }).toISO());

  if (limitResult.blockReason) {
    // eslint-disable-next-line no-magic-numbers
    response.set('Retry-After', Math.round(limitResult.resetMs / 1000));
  }
}

/**
 * Assert properties of object
 *
 * @param {{}} obj
 * @param {string[]} properties
 * @returns {void}
 */
export function assertProperties(obj: any, properties: string[]): void {
  if (!isObject(obj)) throw new Error('not object');
  if (properties.length === 0) throw new Error('no properties provided');

  properties.map((prop) => assert(!isNil(obj[prop]), `${prop} not found on obj`));
}