78 lines
1.5 KiB
Go
Executable File
78 lines
1.5 KiB
Go
Executable File
package storage
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
type Files struct {
|
|
root string
|
|
}
|
|
|
|
func NewFiles(root string) (*Files, error) {
|
|
_, err := os.Stat(path.Dir(root))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
root, err = filepath.Abs(root)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &Files{
|
|
root: root,
|
|
}, os.MkdirAll(root, os.ModePerm)
|
|
}
|
|
|
|
func (b *Files) List(ns []string, limits ...string) ([]string, error) {
|
|
namespace := resolveNamespace(ns)
|
|
files := make([]string, 0)
|
|
err := filepath.Walk(b.root, func(p string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if info.IsDir() {
|
|
return nil
|
|
}
|
|
if !strings.HasPrefix(strings.TrimPrefix(p, b.root+"/"), namespace) {
|
|
return nil
|
|
}
|
|
files = append(files, strings.TrimPrefix(p, path.Join(b.root, namespace)+"/"))
|
|
return nil
|
|
})
|
|
return files, err
|
|
/*
|
|
namespace := resolveNamespace(ns)
|
|
limits = resolveLimits(limits)
|
|
*/
|
|
}
|
|
|
|
func (b *Files) Get(key string, ns ...string) ([]byte, error) {
|
|
namespace := resolveNamespace(ns)
|
|
path := path.Join(b.root, namespace, key)
|
|
return ioutil.ReadFile(path)
|
|
}
|
|
|
|
func (b *Files) Set(key string, value []byte, ns ...string) error {
|
|
namespace := resolveNamespace(ns)
|
|
dir := path.Join(b.root, namespace)
|
|
if value == nil {
|
|
err := os.Remove(path.Join(dir, key))
|
|
if os.IsNotExist(err) {
|
|
err = nil
|
|
}
|
|
return err
|
|
}
|
|
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
path := path.Join(dir, key)
|
|
return ioutil.WriteFile(path, value, os.ModePerm)
|
|
}
|
|
|
|
func (b *Files) Close() error {
|
|
return nil
|
|
}
|