@tryghost/limit-service 1.2.19 → 1.3.0

package/README.md

@@ -158,7 +158,7 @@ db.transaction((transacting) => {
 
 ### Types of limits
 At the moment there are four different types of limits that limit service allows to define. These types are:
-1. `flag` - is an "on/off" switch for certain feature. Example usecase: "disable all emails". It's identified by a `disabled: true` property in the "limits" configuration.
+1. `flag` - is an "on/off" switch for certain feature. Example usecase: "disable all emails". It's identified by a `disabled: true` property in the "limits" configuration. It is possible to overwrite the limit by providing a `currentCountQuery` for it. This is useful in cases where we introduce new limits to existing plans and customers have already been using the feature affected by the limit. By providing a `currentCountQuery` that detects if the feature is already in use, we won't disable it.
 2. `max` - checks if the maximum amount of the resource has been used up.Example usecase: "disable creating a staff user when maximum of 5 has been reached". To configure this limit add `max: NUMBER` to the configuration. The limits that support max checks are: `members`, and `staff`
 3. `maxPeriodic` - it's a variation of `max` type with a difference that the check is done over certain period of time. Example usecase: "disable sending emails when the sent emails count has acceded a limit for last billing period". To enable this limit define `maxPeriodic: NUMBER` in the limit configuration and provide a subscription configuration when initializing the limit service instance. The subscription object comes as a separate parameter and has to contain two properties: `startDate` and `interval`, where `startDate` is a date in  ISO 8601 format and period is `'month'` (other values like `'year'` are not supported yet)
 4. `allowList` - checks if provided value is defined in configured "allowlist". Example usecase: "disable theme activation if it is not an official theme". To configure this limit define ` allowlist: ['VALUE_1', 'VALUE_2', 'VALUE_N']` property in the "limits" parameter.

package/lib/limit.js

@@ -264,6 +264,7 @@ class FlagLimit extends Limit {
      * @param {Number} options.config.disabled - disabled/enabled flag for the limit
      * @param {String} options.config.error - error message to use when limit is reached
      * @param {String} options.helpLink - URL to the resource explaining how the limit works
+     * @param {Function} [options.config.currentCountQuery] - query checking the state that would be compared against the limit
      * @param {Object} [options.db] - instance of knex db connection that currentCountQuery can use to run state check through
      * @param {Object} options.errors - instance of errors compatible with GhostError errors (@tryghost/errors)
      */
@@ -273,6 +274,7 @@ class FlagLimit extends Limit {
 
         this.disabled = config.disabled;
         this.fallbackMessage = `Your plan does not support ${userFacingLimitName}. Please upgrade to enable ${userFacingLimitName}.`;
+        this.currentCountQueryFn = config?.currentCountQuery || null;
     }
 
     generateError() {
@@ -288,20 +290,59 @@ class FlagLimit extends Limit {
     }
 
     /**
-     * Flag limits are on/off so using a feature is always over the limit
+     * @param {Object} [options]
+     * @param {Object} [options.transacting] Transaction to run the count query on
+     * @returns {Promise<boolean>} - returns the current count of items that would be compared against the limit
      */
-    async errorIfWouldGoOverLimit() {
-        if (this.disabled) {
+    async currentCountQuery(options = {}) {
+        if (!this.currentCountQueryFn || typeof this.currentCountQueryFn !== 'function') {
+            return false;
+        }
+
+        return await this.currentCountQueryFn(options.transacting ?? this.db?.knex);
+    }
+
+    // As Flag limits are on/off, we won't check against max values.
+    // `errorIfWouldGoOverLimit` and `errorIfIsOverLimit` end up doing the same thing.
+    async _isOrWouldOverLimitError(options = {}) {
+        if (!this.disabled) {
+            return;
+        }
+
+        // If no currentCountQuery is provided, throw error when disabled
+        if (!this.currentCountQueryFn || typeof this.currentCountQueryFn !== 'function') {
             throw this.generateError();
         }
+
+        // If currentCountQuery is provided, check if feature is in use
+        const featureInUse = await this.currentCountQuery(options);
+
+        // Only throw error if feature is NOT in use (allowing grandfathering)
+        if (!featureInUse) {
+            throw this.generateError();
+        }
+    }
+
+    /**
+     * Flag limits are usually on/off so using a feature is always over the limit,
+     * unless the limit has a currentCountQuery function provided to check if the
+     * feature is in use. This is a use case for when we introduce a new limit and
+     * customers have already been using this feature. We don't want to take it
+     * away from them.
+     */
+    async errorIfWouldGoOverLimit(options = {}) {
+        await this._isOrWouldOverLimitError(options);
     }
 
     /**
      * Flag limits are on/off. They don't necessarily mean the limit wasn't possible to reach
-     * NOTE: this method should not be relied on as it's impossible to check the limit was surpassed!
+     * Exception: the limit has a currentCountQuery function provided to check if the
+     * feature is in use. This is a use case for when we introduce a new limit and
+     * customers have already been using this feature. We don't want to take it
+     * away from them.
      */
-    async errorIfIsOverLimit() {
-        return;
+    async errorIfIsOverLimit(options = {}) {
+        await this._isOrWouldOverLimitError(options);
     }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tryghost/limit-service",
-  "version": "1.2.19",
+  "version": "1.3.0",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/TryGhost/SDK.git",
@@ -34,5 +34,5 @@
     "lodash": "^4.17.21",
     "luxon": "^1.26.0"
   },
-  "gitHead": "b10773947244536b8829fec0540819990c901987"
+  "gitHead": "a4329bd0bf75c10a862bb1c8761b45a6b08c5c2a"
 }