Update isTagUsedInEdgeTreatment to handle the new kwargs

KittyCAD · Feb 15, 2025 · 5b34c48 · 5b34c48
1 parent adf6ef4
commit 5b34c48
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/lang/modifyAst/addEdgeTreatment.ts b/src/lang/modifyAst/addEdgeTreatment.ts
@@ -754,6 +754,16 @@ export const isTagUsedInEdgeTreatment = ({
       ) {
         inEdgeTreatment = true
       }
+      if (inEdgeTreatment && node.type === 'CallExpressionKw') {
+        node.arguments.forEach((prop) => {
+          if (
+            prop.label.name === 'tags' &&
+            prop.arg.type === 'ArrayExpression'
+          ) {
+            inObj = true
+          }
+        })
+      }
       if (inEdgeTreatment && node.type === 'ObjectExpression') {
         node.properties.forEach((prop) => {
           if (
@@ -798,6 +808,16 @@ export const isTagUsedInEdgeTreatment = ({
       ) {
         inEdgeTreatment = false
       }
+      if (inEdgeTreatment && node.type === 'CallExpressionKw') {
+        node.arguments.forEach((prop) => {
+          if (
+            prop.label.name === 'tags' &&
+            prop.arg.type === 'ArrayExpression'
+          ) {
+            inObj = true
+          }
+        })
+      }
       if (inEdgeTreatment && node.type === 'ObjectExpression') {
         node.properties.forEach((prop) => {
           if (