91 lines
2.1 KiB
Go
91 lines
2.1 KiB
Go
package middleware
|
|
|
|
import (
|
|
"fmt"
|
|
"honoka-chan/internal/session"
|
|
honokautils "honoka-chan/internal/utils"
|
|
"net/http"
|
|
"net/url"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func Common(ctx *gin.Context) {
|
|
reqData := ""
|
|
if form, err := ctx.MultipartForm(); err == nil {
|
|
if v, ok := form.Value["request_data"]; ok && len(v) > 0 {
|
|
reqData = v[0]
|
|
}
|
|
}
|
|
if reqData == "" {
|
|
reqData = ctx.PostForm("request_data")
|
|
}
|
|
ctx.Set("request_data", reqData)
|
|
|
|
uid := ctx.GetHeader("User-ID")
|
|
|
|
ss := session.Attach(ctx)
|
|
ctx.Set("session", ss)
|
|
defer ss.FinalizeOrRollback()
|
|
|
|
if ctx.IsAborted() {
|
|
return
|
|
}
|
|
|
|
authorize := ctx.GetHeader("Authorize")
|
|
params, err := url.ParseQuery(authorize)
|
|
if err != nil {
|
|
ss.AbortWithStatus(http.StatusNotFound, honokautils.NewNotFoundContent(ctx.Request.URL.Path))
|
|
return
|
|
}
|
|
|
|
nonce, _ := strconv.Atoi(params.Get("nonce"))
|
|
nonce++
|
|
ctx.Set("nonce", nonce)
|
|
|
|
token := params.Get("token")
|
|
ctx.Set("token", token)
|
|
|
|
if !honokautils.IsMainLoginEndpoint(ctx.Request.URL.Path) {
|
|
userID, err := strconv.Atoi(uid)
|
|
if err != nil || userID <= 0 {
|
|
ss.AbortWithStatus(http.StatusNotFound, honokautils.NewNotFoundContent(ctx.Request.URL.Path))
|
|
return
|
|
}
|
|
|
|
valid, err := ss.IsAuthorizeTokenForUser(token, userID)
|
|
if ss.CheckErr(err) {
|
|
return
|
|
}
|
|
if !valid {
|
|
ss.AbortWithStatus(http.StatusNotFound, honokautils.NewNotFoundContent(ctx.Request.URL.Path))
|
|
return
|
|
}
|
|
|
|
ctx.Set("userid", uid)
|
|
ss.LoadUser(uid)
|
|
if ss.Done() {
|
|
return
|
|
}
|
|
|
|
if ss.UserPref.ForceRelogin {
|
|
ss.AbortWithStatus(http.StatusNotFound, honokautils.NewNotFoundContent(ctx.Request.URL.Path))
|
|
return
|
|
}
|
|
}
|
|
|
|
ctx.Header("user_id", uid)
|
|
ctx.Header("authorize", fmt.Sprintf("consumerKey=lovelive_test&timeStamp=%d&version=1.1&token=%s&nonce=%d&user_id=%s&requestTimeStamp=%d", time.Now().Unix(), token, nonce, uid, time.Now().Unix()))
|
|
|
|
ctx.Header("Content-Type", "application/json; charset=utf-8")
|
|
ctx.Header("X-Powered-By", "KLab Native APP Platform")
|
|
ctx.Header("server_version", "20120129")
|
|
ctx.Header("Server-Version", "97.4.6")
|
|
ctx.Header("version_up", "0")
|
|
ctx.Header("status_code", strconv.Itoa(http.StatusOK))
|
|
|
|
ctx.Next()
|
|
}
|