fix: interfaces are sometimes included in the output (#50)

defi-wonderland · Apr 4, 2024 · 507b6e2 · 507b6e2
1 parent b8fade5
commit 507b6e2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/utils.ts b/src/utils.ts
@@ -261,7 +261,9 @@ export async function getSourceUnits(rootPath: string, contractsDirectories: str
     includePath: [rootPath],
   });
 
-  const sourceUnits = new ASTReader().read(compiledFiles.data, ASTKind.Any, compiledFiles.files);
+  const sourceUnits = new ASTReader().read(compiledFiles.data, ASTKind.Any, compiledFiles.files)
+    // Skip source units that are not in the contracts directories
+    .filter((sourceUnit) => contractsDirectories.some((directory) => sourceUnit.absolutePath.includes(directory)));
 
   return sourceUnits;
 }