@spintax/core 0.5.0 → 0.5.2

package/dist/index.js

@@ -631,6 +631,7 @@ function checkPlurals(text, idx, locale, knownVariables, out) {
 var UNRESOLVED_AT_PLURAL_TIME = /\[|\{(?!\?)/u;
 var ANY_BRACKET = /[[\]{}]/u;
 var FORM_EXPANSION_PASSES = 51;
+var FORM_EXPANSION_MAX_GROWTH = 64 * 1024;
 function expandFormsForCounting(formsRaw, defs, macros, hostNames) {
   const unknown = (directMacroSpintax = false) => ({
     forms: 0,
@@ -638,17 +639,23 @@ function expandFormsForCounting(formsRaw, defs, macros, hostNames) {
     directMacroSpintax
   });
   const seenMacro = /* @__PURE__ */ new Set();
+  const refsOf = (text2) => [...text2.matchAll(/%(\w+)%/gu)].map((m) => (m[1] ?? "").toLowerCase());
   const walkMacros = (source) => {
-    for (const m of source.matchAll(/%(\w+)%/gu)) {
-      const name = (m[1] ?? "").toLowerCase();
+    const stack = [{ refs: refsOf(source), i: 0 }];
+    while (stack.length > 0) {
+      const frame = stack[stack.length - 1];
+      if (frame.i >= frame.refs.length) {
+        stack.pop();
+        continue;
+      }
+      const name = frame.refs[frame.i++];
       if (defs[name] !== void 0 || hostNames.has(name)) continue;
       const macro = macros[name];
       if (macro === void 0 || seenMacro.has(name)) continue;
       seenMacro.add(name);
       if (/\{\?/u.test(macro)) return "opaque";
       if (ANY_BRACKET.test(macro)) return "brackets";
-      const deeper = walkMacros(macro);
-      if (deeper !== "clean") return deeper;
+      stack.push({ refs: refsOf(macro), i: 0 });
     }
     return "clean";
   };
@@ -656,28 +663,33 @@ function expandFormsForCounting(formsRaw, defs, macros, hostNames) {
   if (verbatim === "brackets") return unknown(true);
   if (verbatim === "opaque") return unknown();
   let text = formsRaw;
+  const budget = formsRaw.length + FORM_EXPANSION_MAX_GROWTH;
   for (let pass = 0; pass < FORM_EXPANSION_PASSES; pass++) {
     let sawReference = false;
     let bailed = false;
-    text = text.replace(/%(\w+)%/gu, (whole, raw) => {
+    const parts = [];
+    let total = 0;
+    let cursor = 0;
+    const re = /%(\w+)%/gu;
+    let m;
+    while ((m = re.exec(text)) !== null) {
       sawReference = true;
-      const name = raw.toLowerCase();
-      if (hostNames.has(name)) {
-        bailed = true;
-        return whole;
-      }
-      const value = defs[name] ?? macros[name];
-      if (value === void 0) {
-        bailed = true;
-        return whole;
-      }
-      if (ANY_BRACKET.test(value)) {
+      const name = (m[1] ?? "").toLowerCase();
+      const value = hostNames.has(name) ? void 0 : defs[name] ?? macros[name];
+      if (value === void 0 || ANY_BRACKET.test(value)) {
         bailed = true;
-        return whole;
+        break;
       }
-      return value;
-    });
+      parts.push(text.slice(cursor, m.index), value);
+      total += m.index - cursor + value.length;
+      cursor = m.index + m[0].length;
+      if (total > budget) return unknown();
+    }
     if (bailed) return unknown();
+    parts.push(text.slice(cursor));
+    total += text.length - cursor;
+    if (total > budget) return unknown();
+    text = parts.join("");
     if (!sawReference) {
       return { forms: text.split("|").length, unresolved: false, directMacroSpintax: false };
     }
@@ -897,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);
@@ -904,7 +917,8 @@ function renderAst(ast, ctx) {
     rng: ctx.rng,
     locale: ctx.locale,
     depth: 0,
-    onPluralError: ctx.onPluralError
+    onPluralError: ctx.onPluralError,
+    budget: { left: MAX_EXPANSION_CHARS }
   };
   const vars = Object.keys(ast.defDefs).length > 0 ? { ...base, ...rollDefinitions(ast.defDefs, base, ctx.runtimeContext, walkOpts) } : base;
   const text = renderNodes(ast.nodes, { ...walkOpts, vars });
@@ -1017,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) {
@@ -1026,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;
     });