From d77fa8c8e6b52d4766cdcea26a332059f313e338 Mon Sep 17 00:00:00 2001
From: Emanuel Braz <fastencoding@gmail.com>
Date: Sat, 19 Mar 2022 17:47:24 -0300
Subject: [PATCH] fix: remove 'part of' files from list of imports

---
 CHANGELOG.md                             |  4 ++++
 README.md                                |  2 +-
 lib/usecases/create_file_references.dart | 30 ++++++++++++++++--------
 pubspec.yaml                             |  2 +-
 4 files changed, 26 insertions(+), 12 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index cb3a181..b94f21b 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,7 @@
+# [3.0.1] 2022-03-19
+#### Fix
+- remove 'part of' files from list of imports
+
 # [3.0.0] 2022-03-07
 #### Break
 - Spelling fixes: `--exclude-sufix` becomes `--exclude-suffix`
diff --git a/README.md b/README.md
index 3431308..5fe5a26 100644
--- a/README.md
+++ b/README.md
@@ -1,4 +1,4 @@
-## DLCOV - CLI to add code coverage threshold (CI/CD, git hooks, etc.)
+## DLCOV - CLI to verify code coverage threshold (CI/CD, git hooks, etc.)
 
 ### Usage Example
 #### Long
diff --git a/lib/usecases/create_file_references.dart b/lib/usecases/create_file_references.dart
index e0fe874..8a54ca9 100644
--- a/lib/usecases/create_file_references.dart
+++ b/lib/usecases/create_file_references.dart
@@ -21,8 +21,11 @@ class CreateFileReferences {
     final fileSytemEntities =
         await _helper.getFileSystemEntities(Directory(_sourceDirectory));
 
-    final filteredFilePaths = _helper.getFilteredFilePaths(
-        fileSytemEntities, _removeFileWithSuffixes);
+    final candidateFiles =
+        await _helper.getOnlyCandidateFiles(fileSytemEntities);
+
+    final filteredFilePaths =
+        _helper.getFilteredFilePaths(candidateFiles, _removeFileWithSuffixes);
 
     final fileImports = [
       '/*\n'
@@ -68,18 +71,25 @@ class CreateFileReferencesHelper {
     return completer.future;
   }
 
-  List<String> getFilteredFilePaths(List<FileSystemEntity> fileSytemEntities,
-      List<String> removeFileWithSuffixes) {
-    return fileSytemEntities
-        .where((fileSystemEntity) {
-          return fileSystemEntity.path.endsWith('.dart') &&
-              !removeFileWithSuffixes
-                  .any((suffix) => fileSystemEntity.path.endsWith(suffix));
-        })
+  List<String> getFilteredFilePaths(
+      List<File> files, List<String> removeFileWithSuffixes) {
+    return files
+        .where((file) =>
+            !removeFileWithSuffixes.any((suffix) => file.path.endsWith(suffix)))
         .map((e) => e.path)
         .toList();
   }
 
+  Future<List<File>> getOnlyCandidateFiles(
+      List<FileSystemEntity> fileSytemEntities) async {
+    List<File> candidateFiles = fileSytemEntities.whereType<File>().toList();
+    return candidateFiles
+        .where((file) => file.path.endsWith('.dart'))
+        .where((file) =>
+            !file.readAsLinesSync().any((line) => line.startsWith('part of')))
+        .toList();
+  }
+
   Future<File> writeContentToFile(String content, String path) =>
       fileSystemUtil.writeToFile(content, path);
 }
diff --git a/pubspec.yaml b/pubspec.yaml
index 63c99c8..c7e3ef4 100644
--- a/pubspec.yaml
+++ b/pubspec.yaml
@@ -1,6 +1,6 @@
 name: dlcov
 description: A CLI to add code coverage threshold (CI/CD, git hooks, etc.)
-version: 3.0.0
+version: 3.0.1
 homepage: https://github.com/emanuel-braz/dlcov
 
 environment: