json-p3 1.1.1 → 1.2.0

package/dist/json-p3.iife.js

@@ -1,5 +1,5 @@
 /*
- * json-p3 version 1.1.1
+ * json-p3 version 1.2.0
  * https://github.com/jg-rp/json-p3
  * 
  * MIT License
@@ -1183,6 +1183,8 @@ var json_p3 = (function (exports) {
     TokenKind["GE"] = "TOKEN_GE";
     TokenKind["GT"] = "TOKEN_GT";
     TokenKind["INDEX"] = "TOKEN_INDEX";
+    TokenKind["KEY"] = "TOKEN_KEY";
+    TokenKind["KEYS"] = "TOKEN_KEYS";
     TokenKind["LBRACKET"] = "TOKEN_LBRACKET";
     TokenKind["LE"] = "TOKEN_LE";
     TokenKind["LG"] = "TOKEN_LG";
@@ -1252,6 +1254,9 @@ var json_p3 = (function (exports) {
     }
   }
 
+  /** A lexer that accepts additional, non-standard tokens. */
+
+
   // These regular expressions are to be used with Lexer.acceptMatchRun(),
   // which expects the sticky flag to be set.
   const exponentPattern = /e[+-]?\d+/y;
@@ -1291,7 +1296,8 @@ var json_p3 = (function (exports) {
     /**
      * @param path - A JSONPath query.
      */
-    constructor(path) {
+    constructor(environment, path) {
+      this.environment = environment;
       this.path = path;
     }
     get pos() {
@@ -1391,8 +1397,8 @@ var json_p3 = (function (exports) {
    * @returns A two-tuple containing a lexer for _path_ and an array to populate
    * with tokens.
    */
-  function lex(path) {
-    const lexer = new Lexer(path);
+  function lex(environment, path) {
+    const lexer = new Lexer(environment, path);
     return [lexer, lexer.tokens];
   }
 
@@ -1401,8 +1407,8 @@ var json_p3 = (function (exports) {
    * @param path - A JSONPath query.
    * @returns Tokens to be parsed by the parser.
    */
-  function tokenize(path) {
-    const [lexer, tokens] = lex(path);
+  function tokenize(environment, path) {
+    const [lexer, tokens] = lex(environment, path);
     lexer.run();
     if (tokens.length && tokens[tokens.length - 1].kind === TokenKind.ERROR) {
       throw new JSONPathSyntaxError(tokens[tokens.length - 1].value, tokens[tokens.length - 1]);
@@ -1455,6 +1461,16 @@ var json_p3 = (function (exports) {
    * @returns -
    */
   function lexDescendantSelection(l) {
+    if (l.acceptMatchRun(namePattern)) {
+      // Shorthand name
+      l.emit(TokenKind.NAME);
+      return lexSegment;
+    }
+    if (!l.environment.strict && l.acceptMatchRun(l.environment.keysPattern)) {
+      // Non-standard keys selector
+      l.emit(TokenKind.KEYS);
+      return lexSegment;
+    }
     const ch = l.next();
     switch (ch) {
       case "":
@@ -1468,13 +1484,8 @@ var json_p3 = (function (exports) {
         return lexInsideBracketedSelection;
       default:
         l.backup();
-        if (l.acceptMatchRun(namePattern)) {
-          l.emit(TokenKind.NAME);
-          return lexSegment;
-        } else {
-          l.error(`unexpected descendent selection token '${ch}'`);
-          return null;
-        }
+        l.error(`unexpected descendent selection token '${ch}'`);
+        return null;
     }
   }
   function lexDotSelector(l) {
@@ -1483,23 +1494,34 @@ var json_p3 = (function (exports) {
       l.error("unexpected whitespace after dot");
       return null;
     }
-    const ch = l.next();
-    if (ch === "*") {
-      l.emit(TokenKind.WILD);
+    if (!l.environment.strict && l.acceptMatchRun(l.environment.keysPattern)) {
+      l.emit(TokenKind.KEYS);
       return lexSegment;
     }
-    l.backup();
     if (l.acceptMatchRun(namePattern)) {
       l.emit(TokenKind.NAME);
       return lexSegment;
-    } else {
-      l.error(`unexpected shorthand selector '${ch}'`);
-      return null;
     }
+    const ch = l.next();
+    if (ch === "*") {
+      l.emit(TokenKind.WILD);
+      return lexSegment;
+    }
+    l.backup();
+    l.error(`unexpected shorthand selector '${ch}'`);
+    return null;
   }
   function lexInsideBracketedSelection(l) {
     for (;;) {
       l.ignoreWhitespace();
+      if (l.acceptMatchRun(indexPattern)) {
+        l.emit(TokenKind.INDEX);
+        continue;
+      }
+      if (!l.environment.strict && l.acceptMatchRun(l.environment.keysPattern)) {
+        l.emit(TokenKind.KEYS);
+        continue;
+      }
       const ch = l.next();
       switch (ch) {
         case "]":
@@ -1528,10 +1550,6 @@ var json_p3 = (function (exports) {
           return lexDoubleQuoteStringInsideBracketSelection;
         default:
           l.backup();
-          if (l.acceptMatchRun(indexPattern)) {
-            l.emit(TokenKind.INDEX);
-            continue;
-          }
           l.error(`unexpected token '${ch}' in bracketed selection`);
           return null;
       }
@@ -1545,6 +1563,8 @@ var json_p3 = (function (exports) {
       const ch = l.next();
       switch (ch) {
         case "":
+          l.error("unclosed bracketed selection");
+          return null;
         case "]":
           l.filterLevel -= 1;
           if (l.parenStack.length === 1) {
@@ -1586,6 +1606,9 @@ var json_p3 = (function (exports) {
         case "@":
           l.emit(TokenKind.CURRENT);
           return lexSegment;
+        case "#":
+          l.emit(TokenKind.KEY);
+          return lexSegment;
         case ".":
           l.backup();
           return lexSegment;
@@ -1963,12 +1986,22 @@ var json_p3 = (function (exports) {
     }
     resolve(nodes) {
       const rv = [];
-      for (const node of nodes) {
-        rv.push(node);
-        for (const _node of this.visit(node)) {
-          rv.push(_node);
+      if (this.environment.nondeterministic) {
+        for (const root of nodes) {
+          for (const node of this.nondeterministicVisitor(root)) {
+            rv.push(node);
+          }
+        }
+      } else {
+        for (const node of nodes) {
+          rv.push(node);
+          for (const _node of this.visitor(node)) {
+            rv.push(_node);
+          }
         }
       }
+
+      // console.log(JSON.stringify(rv.map((n: any) => n.value)));
       return this.selector.resolve(rv);
     }
     *lazyResolve(nodes) {
@@ -2021,7 +2054,7 @@ var json_p3 = (function (exports) {
     toString() {
       return `..${this.selector.toString()}`;
     }
-    visit(node) {
+    visitor(node) {
       let depth = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 1;
       if (depth >= this.environment.maxRecursionDepth) {
         throw new JSONPathRecursionLimitError("recursion limit reached", this.token);
@@ -2032,7 +2065,7 @@ var json_p3 = (function (exports) {
         for (let i = 0; i < node.value.length; i++) {
           const _node = new JSONPathNode(node.value[i], node.location.concat(i), node.root);
           rv.push(_node);
-          for (const __node of this.visit(_node, depth + 1)) {
+          for (const __node of this.visitor(_node, depth + 1)) {
             rv.push(__node);
           }
         }
@@ -2040,13 +2073,52 @@ var json_p3 = (function (exports) {
         for (const [key, value] of this.environment.entries(node.value)) {
           const _node = new JSONPathNode(value, node.location.concat(key), node.root);
           rv.push(_node);
-          for (const __node of this.visit(_node, depth + 1)) {
+          for (const __node of this.visitor(_node, depth + 1)) {
             rv.push(__node);
           }
         }
       }
       return rv;
     }
+    nondeterministicVisitor(root) {
+      let depth = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 1;
+      const rv = [root];
+      let queue = this.nondeterministicChildren(root).map(node => [node, depth]);
+      while (queue.length) {
+        const [node, _depth] = queue.shift();
+        rv.push(node);
+        if (_depth >= this.environment.maxRecursionDepth) {
+          throw new JSONPathRecursionLimitError("recursion limit reached", this.token);
+        }
+
+        // Visit child nodes now or queue them for later?
+        const visitChildren = Math.random() < 0.5;
+        for (const child of this.nondeterministicChildren(node)) {
+          if (visitChildren) {
+            rv.push(child);
+            const grandchildren = this.nondeterministicChildren(child).map(n => [n, _depth + 2]);
+            queue = interleave(queue, grandchildren);
+          } else {
+            queue.push([child, _depth + 1]);
+          }
+        }
+      }
+      return rv;
+    }
+    nondeterministicChildren(node) {
+      const _rv = [];
+      if (node.value instanceof String) return _rv;
+      if (isArray(node.value)) {
+        for (let i = 0; i < node.value.length; i++) {
+          _rv.push(new JSONPathNode(node.value[i], node.location.concat(i), node.root));
+        }
+      } else if (isObject(node.value)) {
+        for (const [key, value] of this.environment.entries(node.value)) {
+          _rv.push(new JSONPathNode(value, node.location.concat(key), node.root));
+        }
+      }
+      return _rv;
+    }
   }
   class FilterSelector extends JSONPathSelector {
     constructor(environment, token, expression) {
@@ -2067,7 +2139,8 @@ var json_p3 = (function (exports) {
             const filterContext = {
               environment: this.environment,
               currentValue: value,
-              rootValue: node.root
+              rootValue: node.root,
+              currentKey: i
             };
             if (this.expression.evaluate(filterContext)) {
               rv.push(new JSONPathNode(value, node.location.concat(i), node.root));
@@ -2078,7 +2151,8 @@ var json_p3 = (function (exports) {
             const filterContext = {
               environment: this.environment,
               currentValue: value,
-              rootValue: node.root
+              rootValue: node.root,
+              currentKey: key
             };
             if (this.expression.evaluate(filterContext)) {
               rv.push(new JSONPathNode(value, node.location.concat(key), node.root));
@@ -2100,7 +2174,8 @@ var json_p3 = (function (exports) {
               environment: this.environment,
               currentValue: value,
               rootValue: node.root,
-              lazy: true
+              lazy: true,
+              currentKey: i
             };
             if (this.expression.evaluate(filterContext)) {
               yield new JSONPathNode(value, node.location.concat(i), node.root);
@@ -2112,7 +2187,8 @@ var json_p3 = (function (exports) {
               environment: this.environment,
               currentValue: value,
               rootValue: node.root,
-              lazy: true
+              lazy: true,
+              currentKey: key
             };
             if (this.expression.evaluate(filterContext)) {
               yield new JSONPathNode(value, node.location.concat(key), node.root);
@@ -2155,6 +2231,41 @@ var json_p3 = (function (exports) {
     }
   }
 
+  /**
+   * Randomly interleave elements from two arrays while maintaining relative
+   * order of each input array.
+   *
+   * If _arrayA_ is empty, _arrayB_ is returned, and vice versa.
+   */
+  function interleave(arrayA, arrayB) {
+    if (arrayA.length === 0) {
+      return arrayB;
+    }
+    if (arrayB.length === 0) {
+      return arrayA;
+    }
+
+    // An array of iterators
+    const iterators = [];
+    const itA = arrayA[Symbol.iterator]();
+    const itB = arrayB[Symbol.iterator]();
+    for (let i = 0; i < arrayA.length; i++) {
+      iterators.push(itA);
+    }
+    for (let i = 0; i < arrayB.length; i++) {
+      iterators.push(itB);
+    }
+    shuffle(iterators);
+    return iterators.map(it => it.next().value);
+  }
+  function shuffle(entries) {
+    for (let i = entries.length - 1; i > 0; i--) {
+      const j = Math.floor(Math.random() * (i + 1));
+      [entries[i], entries[j]] = [entries[j], entries[i]];
+    }
+    return entries;
+  }
+
   var selectors = /*#__PURE__*/Object.freeze({
     __proto__: null,
     BracketedSelection: BracketedSelection,
@@ -2238,6 +2349,57 @@ var json_p3 = (function (exports) {
     }
   }
 
+  class CurrentKey extends FilterExpression {
+    evaluate(context) {
+      return context.currentKey ?? Nothing;
+    }
+    toString() {
+      return "#";
+    }
+  }
+
+  /**
+   * Object property name selector or array index selector.
+   */
+  class KeysSelector extends JSONPathSelector {
+    constructor(environment, token) {
+      let shorthand = arguments.length > 2 && arguments[2] !== undefined ? arguments[2] : false;
+      super(environment, token);
+      this.environment = environment;
+      this.token = token;
+      this.shorthand = shorthand;
+    }
+    resolve(nodes) {
+      const rv = [];
+      for (const node of nodes) {
+        if (node.value instanceof String || isArray(node.value)) continue;
+        if (isObject(node.value)) {
+          let i = 0;
+          for (const [key, _] of this.environment.entries(node.value)) {
+            rv.push(new JSONPathNode(key, node.location.concat("[~]", `[${i}]`), node.root));
+            i++;
+          }
+        }
+      }
+      return rv;
+    }
+    *lazyResolve(nodes) {
+      for (const node of nodes) {
+        if (node.value instanceof String || isArray(node.value)) continue;
+        if (isObject(node.value)) {
+          let i = 0;
+          for (const [key, _] of this.environment.entries(node.value)) {
+            yield new JSONPathNode(key, node.location.concat("[~]", `[${i}]`), node.root);
+            i++;
+          }
+        }
+      }
+    }
+    toString() {
+      return this.shorthand ? "[~]" : "~";
+    }
+  }
+
   const PRECEDENCE_LOWEST = 1;
   const PRECEDENCE_LOGICAL_OR = 4;
   const PRECEDENCE_LOGICAL_AND = 5;
@@ -2253,7 +2415,7 @@ var json_p3 = (function (exports) {
   class Parser {
     constructor(environment) {
       this.environment = environment;
-      this.tokenMap = new Map([[TokenKind.FALSE, this.parseBoolean], [TokenKind.NUMBER, this.parseNumber], [TokenKind.LPAREN, this.parseGroupedExpression], [TokenKind.NOT, this.parsePrefixExpression], [TokenKind.NULL, this.parseNull], [TokenKind.ROOT, this.parseRootQuery], [TokenKind.CURRENT, this.parseRelativeQuery], [TokenKind.SINGLE_QUOTE_STRING, this.parseString], [TokenKind.DOUBLE_QUOTE_STRING, this.parseString], [TokenKind.TRUE, this.parseBoolean], [TokenKind.FUNCTION, this.parseFunction]]);
+      this.tokenMap = new Map([[TokenKind.FALSE, this.parseBoolean], [TokenKind.NUMBER, this.parseNumber], [TokenKind.LPAREN, this.parseGroupedExpression], [TokenKind.NOT, this.parsePrefixExpression], [TokenKind.NULL, this.parseNull], [TokenKind.ROOT, this.parseRootQuery], [TokenKind.CURRENT, this.parseRelativeQuery], [TokenKind.SINGLE_QUOTE_STRING, this.parseString], [TokenKind.DOUBLE_QUOTE_STRING, this.parseString], [TokenKind.TRUE, this.parseBoolean], [TokenKind.FUNCTION, this.parseFunction], [TokenKind.KEY, this.parseCurrentKey], [TokenKind.KEY, this.parseCurrentKey]]);
     }
     parse(stream) {
       if (stream.current.kind === TokenKind.ROOT) stream.next();
@@ -2285,6 +2447,8 @@ var json_p3 = (function (exports) {
           return new NameSelector(this.environment, stream.current, stream.current.value, true);
         case TokenKind.WILD:
           return new WildcardSelector(this.environment, stream.current, true);
+        case TokenKind.KEYS:
+          return new KeysSelector(this.environment, stream.current, true);
         case TokenKind.DDOT:
           {
             const segmentToken = stream.current;
@@ -2378,6 +2542,9 @@ var json_p3 = (function (exports) {
           case TokenKind.WILD:
             items.push(new WildcardSelector(this.environment, stream.current));
             break;
+          case TokenKind.KEYS:
+            items.push(new KeysSelector(this.environment, stream.current));
+            break;
           case TokenKind.EOF:
             throw new JSONPathSyntaxError("unexpected end of query", stream.current);
           default:
@@ -2458,6 +2625,9 @@ var json_p3 = (function (exports) {
       const tok = stream.next();
       return new RelativeQuery(tok, new JSONPath(this.environment, this.parsePath(stream, true)));
     }
+    parseCurrentKey(stream) {
+      return new CurrentKey(stream.current);
+    }
     parseFunction(stream) {
       const args = [];
       const tok = stream.next();
@@ -2574,6 +2744,10 @@ var json_p3 = (function (exports) {
      * If `true`, enable nondeterministic ordering when iterating JSON object data.
      */
 
+    /**
+     * The pattern to use for the non-standard _keys selector_.
+     */
+
     /**
      * A map of function names to objects implementing the {@link FilterFunction}
      * interface. You are free to set or delete custom filter functions directly.
@@ -2589,6 +2763,7 @@ var json_p3 = (function (exports) {
       this.minIntIndex = options.maxIntIndex ?? -Math.pow(2, 53) - 1;
       this.maxRecursionDepth = options.maxRecursionDepth ?? 50;
       this.nondeterministic = options.nondeterministic ?? false;
+      this.keysPattern = options.keysPattern ?? /~/y;
       this.parser = new Parser(this);
       this.setupFilterFunctions();
     }
@@ -2598,7 +2773,7 @@ var json_p3 = (function (exports) {
      * @returns A new {@link JSONPath} object, bound to this environment.
      */
     compile(path) {
-      return new JSONPath(this, this.parser.parse(new TokenStream(tokenize(path))));
+      return new JSONPath(this, this.parser.parse(new TokenStream(tokenize(this, path))));
     }
 
     /**
@@ -2680,7 +2855,7 @@ var json_p3 = (function (exports) {
       for (const [typ, arg, idx] of func.argTypes.map((t, i) => [t, args[i], i])) {
         switch (typ) {
           case FunctionExpressionType.ValueType:
-            if (!(arg instanceof FilterExpressionLiteral || arg instanceof JSONPathQuery && arg.path.singularQuery() || arg instanceof FunctionExtension && this.functionRegister.get(arg.name)?.returnType === FunctionExpressionType.ValueType)) {
+            if (!(arg instanceof FilterExpressionLiteral || arg instanceof CurrentKey || arg instanceof JSONPathQuery && arg.path.singularQuery() || arg instanceof FunctionExtension && this.functionRegister.get(arg.name)?.returnType === FunctionExpressionType.ValueType)) {
               throw new JSONPathTypeError(`${token.value}() argument ${idx} must be of ValueType`, arg.token);
             }
             break;
@@ -3308,7 +3483,7 @@ var json_p3 = (function (exports) {
     apply: apply
   });
 
-  const version = "1.1.1";
+  const version = "1.2.0";
 
   exports.DEFAULT_ENVIRONMENT = DEFAULT_ENVIRONMENT;
   exports.FunctionExpressionType = FunctionExpressionType;