mirror of
https://github.com/whitechi73/OpenShamrock.git
synced 2024-08-14 13:12:17 +08:00
Shamrock
: fix #159
This commit is contained in:
parent
3f9613c43c
commit
538db69754
@ -5,6 +5,7 @@ import moe.fuqiuluo.shamrock.helper.ErrorTokenException
|
||||
import io.ktor.server.application.createApplicationPlugin
|
||||
import moe.fuqiuluo.shamrock.remote.service.config.ShamrockConfig
|
||||
import moe.fuqiuluo.shamrock.tools.fetchOrNull
|
||||
import java.net.URLDecoder
|
||||
|
||||
private suspend fun ApplicationCall.checkToken() {
|
||||
val token = ShamrockConfig.getToken()
|
||||
@ -12,8 +13,12 @@ private suspend fun ApplicationCall.checkToken() {
|
||||
return
|
||||
}
|
||||
var accessToken = request.headers["Authorization"]
|
||||
?: fetchOrNull("ticket")
|
||||
?: fetchOrNull("access_token")
|
||||
?: fetchOrNull("ticket")?.let {
|
||||
URLDecoder.decode(it, "UTF-8")
|
||||
}
|
||||
?: fetchOrNull("access_token")?.let {
|
||||
URLDecoder.decode(it, "UTF-8")
|
||||
}
|
||||
?: throw ErrorTokenException
|
||||
if (accessToken.startsWith("Bearer ", ignoreCase = true)) {
|
||||
accessToken = accessToken.substring(7)
|
||||
|
Loading…
x
Reference in New Issue
Block a user