@spintax/core 0.5.1 → 0.5.3

package/dist/index.js

@@ -909,6 +909,7 @@ var NotImplementedError = class extends SpintaxError {
 
 // src/internal/render.ts
 var MAX_VARIABLE_DEPTH = 50;
+var MAX_EXPANSION_CHARS = 1024 * 1024;
 var INCLUDE_LINE_RE = /^[ \t]*#include[ \t\n\r\f\x0B]+"([^"]+)"[ \t\n\r\f\x0B]*$/gmu;
 function renderAst(ast, ctx) {
   const base = buildVars(ast.setDefs, ctx.runtimeContext);
@@ -916,7 +917,8 @@ function renderAst(ast, ctx) {
     rng: ctx.rng,
     locale: ctx.locale,
     depth: 0,
-    onPluralError: ctx.onPluralError
+    onPluralError: ctx.onPluralError,
+    budget: ctx.budget
   };
   const vars = Object.keys(ast.defDefs).length > 0 ? { ...base, ...rollDefinitions(ast.defDefs, base, ctx.runtimeContext, walkOpts) } : base;
   const text = renderNodes(ast.nodes, { ...walkOpts, vars });
@@ -1029,6 +1031,8 @@ function resolveVariable(name, opts) {
   const value = opts.vars[name.toLowerCase()];
   if (value === void 0) return `%${name}%`;
   if (opts.depth >= MAX_VARIABLE_DEPTH || !/[{[%]/u.test(value)) return value;
+  if (opts.budget.left <= 0) return `%${name}%`;
+  opts.budget.left -= value.length;
   return renderNodes(parseSequence(value), { ...opts, depth: opts.depth + 1 });
 }
 function expandVarsOnly(text, opts) {
@@ -1038,6 +1042,8 @@ function expandVarsOnly(text, opts) {
     out = out.replace(/%(\w+)%/gu, (m, name) => {
       const value = opts.vars[name.toLowerCase()];
       if (value === void 0) return m;
+      if (opts.budget.left <= 0) return m;
+      opts.budget.left -= value.length;
       changed = true;
       return value;
     });
@@ -1352,7 +1358,10 @@ function renderWith(input, rng, opts = {}) {
     resolver: opts.includeResolver,
     maxDepth: opts.maxDepth ?? DEFAULT_MAX_DEPTH,
     includeStack: [],
-    onPluralError: opts.onPluralError
+    onPluralError: opts.onPluralError,
+    // One allowance for the whole call. Created here rather than in renderAst because
+    // includes re-enter renderAst, and a budget made there is a budget per subtree.
+    budget: { left: MAX_EXPANSION_CHARS }
   };
   const ast = resolveAst(input);
   let out = renderAst(ast, ctx);