AppUpdater
This commit is contained in:
sathwikcs
@@ -0,0 +1,240 @@
|
|||||||
|
package `in`.sminnovations.attendanceapp.main
|
||||||
|
|
||||||
|
import android.content.Context
|
||||||
|
import android.content.Intent
|
||||||
|
import androidx.core.content.FileProvider
|
||||||
|
import com.google.firebase.firestore.ktx.firestore
|
||||||
|
import com.google.firebase.ktx.Firebase
|
||||||
|
import kotlinx.coroutines.CoroutineScope
|
||||||
|
import kotlinx.coroutines.Dispatchers
|
||||||
|
import kotlinx.coroutines.flow.MutableStateFlow
|
||||||
|
import kotlinx.coroutines.flow.asStateFlow
|
||||||
|
import kotlinx.coroutines.launch
|
||||||
|
import kotlinx.coroutines.tasks.await
|
||||||
|
import kotlinx.coroutines.withContext
|
||||||
|
import okhttp3.OkHttpClient
|
||||||
|
import okhttp3.Request
|
||||||
|
import okhttp3.ResponseBody
|
||||||
|
import okio.Buffer
|
||||||
|
import okio.BufferedSource
|
||||||
|
import okio.ForwardingSource
|
||||||
|
import okio.Source
|
||||||
|
import okio.buffer
|
||||||
|
import okio.source
|
||||||
|
import java.io.File
|
||||||
|
import java.io.IOException
|
||||||
|
|
||||||
|
data class UpdateInfo(
|
||||||
|
val version: String = "",
|
||||||
|
val url: String = "",
|
||||||
|
val isNew: Boolean = false
|
||||||
|
)
|
||||||
|
|
||||||
|
object UpdateChecker {
|
||||||
|
private const val COLLECTION_NAME = "app_updates"
|
||||||
|
private const val DOCUMENT_NAME = "latest_version"
|
||||||
|
|
||||||
|
suspend fun checkForUpdate(
|
||||||
|
context: Context
|
||||||
|
): Result<UpdateInfo> = runCatching {
|
||||||
|
val currentVersion = getCurrentVersion(context)
|
||||||
|
val updateInfo = getLatestVersion()
|
||||||
|
|
||||||
|
if (isNewerVersion(currentVersion, updateInfo.version)) {
|
||||||
|
updateInfo.copy(isNew = true)
|
||||||
|
} else {
|
||||||
|
updateInfo.copy(isNew = false)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun getCurrentVersion(context: Context): String {
|
||||||
|
return try {
|
||||||
|
context.packageManager.getPackageInfo(context.packageName, 0).versionName
|
||||||
|
} catch (e: Exception) {
|
||||||
|
throw UpdateCheckException("Failed to get current app version", e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private suspend fun getLatestVersion(): UpdateInfo {
|
||||||
|
return try {
|
||||||
|
val document = Firebase.firestore
|
||||||
|
.collection(COLLECTION_NAME)
|
||||||
|
.document(DOCUMENT_NAME)
|
||||||
|
.get()
|
||||||
|
.await()
|
||||||
|
|
||||||
|
if (!document.exists()) {
|
||||||
|
throw UpdateCheckException("Update document does not exist")
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateInfo(
|
||||||
|
version = document.getString("version") ?: throw UpdateCheckException("Version field missing"),
|
||||||
|
url = document.getString("url") ?: throw UpdateCheckException("URL field missing")
|
||||||
|
)
|
||||||
|
} catch (e: Exception) {
|
||||||
|
throw UpdateCheckException("Failed to fetch update info", e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun isNewerVersion(currentVersion: String, latestVersion: String): Boolean {
|
||||||
|
return try {
|
||||||
|
val current = currentVersion.split(".").map { it.toIntOrNull() ?: 0 }
|
||||||
|
val latest = latestVersion.split(".").map { it.toIntOrNull() ?: 0 }
|
||||||
|
|
||||||
|
for (i in 0 until maxOf(current.size, latest.size)) {
|
||||||
|
val currentNum = current.getOrNull(i) ?: 0
|
||||||
|
val latestNum = latest.getOrNull(i) ?: 0
|
||||||
|
|
||||||
|
when {
|
||||||
|
latestNum > currentNum -> return true
|
||||||
|
latestNum < currentNum -> return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
false
|
||||||
|
} catch (e: Exception) {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class UpdateCheckException(
|
||||||
|
message: String,
|
||||||
|
cause: Throwable? = null
|
||||||
|
) : Exception(message, cause){
|
||||||
|
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
class AppUpdater(private val context: Context) {
|
||||||
|
sealed class UpdaterState {
|
||||||
|
object Idle : UpdaterState()
|
||||||
|
data class Downloading(val progress: Float) : UpdaterState()
|
||||||
|
data class Success(val file: File) : UpdaterState()
|
||||||
|
data class Error(val message: String) : UpdaterState()
|
||||||
|
}
|
||||||
|
|
||||||
|
private val _updateState = MutableStateFlow<UpdaterState>(UpdaterState.Idle)
|
||||||
|
val updateState = _updateState.asStateFlow()
|
||||||
|
|
||||||
|
fun downloadAndInstall(url: String) {
|
||||||
|
CoroutineScope(Dispatchers.IO).launch {
|
||||||
|
try {
|
||||||
|
val file = downloadApk(url) { progress ->
|
||||||
|
_updateState.value = UpdaterState.Downloading(progress)
|
||||||
|
}
|
||||||
|
_updateState.value = UpdaterState.Success(file)
|
||||||
|
installApk(file)
|
||||||
|
} catch (e: Exception) {
|
||||||
|
_updateState.value = UpdaterState.Error(e.message ?: "Unknown error occurred")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private suspend fun downloadApk(url: String, onProgress: (Float) -> Unit): File {
|
||||||
|
return withContext(Dispatchers.IO) {
|
||||||
|
val client = OkHttpClient.Builder()
|
||||||
|
.addNetworkInterceptor { chain ->
|
||||||
|
val originalResponse = chain.proceed(chain.request())
|
||||||
|
// Verify content type
|
||||||
|
val contentType = originalResponse.header("Content-Type")
|
||||||
|
if (contentType?.contains("application/vnd.android.package-archive") != true &&
|
||||||
|
contentType?.contains("application/octet-stream") != true) {
|
||||||
|
throw IOException("Invalid content type: $contentType")
|
||||||
|
}
|
||||||
|
originalResponse.newBuilder()
|
||||||
|
.body(originalResponse.body?.let { body ->
|
||||||
|
ProgressResponseBody(body) { progress ->
|
||||||
|
CoroutineScope(Dispatchers.Main).launch {
|
||||||
|
onProgress(progress)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.build()
|
||||||
|
}
|
||||||
|
.build()
|
||||||
|
|
||||||
|
val request = Request.Builder()
|
||||||
|
.url(url)
|
||||||
|
.build()
|
||||||
|
|
||||||
|
val file = File(context.cacheDir, "update.apk")
|
||||||
|
|
||||||
|
client.newCall(request).execute().use { response ->
|
||||||
|
if (!response.isSuccessful) throw IOException("Failed to download update")
|
||||||
|
|
||||||
|
response.body?.let { body ->
|
||||||
|
file.outputStream().use { fileOut ->
|
||||||
|
body.byteStream().use { bodyIn ->
|
||||||
|
bodyIn.copyTo(fileOut)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} ?: throw IOException("Empty response")
|
||||||
|
}
|
||||||
|
|
||||||
|
file
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private fun installApk(file: File) {
|
||||||
|
try {
|
||||||
|
val uri = FileProvider.getUriForFile(
|
||||||
|
context,
|
||||||
|
"${context.packageName}.provider",
|
||||||
|
file
|
||||||
|
)
|
||||||
|
|
||||||
|
// Verify the file is actually an APK
|
||||||
|
if (file.length() == 0L) {
|
||||||
|
throw IOException("Downloaded file is empty")
|
||||||
|
}
|
||||||
|
|
||||||
|
val intent = Intent(Intent.ACTION_VIEW).apply {
|
||||||
|
setDataAndType(uri, "application/vnd.android.package-archive")
|
||||||
|
addFlags(Intent.FLAG_ACTIVITY_NEW_TASK)
|
||||||
|
addFlags(Intent.FLAG_GRANT_READ_URI_PERMISSION)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify that there's an app to handle the install intent
|
||||||
|
if (intent.resolveActivity(context.packageManager) != null) {
|
||||||
|
context.startActivity(intent)
|
||||||
|
} else {
|
||||||
|
throw IOException("No app found to handle APK installation")
|
||||||
|
}
|
||||||
|
} catch (e: Exception) {
|
||||||
|
_updateState.value = UpdaterState.Error("Installation failed: ${e.message}")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private class ProgressResponseBody(
|
||||||
|
private val responseBody: ResponseBody,
|
||||||
|
private val onProgressUpdate: (Float) -> Unit
|
||||||
|
) : ResponseBody() {
|
||||||
|
private val bufferedSource: BufferedSource by lazy {
|
||||||
|
source(responseBody.source()).buffer()
|
||||||
|
}
|
||||||
|
|
||||||
|
override fun contentType() = responseBody.contentType()
|
||||||
|
override fun contentLength() = responseBody.contentLength()
|
||||||
|
override fun source(): BufferedSource = bufferedSource
|
||||||
|
|
||||||
|
private fun source(source: Source): Source {
|
||||||
|
return object : ForwardingSource(source) {
|
||||||
|
var totalBytesRead = 0L
|
||||||
|
|
||||||
|
override fun read(sink: Buffer, byteCount: Long): Long {
|
||||||
|
val bytesRead = super.read(sink, byteCount)
|
||||||
|
totalBytesRead += if (bytesRead != -1L) bytesRead else 0
|
||||||
|
|
||||||
|
val progress = if (contentLength() > 0) {
|
||||||
|
totalBytesRead.toFloat() / contentLength()
|
||||||
|
} else {
|
||||||
|
-1f
|
||||||
|
}
|
||||||
|
onProgressUpdate(progress)
|
||||||
|
|
||||||
|
return bytesRead
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user