shkong/OpenShamrock
1
0
Fork 0
mirror of https://github.com/whitechi73/OpenShamrock.git synced 2024-08-14 13:12:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #303 from tobycroft/master

Browse Source 
GetFile的type新增gzip，会将数据流压缩后再b64降低带宽占用
This commit is contained in:
ikechan8370 2024-03-18 22:43:18 +08:00 committed by GitHub
commit 7782feb6ac
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
xposed/src/main/java/moe/fuqiuluo/shamrock/remote/action/handlers/GetFile.kt
View File

@ -7,10 +7,11 @@ import moe.fuqiuluo.shamrock.remote.service.data.OutResourceByBase64
import moe.fuqiuluo.shamrock.tools.EmptyJsonString
import moe.fuqiuluo.shamrock.utils.FileUtils
import moe.fuqiuluo.symbols.OneBotHandler
import java.io.ByteArrayOutputStream
import java.util.Base64
import java.util.zip.GZIPOutputStream
@OneBotHandler("get_file")
internal object GetFile : IActionHandler() {
@OneBotHandler("get_file") internal object GetFile : IActionHandler() {
override suspend fun internalHandle(session: ActionSession): String {
val file = session.getString("file")
.replace(regex = "[{}\\-]".toRegex(), replacement = "")
@ -33,7 +34,15 @@ internal object GetFile : IActionHandler() {
), echo
)
else -> error("only support base64", echo)
"gzip" -> ok(
OutResourceByBase64(
"/res/${targetFile.nameWithoutExtension}",
compressAndEncode(targetFile.readBytes()),
targetFile.nameWithoutExtension,
), echo
)
else -> error("file_type error", echo)
}
} else {
@ -41,5 +50,19 @@ internal object GetFile : IActionHandler() {
}
}
fun compressAndEncode(input: ByteArray): String {
// 压缩数据
val outputStream = ByteArrayOutputStream()
val gzip = GZIPOutputStream(outputStream)
gzip.write(input)
gzip.close()
val compressedBytes = outputStream.toByteArray()
// 编码为 Base64 字符串
return Base64.getEncoder()
.encodeToString(compressedBytes)
}
override val requiredParams: Array<String> = arrayOf("file")
}