switch to config file and implement src path filter
This commit is contained in:
parent
055e9db8e2
commit
a6b9602821
30 changed files with 11907 additions and 60 deletions
53
pkg/provider/filter/file.go
Normal file
53
pkg/provider/filter/file.go
Normal file
|
|
@ -0,0 +1,53 @@
|
|||
package filter
|
||||
|
||||
import (
|
||||
"cachefs/pkg/provider"
|
||||
"io/fs"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
var _ provider.File = (*file)(nil)
|
||||
|
||||
type file struct {
|
||||
provider.File
|
||||
fs *FS
|
||||
path string
|
||||
}
|
||||
|
||||
func newFile(f provider.File, fs *FS, path string) (nf provider.File, err error) {
|
||||
nf = &file{
|
||||
File: f,
|
||||
fs: fs,
|
||||
path: path,
|
||||
}
|
||||
return nf, nil
|
||||
}
|
||||
|
||||
func (f *file) Readdir(n int) ([]fs.FileInfo, error) {
|
||||
fis, err := f.File.Readdir(n)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fio := make([]fs.FileInfo, len(fis))
|
||||
i := 0
|
||||
for _, fi := range fis {
|
||||
file := filepath.Join(f.path, fi.Name())
|
||||
if f.fs.match(file) == nil {
|
||||
fio[i] = fi
|
||||
i++
|
||||
}
|
||||
}
|
||||
return fio[:i], nil
|
||||
}
|
||||
|
||||
func (f *file) ReadDir(n int) ([]fs.DirEntry, error) {
|
||||
fis, err := f.Readdir(n)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
fio := make([]fs.DirEntry, len(fis))
|
||||
for i, fi := range fis {
|
||||
fio[i] = fs.FileInfoToDirEntry(fi)
|
||||
}
|
||||
return fio, nil
|
||||
}
|
||||
78
pkg/provider/filter/fs.go
Normal file
78
pkg/provider/filter/fs.go
Normal file
|
|
@ -0,0 +1,78 @@
|
|||
package filter
|
||||
|
||||
import (
|
||||
stdfs "io/fs"
|
||||
"os"
|
||||
"regexp"
|
||||
|
||||
"cachefs/pkg/provider"
|
||||
)
|
||||
|
||||
var _ provider.FS = (*FS)(nil)
|
||||
|
||||
type FS struct {
|
||||
provider.FS
|
||||
re *regexp.Regexp
|
||||
}
|
||||
|
||||
func NewFS(fs provider.FS, re *regexp.Regexp) (*FS, error) {
|
||||
return &FS{
|
||||
FS: fs,
|
||||
re: re,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (fs *FS) match(p string) error {
|
||||
if fs.re.MatchString(p) {
|
||||
return nil
|
||||
}
|
||||
return stdfs.ErrNotExist
|
||||
}
|
||||
|
||||
func (fs *FS) Stat(p string) (stdfs.FileInfo, error) {
|
||||
err := fs.match(p)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return fs.FS.Stat(p)
|
||||
}
|
||||
|
||||
func (fs *FS) Remove(p string) error {
|
||||
err := fs.match(p)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return fs.FS.Remove(p)
|
||||
}
|
||||
|
||||
func (fs *FS) Open(p string) (provider.File, error) {
|
||||
err := fs.match(p)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
f, err := fs.FS.Open(p)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return newFile(f, fs, p)
|
||||
}
|
||||
|
||||
func (fs *FS) OpenFile(p string, flags int, mode os.FileMode) (provider.File, error) {
|
||||
err := fs.match(p)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
f, err := fs.FS.OpenFile(p, flags, mode)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return newFile(f, fs, p)
|
||||
}
|
||||
|
||||
func (fs *FS) MkdirAll(p string, mode os.FileMode) error {
|
||||
err := fs.match(p)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return fs.FS.MkdirAll(p, mode)
|
||||
}
|
||||
|
|
@ -5,12 +5,14 @@ package parse
|
|||
import (
|
||||
"cachefs/pkg/provider"
|
||||
"cachefs/pkg/provider/crypto"
|
||||
"cachefs/pkg/provider/filter"
|
||||
"cachefs/pkg/provider/os"
|
||||
"cachefs/pkg/provider/sftp"
|
||||
"errors"
|
||||
"fmt"
|
||||
neturl "net/url"
|
||||
"path/filepath"
|
||||
"regexp"
|
||||
|
||||
"k8s.io/klog/v2/klogr"
|
||||
)
|
||||
|
|
@ -36,7 +38,7 @@ func TryParse(url string) error {
|
|||
return nil
|
||||
}
|
||||
|
||||
func FS(url string, key []byte, block bool) (provider.FS, error) {
|
||||
func FS(url string, key []byte, re *regexp.Regexp, block bool) (provider.FS, error) {
|
||||
u, err := neturl.Parse(url)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
|
|
@ -58,6 +60,12 @@ func FS(url string, key []byte, block bool) (provider.FS, error) {
|
|||
}
|
||||
if key != nil {
|
||||
fs, err = crypto.NewFS(fs, key)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
if re != nil {
|
||||
fs, err = filter.NewFS(fs, re)
|
||||
}
|
||||
return fs, err
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue