Support downloading resources from CDN server

Signed-off-by: Yuan Si <do4suki@gmail.com>
2023-04-10 05:35:13 +08:00
parent 8a69a9391b
commit 0fcb44dbd4
10 changed files with 464 additions and 10 deletions
@@ -0,0 +1,98 @@
+package handler
+
+import (
+	"encoding/base64"
+	"encoding/json"
+	"fmt"
+	"honoka-chan/config"
+	"honoka-chan/database"
+	"honoka-chan/encrypt"
+	"honoka-chan/utils"
+	"net/http"
+	"time"
+
+	"github.com/gin-gonic/gin"
+)
+
+type MultiUnitStartUpResp struct {
+	ResponseData MultiUnitStartUpData `json:"response_data"`
+	ReleaseInfo  []interface{}        `json:"release_info"`
+	StatusCode   int                  `json:"status_code"`
+}
+
+type MultiUnitStartUpData struct {
+	MultiUnitScenarioID int   `json:"multi_unit_scenario_id"`
+	ScenarioAdjustment  int   `json:"scenario_adjustment"`
+	ServerTimestamp     int64 `json:"server_timestamp"`
+}
+
+type MultiUnitStartUpReq struct {
+	Module              string `json:"module"`
+	Action              string `json:"action"`
+	TimeStamp           int    `json:"timeStamp"`
+	Mgd                 int    `json:"mgd"`
+	MultiUnitScenarioID int    `json:"multi_unit_scenario_id"`
+	CommandNum          string `json:"commandNum"`
+}
+
+func MultiUnitStartUpHandler(ctx *gin.Context) {
+	reqTime := time.Now().Unix()
+
+	startReq := MultiUnitStartUpReq{}
+	err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &startReq)
+	if err != nil {
+		panic(err)
+	}
+
+	authorizeStr := ctx.Request.Header["Authorize"]
+	authToken, err := utils.GetAuthorizeToken(authorizeStr)
+	if err != nil {
+		ctx.String(http.StatusForbidden, "Fuck you!")
+		return
+	}
+	userId := ctx.Request.Header[http.CanonicalHeaderKey("User-ID")]
+	if len(userId) == 0 {
+		ctx.String(http.StatusForbidden, "Fuck you!")
+		return
+	}
+
+	if !database.MatchTokenUid(authToken, userId[0]) {
+		ctx.String(http.StatusForbidden, "Fuck you!")
+		return
+	}
+
+	nonce, err := utils.GetAuthorizeNonce(authorizeStr)
+	if err != nil {
+		fmt.Println(err)
+		ctx.String(http.StatusForbidden, "Fuck you!")
+		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)
+
+	startResp := MultiUnitStartUpResp{
+		ResponseData: MultiUnitStartUpData{
+			MultiUnitScenarioID: startReq.MultiUnitScenarioID,
+			ScenarioAdjustment:  50,
+			ServerTimestamp:     time.Now().Unix(),
+		},
+		ReleaseInfo: []interface{}{},
+		StatusCode:  200,
+	}
+	resp, err := json.Marshal(startResp)
+	if err != nil {
+		panic(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)
+
+	ctx.String(http.StatusOK, string(resp))
+}