ProjectHentai/alist
6
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🍃 Optimization download config

Browse Source
This commit is contained in:
微凉
2021-08-21 17:10:56 +08:00
parent f927c974cf
commit eefe53ff01
4 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
server/controllers/down.go
View File

@ -2,6 +2,7 @@ package controllers
import (
"github.com/Xhofe/alist/alidrive"
"github.com/Xhofe/alist/conf"
"github.com/Xhofe/alist/server/models"
"github.com/Xhofe/alist/utils"
"github.com/gin-gonic/gin"
@ -14,8 +15,12 @@ type DownReq struct {
Password string `form:"pw"`
}
// handle download request
// Down handle download request
func Down(c *gin.Context) {
if !conf.Conf.Server.Download {
c.JSON(200,MetaResponse(403,"not allowed download and preview"))
return
}
filePath := c.Param("path")[1:]
var down DownReq
if err := c.ShouldBindQuery(&down); err != nil {

5
server/controllers/offie_preview.go
View File

@ -2,6 +2,7 @@ package controllers
import (
"github.com/Xhofe/alist/alidrive"
"github.com/Xhofe/alist/conf"
"github.com/Xhofe/alist/utils"
"github.com/gin-gonic/gin"
log "github.com/sirupsen/logrus"
@ -13,6 +14,10 @@ type OfficePreviewReq struct {
// handle office_preview request
func OfficePreview(c *gin.Context) {
if !conf.Conf.Server.Download {
c.JSON(200,MetaResponse(403,"not allowed download and preview"))
return
}
drive := utils.GetDriveByName(c.Param("drive"))
if drive == nil {
c.JSON(200, MetaResponse(400, "drive isn't exist."))

9
server/controllers/video_preview.go
View File

@ -2,6 +2,7 @@ package controllers
import (
"github.com/Xhofe/alist/alidrive"
"github.com/Xhofe/alist/conf"
"github.com/Xhofe/alist/utils"
"github.com/gin-gonic/gin"
log "github.com/sirupsen/logrus"
@ -13,6 +14,10 @@ type VideoPreviewReq struct {
// VideoPreview handle video_preview request
func VideoPreview(c *gin.Context) {
if !conf.Conf.Server.Download {
c.JSON(200,MetaResponse(403,"not allowed download and preview"))
return
}
drive := utils.GetDriveByName(c.Param("drive"))
if drive == nil {
c.JSON(200, MetaResponse(400, "drive isn't exist."))
@ -33,6 +38,10 @@ func VideoPreview(c *gin.Context) {
}
func VideoPreviewPlayInfo(c *gin.Context) {
if !conf.Conf.Server.Download {
c.JSON(200,MetaResponse(403,"not allowed download and preview"))
return
}
drive := utils.GetDriveByName(c.Param("drive"))
if drive == nil {
c.JSON(200, MetaResponse(400, "drive isn't exist."))

10
server/router.go
View File

@ -8,7 +8,7 @@ import (
log "github.com/sirupsen/logrus"
)
// init router
// InitRouter init router
func InitRouter(engine *gin.Engine) {
log.Infof("初始化路由...")
engine.Use(CorsHandler())
@ -16,11 +16,11 @@ func InitRouter(engine *gin.Engine) {
engine.NoRoute(func(c *gin.Context) {
c.File(conf.Conf.Server.Static + "/index.html")
})
InitApiRouter(engine, conf.Conf.Server.Download)
InitApiRouter(engine)
}
// init api router
func InitApiRouter(engine *gin.Engine, download bool) {
// InitApiRouter init api router
func InitApiRouter(engine *gin.Engine) {
apiV2 := engine.Group("/api")
{
apiV2.GET("/info", controllers.Info)
@ -29,9 +29,7 @@ func InitApiRouter(engine *gin.Engine, download bool) {
apiV2.POST("/local_search", controllers.LocalSearch)
apiV2.POST("/global_search", controllers.GlobalSearch)
apiV2.POST("/rebuild", controllers.RebuildTree)
}
if download {
apiV2.POST("/office_preview/:drive", controllers.OfficePreview)
apiV2.POST("/video_preview/:drive", controllers.VideoPreview)
apiV2.POST("/video_preview_play_info/:drive", controllers.VideoPreviewPlayInfo)