Fix: auto close stream open dirs

fioan89 · fioan89 · commit 8c21dd8aa994 · 2022-12-02T23:37:56.000+02:00
- we also limit the walking depth to just one level.
  Coder CLI is placed at the root of the tmp dir.
- make the walking more resilient to security exceptions
diff --git a/src/main/kotlin/com/coder/gateway/sdk/CoderCLIManager.kt b/src/main/kotlin/com/coder/gateway/sdk/CoderCLIManager.kt
@@ -3,6 +3,7 @@ package com.coder.gateway.sdk
 import com.intellij.openapi.diagnostic.Logger
 import java.io.InputStream
 import java.net.URL
+import java.nio.file.FileVisitOption
 import java.nio.file.Files
 import java.nio.file.Path
 import java.nio.file.Paths
@@ -67,9 +68,14 @@ class CoderCLIManager(url: URL, buildVersion: String) {
     }
 
     fun removeOldCli() {
-        Files.walk(Path.of(tmpDir)).sorted().map { it.toFile() }.filter { it.name.contains(cliNamePrefix) && !it.name.contains(cliFileName) }.forEach {
-            logger.info("Removing $it because it is an old coder cli")
-            it.delete()
+        val tmpPath = Path.of(tmpDir)
+        if (Files.isReadable(tmpPath)) {
+            Files.walk(tmpPath, 1).use {
+                it.sorted().map { pt -> pt.toFile() }.filter { fl -> fl.name.contains(cliNamePrefix) && !fl.name.contains(cliFileName) }.forEach { fl ->
+                    logger.info("Removing $fl because it is an old coder cli")
+                    fl.delete()
+                }
+            }
         }
     }
 
