Ghost/ghost/limit-service/lib/limit-service.js

const errors = require('@tryghost/errors');
const {MaxLimit, FlagLimit} = require('./limit');
const config = require('./config');
const _ = require('lodash');

class LimitService {
    constructor() {
        this.limits = {};
    }

    loadLimits({limits, helpLink, db}) {
        Object.keys(limits).forEach((name) => {
            if (config[name]) {
                let limitConfig = _.merge({}, limits[name], config[name]);

                if (_.has(limitConfig, 'max')) {
                    this.limits[name] = new MaxLimit({name: name, config: limitConfig, helpLink, db});
                } else {
                    this.limits[name] = new FlagLimit({name: name, config: limitConfig, helpLink});
                }
            }
        });
    }

    isLimited(limitName) {
        return !!this.limits[limitName];
    }

    async checkIsOverLimit(limitName) {
        if (!this.isLimited(limitName)) {
            return;
        }

        try {
            await this.limits[limitName].errorIfIsOverLimit();
            return false;
        } catch (error) {
            if (error instanceof errors.HostLimitError) {
                return true;
            }
        }
    }

    async checkWouldGoOverLimit(limitName) {
        if (!this.isLimited(limitName)) {
            return;
        }

        try {
            await this.limits[limitName].errorIfWouldGoOverLimit();
            return false;
        } catch (error) {
            if (error instanceof errors.HostLimitError) {
                return true;
            }
        }
    }

    async errorIfIsOverLimit(limitName) {
        if (!this.isLimited(limitName)) {
            return;
        }

        await this.limits[limitName].errorIfIsOverLimit();
    }

    async errorIfWouldGoOverLimit(limitName) {
        if (!this.isLimited(limitName)) {
            return;
        }

        await this.limits[limitName].errorIfWouldGoOverLimit();
    }
}

module.exports = LimitService;