file size filtering change

Privado-Inc · Mar 26, 2024 · 4173e78 · 4173e78
1 parent 79866b2
commit 4173e78
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/src/main/scala/ai/privado/passes/PropertyParserPass.scala b/src/main/scala/ai/privado/passes/PropertyParserPass.scala
@@ -423,21 +423,20 @@ class PropertyParserPass(
       )
       .filter(file => !file.contains("delombok"))
       .filter(file => Utilities.isFileProcessable(file, ruleCache) && (!file.matches(".*node_modules.*")))
+      .filter(file => filterUsingFileSize(file, ruleCache))
       .distinct
 
-    filterFilesWithDir(filterUsingFileSize(propertyFiles, ruleCache), ruleCache)
+    filterFilesWithDir(propertyFiles, ruleCache)
   }
 
-  private def filterUsingFileSize(filePaths: List[String], ruleCache: RuleCache): List[String] = {
+  private def filterUsingFileSize(filePath: String, ruleCache: RuleCache): Boolean = {
     val fileLimit = ruleCache.getSystemConfigByKey(Constants.PropertyFileSizeLimit, true)
     if (fileLimit.nonEmpty) {
-      filePaths.filter(filePath => {
-        val file               = new File(filePath)
-        val fileSizeInKiloByte = file.length() / 1024 // Get the size in KB
-        fileSizeInKiloByte <= fileLimit.toInt
-      })
+      val file               = new File(filePath)
+      val fileSizeInKiloByte = file.length() / 1024 // Get the size in KB
+      fileSizeInKiloByte <= fileLimit.toInt
     } else {
-      filePaths
+      true
     }
   }