feature: braces: add maxSymbols (#36 (comment))

coderaiser · coderaiser · commit 2092bd1fb108 · 2024-05-15T17:55:51.000+03:00
diff --git a/README.md b/README.md
@@ -186,6 +186,18 @@ console.log(braces.expand('a{b}c'));
 console.log(braces('a/{b,c}/d', { maxLength: 3 }));  //=> throws an error
 ```
 
+### options.maxSymbols
+
+**Type**: `Number`
+
+**Default**: `1024`
+
+**Description**: Limit the count of unique symbols the input string. 
+
+```js
+console.log(braces('a/{b,c}/d', { maxSymbols: 2 }));  //=> throws an error
+```
+
 ### options.expand
 
 **Type**: `Boolean`
diff --git a/lib/constants.js b/lib/constants.js
@@ -2,6 +2,7 @@
 
 module.exports = {
   MAX_LENGTH: 1024 * 64,
+  MAX_SYMBOLS: 1024,
 
   // Digits
   CHAR_0: '0', /* 0 */
diff --git a/lib/parse.js b/lib/parse.js
@@ -1,13 +1,15 @@
 'use strict';
 
 const stringify = require('./stringify');
+const {isCorrectBraces, validateInput} = require('./validate-input');
 
 /**
  * Constants
  */
 
 const {
   MAX_LENGTH,
+  MAX_SYMBOLS,
   CHAR_BACKSLASH, /* \ */
   CHAR_BACKTICK, /* ` */
   CHAR_COMMA, /* , */
@@ -34,6 +36,11 @@ const parse = (input, options = {}) => {
   }
 
   let opts = options || {};
+
+  validateInput(input, {
+    maxSymbols: opts.maxSymbols || MAX_SYMBOLS,
+  });
+
   let max = typeof opts.maxLength === 'number' ? Math.min(MAX_LENGTH, opts.maxLength) : MAX_LENGTH;
   if (input.length > max) {
     throw new SyntaxError(`Input length (${input.length}), exceeds max characters (${max})`);
@@ -316,15 +323,16 @@ module.exports = parse;
 function markImbalancedBraces({nodes}) {
   // Mark imbalanced braces and brackets as invalid
   for (const node of nodes) {
-    if (node.nodes || node.invalid)
-      continue;
+    if (!node.nodes && !node.invalid) {
+      if (node.type === 'open') node.isOpen = true;
+      if (node.type === 'close') node.isClose = true;
+      if (!node.nodes) node.type = 'text';
 
-    if (node.type === 'open') node.isOpen = true;
-    if (node.type === 'close') node.isClose = true;
-    if (!node.nodes) node.type = 'text';
+      node.invalid = true;
+    }
 
-    node.invalid = true;
     delete node.parent;
+    delete node.prev;
   }
 }
 
diff --git a/lib/validate-input.js b/lib/validate-input.js
@@ -0,0 +1,12 @@
+module.exports.validateInput = (line, {maxSymbols}) => {
+  const symbols = {};
+
+    for (const current of line) {
+        symbols[current] = (symbols[current] || 0) + 1;
+    }
+
+    for (const [value, count] of Object.entries(symbols)) {
+      if (count > maxSymbols)
+        throw SyntaxError(`To many symbols '${value}'. Maximum: ${maxSymbols} allowed. Received: ${count}`);
+    }
+};
diff --git a/test/braces.parse.js b/test/braces.parse.js
@@ -10,6 +10,16 @@ describe('braces.parse()', () => {
       let MAX_LENGTH = 1024 * 64;
       assert.throws(() => parse('.'.repeat(MAX_LENGTH + 2)));
     });
+    it('should throw an error when symbols exceeds max symbols count default', () => {
+      let SYMBOLS= 1024;
+      assert.throws(() => parse('.'.repeat(MAX_SYMBOLS * 2)));
+    });
+    it('should throw an error when symbols exceeds max symbols count ', () => {
+      let SYMBOLS= 2;
+      assert.throws(() => parse('...', {
+        maxSymbols: 2,
+      }));
+    });
   });
 
   describe('valid', () => {
