ReVanced/revanced-manager
1
0
Fork 0
mirror of https://github.com/revanced/revanced-manager.git synced 2025-04-30 05:54:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

add assertions

Browse Source
This commit is contained in:
Ax333l 2024-07-13 22:57:49 +02:00
parent c0f6699c0d
commit c8975f55bf
No known key found for this signature in database
GPG Key ID: D2B4D85271127D23
3 changed files with 18 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
app/src/main/java/app/revanced/manager/domain/repository/DownloadedAppRepository.kt
View File

@ -31,16 +31,25 @@ class DownloadedAppRepository(app: Application, db: AppDatabase) {
// Converted integers cannot contain / or .. unlike the package name or version, so they are safer to use here.
val relativePath = File(generateUid().toString())
val savePath = dir.resolve(relativePath).also { it.mkdirs() }
val saveDir = dir.resolve(relativePath).also { it.mkdirs() }
val targetFile = saveDir.resolve("base.apk")
try {
val scope = object : DownloadScope {
override val saveLocation = savePath.resolve("base.apk")
override suspend fun reportProgress(bytesReceived: Int, bytesTotal: Int?) = onDownload(bytesReceived.megaBytes to bytesTotal?.megaBytes)
override val targetFile = targetFile
override suspend fun reportProgress(bytesReceived: Int, bytesTotal: Int?) {
require(bytesReceived >= 0) { "bytesReceived must not be negative" }
require(bytesTotal == null || bytesTotal >= bytesReceived) { "bytesTotal must be greater than or equal to bytesReceived" }
require(bytesTotal != 0) { "bytesTotal must not be zero" }
onDownload(bytesReceived.megaBytes to bytesTotal?.megaBytes)
}
}
plugin.download(scope, app)
if (!targetFile.exists()) throw Exception("Downloader did not download any files")
dao.insert(
DownloadedApp(
packageName = app.packageName,
@ -49,12 +58,12 @@ class DownloadedAppRepository(app: Application, db: AppDatabase) {
)
)
} catch (e: Exception) {
savePath.deleteRecursively()
saveDir.deleteRecursively()
throw e
}
// Return the Apk file.
return getApkFileForDir(savePath)
return getApkFileForDir(saveDir)
}
suspend fun get(packageName: String, version: String) = dao.get(packageName, version)

2
downloader-plugin/src/main/java/app/revanced/manager/plugin/downloader/DownloadScope.kt
View File

@ -6,7 +6,7 @@ interface DownloadScope {
/**
* The location where the downloaded APK should be saved.
*/
val saveLocation: File
val targetFile: File
/**
* A callback for reporting download progress

6
example-downloader-plugin/src/main/java/app/revanced/manager/plugin/downloader/example/ExamplePlugins.kt
View File

@ -43,11 +43,11 @@ private fun installedAppDownloader(context: DownloaderContext) = downloader<Inst
}
download {
Files.copy(Path(it.apkPath), saveLocation.toPath(), StandardCopyOption.REPLACE_EXISTING)
Files.copy(Path(it.apkPath), targetFile.toPath(), StandardCopyOption.REPLACE_EXISTING)
}
}
private val Int.megabytes get() = times(1_000_000)
private val Int.megaBytes get() = times(1_000_000)
val examplePaginatedDownloader = paginatedDownloader {
versionPager { packageName, versionHint ->
@ -81,7 +81,7 @@ val examplePaginatedDownloader = paginatedDownloader {
download {
for (i in 0..5) {
reportProgress(i.megabytes , 5.megabytes)
reportProgress(i.megaBytes , 5.megaBytes)
delay(1000L)
}