add path pattern filter option to linesOfCode

danger · Dec 23, 2020 · fe53d94 · fe53d94
1 parent f502880
commit fe53d94
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/source/dsl/GitDSL.ts b/source/dsl/GitDSL.ts
@@ -186,5 +186,5 @@ export interface GitDSL extends GitJSONDSL {
   /**
    * Offers the overall lines of code added/removed in the diff
    */
-  linesOfCode(): Promise<number | null>
+  linesOfCode(pattern?: string): Promise<number | null>
 }
diff --git a/source/platforms/git/gitJSONToGitDSL.ts b/source/platforms/git/gitJSONToGitDSL.ts
@@ -10,6 +10,8 @@ import * as jsonDiff from "fast-json-patch"
 import jsonpointer from "jsonpointer"
 import JSON5 from "json5"
 
+import micromatch from "micromatch"
+
 import { GitDSL, JSONPatchOperation, GitJSONDSL, StructuredDiff } from "../../dsl/GitDSL"
 import chainsmoker from "../../commands/utils/chainsmoker"
 
@@ -155,11 +157,13 @@ export const gitJSONToGitDSL = (gitJSONRep: GitJSONDSL, config: GitJSONToGitDSLC
     }, Object.create(null))
   }
 
-  const linesOfCode = async () => {
+  const linesOfCode = async (pattern?: string) => {
+    const isPatternMatch = (path: string) => pattern === undefined || micromatch.isMatch(path, pattern)
+
     const [createdFilesDiffs, modifiedFilesDiffs, deletedFilesDiffs] = await Promise.all([
-      Promise.all(gitJSONRep.created_files.map(path => diffForFile(path))),
-      Promise.all(gitJSONRep.modified_files.map(path => diffForFile(path))),
-      Promise.all(gitJSONRep.deleted_files.map(path => diffForFile(path))),
+      Promise.all(gitJSONRep.created_files.filter(isPatternMatch).map(path => diffForFile(path))),
+      Promise.all(gitJSONRep.modified_files.filter(isPatternMatch).map(path => diffForFile(path))),
+      Promise.all(gitJSONRep.deleted_files.filter(isPatternMatch).map(path => diffForFile(path))),
     ])
 
     let additions = createdFilesDiffs