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 io.ktor.server.application.createApplicationPlugin
|
||||||
import moe.fuqiuluo.shamrock.remote.service.config.ShamrockConfig
|
import moe.fuqiuluo.shamrock.remote.service.config.ShamrockConfig
|
||||||
import moe.fuqiuluo.shamrock.tools.fetchOrNull
|
import moe.fuqiuluo.shamrock.tools.fetchOrNull
|
||||||
|
import java.net.URLDecoder
|
||||||
|
|
||||||
private suspend fun ApplicationCall.checkToken() {
|
private suspend fun ApplicationCall.checkToken() {
|
||||||
val token = ShamrockConfig.getToken()
|
val token = ShamrockConfig.getToken()
|
||||||
@ -12,8 +13,12 @@ private suspend fun ApplicationCall.checkToken() {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
var accessToken = request.headers["Authorization"]
|
var accessToken = request.headers["Authorization"]
|
||||||
?: fetchOrNull("ticket")
|
?: fetchOrNull("ticket")?.let {
|
||||||
?: fetchOrNull("access_token")
|
URLDecoder.decode(it, "UTF-8")
|
||||||
|
}
|
||||||
|
?: fetchOrNull("access_token")?.let {
|
||||||
|
URLDecoder.decode(it, "UTF-8")
|
||||||
|
}
|
||||||
?: throw ErrorTokenException
|
?: throw ErrorTokenException
|
||||||
if (accessToken.startsWith("Bearer ", ignoreCase = true)) {
|
if (accessToken.startsWith("Bearer ", ignoreCase = true)) {
|
||||||
accessToken = accessToken.substring(7)
|
accessToken = accessToken.substring(7)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user