feat: driver and account operate

internal/operations/driver.go

@@ -0,0 +1,111 @@
+package operations
+
+import (
+	"github.com/alist-org/alist/v3/internal/driver"
+	"github.com/pkg/errors"
+	log "github.com/sirupsen/logrus"
+	"reflect"
+	"strings"
+)
+
+type New func() driver.Driver
+
+var driverNewMap = map[string]New{}
+var driverItemsMap = map[string]driver.Items{}
+
+func RegisterDriver(config driver.Config, driver New) {
+	log.Infof("register driver: [%s]", config.Name)
+	registerDriverItems(config, driver().GetAddition())
+	driverNewMap[config.Name] = driver
+}
+
+func GetDriverNew(name string) (New, error) {
+	n, ok := driverNewMap[name]
+	if !ok {
+		return nil, errors.Errorf("no driver named: %s", name)
+	}
+	return n, nil
+}
+
+func registerDriverItems(config driver.Config, addition driver.Additional) {
+	tAddition := reflect.TypeOf(addition)
+	mainItems := getMainItems(config)
+	additionalItems := getAdditionalItems(tAddition)
+	driverItemsMap[config.Name] = driver.Items{mainItems, additionalItems}
+}
+
+func getMainItems(config driver.Config) []driver.Item {
+	items := []driver.Item{{
+		Name:     "virtual_path",
+		Type:     driver.TypeString,
+		Required: true,
+		Help:     "",
+	}, {
+		Name: "index",
+		Type: driver.TypeNumber,
+		Help: "use to sort",
+	}, {
+		Name: "down_proxy_url",
+		Type: driver.TypeText,
+	}, {
+		Name: "webdav_direct",
+		Type: driver.TypeBool,
+		Help: "Transfer the WebDAV of this account through the native without redirect",
+	}}
+	if !config.OnlyProxy && !config.OnlyLocal {
+		items = append(items, []driver.Item{{
+			Name: "web_proxy",
+			Type: driver.TypeBool,
+		}, {
+			Name: "webdav_proxy",
+			Type: driver.TypeBool,
+		},
+		}...)
+	}
+	if config.LocalSort {
+		items = append(items, []driver.Item{{
+			Name:   "order_by",
+			Type:   driver.TypeSelect,
+			Values: "name,size,modified",
+		}, {
+			Name:   "order_direction",
+			Type:   driver.TypeSelect,
+			Values: "ASC,DESC",
+		}}...)
+	}
+	items = append(items, driver.Item{
+		Name:   "extract_folder",
+		Type:   driver.TypeSelect,
+		Values: "front,back",
+	})
+	return items
+}
+
+func getAdditionalItems(t reflect.Type) []driver.Item {
+	var items []driver.Item
+	for i := 0; i < t.NumField(); i++ {
+		field := t.Field(i)
+		tag := field.Tag
+		ignore, ok := tag.Lookup("ignore")
+		if !ok || ignore == "false" {
+			continue
+		}
+		item := driver.Item{
+			Name:     tag.Get("json"),
+			Type:     strings.ToLower(field.Type.Name()),
+			Default:  tag.Get("default"),
+			Values:   tag.Get("values"),
+			Required: tag.Get("required") == "true",
+			Help:     tag.Get("help"),
+		}
+		if tag.Get("type") != "" {
+			item.Type = tag.Get("type")
+		}
+		// set default type to string
+		if item.Type == "" {
+			item.Type = "string"
+		}
+		items = append(items, item)
+	}
+	return items
+}