refactor: connection callback when downloading

FunkyMuse · Oct 16, 2020 · 0bfe8dd · 0bfe8dd
1 parent 4ecfb67
commit 0bfe8dd
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/kotlinextensions/src/main/java/com/crazylegend/kotlinextensions/file/FileExtensions.kt b/kotlinextensions/src/main/java/com/crazylegend/kotlinextensions/file/FileExtensions.kt
@@ -337,16 +337,19 @@ fun downloadFile(urlPath: String, localPath: String, callback: (Uri?) -> Unit =
     return uri
 }
 
-fun downloadFileWithProgress(urlPath: String, localPath: String, onError: () -> Unit = {}, progress: (Int) -> Unit = {}, callback: (Uri?) -> Unit = {}) {
+fun downloadFileWithProgress(urlPath: String, localPath: String,
+                             connectionCallBack: (responseCode: Int) -> Unit = {},
+                             onError: (Exception) -> Unit = {}, progress: (Int) -> Unit = {}, callback: (Uri?) -> Unit = {}) {
     val uri = localPath.toFile().toUri()
     val connection = URL(urlPath).openConnection() as HttpURLConnection
     val input = connection.inputStream
     val output = FileOutputStream(uri.toFile())
     try {
         connection.connect()
 
-        if (connection.responseCode != HttpURLConnection.HTTP_OK) {
-            onError()
+        val responseCode = connection.responseCode
+        connectionCallBack(responseCode)
+        if (responseCode != HttpURLConnection.HTTP_OK) {
             return
         }
 
@@ -361,7 +364,7 @@ fun downloadFileWithProgress(urlPath: String, localPath: String, onError: () ->
             output.write(data, 0, count)
         }
     } catch (e: Exception) {
-        onError()
+        onError(e)
         return
     } finally {
         tryOrIgnore {