From ed1e231c7c9502cab6f6901b8c65d0f1c3f6518b Mon Sep 17 00:00:00 2001
From: Eric Eldredge <lettertwo@gmail.com>
Date: Tue, 13 Dec 2022 16:03:21 -0500
Subject: [PATCH] Add  parcel-query function to locate a symbol by local name
 (#8452)

---
 packages/dev/query/src/cli.js | 115 +++++++++++++++++++++++++++++++---
 1 file changed, 106 insertions(+), 9 deletions(-)

diff --git a/packages/dev/query/src/cli.js b/packages/dev/query/src/cli.js
index a759ce4af1f..edec0020a5c 100644
--- a/packages/dev/query/src/cli.js
+++ b/packages/dev/query/src/cli.js
@@ -121,23 +121,113 @@ function getAsset(v: string) {
   }
 }
 
-function findAsset(v: string) {
+function _findAssetNode(v: string) {
   let assetRegex = new RegExp(v);
   for (let node of assetGraph.nodes.values()) {
     if (
       node.type === 'asset' &&
       assetRegex.test(fromProjectPathRelative(node.value.filePath))
     ) {
-      try {
-        console.log(
-          `${bundleGraph.getAssetPublicId(
-            bundleGraph.getAssetById(node.id),
-          )} ${fromProjectPathRelative(node.value.filePath)}`,
-        );
-      } catch (e) {
-        console.log(fromProjectPathRelative(node.value.filePath));
+      return node;
+    }
+  }
+}
+
+function findAsset(v: string) {
+  let node = _findAssetNode(v);
+  if (node) {
+    try {
+      console.log(
+        `${bundleGraph.getAssetPublicId(
+          bundleGraph.getAssetById(node.id),
+        )} ${fromProjectPathRelative(node.value.filePath)}`,
+      );
+    } catch (e) {
+      console.log(fromProjectPathRelative(node.value.filePath));
+    }
+  }
+}
+
+function findAssetWithSymbol(local: string) {
+  let [, assetId, binding, ref] = nullthrows(
+    local.match(/^\$([^$]+)\$([^$]+)\$(.*)$/),
+    `symbol ${local} could not be resolved`,
+  );
+
+  let asset;
+  // Search against the id used by the JSTransformer and ScopeHoistingPackager,
+  // not the final asset id, as it may have changed with further transformation.
+  for (let node of assetGraph.nodes.values()) {
+    if (node.type === 'asset' && node.value.meta.id === assetId) {
+      asset = node;
+      break;
+    }
+  }
+
+  // If the asset couldn't be found by searching for the id,
+  // search for the local name in asset used symbols.
+  if (asset == null) {
+    outer: for (let node of assetGraph.nodes.values()) {
+      if (node.type === 'asset' && node.value.symbols) {
+        for (let symbol of node.value.symbols.values()) {
+          if (symbol.local === local) {
+            asset = node;
+            break outer;
+          }
+        }
+      }
+    }
+  }
+
+  invariant(asset, `An asset for ${assetId} could not be found`);
+  invariant(
+    asset.type === 'asset',
+    `Expected ${assetId} to be an asset, but found a ${asset.type}`,
+  );
+
+  try {
+    console.log(
+      `${bundleGraph.getAssetPublicId(
+        bundleGraph.getAssetById(asset.id),
+      )} ${fromProjectPathRelative(asset.value.filePath)}`,
+    );
+  } catch (e) {
+    console.log(fromProjectPathRelative(asset.value.filePath));
+  }
+  if (binding === 'export' && asset.value.symbols) {
+    for (let [symbolName, symbol] of asset.value.symbols) {
+      if (symbol.local === local) {
+        if (symbol.loc) {
+          let locPath = symbol.loc.filePath;
+          let locAsset = _findAssetNode(String(locPath));
+          if (locAsset != null) {
+            try {
+              console.log(
+                `${bundleGraph.getAssetPublicId(
+                  bundleGraph.getAssetById(locAsset.id),
+                )} ${fromProjectPathRelative(locAsset.value.filePath)}`,
+              );
+            } catch (e) {
+              console.log(
+                `imported as ${symbolName} from ${fromProjectPathRelative(
+                  locAsset.value.filePath,
+                )}`,
+              );
+            }
+          } else {
+            console.log(
+              `imported as ${symbolName} from ${fromProjectPathRelative(
+                locPath,
+              )}`,
+            );
+          }
+        } else {
+          console.log(`imported as ${symbolName}`);
+        }
       }
     }
+  } else if (ref) {
+    console.log(`possibly defined as ${ref}`);
   }
 }
 
@@ -666,6 +756,13 @@ if (initialCmd != null) {
         action: findAsset,
       },
     ],
+    [
+      'findAssetWithSymbol',
+      {
+        help: 'args: <local>. Get the asset that defines the symbol with the given local name',
+        action: findAssetWithSymbol,
+      },
+    ],
   ])) {
     // $FlowFixMe
     server.context[name] = cmd.action;