@spintax/core 0.5.0 → 0.5.1

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 };
     }