🎨 Improve the code structure

drivers/base/driver.go

@@ -0,0 +1,99 @@
+package base
+
+import (
+	"github.com/Xhofe/alist/model"
+	"github.com/gin-gonic/gin"
+	"github.com/go-resty/resty/v2"
+	log "github.com/sirupsen/logrus"
+	"net/http"
+)
+
+type DriverConfig struct {
+	Name string
+	OnlyProxy bool
+}
+
+type Driver interface {
+	Config() DriverConfig
+	Items() []Item
+	Save(account *model.Account, old *model.Account) error
+	File(path string, account *model.Account) (*model.File, error)
+	Files(path string, account *model.Account) ([]model.File, error)
+	Link(path string, account *model.Account) (string, error)
+	Path(path string, account *model.Account) (*model.File, []model.File, error)
+	Proxy(c *gin.Context, account *model.Account)
+	Preview(path string, account *model.Account) (interface{}, error)
+	// TODO
+	//Search(path string, keyword string, account *model.Account) ([]*model.File, error)
+	MakeDir(path string, account *model.Account) error
+	Move(src string, dst string, account *model.Account) error
+	Copy(src string, dst string, account *model.Account) error
+	Delete(path string, account *model.Account) error
+	Upload(file *model.FileStream, account *model.Account) error
+}
+
+type Item struct {
+	Name        string `json:"name"`
+	Label       string `json:"label"`
+	Type        string `json:"type"`
+	Values      string `json:"values"`
+	Required    bool   `json:"required"`
+	Description string `json:"description"`
+}
+
+type TokenResp struct {
+	AccessToken  string `json:"access_token"`
+	RefreshToken string `json:"refresh_token"`
+}
+
+var driversMap = map[string]Driver{}
+
+func RegisterDriver(driver Driver) {
+	log.Infof("register driver: [%s]", driver.Config().Name)
+	driversMap[driver.Config().Name] = driver
+}
+
+func GetDriver(name string) (driver Driver, ok bool) {
+	driver, ok = driversMap[name]
+	return
+}
+
+func GetDrivers() map[string][]Item {
+	res := make(map[string][]Item, 0)
+	for k, v := range driversMap {
+		if v.Config().OnlyProxy {
+			res[k] = v.Items()
+		} else {
+			res[k] = append([]Item{
+				{
+					Name:        "proxy",
+					Label:       "proxy",
+					Type:        "bool",
+					Required:    true,
+					Description: "allow proxy",
+				},
+				{
+					Name:        "webdav_proxy",
+					Label:       "webdav proxy",
+					Type:        "bool",
+					Required:    true,
+					Description: "Transfer the WebDAV of this account through the server",
+				},
+			}, v.Items()...)
+		}
+	}
+	return res
+}
+
+type Json map[string]interface{}
+
+var NoRedirectClient *resty.Client
+
+func init() {
+	NoRedirectClient = resty.New().SetRedirectPolicy(
+		resty.RedirectPolicyFunc(func(req *http.Request, via []*http.Request) error {
+			return http.ErrUseLastResponse
+		}),
+	)
+	NoRedirectClient.SetHeader("user-agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.88 Safari/537.36")
+}

drivers/base/types.go

@@ -0,0 +1,17 @@
+package base
+
+import "fmt"
+
+var (
+	ErrPathNotFound = fmt.Errorf("path not found")
+	ErrNotFile      = fmt.Errorf("not file")
+	ErrNotImplement = fmt.Errorf("not implement")
+	ErrNotSupport   = fmt.Errorf("not support")
+)
+
+const (
+	TypeString = "string"
+	TypeSelect = "select"
+	TypeBool   = "bool"
+	TypeNumber = "number"
+)