Fixed FileNotFoundException while scanning FS (#36)

pinterest · Mar 7, 2017 · bb3a55a · bb3a55a
1 parent 4fe6a3e
commit bb3a55a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/ktlint/src/main/kotlin/com/github/shyiko/ktlint/Main.kt b/ktlint/src/main/kotlin/com/github/shyiko/ktlint/Main.kt
@@ -291,14 +291,15 @@ ${ByteArrayOutputStream().let { this.printUsage(it); it }.toString().trimEnd().s
         return generateSequence(fun (): File? {
             while (true) {
                 val file = stack.pollLast()
-                if (file != null && file.isDirectory) {
+                if (file == null || file.isFile) {
+                    return file
+                }
+                if (file.isDirectory) {
                     val fileList = file.listFiles(filter)
                     if (fileList != null) {
                         stack.addAll(fileList)
                     }
-                    continue
                 }
-                return file
             }
         })
     }