From 042f184d80cda1a19dd37e4fd32de0370e3490f7 Mon Sep 17 00:00:00 2001
From: Cedric Beust <cedric@beust.com>
Date: Wed, 28 Oct 2015 08:22:30 -0700
Subject: [PATCH] Copy stuff with closure progress

---
 .../kotlin/com/beust/kobalt/maven/Kurl.kt     | 29 +++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/src/main/kotlin/com/beust/kobalt/maven/Kurl.kt b/src/main/kotlin/com/beust/kobalt/maven/Kurl.kt
index f0966a89..44d4b695 100644
--- a/src/main/kotlin/com/beust/kobalt/maven/Kurl.kt
+++ b/src/main/kotlin/com/beust/kobalt/maven/Kurl.kt
@@ -15,6 +15,7 @@ public class Kurl @Inject constructor(@Assisted val url: String, val http: Http)
     val connection : URLConnection by lazy {
         URL(url).openConnection()
     }
+
     val exists : Boolean
         get() {
             if (url.contains("android")) {
@@ -37,9 +38,33 @@ public class Kurl @Inject constructor(@Assisted val url: String, val http: Http)
     private val estimatedSize: Int
         get() = if (url.contains("kotlin-compiler")) 18000000 else 1000000
 
-    fun toOutputStream(os: OutputStream) = ByteStreams.copy(connection.inputStream, os)
+    fun toOutputStream(os: OutputStream, progress: (Long) -> Unit) = copy(connection.inputStream, os, progress)
 
-    fun toFile(file: File) = toOutputStream(FileOutputStream(file))
+    fun toFile(file: File, progress: (Long) -> Unit = {}) = toOutputStream(FileOutputStream(file), progress)
+
+    private fun copy(from: InputStream, to: OutputStream, progress: (Long) -> Unit = {}) : Long {
+        val estimate =
+            if (connection is HttpURLConnection) {
+                (connection as HttpURLConnection).let {
+                    it.contentLength
+                }
+            } else {
+                estimatedSize
+            }
+
+        val buf = ByteArray(estimatedSize)
+        var total: Long = 0
+        while (true) {
+            val r = from.read(buf)
+            if (r == -1) {
+                break
+            }
+            to.write(buf, 0, r)
+            total += r.toLong()
+            progress(total * 100 / estimate)
+        }
+        return total
+    }
 
     val string: String
         get() {