[add] load dsl & script filesystem
This commit is contained in:
parent
bb3414f0eb
commit
a9bdf8e7f6
2 changed files with 51 additions and 6 deletions
21
fs/fs.go
21
fs/fs.go
|
|
@ -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
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue