@poe-platform/safe-bash 0.1.799 → 0.1.800

package/dist/safe-bash/command-chunks/{chunk-OPUJSTHS.js → chunk-67IQF5GC.js}

@@ -1,29 +1,34 @@
 import {
   recurse
-} from "./chunk-ZT4KSS2V.js";
+} from "./chunk-PTT3I223.js";
 import {
   splitRegex
-} from "./chunk-EIVW3X3N.js";
+} from "./chunk-3DT2YOGW.js";
 import {
+  matchRegex,
   scanRegex,
   substituteRegex
-} from "./chunk-KMCZ6P4O.js";
+} from "./chunk-BKCNF5U3.js";
+import {
+  formatValue
+} from "./chunk-53P53XFM.js";
+import {
+  indices
+} from "./chunk-2QHG43WJ.js";
 import {
   capture
-} from "./chunk-JCNNOMBV.js";
+} from "./chunk-GI4CJED2.js";
 import {
   fromDateIso8601,
+  gmtime,
+  mktime,
+  strftime,
+  strptime,
   toDateIso8601
-} from "./chunk-NETECMOF.js";
+} from "./chunk-GMIHJYDG.js";
 import {
   delpaths
-} from "./chunk-ZXKW7SJ2.js";
-import {
-  formatValue
-} from "./chunk-4ZALPDBV.js";
-import {
-  indices
-} from "./chunk-LLCXZKZ6.js";
+} from "./chunk-TYCGMRLM.js";
 import {
   binary,
   compare,
@@ -37,16 +42,16 @@ import {
   sortedKeys,
   stableSort,
   type
-} from "./chunk-YZ5SVZFD.js";
+} from "./chunk-UMDIOL3O.js";
 import {
   splitString
-} from "./chunk-UI32HWGS.js";
+} from "./chunk-HMOZ6XVJ.js";
 import {
   JqParseError,
   measureValue,
   parseJson,
   stringify
-} from "./chunk-5ZSM4UFN.js";
+} from "./chunk-YGC6OS5O.js";
 import {
   JqError,
   JqHalt,
@@ -61,7 +66,7 @@ import {
   put,
   remove,
   truth
-} from "./chunk-VVI4Z73I.js";
+} from "./chunk-QP4R6GLA.js";
 import {
   utf8ByteLength
 } from "./chunk-PDKXAR4Y.js";
@@ -121,6 +126,8 @@ var Interpreter = class _Interpreter {
   budget;
   variables;
   frame;
+  definitions = /* @__PURE__ */ new Map();
+  callDepth = 0;
   _filters;
   _labels;
   get filters() {
@@ -137,6 +144,8 @@ var Interpreter = class _Interpreter {
   resetForRun(budget, variables) {
     this.budget = budget;
     this.variables = variables;
+    this.definitions.clear();
+    this.callDepth = 0;
     this._filters?.clear();
     this._labels?.clear();
     this.labelSequence.next = 0;
@@ -150,14 +159,27 @@ var Interpreter = class _Interpreter {
   }
   tryRunSync(ast, input) {
     const savedSteps = this.budget.currentSteps;
-    const res = this.tryRunSyncInternal(ast, input, true);
-    if (res === void 0) this.budget.restoreSteps(savedSteps);
+    let res;
+    try {
+      res = this.tryRunSyncInternal(ast, input, true);
+    } catch (error) {
+      if (!(error instanceof JqLimitError) || this.budget.signal.aborted) throw error;
+    }
+    if (res === void 0) {
+      this.budget.restoreSteps(savedSteps);
+      this.releaseScratch();
+    }
     return res;
   }
   tryRunSyncNoScratch(ast, input) {
     if (!this.budget.unlimitedValueCheck) return void 0;
     const savedSteps = this.budget.currentSteps;
-    const res = this.tryRunSyncInternal(ast, input, false);
+    let res;
+    try {
+      res = this.tryRunSyncInternal(ast, input, false);
+    } catch (error) {
+      if (!(error instanceof JqLimitError) || this.budget.signal.aborted) throw error;
+    }
     if (res === void 0) {
       this.budget.restoreSteps(savedSteps);
       return void 0;
@@ -183,7 +205,7 @@ var Interpreter = class _Interpreter {
         if (!truth(cond)) return EMPTY_RESULTS;
         return this.tryRunSyncInternal(ast.right, input, allowScratch);
       }
-      const leftResults = this.tryRunSyncInternal(ast.left, input, allowScratch);
+      const leftResults = this.tryRunSyncInternal(ast.left, input, false);
       if (!leftResults) return void 0;
       if (leftResults.length === 0) return EMPTY_RESULTS;
       if (leftResults.length === 1) return this.tryRunSyncInternal(ast.right, leftResults[0], allowScratch);
@@ -456,12 +478,15 @@ var Interpreter = class _Interpreter {
               } else {
                 const nk = exactFiniteNumber(k);
                 if (nk === void 0) return NOT_SINGLE;
-                keyBytes += this.budget.value(k);
+                keyBytes += this.budget.value([k]);
                 if (keyBytes > this.budget.limits.maxValueBytes) throw new JqLimitError("maxValueBytes");
                 keyed.push({ key: nk, value: val, idx: i });
               }
             }
-            keyed.sort((a, b) => a.key < b.key ? -1 : a.key > b.key ? 1 : a.idx - b.idx);
+            keyed.sort((a, b) => {
+              this.budget.step();
+              return a.key < b.key ? -1 : a.key > b.key ? 1 : a.idx - b.idx;
+            });
             if (name === "sort_by") {
               const res = keyed.map((x) => x.value);
               this.budget.value(res);
@@ -576,8 +601,12 @@ var Interpreter = class _Interpreter {
         yield* filter.scope.run(filter.ast, input);
         return;
       }
+      case "define":
+        yield* this.definitionScope(ast).run(ast.body, input);
+        return;
       case "invoke":
-        yield* this.invocation(ast).run(ast.body, input);
+        await Promise.resolve();
+        yield* this.invocation(ast).run(ast.definition.body, input);
         return;
       case "label": {
         const target = this.labelSequence.next++;
@@ -881,14 +910,24 @@ var Interpreter = class _Interpreter {
       }
     }
   }
+  definitionScope(ast) {
+    const scope = Object.create(_Interpreter.prototype);
+    const definitions = new Map(this.definitions);
+    Object.assign(scope, this, { definitions });
+    for (const definition of ast.definitions) definitions.set(definition, scope);
+    return scope;
+  }
   invocation(ast) {
     const scope = Object.create(_Interpreter.prototype);
-    const filters = new Map(this.filters);
-    for (let index = 0; index < ast.parameters.length; index++) {
+    const lexical = this.definitions.get(ast.definition) ?? this;
+    const filters = new Map(lexical.filters);
+    const callDepth = this.callDepth + 1;
+    if (callDepth > this.budget.limits.maxDepth) throw new JqLimitError("maxDepth");
+    for (let index = 0; index < ast.definition.parameters.length; index++) {
       this.budget.step();
-      filters.set(ast.parameters[index], { ast: ast.args[index], scope: this });
+      filters.set(ast.definition.parameters[index], { ast: ast.args[index], scope: this });
     }
-    Object.assign(scope, this, { _filters: filters });
+    Object.assign(scope, lexical, { _filters: filters, callDepth });
     return scope;
   }
   async read(input, path) {
@@ -912,8 +951,12 @@ var Interpreter = class _Interpreter {
       yield* filter.scope.paths(filter.ast, input);
       return;
     }
+    if (ast.kind === "define") {
+      yield* this.definitionScope(ast).paths(ast.body, input);
+      return;
+    }
     if (ast.kind === "invoke") {
-      yield* this.invocation(ast).paths(ast.body, input);
+      yield* this.invocation(ast).paths(ast.definition.body, input);
       return;
     }
     if (ast.kind === "bind") {
@@ -1127,11 +1170,100 @@ var Interpreter = class _Interpreter {
       yield* visit(input);
       return;
     }
-    if (name === "fromdateiso8601") {
+    if (name === "test" || name === "match") {
+      for await (const pattern of this.run(args[0], input)) {
+        const source = !args[1] && Array.isArray(pattern) ? pattern[0] ?? null : pattern;
+        for await (const flags of args[1] ? this.run(args[1], input) : [Array.isArray(pattern) ? pattern[1] ?? null : null]) {
+          yield* matchRegex(input, source, flags, budget, name === "test");
+        }
+      }
+      return;
+    }
+    if (name === "isempty") {
+      for await (const value of this.run(args[0], input)) {
+        void value;
+        yield false;
+        return;
+      }
+      yield true;
+      return;
+    }
+    if (name === "nth") {
+      for await (const value of this.run(args[0], input)) {
+        if (!args[1]) {
+          yield indexValue(input, value);
+          continue;
+        }
+        if (!isNumber(value) || numberValue(value) < 0) throw new JqError("nth doesn't support negative indices");
+        const target = Math.ceil(numberValue(value));
+        let index = 0;
+        for await (const item of this.run(args[1], input)) {
+          if (index++ === target) {
+            yield item;
+            break;
+          }
+        }
+      }
+      return;
+    }
+    if (name === "in") {
+      for await (const container of this.run(args[0], input)) {
+        yield* this.run({ kind: "call", name: "has", args: [{ kind: "literal", value: input }] }, container);
+      }
+      return;
+    }
+    if (name === "IN") {
+      for await (const candidate of args.length === 2 ? this.run(args[0], input) : [input]) {
+        for await (const value of this.run(args.at(-1), input)) {
+          if (equal(candidate, value, budget)) {
+            yield true;
+            return;
+          }
+        }
+      }
+      yield false;
+      return;
+    }
+    if (name === "INDEX") {
+      const result = object();
+      const source = args.length === 2 ? args[0] : { kind: "iterate", base: { kind: "identity" } };
+      for await (const value of this.run(source, input)) {
+        for await (const key of this.run(args.at(-1), value)) {
+          const text = typeof key === "string" ? key : await stringify(key, budget);
+          put(result, text, value);
+          budget.collection(objectKeys(result).length);
+          budget.value(result);
+        }
+      }
+      yield result;
+      return;
+    }
+    if (name === "pick") {
+      let result = null;
+      for await (const path of this.paths(args[0], input)) {
+        result = await this.set(result, path, await this.read(input, path));
+        budget.value(result);
+      }
+      yield result;
+      return;
+    }
+    if (name === "gmtime") {
+      yield gmtime(input);
+      return;
+    }
+    if (name === "mktime") {
+      yield mktime(input);
+      return;
+    }
+    if (name === "strftime" || name === "strptime") {
+      for await (const format of this.run(args[0], input)) yield name === "strftime" ? strftime(input, format, budget) : strptime(input, format, budget);
+      return;
+    }
+    if (name === "fromdateiso8601" || name === "fromdate") {
       yield fromDateIso8601(input);
       return;
     }
-    if (name === "todateiso8601") {
+    if (name === "todateiso8601" || name === "todate") {
       yield toDateIso8601(input);
       return;
     }
@@ -1170,7 +1302,7 @@ var Interpreter = class _Interpreter {
       return;
     }
     if (name === "scan") {
-      for await (const source of this.run(args[0], input)) yield* scanRegex(input, source, budget);
+      for await (const source of this.run(args[0], input)) for await (const flags of args[1] ? this.run(args[1], input) : [null]) yield* scanRegex(input, source, budget, flags);
       return;
     }
     if (name === "recurse") {
@@ -1334,11 +1466,11 @@ var Interpreter = class _Interpreter {
       }
       return;
     }
-    if (name === "floor" || name === "ceil" || name === "round" || name === "abs") {
+    if (name === "floor" || name === "ceil" || name === "round" || name === "abs" || name === "sqrt") {
       if (!isNumber(input)) throw new JqError(`${describe(input, budget)} cannot be ${name}ed`);
       budget.step();
       const n = numberValue(input);
-      yield name === "floor" ? Math.floor(n) : name === "ceil" ? Math.ceil(n) : name === "round" ? Math.sign(n) * Math.round(Math.abs(n)) : Math.abs(n);
+      yield name === "floor" ? Math.floor(n) : name === "ceil" ? Math.ceil(n) : name === "round" ? Math.sign(n) * Math.round(Math.abs(n)) : name === "sqrt" ? Math.sqrt(n) : Math.abs(n);
       return;
     }
     if (name === "implode") {
@@ -1478,39 +1610,41 @@ var Interpreter = class _Interpreter {
       return;
     }
     if (name === "combinations") {
-      for await (const length of this.run({ kind: "call", name: "length", args: [] }, input)) {
-        const width = Math.ceil(numberValue(length));
+      let width = 0;
+      for await (const length of this.run(args[0] ?? { kind: "call", name: "length", args: [] }, input)) {
+        if (!isNumber(length)) throw new JqError("combinations requires a number");
+        width += Math.max(0, Math.ceil(numberValue(length)));
         budget.collection(width);
-        if (width === 0) {
-          yield [];
-          return;
-        }
-        const values = [];
-        const iterators = [];
-        let depth = 0;
-        try {
-          while (depth >= 0) {
-            {
-              const _p = budget.tickSync();
-              if (_p) await _p;
-            }
-            if (!iterators[depth]) iterators[depth] = entries(indexValue(input, depth), budget);
-            const next = await iterators[depth].next();
-            if (next.done) {
-              iterators.pop();
-              values.pop();
-              depth--;
-            } else {
-              values[depth] = next.value[1];
-              if (depth + 1 === width) {
-                budget.value(values);
-                yield [...values];
-              } else depth++;
-            }
+      }
+      if (width === 0) {
+        yield [];
+        return;
+      }
+      const values = [];
+      const iterators = [];
+      let depth = 0;
+      try {
+        while (depth >= 0) {
+          {
+            const _p = budget.tickSync();
+            if (_p) await _p;
+          }
+          if (!iterators[depth]) iterators[depth] = entries(args.length ? input : indexValue(input, depth), budget);
+          const next = await iterators[depth].next();
+          if (next.done) {
+            iterators.pop();
+            values.pop();
+            depth--;
+          } else {
+            values[depth] = next.value[1];
+            if (depth + 1 === width) {
+              budget.value(values);
+              yield [...values];
+            } else depth++;
           }
-        } finally {
-          for (const iterator of iterators) await iterator.return(void 0);
         }
+      } finally {
+        for (const iterator of iterators) await iterator.return(void 0);
       }
       return;
     }
@@ -1660,17 +1794,17 @@ var Interpreter = class _Interpreter {
       }
       return;
     }
-    if (name === "gsub") {
+    if (name === "gsub" || name === "sub") {
       for await (const pattern of this.run(args[0], input)) {
         for await (const flags of args[2] ? this.run(args[2], input) : [""]) {
-          yield* substituteRegex(input, pattern, flags, (captures) => this.run(args[1], captures), budget);
+          yield* substituteRegex(input, pattern, flags, (captures) => this.run(args[1], captures), budget, name === "gsub");
         }
       }
       return;
     }
     if (name === "splits") {
       for await (const separator of this.run(args[0], input)) {
-        yield* splitRegex(input, separator, budget);
+        for await (const flags of args[1] ? this.run(args[1], input) : [null]) yield* splitRegex(input, separator, budget, flags);
       }
       return;
     }