fix: empty code error

jehna · Sep 29, 2024 · 4a4b05c · 4a4b05c
1 parent 64a1b95
commit 4a4b05c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/plugins/local-llm-rename/visit-all-identifiers.test.ts b/src/plugins/local-llm-rename/visit-all-identifiers.test.ts
@@ -7,6 +7,11 @@ test("no-op returns the same code", async () => {
   assert.equal(code, await visitAllIdentifiers(code, async (name) => name));
 });
 
+test("no-op returns the same empty code", async () => {
+  const code = "";
+  assert.equal(code, await visitAllIdentifiers(code, async (name) => name));
+});
+
 test("renames a simple variable", async () => {
   const code = `const a = 1;`;
   assert.equal(

diff --git a/src/plugins/local-llm-rename/visit-all-identifiers.ts b/src/plugins/local-llm-rename/visit-all-identifiers.ts
@@ -55,10 +55,10 @@ export async function visitAllIdentifiers(
   onProgress?.(1);
 
   const stringified = await transformFromAstAsync(ast);
-  if (!stringified?.code) {
+  if (stringified?.code == null) {
     throw new Error("Failed to stringify code");
   }
-  return stringified?.code;
+  return stringified.code;
 }
 
 function findIdentifierWithLargestScopeNotVisited(