Merge pull request #193 from trheyi/main

[add] load dsl & script filesystem
This commit is contained in:
Max 2022-10-12 22:22:54 +08:00 committed by GitHub
commit 9c4b51d331
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 51 additions and 6 deletions

View file

@ -5,6 +5,7 @@ import (
"path/filepath" "path/filepath"
"github.com/yaoapp/gou/fs" "github.com/yaoapp/gou/fs"
"github.com/yaoapp/gou/fs/dsl"
"github.com/yaoapp/gou/fs/system" "github.com/yaoapp/gou/fs/system"
"github.com/yaoapp/yao/config" "github.com/yaoapp/yao/config"
) )
@ -12,20 +13,30 @@ import (
// Load system fs // Load system fs
func Load(cfg config.Config) error { func Load(cfg config.Config) error {
root, err := Root(cfg) root, err := filepath.Abs(cfg.Root)
if err != nil { if err != nil {
return err return err
} }
if _, err := os.Stat(root); os.IsNotExist(err) { dataRoot, err := Root(cfg)
err := os.MkdirAll(root, os.ModePerm) 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 { if err != nil {
return err return err
} }
} }
fs.Register("system", system.New(root)) fs.Register("system", system.New(dataRoot))
fs.Register("binary", system.New(root)) // next 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 return nil
} }

View file

@ -26,5 +26,39 @@ func TestLoad(t *testing.T) {
assert.Equal(t, int64(2), info.Size()) assert.Equal(t, int64(2), info.Size())
err = fs.Remove(data, "test.file") 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)
}
} }