🎨 Improve the code structure

server/controllers/account.go

@@ -0,0 +1,95 @@
+package controllers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/drivers/base"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+	"strconv"
+	"time"
+)
+
+func GetAccounts(c *gin.Context) {
+	accounts, err := model.GetAccounts()
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	common.SuccessResp(c, accounts)
+}
+
+func CreateAccount(c *gin.Context) {
+	var req model.Account
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	driver, ok := base.GetDriver(req.Type)
+	if !ok {
+		common.ErrorResp(c, fmt.Errorf("no [%s] driver", req.Type), 400)
+		return
+	}
+	now := time.Now()
+	req.UpdatedAt = &now
+	if err := model.CreateAccount(&req); err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		log.Debugf("new account: %+v", req)
+		err = driver.Save(&req, nil)
+		if err != nil {
+			common.ErrorResp(c, err, 500)
+			return
+		}
+		common.SuccessResp(c)
+	}
+}
+
+func SaveAccount(c *gin.Context) {
+	var req model.Account
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	driver, ok := base.GetDriver(req.Type)
+	if !ok {
+		common.ErrorResp(c, fmt.Errorf("no [%s] driver", req.Type), 400)
+		return
+	}
+	old, err := model.GetAccountById(req.ID)
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	now := time.Now()
+	req.UpdatedAt = &now
+	if old.Name != req.Name {
+		model.DeleteAccountFromMap(old.Name)
+	}
+	if err := model.SaveAccount(&req); err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		log.Debugf("save account: %+v", req)
+		err = driver.Save(&req, old)
+		if err != nil {
+			common.ErrorResp(c, err, 500)
+			return
+		}
+		common.SuccessResp(c)
+	}
+}
+
+func DeleteAccount(c *gin.Context) {
+	idStr := c.Query("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	if err := model.DeleteAccount(uint(id)); err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	common.SuccessResp(c)
+}

server/controllers/cache.go

@@ -0,0 +1,16 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/gin-gonic/gin"
+)
+
+func ClearCache(c *gin.Context) {
+	err := conf.Cache.Clear(conf.Ctx)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		common.SuccessResp(c)
+	}
+}

server/controllers/down.go

@@ -0,0 +1,110 @@
+package controllers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/conf"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+	"github.com/go-resty/resty/v2"
+	log "github.com/sirupsen/logrus"
+	"net/http/httputil"
+	"net/url"
+	"path/filepath"
+	"strings"
+)
+
+func Down(c *gin.Context) {
+	rawPath := c.Param("path")
+	rawPath = utils.ParsePath(rawPath)
+	log.Debugf("down: %s", rawPath)
+	account, path, driver, err := common.ParsePath(rawPath)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	if driver.Config().OnlyProxy {
+		Proxy(c)
+		return
+	}
+	link, err := driver.Link(path, account)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	c.Redirect(302, link)
+	return
+}
+
+func Proxy(c *gin.Context) {
+	rawPath := c.Param("path")
+	rawPath = utils.ParsePath(rawPath)
+	log.Debugf("proxy: %s", rawPath)
+	account, path, driver, err := common.ParsePath(rawPath)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	if !account.Proxy && utils.GetFileType(filepath.Ext(rawPath)) != conf.TEXT {
+		common.ErrorResp(c, fmt.Errorf("[%s] not allowed proxy", account.Name), 403)
+		return
+	}
+	link, err := driver.Link(path, account)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	if account.Type == "Native" {
+		c.File(link)
+		return
+	} else {
+		if utils.GetFileType(filepath.Ext(rawPath)) == conf.TEXT {
+			Text(c, link)
+			return
+		}
+		driver.Proxy(c, account)
+		r := c.Request
+		w := c.Writer
+		target, err := url.Parse(link)
+		if err != nil {
+			common.ErrorResp(c, err, 500)
+			return
+		}
+		protocol := "http://"
+		if strings.HasPrefix(link, "https://") {
+			protocol = "https://"
+		}
+		targetHost, err := url.Parse(fmt.Sprintf("%s%s", protocol, target.Host))
+		proxy := httputil.NewSingleHostReverseProxy(targetHost)
+		r.URL = target
+		r.Host = target.Host
+		proxy.ServeHTTP(w, r)
+	}
+}
+
+var client *resty.Client
+
+func init() {
+	client = resty.New()
+	client.SetRetryCount(3)
+}
+
+func Text(c *gin.Context, link string) {
+	res, err := client.R().Get(link)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	text := res.String()
+	t := utils.GetStrCoding(res.Body())
+	log.Debugf("text type: %s", t)
+	if t != utils.UTF8 {
+		body, err := utils.GbkToUtf8(res.Body())
+		if err != nil {
+			common.ErrorResp(c, err, 500)
+			return
+		}
+		text = string(body)
+	}
+	c.String(200, text)
+}

server/controllers/driver.go

@@ -0,0 +1,11 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/drivers/base"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/gin-gonic/gin"
+)
+
+func GetDrivers(c *gin.Context) {
+	common.SuccessResp(c, base.GetDrivers())
+}

server/controllers/meta.go

@@ -0,0 +1,61 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+	"strconv"
+)
+
+func GetMetas(c *gin.Context)  {
+	metas,err := model.GetMetas()
+	if err != nil {
+		common.ErrorResp(c,err,500)
+		return
+	}
+	common.SuccessResp(c, metas)
+}
+
+func CreateMeta(c *gin.Context)  {
+	var req model.Meta
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	req.Path = utils.ParsePath(req.Path)
+	if err := model.CreateMeta(req); err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		common.SuccessResp(c)
+	}
+}
+
+func SaveMeta(c *gin.Context)  {
+	var req model.Meta
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	req.Path = utils.ParsePath(req.Path)
+	if err := model.SaveMeta(req); err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		common.SuccessResp(c)
+	}
+}
+
+func DeleteMeta(c *gin.Context) {
+	idStr := c.Query("id")
+	id, err := strconv.Atoi(idStr)
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	//path = utils.ParsePath(path)
+	if err := model.DeleteMeta(uint(id)); err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	common.SuccessResp(c)
+}

server/controllers/path.go

@@ -0,0 +1,114 @@
+package controllers
+
+import (
+	"fmt"
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/Xhofe/alist/utils"
+	"github.com/gin-gonic/gin"
+	log "github.com/sirupsen/logrus"
+	"strings"
+)
+
+func Path(c *gin.Context) {
+	reqV,_ := c.Get("req")
+	req := reqV.(common.PathReq)
+	if model.AccountsCount() > 1 && req.Path == "/" {
+		files, err := model.GetAccountFiles()
+		if err != nil {
+			common.ErrorResp(c, err, 500)
+			return
+		}
+		c.JSON(200, common.Resp{
+			Code:    200,
+			Message: "folder",
+			Data:    files,
+		})
+		return
+	}
+	account, path, driver, err := common.ParsePath(req.Path)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	file, files, err := driver.Path(path, account)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	if file != nil {
+		if driver.Config().OnlyProxy {
+			file.Url = fmt.Sprintf("//%s/d%s", c.Request.Host, req.Path)
+		}
+		c.JSON(200, common.Resp{
+			Code:    200,
+			Message: "file",
+			Data:    []*model.File{file},
+		})
+	} else {
+		meta, _ := model.GetMetaByPath(req.Path)
+		if meta != nil && meta.Hide != "" {
+			tmpFiles := make([]model.File, 0)
+			hideFiles := strings.Split(meta.Hide, ",")
+			for _, item := range files {
+				if !utils.IsContain(hideFiles, item.Name) {
+					tmpFiles = append(tmpFiles, item)
+				}
+			}
+			files = tmpFiles
+		}
+		c.JSON(200, common.Resp{
+			Code:    200,
+			Message: "folder",
+			Data:    files,
+		})
+	}
+}
+
+func Link(c *gin.Context) {
+	reqV,_ := c.Get("req")
+	req := reqV.(common.PathReq)
+	rawPath := req.Path
+	rawPath = utils.ParsePath(rawPath)
+	log.Debugf("link: %s", rawPath)
+	account, path, driver, err := common.ParsePath(rawPath)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	link, err := driver.Link(path, account)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	if driver.Config().OnlyProxy {
+		common.SuccessResp(c, gin.H{
+			"url": fmt.Sprintf("//%s/d%s", c.Request.Host, req.Path),
+		})
+		return
+	} else {
+		common.SuccessResp(c, gin.H{
+			"url": link,
+		})
+		return
+	}
+}
+
+func Preview(c *gin.Context) {
+	reqV,_ := c.Get("req")
+	req := reqV.(common.PathReq)
+	rawPath := req.Path
+	rawPath = utils.ParsePath(rawPath)
+	log.Debugf("preview: %s", rawPath)
+	account, path, driver, err := common.ParsePath(rawPath)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+		return
+	}
+	data, err := driver.Preview(path, account)
+	if err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		common.SuccessResp(c, data)
+	}
+}

server/controllers/setting.go

@@ -0,0 +1,39 @@
+package controllers
+
+import (
+	"github.com/Xhofe/alist/model"
+	"github.com/Xhofe/alist/server/common"
+	"github.com/gin-gonic/gin"
+)
+
+func SaveSettings(c *gin.Context) {
+	var req []model.SettingItem
+	if err := c.ShouldBind(&req); err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	if err := model.SaveSettings(req); err != nil {
+		common.ErrorResp(c, err, 500)
+	} else {
+		model.LoadSettings()
+		common.SuccessResp(c)
+	}
+}
+
+func GetSettings(c *gin.Context) {
+	settings, err := model.GetSettings()
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	common.SuccessResp(c, settings)
+}
+
+func GetSettingsPublic(c *gin.Context) {
+	settings, err := model.GetSettingsPublic()
+	if err != nil {
+		common.ErrorResp(c, err, 400)
+		return
+	}
+	common.SuccessResp(c, settings)
+}