fix: support template literals in require

fixes #20

Author: connor4312

src/__snapshots__/parsing.test.ts.snap

@@ -593,3 +593,39 @@ exports[`extract > extracts ts import mangled 1`] = `
   },
 ]
 `;
+
+exports[`extract > works with string literals 1`] = `
+[
+  {
+    "children": [
+      {
+        "children": [],
+        "fn": "it",
+        "location": SourceLocation {
+          "end": Position {
+            "column": 35,
+            "line": 5,
+          },
+          "start": Position {
+            "column": 8,
+            "line": 5,
+          },
+        },
+        "name": "addition",
+      },
+    ],
+    "fn": "describe",
+    "location": SourceLocation {
+      "end": Position {
+        "column": 8,
+        "line": 6,
+      },
+      "start": Position {
+        "column": 6,
+        "line": 4,
+      },
+    },
+    "name": "math",
+  },
+]
+`;

src/parsing.test.ts

@@ -101,4 +101,16 @@ const node_test_1 = require("node:test");
 
     expect(parseSource(src)).toMatchSnapshot();
   });
+
+  it("works with string literals", () => {
+    const src = `
+      const nt = require(\`node:test\`);
+
+      nt.describe(\`math\`, () => {
+        nt.it(\`addition\`, () => {});
+      });
+    `;
+
+    expect(parseSource(src)).toMatchSnapshot();
+  });
 });

src/parsing.ts

@@ -56,6 +56,15 @@ const matchNamespaced =
       : undefined;
   };
 
+const getStringish = (nameArg: Node | undefined): string | undefined => {
+  if (nameArg?.type === C.Literal && typeof nameArg.value === "string") {
+    return nameArg.value;
+  }
+  if (nameArg?.type === C.TemplateLiteral && nameArg.quasis.length === 1) {
+    return nameArg.quasis[0].value.cooked || nameArg.quasis[0].value.raw;
+  }
+};
+
 export interface IParsedNode {
   fn: string;
   name: string;
@@ -91,8 +100,8 @@ export const parseSource = (text: string) => {
         node.init.callee.type === C.Identifier &&
         node.init.callee.name === "require"
       ) {
-        const firstArg = node.init.arguments[0];
-        if (firstArg?.type === C.Literal && firstArg.value === C.NodeTest) {
+        const firstArg = getStringish(node.init.arguments[0]);
+        if (firstArg === C.NodeTest) {
           if (node.id.type === C.ObjectPattern) {
             for (const prop of node.id.properties) {
               if (
@@ -108,13 +117,8 @@ export const parseSource = (text: string) => {
           }
         }
       } else if (node.type === C.CallExpression) {
-        const nameArg = node.arguments[0];
-        let name: string;
-        if (nameArg?.type === C.Literal && typeof nameArg.value === "string") {
-          name = nameArg.value;
-        } else if (nameArg?.type === C.TemplateLiteral && nameArg.quasis.length === 1) {
-          name = nameArg.quasis[0].value.cooked || nameArg.quasis[0].value.raw;
-        } else {
+        const name = getStringish(node.arguments[0]);
+        if (name === undefined) {
           return;
         }