[add] load dsl & script filesystem

This commit is contained in:
Max 2022-10-12 22:11:45 +08:00
parent bb3414f0eb
commit a9bdf8e7f6
2 changed files with 51 additions and 6 deletions

View file

@ -5,6 +5,7 @@ import (
"path/filepath"
"github.com/yaoapp/gou/fs"
"github.com/yaoapp/gou/fs/dsl"
"github.com/yaoapp/gou/fs/system"
"github.com/yaoapp/yao/config"
)
@ -12,20 +13,30 @@ import (
// Load system fs
func Load(cfg config.Config) error {
root, err := Root(cfg)
root, err := filepath.Abs(cfg.Root)
if err != nil {
return err
}
if _, err := os.Stat(root); os.IsNotExist(err) {
err := os.MkdirAll(root, os.ModePerm)
dataRoot, err := Root(cfg)
if err != nil {
return err
}
scriptRoot := filepath.Join(root, "scripts")
dslDenyList := []string{scriptRoot, dataRoot}
if _, err := os.Stat(dataRoot); os.IsNotExist(err) {
err := os.MkdirAll(dataRoot, os.ModePerm)
if err != nil {
return err
}
}
fs.Register("system", system.New(root))
fs.Register("binary", system.New(root)) // next
fs.Register("system", system.New(dataRoot))
fs.Register("dsl", dsl.New(root).DenyAbs(dslDenyList...)) // DSL
fs.Register("script", system.New(scriptRoot)) // Script
fs.Register("binary", system.New(root)) // Binary
return nil
}

View file

@ -26,5 +26,39 @@ func TestLoad(t *testing.T) {
assert.Equal(t, int64(2), info.Size())
err = fs.Remove(data, "test.file")
assert.Nil(t, err)
if err != nil {
t.Fatal(err)
}
}
func TestDSL(t *testing.T) {
Load(config.Conf)
dsl := fs.MustGet("dsl")
name := filepath.Join("models", "test.mod.json")
_, err := fs.WriteFile(dsl, name, []byte(`{"foo": "bar", "hello":{ "int": 1, "float": 0.618}}`), 0644)
assert.Nil(t, err)
info, err := os.Stat(filepath.Join(config.Conf.Root, name))
assert.Nil(t, err)
assert.Equal(t, int64(69), info.Size())
dsl.Remove(name)
if err != nil {
t.Fatal(err)
}
}
func TestScirpt(t *testing.T) {
Load(config.Conf)
script := fs.MustGet("script")
name := "test.js"
_, err := fs.WriteFile(script, name, []byte(`console.log("hello")`), 0644)
assert.Nil(t, err)
info, err := os.Stat(filepath.Join(config.Conf.Root, "scripts", name))
assert.Nil(t, err)
assert.Equal(t, int64(20), info.Size())
script.Remove(name)
if err != nil {
t.Fatal(err)
}
}