Allow for include/exclude to check for contract name

b00ste · Jul 10, 2023 · ab070cb · ab070cb
1 parent fb162a9
commit ab070cb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -32,10 +32,10 @@ async function generateDocumentation(hre: HardhatRuntimeEnvironment): Promise<vo
 
   const qualifiedNames = await hre.artifacts.getAllFullyQualifiedNames();
   const filteredQualifiedNames = qualifiedNames.filter((filePath: string) => {
-    const relativeFilePath = filePath.split(':')[0];
+    const [relativeFilePath, contractName] = filePath.split(':')[0];
     // Checks if the documentation has to be generated for this contract
-    const includesPath = config.include.some((str) => relativeFilePath === str);
-    const excludesPath = config.exclude.some((str) => relativeFilePath === str);
+    const includesPath = config.include.some((str) => relativeFilePath === str || contractName === str);
+    const excludesPath = config.exclude.some((str) => relativeFilePath === str || contractName === str);
     return (config.include.length === 0 || includesPath) && !excludesPath;
   });