tag as top level part 2 (#2773)

* updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * A snapshot a day keeps the bugs away! 📷🐛 (OS: ubuntu) * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * fmt Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * unit tests pass Signed-off-by: Jess Frazelle <github@jessfraz.com> * playwright Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * format Signed-off-by: Jess Frazelle <github@jessfraz.com> * format Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * more literals gone Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * remove console log Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * remove only Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix some recast shit Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * last Signed-off-by: Jess Frazelle <github@jessfraz.com> --------- Signed-off-by: Jess Frazelle <github@jessfraz.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
2024-06-24 22:39:04 -07:00
parent ad5bfa1a29
commit da6cd5cf9f
44 changed files with 720 additions and 536 deletions
--- a/src/lang/queryAst.ts
+++ b/src/lang/queryAst.ts
@ -365,6 +365,8 @@ export function traverse(
    // do nothing
  } else if (_node.type === 'Literal') {
    // do nothing
+  } else if (_node.type === 'TagDeclarator') {
+    // do nothing
  } else if (_node.type === 'ArrayExpression') {
    _node.elements.forEach((el, index) =>
      _traverse(el, [
@ -785,8 +787,14 @@ export function findUsesOfTagInPipe(
  if (node.type !== 'CallExpression') return []
  const tagIndex = node.callee.name === 'close' ? 1 : 2
  const thirdParam = node.arguments[tagIndex]
-  if (thirdParam?.type !== 'Literal') return []
-  const tag = String(thirdParam.value)
+  if (
+    !(thirdParam?.type === 'TagDeclarator' || thirdParam?.type === 'Identifier')
+  )
+    return []
+  const tag =
+    thirdParam?.type === 'TagDeclarator'
+      ? String(thirdParam.value)
+      : thirdParam.name

  const varDec = getNodeFromPath<VariableDeclaration>(
    ast,
@ -807,9 +815,11 @@ export function findUsesOfTagInPipe(
      )
        return
      const tagArg = node.arguments[0]
-      if (tagArg.type !== 'Literal') return
-      if (String(tagArg.value) === tag)
-        dependentRanges.push([node.start, node.end])
+      if (!(tagArg.type === 'TagDeclarator' || tagArg.type === 'Identifier'))
+        return
+      const tagArgValue =
+        tagArg.type === 'TagDeclarator' ? String(tagArg.value) : tagArg.name
+      if (tagArgValue === tag) dependentRanges.push([node.start, node.end])
    },
  })
  return dependentRanges