ef03b1e3df
Upgrade dependent version: github.com/open-policy-agent/opa v0.18.0 -> v0.45.0 Signed-off-by: hongzhouzi <hongzhouzi@kubesphere.io> Signed-off-by: hongzhouzi <hongzhouzi@kubesphere.io>
101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
//go:build go1.16
|
|
// +build go1.16
|
|
|
|
package bundle
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"io/fs"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/open-policy-agent/opa/loader/filter"
|
|
)
|
|
|
|
const (
|
|
defaultFSLoaderRoot = "."
|
|
)
|
|
|
|
type dirLoaderFS struct {
|
|
sync.Mutex
|
|
filesystem fs.FS
|
|
files []string
|
|
idx int
|
|
filter filter.LoaderFilter
|
|
}
|
|
|
|
// NewFSLoader returns a basic DirectoryLoader implementation
|
|
// that will load files from a fs.FS interface
|
|
func NewFSLoader(filesystem fs.FS) (DirectoryLoader, error) {
|
|
d := dirLoaderFS{
|
|
filesystem: filesystem,
|
|
}
|
|
|
|
return &d, nil
|
|
}
|
|
|
|
func (d *dirLoaderFS) walkDir(path string, dirEntry fs.DirEntry, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if dirEntry != nil {
|
|
info, err := dirEntry.Info()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if dirEntry.Type().IsRegular() {
|
|
if d.filter != nil && d.filter(filepath.ToSlash(path), info, getdepth(path, false)) {
|
|
return nil
|
|
}
|
|
|
|
d.files = append(d.files, path)
|
|
} else if dirEntry.Type().IsDir() {
|
|
if d.filter != nil && d.filter(filepath.ToSlash(path), info, getdepth(path, true)) {
|
|
return fs.SkipDir
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// WithFilter specifies the filter object to use to filter files while loading bundles
|
|
func (d *dirLoaderFS) WithFilter(filter filter.LoaderFilter) DirectoryLoader {
|
|
d.filter = filter
|
|
return d
|
|
}
|
|
|
|
// NextFile iterates to the next file in the directory tree
|
|
// and returns a file Descriptor for the file.
|
|
func (d *dirLoaderFS) NextFile() (*Descriptor, error) {
|
|
d.Lock()
|
|
defer d.Unlock()
|
|
|
|
if d.files == nil {
|
|
err := fs.WalkDir(d.filesystem, defaultFSLoaderRoot, d.walkDir)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to list files: %w", err)
|
|
}
|
|
}
|
|
|
|
// If done reading files then just return io.EOF
|
|
// errors for each NextFile() call
|
|
if d.idx >= len(d.files) {
|
|
return nil, io.EOF
|
|
}
|
|
|
|
fileName := d.files[d.idx]
|
|
d.idx++
|
|
|
|
fh, err := d.filesystem.Open(fileName)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to open file %s: %w", fileName, err)
|
|
}
|
|
|
|
fileNameWithSlash := fmt.Sprintf("/%s", fileName)
|
|
f := newDescriptor(fileNameWithSlash, fileNameWithSlash, fh).withCloser(fh)
|
|
return f, nil
|
|
}
|