aboutsummaryrefslogtreecommitdiff
path: root/app/src/main/kotlin
diff options
context:
space:
mode:
Diffstat (limited to 'app/src/main/kotlin')
-rw-r--r--app/src/main/kotlin/com/pitchedapps/frost/services/DownloadService.kt184
-rw-r--r--app/src/main/kotlin/com/pitchedapps/frost/utils/Downloader.kt21
2 files changed, 193 insertions, 12 deletions
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/services/DownloadService.kt b/app/src/main/kotlin/com/pitchedapps/frost/services/DownloadService.kt
new file mode 100644
index 00000000..ee0c2027
--- /dev/null
+++ b/app/src/main/kotlin/com/pitchedapps/frost/services/DownloadService.kt
@@ -0,0 +1,184 @@
+package com.pitchedapps.frost.services
+
+import android.app.IntentService
+import android.app.Notification
+import android.app.PendingIntent
+import android.app.Service
+import android.content.Context
+import android.content.Intent
+import android.support.v4.app.NotificationCompat
+import android.support.v4.app.NotificationManagerCompat
+import android.support.v4.content.FileProvider
+import ca.allanwang.kau.utils.copyFromInputStream
+import ca.allanwang.kau.utils.string
+import com.pitchedapps.frost.BuildConfig
+import com.pitchedapps.frost.R
+import com.pitchedapps.frost.utils.L
+import com.pitchedapps.frost.utils.createMediaFile
+import okhttp3.MediaType
+import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.ResponseBody
+import okio.*
+import org.jetbrains.anko.toast
+import java.io.File
+
+/**
+ * Created by Allan Wang on 2017-08-08.
+ *
+ * Background file downloader
+ * All we are given is a link and a mime type
+ */
+class DownloadService : IntentService("FrostVideoDownloader") {
+
+ companion object {
+ const val EXTRA_URL = "download_url"
+ private const val MAX_PROGRESS = 1000
+ private const val DOWNLOAD_GROUP = "frost_downloads"
+ }
+
+ val client: OkHttpClient by lazy { initClient() }
+
+ val start = System.currentTimeMillis()
+ var totalSize = 0L
+ val downloaded = mutableSetOf<String>()
+
+ private lateinit var notifBuilder: NotificationCompat.Builder
+ private var notifId: Int = -1
+
+ override fun onStartCommand(intent: Intent?, flags: Int, startId: Int): Int {
+ if (intent != null && intent.flags == PendingIntent.FLAG_CANCEL_CURRENT) {
+ L.i("Cancelling download service")
+ cancelDownload()
+ return Service.START_NOT_STICKY
+ }
+ return super.onStartCommand(intent, flags, startId)
+ }
+
+ override fun onHandleIntent(intent: Intent?) {
+ val url: String = intent?.getStringExtra(EXTRA_URL) ?: return
+
+ if (downloaded.contains(url)) return
+
+ val request: Request = Request.Builder()
+ .url(url)
+ .build()
+
+ notifBuilder = frostNotification.quiet
+ notifId = Math.abs(url.hashCode() + System.currentTimeMillis().toInt())
+ val cancelIntent = PendingIntent.getService(this, 0, intent, PendingIntent.FLAG_CANCEL_CURRENT)
+
+ notifBuilder.setContentTitle(string(R.string.downloading_video))
+ .setCategory(Notification.CATEGORY_PROGRESS)
+ .setWhen(System.currentTimeMillis())
+ .setProgress(MAX_PROGRESS, 0, false)
+ .setOngoing(true)
+ .addAction(R.drawable.ic_action_cancel, string(R.string.kau_cancel), cancelIntent)
+ .setGroup(DOWNLOAD_GROUP)
+
+ client.newCall(request).execute().use { response ->
+ if (!response.isSuccessful) {
+ L.e("Video download failed")
+ toast("Video download failed")
+ return@use
+ }
+
+ val stream = response.body()?.byteStream() ?: return@use
+ val extension = response.request().body()?.contentType()?.subtype()
+ val destination = createMediaFile(if (extension == null) "" else ".$extension")
+ destination.copyFromInputStream(stream)
+
+ notifBuilder.setContentIntent(getPendingIntent(this, destination))
+ notifBuilder.show()
+ }
+ }
+
+ private fun NotificationCompat.Builder.show() {
+ NotificationManagerCompat.from(this@DownloadService).notify(DOWNLOAD_GROUP, notifId, build())
+ }
+
+
+ private fun getPendingIntent(context: Context, file: File): PendingIntent {
+ val uri = FileProvider.getUriForFile(context, BuildConfig.APPLICATION_ID + ".provider", file)
+ val type = context.contentResolver.getType(uri)
+ L.i("DownloadType: retrieved pending intent - $uri $type")
+ val intent = Intent(Intent.ACTION_VIEW, uri)
+ .addFlags(Intent.FLAG_GRANT_READ_URI_PERMISSION)
+ .setDataAndType(uri, type)
+ return PendingIntent.getActivity(context, 0, intent, 0)
+ }
+
+ /**
+ * Adds url to downloaded list and modifies the notif builder for the finished state
+ * Does not show the new notification
+ */
+ private fun finishDownload(url: String) {
+ L.i("Video download finished", url)
+ downloaded.add(url)
+ notifBuilder.setContentTitle(string(R.string.downloaded_video))
+ .setProgress(0, 0, false).setOngoing(false).setAutoCancel(true)
+ .apply { mActions.clear() }
+ }
+
+ private fun cancelDownload() {
+ client.dispatcher().cancelAll()
+ NotificationManagerCompat.from(this).cancel(DOWNLOAD_GROUP, notifId)
+ }
+
+ private fun onProgressUpdate(url: String, type: MediaType?, percentage: Float, done: Boolean) {
+ L.v("Download request progress $percentage", url)
+ notifBuilder.setProgress(MAX_PROGRESS, (percentage * MAX_PROGRESS).toInt(), false)
+ if (done) finishDownload(url)
+ notifBuilder.show()
+ }
+
+ override fun onTaskRemoved(rootIntent: Intent?) {
+ super.onTaskRemoved(rootIntent)
+ }
+
+ private fun initClient(): OkHttpClient = OkHttpClient.Builder()
+ .addNetworkInterceptor { chain ->
+ val original = chain.proceed(chain.request())
+ val body = original.body() ?: return@addNetworkInterceptor original
+ if (body.contentLength() > 0L) totalSize += body.contentLength()
+ return@addNetworkInterceptor original.newBuilder()
+ .body(ProgressResponseBody(
+ original.request().url().toString(),
+ body,
+ this@DownloadService::onProgressUpdate))
+ .build()
+ }
+ .build()
+
+ private class ProgressResponseBody(
+ val url: String,
+ val responseBody: ResponseBody,
+ val listener: (url: String, type: MediaType?, percentage: Float, done: Boolean) -> Unit) : ResponseBody() {
+
+ private val bufferedSource: BufferedSource by lazy { Okio.buffer(source(responseBody.source())) }
+
+ override fun contentLength(): Long = responseBody.contentLength()
+
+ override fun contentType(): MediaType? = responseBody.contentType()
+
+ override fun source(): BufferedSource = bufferedSource
+
+ private fun source(source: Source): Source = object : ForwardingSource(source) {
+
+ private var totalBytesRead = 0L
+
+ override fun read(sink: Buffer?, byteCount: Long): Long {
+ val bytesRead = super.read(sink, byteCount)
+ // read() returns the number of bytes read, or -1 if this source is exhausted.
+ totalBytesRead += if (bytesRead != -1L) bytesRead else 0
+ listener(
+ url,
+ contentType(),
+ totalBytesRead.toFloat() / responseBody.contentLength(),
+ bytesRead == -1L
+ )
+ return bytesRead
+ }
+ }
+ }
+} \ No newline at end of file
diff --git a/app/src/main/kotlin/com/pitchedapps/frost/utils/Downloader.kt b/app/src/main/kotlin/com/pitchedapps/frost/utils/Downloader.kt
index 60d709fb..566bffde 100644
--- a/app/src/main/kotlin/com/pitchedapps/frost/utils/Downloader.kt
+++ b/app/src/main/kotlin/com/pitchedapps/frost/utils/Downloader.kt
@@ -13,6 +13,7 @@ import ca.allanwang.kau.utils.copyFromInputStream
import ca.allanwang.kau.utils.string
import com.pitchedapps.frost.BuildConfig
import com.pitchedapps.frost.R
+import com.pitchedapps.frost.services.DownloadService
import com.pitchedapps.frost.services.frostNotification
import com.pitchedapps.frost.services.getNotificationPendingCancelIntent
import com.pitchedapps.frost.services.quiet
@@ -22,7 +23,7 @@ import okhttp3.Request
import okhttp3.ResponseBody
import okio.*
import org.jetbrains.anko.AnkoAsyncContext
-import org.jetbrains.anko.doAsync
+import org.jetbrains.anko.startService
import java.io.File
import java.io.IOException
import java.lang.ref.WeakReference
@@ -34,11 +35,10 @@ import java.lang.ref.WeakReference
*/
fun Context.frostDownload(url: String) {
L.d("Received download request", "Download $url")
- val type = if (url.contains("video")) DownloadType.VIDEO
- else return L.d("Download request does not match any type")
- kauRequestPermissions(PERMISSION_WRITE_EXTERNAL_STORAGE) {
- granted, _ ->
- if (granted) doAsync { frostDownloadImpl(url, type) }
+ kauRequestPermissions(PERMISSION_WRITE_EXTERNAL_STORAGE) { granted, _ ->
+ if (granted)
+// doAsync { frostDownloadImpl(url, type) }
+ startService<DownloadService>(DownloadService.EXTRA_URL to url)
}
}
@@ -85,11 +85,9 @@ private fun AnkoAsyncContext<Context>.frostDownloadImpl(url: String, type: Downl
var client: OkHttpClient? = null
client = OkHttpClient.Builder()
- .addNetworkInterceptor {
- chain ->
+ .addNetworkInterceptor { chain ->
val original = chain.proceed(chain.request())
- return@addNetworkInterceptor original.newBuilder().body(ProgressResponseBody(original.body()!!) {
- bytesRead, contentLength, done ->
+ return@addNetworkInterceptor original.newBuilder().body(ProgressResponseBody(original.body()!!) { bytesRead, contentLength, done ->
//cancel request if context reference is now invalid
if (weakRef.get() == null) {
client?.cancel(url)
@@ -106,8 +104,7 @@ private fun AnkoAsyncContext<Context>.frostDownloadImpl(url: String, type: Downl
}).build()
}
.build()
- client.newCall(request).execute().use {
- response ->
+ client.newCall(request).execute().use { response ->
if (!response.isSuccessful) throw IOException("Unexpected code $response")
val stream = response.body()?.byteStream()
if (stream != null) {