68 lines
1.9 KiB
Go
68 lines
1.9 KiB
Go
package download
|
|
|
|
import (
|
|
"fmt"
|
|
"honoka-chan/config"
|
|
"honoka-chan/internal/middleware"
|
|
"honoka-chan/internal/router"
|
|
downloadschema "honoka-chan/internal/schema/download"
|
|
"honoka-chan/internal/session"
|
|
honokautils "honoka-chan/internal/utils"
|
|
"net/http"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"xorm.io/builder"
|
|
)
|
|
|
|
func batch(ctx *gin.Context) {
|
|
ss := session.Get(ctx)
|
|
defer ss.FinalizeOrRollback()
|
|
|
|
downloadReq := downloadschema.BatchReq{}
|
|
err := honokautils.ParseRequestData(ctx, &downloadReq)
|
|
if ss.CheckErr(err) {
|
|
return
|
|
}
|
|
|
|
pkgList := []downloadschema.BatchData{}
|
|
if downloadReq.ClientVersion == config.PackageVersion {
|
|
pkgType := downloadReq.PackageType
|
|
|
|
// 首次下载时客户端发送的 downloadReq.PackageType = 0 但是实际下载的是 [0,4]
|
|
// 此时直接使用 downloadReq.PackageType 会导致下载数据量计算错误
|
|
// 游戏会自动将 [0,4] 的包大小合并计算
|
|
// downloadReq.PackageType = 0 的包已经在 /download/additional 中获取
|
|
// 所以这里只需要返回 downloadReq.PackageType = 4 的包即可
|
|
if downloadReq.PackageType == 0 {
|
|
pkgType = 4
|
|
}
|
|
|
|
var pkgInfo []PkgInfo
|
|
err := ss.MainEng.Table("download_m").
|
|
Where(builder.NotIn("pkg_id", downloadReq.ExcludedPackageIds)).
|
|
Where("pkg_type = ? AND pkg_os = ?", pkgType, downloadReq.Os).
|
|
OrderBy("pkg_id ASC, pkg_order ASC").Find(&pkgInfo)
|
|
if ss.CheckErr(err) {
|
|
return
|
|
}
|
|
|
|
for _, pkg := range pkgInfo {
|
|
pkgList = append(pkgList, downloadschema.BatchData{
|
|
Size: pkg.Size,
|
|
URL: fmt.Sprintf("%s/%s/archives/%d_%d_%d.zip",
|
|
config.Conf.Settings.CdnServer, downloadReq.Os, pkg.PkgType, pkg.PkgID, pkg.Order),
|
|
})
|
|
}
|
|
}
|
|
|
|
ss.Respond(downloadschema.BatchResp{
|
|
ResponseData: pkgList,
|
|
ReleaseInfo: []any{},
|
|
StatusCode: http.StatusOK,
|
|
})
|
|
}
|
|
|
|
func init() {
|
|
router.AddHandler("main.php", "POST", "/download/batch", middleware.Common, batch)
|
|
}
|