From 8f1aa59e6705bc8ce1854cf148792edd30cd3c08 Mon Sep 17 00:00:00 2001 From: Yuan Si Date: Mon, 10 Apr 2023 11:03:06 +0800 Subject: [PATCH] Implement download/getUrl Signed-off-by: Yuan Si --- config.yml | 2 +- handler/download.go | 71 ++++++++++++++++++++++++++++++++++++++++++--- main.go | 1 + model/download.go | 18 ++++++++++++ 4 files changed, 87 insertions(+), 5 deletions(-) diff --git a/config.yml b/config.yml index 867b9dc..5e33582 100644 --- a/config.yml +++ b/config.yml @@ -11,4 +11,4 @@ log: leveldb: data_path: ./data/honoka-chan.db cdn: - cdn_url: http://192.168.1.12:8081/static + cdn_url: http://192.168.1.9:8081/static diff --git a/handler/download.go b/handler/download.go index 5408f4a..05f3ed3 100644 --- a/handler/download.go +++ b/handler/download.go @@ -78,7 +78,7 @@ func DownloadAdditionalHandler(ctx *gin.Context) { } pkgList = append(pkgList, model.AdditionalResult{ Size: pkgSize, - URL: fmt.Sprintf("%s/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), + URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), }) } } @@ -154,7 +154,7 @@ func DownloadBatchHandler(ctx *gin.Context) { exPkgId = fmt.Sprintf("AND pkg_id NOT IN (%s)", exPkgId) } - fmt.Println(exPkgId) + // fmt.Println(exPkgId) stmt, err := db.Prepare("SELECT pkg_id,pkg_order,pkg_size FROM download_db WHERE pkg_type = ? " + exPkgId + " ORDER BY pkg_id ASC, pkg_order ASC") if err != nil { panic(err) @@ -171,7 +171,7 @@ func DownloadBatchHandler(ctx *gin.Context) { } pkgList = append(pkgList, model.BatchResult{ Size: pkgSize, - URL: fmt.Sprintf("%s/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), + URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), }) } } @@ -252,7 +252,7 @@ func DownloadUpdateHandler(ctx *gin.Context) { } pkgList = append(pkgList, model.UpdateResult{ Size: pkgSize, - URL: fmt.Sprintf("%s/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), + URL: fmt.Sprintf("%s/archives/%d_%d_%d.zip", CdnUrl, pkgType, pkgId, pkgOrder), Version: PackageVersion, }) } @@ -277,6 +277,69 @@ func DownloadUpdateHandler(ctx *gin.Context) { ctx.String(http.StatusOK, string(resp)) } +func DownloadUrlHandler(ctx *gin.Context) { + // Extract SQL: SELECT CAST(pkg_type AS TEXT) || '_' || CAST(pkg_id AS TEXT) || '_' || CAST(pkg_order AS TEXT) || '.zip' AS zip_name FROM download_db ORDER BY pkg_type ASC,pkg_id ASC, pkg_order ASC; + // Extract Cmd: cat list.txt | while read line; do; unzip -o $line; done + 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) + + downloadReq := model.UrlReq{} + if err := json.Unmarshal([]byte(ctx.PostForm("request_data")), &downloadReq); err != nil { + panic(err) + } + urlList := []string{} + for _, v := range downloadReq.PathList { + urlList = append(urlList, fmt.Sprintf("%s/extracted/%s", CdnUrl, strings.ReplaceAll(v, "\\", ""))) + } + urlResp := model.UrlResp{ + ResponseData: model.UrlResult{ + UrlList: urlList, + }, + ReleaseInfo: []interface{}{}, + StatusCode: 200, + } + resp, err := json.Marshal(urlResp) + 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 DownloadEventHandler(ctx *gin.Context) { reqTime := time.Now().Unix() diff --git a/main.go b/main.go index 288bc1b..3ebd096 100644 --- a/main.go +++ b/main.go @@ -56,6 +56,7 @@ func main() { m.POST("/tos/tosCheck", handler.TosCheckHandler) m.POST("/download/additional", handler.DownloadAdditionalHandler) m.POST("/download/batch", handler.DownloadBatchHandler) + m.POST("/download/getUrl", handler.DownloadUrlHandler) m.POST("/download/update", handler.DownloadUpdateHandler) m.POST("/download/event", handler.DownloadEventHandler) m.POST("/lbonus/execute", handler.LBonusExecuteHandler) diff --git a/model/download.go b/model/download.go index dee06dd..7c5bd8d 100644 --- a/model/download.go +++ b/model/download.go @@ -64,6 +64,24 @@ type UpdateResp struct { StatusCode int `json:"status_code"` } +type UrlReq struct { + Module string `json:"module"` + Os string `json:"os"` + Mgd int `json:"mgd"` + PathList []string `json:"path_list"` + Action string `json:"action"` +} + +type UrlResult struct { + UrlList []string `json:"url_list"` +} + +type UrlResp struct { + ResponseData UrlResult `json:"response_data"` + ReleaseInfo []interface{} `json:"release_info"` + StatusCode int `json:"status_code"` +} + type EventResp struct { ResponseData []interface{} `json:"response_data"` ReleaseInfo []interface{} `json:"release_info"`