@@ -4,7 +4,7 @@ import (
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"honoka-chan/database"
|
||||
"honoka-chan/db"
|
||||
"honoka-chan/encrypt"
|
||||
"honoka-chan/utils"
|
||||
"time"
|
||||
@@ -37,7 +37,7 @@ func AuthKey(ctx *gin.Context) {
|
||||
"server_token": serverToken,
|
||||
})
|
||||
CheckErr(err)
|
||||
err = database.LevelDb.Put([]byte(authorizeToken), authJson)
|
||||
err = db.DB.Set([]byte(authorizeToken), authJson)
|
||||
CheckErr(err)
|
||||
|
||||
nonce := ctx.GetInt("nonce")
|
||||
|
||||
@@ -2,7 +2,7 @@ package middleware
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"honoka-chan/database"
|
||||
"honoka-chan/db"
|
||||
"net/http"
|
||||
"net/url"
|
||||
"strconv"
|
||||
@@ -54,14 +54,14 @@ func Common(ctx *gin.Context) {
|
||||
}
|
||||
ctx.Set("userid", userId)
|
||||
|
||||
rToken, err := database.LevelDb.Get([]byte(userId))
|
||||
rToken, err := db.DB.Get([]byte(userId))
|
||||
CheckErr(err)
|
||||
if token != string(rToken) {
|
||||
ctx.String(http.StatusForbidden, ErrorMsg)
|
||||
ctx.Abort()
|
||||
}
|
||||
|
||||
if !database.MatchTokenUid(token, userId) {
|
||||
if !db.MatchTokenUid(token, userId) {
|
||||
ctx.String(http.StatusForbidden, ErrorMsg)
|
||||
ctx.Abort()
|
||||
}
|
||||
|
||||
+2
-2
@@ -2,7 +2,7 @@ package middleware
|
||||
|
||||
import (
|
||||
"encoding/base64"
|
||||
"honoka-chan/database"
|
||||
"honoka-chan/db"
|
||||
"honoka-chan/encrypt"
|
||||
"honoka-chan/utils"
|
||||
|
||||
@@ -11,7 +11,7 @@ import (
|
||||
)
|
||||
|
||||
func Login(ctx *gin.Context) {
|
||||
authData, err := database.LevelDb.Get([]byte(ctx.GetString("token")))
|
||||
authData, err := db.DB.Get([]byte(ctx.GetString("token")))
|
||||
CheckErr(err)
|
||||
|
||||
clientToken, err := base64.StdEncoding.DecodeString(gjson.Get(string(authData), "client_token").String())
|
||||
|
||||
Reference in New Issue
Block a user