🎨 Improve the code structure

server/middlewares/account.go

@@ -0,0 +1,16 @@
+package middlewares
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/gin-gonic/gin"
+)
+
+func CheckAccount(c *gin.Context) {
+	if model.AccountsCount() == 0 {
+		common.ErrorResp(c, fmt.Errorf("no accounts,please add one first"), 1001)
+		return
+	}
+	c.Next()
+}

server/middlewares/auth.go

@@ -0,0 +1,28 @@
+package middlewares
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+func Auth(c *gin.Context) {
+	token := c.GetHeader("Authorization")
+	password, err := model.GetSettingByKey("password")
+	if err != nil {
+		if err == gorm.ErrRecordNotFound {
+			common.ErrorResp(c, fmt.Errorf("password not set"), 400)
+			return
+		}
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	if token != utils.GetMD5Encode(password.Value) {
+		common.ErrorResp(c, fmt.Errorf("wrong password"), 401)
+		return
+	}
+	c.Next()
+}

server/middlewares/down.go

@@ -0,0 +1,20 @@
+package middlewares
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+)
+
+func DownCheck(c *gin.Context) {
+	rawPath := c.Param("path")
+	rawPath = utils.ParsePath(rawPath)
+	pw := c.Query("pw")
+	if !common.CheckDownLink(utils.Dir(rawPath), pw, utils.Base(rawPath)) {
+		common.ErrorResp(c, fmt.Errorf("wrong password"), 401)
+		c.Abort()
+		return
+	}
+	c.Next()
+}

server/middlewares/path.go

@@ -0,0 +1,35 @@
+package middlewares
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+)
+
+func PathCheck(c *gin.Context) {
+	var req common.PathReq
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	req.Path = utils.ParsePath(req.Path)
+	c.Set("req",req)
+	meta, err := model.GetMetaByPath(req.Path)
+	if err == nil {
+		if meta.Password != "" && meta.Password != req.Password {
+			common.ErrorResp(c, fmt.Errorf("wrong password"), 401)
+			c.Abort()
+			return
+		}
+	} else if conf.CheckParent {
+		if !common.CheckParent(utils.Dir(req.Path), req.Password) {
+			common.ErrorResp(c, fmt.Errorf("wrong password"), 401)
+			c.Abort()
+			return
+		}
+	}
+	c.Next()
+}