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

feat: add driver config in driver info

Browse Source
This commit is contained in:
Noah Hsu
2022-08-30 14:39:10 +08:00
parent fec98e7f69
commit 59ec17a353
8 changed files with 31 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/operations/driver.go
View File

@ -13,7 +13,7 @@ import (
type New func() driver.Driver
var driverNewMap = map[string]New{}
var driverItemsMap = map[string]driver.Items{}
var driverInfoMap = map[string]driver.Info{}
func RegisterDriver(config driver.Config, driver New) {
// log.Infof("register driver: [%s]", config.Name)
@ -31,14 +31,14 @@ func GetDriverNew(name string) (New, error) {
func GetDriverNames() []string {
var driverNames []string
for k := range driverItemsMap {
for k := range driverInfoMap {
driverNames = append(driverNames, k)
}
return driverNames
}
func GetDriverItemsMap() map[string]driver.Items {
return driverItemsMap
func GetDriverInfoMap() map[string]driver.Info {
return driverInfoMap
}
func registerDriverItems(config driver.Config, addition driver.Additional) {
@ -46,9 +46,10 @@ func registerDriverItems(config driver.Config, addition driver.Additional) {
tAddition := reflect.TypeOf(addition)
mainItems := getMainItems(config)
additionalItems := getAdditionalItems(tAddition, config.DefaultRoot)
driverItemsMap[config.Name] = driver.Items{
driverInfoMap[config.Name] = driver.Info{
Common: mainItems,
Additional: additionalItems,
Config: config,
}
}

6
internal/operations/driver_test.go
View File

@ -8,10 +8,10 @@ import (
)
func TestDriverItemsMap(t *testing.T) {
itemsMap := operations.GetDriverItemsMap()
itemsMap := operations.GetDriverInfoMap()
if len(itemsMap) != 0 {
t.Logf("driverItemsMap: %v", itemsMap)
t.Logf("driverInfoMap: %v", itemsMap)
} else {
t.Errorf("expected driverItemsMap not empty, but got empty")
t.Errorf("expected driverInfoMap not empty, but got empty")
}
}