diff --git a/app/build.gradle.kts b/app/build.gradle.kts index 790e8a7..9570d8a 100644 --- a/app/build.gradle.kts +++ b/app/build.gradle.kts @@ -14,8 +14,8 @@ android { applicationId = "com.donut.mixfile" minSdk = 24 targetSdk = 34 - versionCode = 21 - versionName = "1.1.2" + versionCode = 22 + versionName = "1.1.3" testInstrumentationRunner = "androidx.test.runner.AndroidJUnitRunner" vectorDrawables { diff --git a/app/src/main/java/com/donut/mixfile/server/uploaders/CustomUploader.kt b/app/src/main/java/com/donut/mixfile/server/uploaders/CustomUploader.kt index 4e3fe48..bdbc361 100644 --- a/app/src/main/java/com/donut/mixfile/server/uploaders/CustomUploader.kt +++ b/app/src/main/java/com/donut/mixfile/server/uploaders/CustomUploader.kt @@ -10,6 +10,8 @@ import io.ktor.client.request.url import io.ktor.client.statement.bodyAsText import io.ktor.client.statement.readBytes import io.ktor.http.isSuccess +import kotlinx.coroutines.Dispatchers +import kotlinx.coroutines.withContext var CUSTOM_UPLOAD_URL by cachedMutableOf("", "CUSTOM_UPLOAD_URL") @@ -20,6 +22,13 @@ object CustomUploader : Uploader("自定义") { override suspend fun genHead(): ByteArray { return uploadClient.get { url(CUSTOM_UPLOAD_URL) + }.also { + val referer = it.headers["referer"] + if (!referer.isNullOrEmpty()) { + withContext(Dispatchers.Main) { + CUSTOM_REFERER = referer + } + } }.readBytes() }