Fix #309: Recursive delete on exit

pathikrit · May 8, 2019 · 5cd4680 · 5cd4680
1 parent baf2a67
commit 5cd4680
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/core/src/main/scala/better/files/File.scala b/core/src/main/scala/better/files/File.scala
@@ -1182,8 +1182,16 @@ class File private (val path: Path)(implicit val fileSystem: FileSystem = path.g
     this
   }
 
-  def deleteOnExit(): this.type = {
-    toJava.deleteOnExit()
+  def deleteOnExit(
+      swallowIOExceptions: Boolean = false,
+      linkOption: File.LinkOptions = File.LinkOptions.noFollow
+    ): this.type = {
+    try {
+      if (isDirectory(linkOption)) list.toList.foreach(_.deleteOnExit(swallowIOExceptions, linkOption))
+      toJava.deleteOnExit()
+    } catch {
+      case _: IOException if swallowIOExceptions => //e.printStackTrace() //swallow
+    }
     this
   }