feat: can extract attribute key from FlatMap

sanity-io · Feb 21, 2024 · c5cd5cb · c5cd5cb
1 parent c1526c9
commit c5cd5cb
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/src/parser.ts b/src/parser.ts
@@ -835,6 +835,7 @@ function extractPropertyKey(node: ExprNode): string {
   if (
     node.type === 'Deref' ||
     node.type === 'Map' ||
+    node.type === 'FlatMap' ||
     node.type === 'Projection' ||
     node.type === 'Slice' ||
     node.type === 'Filter' ||

diff --git a/tap-snapshots/test/parse.test.ts.test.cjs b/tap-snapshots/test/parse.test.ts.test.cjs
@@ -212,6 +212,49 @@ Object {
 }
 `
 
+exports[`test/parse.test.ts TAP Expression parsing when extracting property keys can extract from flatmap > must match snapshot 1`] = `
+Object {
+  "base": Object {
+    "type": "Everything",
+  },
+  "expr": Object {
+    "base": Object {
+      "type": "This",
+    },
+    "expr": Object {
+      "attributes": Array [
+        Object {
+          "name": "a",
+          "type": "ObjectAttributeValue",
+          "value": Object {
+            "base": Object {
+              "base": Object {
+                "name": "a",
+                "type": "AccessAttribute",
+              },
+              "type": "ArrayCoerce",
+            },
+            "expr": Object {
+              "base": Object {
+                "base": Object {
+                  "type": "This",
+                },
+                "type": "Deref",
+              },
+              "type": "ArrayCoerce",
+            },
+            "type": "FlatMap",
+          },
+        },
+      ],
+      "type": "Object",
+    },
+    "type": "Projection",
+  },
+  "type": "Map",
+}
+`
+
 exports[`test/parse.test.ts TAP Expression parsing when extracting property keys can extract from group > must match snapshot 1`] = `
 Object {
   "base": Object {

diff --git a/test/parse.test.ts b/test/parse.test.ts
@@ -138,6 +138,10 @@ t.test('Expression parsing', async (t) => {
     t.test('can extract from group', async (t) => {
       t.matchSnapshot(parse('*{(id)}'))
     })
+
+    t.test('can extract from flatmap', async (t) => {
+      t.matchSnapshot(parse('*{a[]->[]}'))
+    })
   })
 
   t.test('when parsing operators', async (t) => {