71 lines
1.3 KiB
Go
71 lines
1.3 KiB
Go
package sui
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/yaoapp/gou/application"
|
|
"github.com/yaoapp/kun/log"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/share"
|
|
"github.com/yaoapp/yao/sui/core"
|
|
"github.com/yaoapp/yao/sui/storages/azure"
|
|
"github.com/yaoapp/yao/sui/storages/local"
|
|
)
|
|
|
|
// SUIs the loaded SUI instances
|
|
var SUIs = map[string]core.SUI{}
|
|
|
|
// New create a new sui
|
|
func New(dsl *core.DSL) (core.SUI, error) {
|
|
|
|
if dsl.Storage == nil {
|
|
return nil, fmt.Errorf("storage is not required")
|
|
}
|
|
|
|
switch strings.ToLower(dsl.Storage.Driver) {
|
|
|
|
case "local":
|
|
return local.New(dsl)
|
|
|
|
case "azure":
|
|
return azure.New(dsl)
|
|
|
|
default:
|
|
return nil, fmt.Errorf("%s is not a valid driver", dsl.Storage.Driver)
|
|
}
|
|
}
|
|
|
|
// Load load the sui
|
|
func Load(cfg config.Config) error {
|
|
exts := []string{"*.sui.yao", "*.sui.jsonc", "*.sui.json"}
|
|
return application.App.Walk("suis", func(root, file string, isdir bool) error {
|
|
if isdir {
|
|
return nil
|
|
}
|
|
|
|
id := share.ID(root, file)
|
|
_, err := loadFile(file, id)
|
|
if err != nil {
|
|
log.Error("[sui] Load sui %s error: %s", id, err.Error())
|
|
return err
|
|
}
|
|
return nil
|
|
}, exts...)
|
|
}
|
|
|
|
func loadFile(file string, id string) (core.SUI, error) {
|
|
|
|
dsl, err := core.Load(file, id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sui, err := New(dsl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
SUIs[id] = sui
|
|
return SUIs[id], nil
|
|
}
|