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
@@ -1,25 +1,28 @@
 package handler

 import (
+	"database/sql"
 	"encoding/base64"
 	"encoding/json"
 	"fmt"
 	"honoka-chan/database"
 	"honoka-chan/encrypt"
+	"honoka-chan/model"
 	"honoka-chan/utils"
 	"net/http"
 	"time"

 	"github.com/gin-gonic/gin"
+	_ "github.com/mattn/go-sqlite3"
 )

-type DownloadResp struct {
-	ResponseData []interface{} `json:"response_data"`
-	ReleaseInfo  []interface{} `json:"release_info"`
-	StatusCode   int           `json:"status_code"`
-}
+func DownloadAdditionalHandler(ctx *gin.Context) {
+	db, err := sql.Open("sqlite3", "assets/main.db")
+	if err != nil {
+		panic(err)
+	}
+	defer db.Close()

-func DownloadBatchHandler(ctx *gin.Context) {
 	reqTime := time.Now().Unix()

 	authorizeStr := ctx.Request.Header["Authorize"]
@@ -51,8 +54,121 @@ func DownloadBatchHandler(ctx *gin.Context) {
 	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)

-	batchResp := DownloadResp{
-		ResponseData: []interface{}{},
+	downloadReq := model.AdditionalReq{}
+	if err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &downloadReq); err != nil {
+		panic(err)
+	}
+	pkgList := []model.AdditionalResult{}
+	if CdnUrl != "" {
+		pkgType, pkgId := downloadReq.PackageType, downloadReq.PackageID
+		stmt, err := db.Prepare("SELECT pkg_order,pkg_size FROM download_db WHERE pkg_type = ? AND pkg_id = ? ORDER BY pkg_order ASC")
+		if err != nil {
+			panic(err)
+		}
+		rows, err := stmt.Query(pkgType, pkgId)
+		if err != nil {
+			panic(err)
+		}
+		for rows.Next() {
+			var pkgOrder, pkgSize int
+			err = rows.Scan(&pkgOrder, &pkgSize)
+			if err != nil {
+				panic(err)
+			}
+			pkgList = append(pkgList, model.AdditionalResult{
+				Size: pkgSize,
+				URL:  fmt.Sprintf("%s/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder),
+			})
+		}
+	}
+
+	addResp := model.AdditionalResp{
+		ResponseData: pkgList,
+		ReleaseInfo:  []interface{}{},
+		StatusCode:   200,
+	}
+	resp, err := json.Marshal(addResp)
+	if err != nil {
+		panic(err)
+	}
+	// fmt.Println(string(resp))
+	xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
+	xms64 := base64.RawStdEncoding.EncodeToString(xms)
+
+	ctx.Header("user_id", userId[0])
+	ctx.Header("authorize", newAuthorizeStr)
+	ctx.Header("X-Message-Sign", xms64)
+	ctx.String(http.StatusOK, string(resp))
+}
+
+func DownloadBatchHandler(ctx *gin.Context) {
+	db, err := sql.Open("sqlite3", "assets/main.db")
+	if err != nil {
+		panic(err)
+	}
+	defer db.Close()
+
+	reqTime := time.Now().Unix()
+
+	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)
+
+	downloadReq := model.BatchReq{}
+	if err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &downloadReq); err != nil {
+		panic(err)
+	}
+	pkgList := []model.BatchResult{}
+	if downloadReq.ClientVersion == PackageVersion && CdnUrl != "" {
+		pkgType := downloadReq.PackageType
+		stmt, err := db.Prepare("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = ? ORDER BY pkg_id ASC, pkg_order ASC")
+		if err != nil {
+			panic(err)
+		}
+		rows, err := stmt.Query(pkgType)
+		if err != nil {
+			panic(err)
+		}
+		for rows.Next() {
+			var pkgId, pkgOrder, pkgSize int
+			err = rows.Scan(&pkgId, &pkgOrder, &pkgSize)
+			if err != nil {
+				panic(err)
+			}
+			pkgList = append(pkgList, model.BatchResult{
+				Size: pkgSize,
+				URL:  fmt.Sprintf("%s/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder),
+			})
+		}
+	}
+
+	batchResp := model.BatchResp{
+		ResponseData: pkgList,
 		ReleaseInfo:  []interface{}{},
 		StatusCode:   200,
 	}
@@ -60,6 +176,89 @@ func DownloadBatchHandler(ctx *gin.Context) {
 	if err != nil {
 		panic(err)
 	}
+	// fmt.Println(string(resp))
+	xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
+	xms64 := base64.RawStdEncoding.EncodeToString(xms)
+
+	ctx.Header("user_id", userId[0])
+	ctx.Header("authorize", newAuthorizeStr)
+	ctx.Header("X-Message-Sign", xms64)
+	ctx.String(http.StatusOK, string(resp))
+}
+
+func DownloadUpdateHandler(ctx *gin.Context) {
+	db, err := sql.Open("sqlite3", "assets/main.db")
+	if err != nil {
+		panic(err)
+	}
+	defer db.Close()
+
+	reqTime := time.Now().Unix()
+
+	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)
+
+	downloadReq := model.UpdateReq{}
+	if err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &downloadReq); err != nil {
+		panic(err)
+	}
+	pkgList := []model.UpdateResult{}
+	if downloadReq.ExternalVersion != PackageVersion && CdnUrl != "" {
+		pkgType := 99
+		rows, err := db.Query("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = 99 ORDER BY pkg_id ASC, pkg_order ASC")
+		if err != nil {
+			panic(err)
+		}
+		for rows.Next() {
+			var pkgId, pkgOrder, pkgSize int
+			err = rows.Scan(&pkgId, &pkgOrder, &pkgSize)
+			if err != nil {
+				panic(err)
+			}
+			pkgList = append(pkgList, model.UpdateResult{
+				Size:    pkgSize,
+				URL:     fmt.Sprintf("%s/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder),
+				Version: PackageVersion,
+			})
+		}
+	}
+
+	updateResp := model.UpdateResp{
+		ResponseData: pkgList,
+		ReleaseInfo:  []interface{}{},
+		StatusCode:   200,
+	}
+	resp, err := json.Marshal(updateResp)
+	if err != nil {
+		panic(err)
+	}
+	// fmt.Println(string(resp))
 	xms := encrypt.RSA_Sign_SHA1(resp, "privatekey.pem")
 	xms64 := base64.RawStdEncoding.EncodeToString(xms)

@@ -101,7 +300,7 @@ func DownloadEventHandler(ctx *gin.Context) {
 	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)

-	eventResp := DownloadResp{
+	eventResp := model.EventResp{
 		ResponseData: []interface{}{},
 		ReleaseInfo:  []interface{}{},
 		StatusCode:   200,