Code cleanup

Signed-off-by: Yuan Si <do4suki@gmail.com>

handler/unit.go

@@ -4,12 +4,9 @@ import (
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
-	"honoka-chan/config"
-	"honoka-chan/database"
 	"honoka-chan/encrypt"
 	"honoka-chan/model"
 	"honoka-chan/tools"
-	"honoka-chan/utils"
 	"net/http"
 	"strconv"
 	"time"
@@ -17,62 +14,26 @@ import (
 	"github.com/gin-gonic/gin"
 )
 
-type SetDisplayRankResp struct {
-	ResponseData []interface{} `json:"response_data"`
-	ReleaseInfo  []interface{} `json:"release_info"`
-	StatusCode   int           `json:"status_code"`
-}
-
-func SetDisplayRankHandler(ctx *gin.Context) {
-	reqTime := time.Now().Unix()
-
-	authorizeStr := ctx.Request.Header["Authorize"]
-	authToken, err := utils.GetAuthorizeToken(authorizeStr)
-	if err != nil {
-		ctx.String(http.StatusForbidden, ErrorMsg)
-		return
-	}
-	userId := ctx.Request.Header[http.CanonicalHeaderKey("User-ID")]
-	if len(userId) == 0 {
-		ctx.String(http.StatusForbidden, ErrorMsg)
-		return
-	}
-
-	if !database.MatchTokenUid(authToken, userId[0]) {
-		ctx.String(http.StatusForbidden, ErrorMsg)
-		return
-	}
-
-	nonce, err := utils.GetAuthorizeNonce(authorizeStr)
-	if err != nil {
-		fmt.Println(err)
-		ctx.String(http.StatusForbidden, ErrorMsg)
-		return
-	}
-	nonce++
-
-	respTime := time.Now().Unix()
-	newAuthorizeStr := fmt.Sprintf("consumerKey=lovelive_test&timeStamp=%d&version=1.1&token=%s&nonce=%d&user_id=%s&requestTimeStamp=%d", respTime, authToken, nonce, userId[0], reqTime)
-	// fmt.Println(newAuthorizeStr)
-
-	dispResp := SetDisplayRankResp{
+func SetDisplayRank(ctx *gin.Context) {
+	dispResp := model.SetDisplayRankResp{
 		ResponseData: []interface{}{},
 		ReleaseInfo:  []interface{}{},
 		StatusCode:   200,
 	}
 	resp, err := json.Marshal(dispResp)
 	CheckErr(err)
-	xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
-	xms64 := base64.RawStdEncoding.EncodeToString(xms)
 
-	ctx.Header("Server-Version", config.Conf.Server.VersionNumber)
-	ctx.Header("user_id", userId[0])
-	ctx.Header("authorize", newAuthorizeStr)
-	ctx.Header("X-Message-Sign", xms64)
+	nonce := ctx.GetInt("nonce")
+	nonce++
+
+	ctx.Header("user_id", ctx.GetString("userid"))
+	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(), ctx.GetString("token"), nonce, ctx.GetString("userid"), ctx.GetInt64("req_time")))
+	ctx.Header("X-Message-Sign", base64.StdEncoding.EncodeToString(encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")))
+
 	ctx.String(http.StatusOK, string(resp))
 }
 
-func SetDeckHandler(ctx *gin.Context) {
+func SetDeck(ctx *gin.Context) {
 	userId, err := strconv.Atoi(ctx.GetString("userid"))
 	CheckErr(err)
 
@@ -197,7 +158,7 @@ func SetDeckHandler(ctx *gin.Context) {
 		panic(err)
 	}
 
-	dispResp := SetDisplayRankResp{
+	dispResp := model.SetDeckResp{
 		ResponseData: []interface{}{},
 		ReleaseInfo:  []interface{}{},
 		StatusCode:   200,
@@ -215,7 +176,7 @@ func SetDeckHandler(ctx *gin.Context) {
 	ctx.String(http.StatusOK, string(resp))
 }
 
-func SetDeckNameHandler(ctx *gin.Context) {
+func SetDeckName(ctx *gin.Context) {
 	userId, err := strconv.Atoi(ctx.GetString("userid"))
 	CheckErr(err)
 
@@ -239,7 +200,7 @@ func SetDeckNameHandler(ctx *gin.Context) {
 	})
 	CheckErr(err)
 
-	dispResp := SetDisplayRankResp{
+	dispResp := model.SetDeckResp{
 		ResponseData: []interface{}{},
 		ReleaseInfo:  []interface{}{},
 		StatusCode:   200,