mirror of https://github.com/muety/wakapi.git
75 lines
1.2 KiB
Go
75 lines
1.2 KiB
Go
package fs
|
|
|
|
import (
|
|
lru "github.com/hashicorp/golang-lru"
|
|
"io/fs"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func NewExistsFS(fs fs.FS) ExistsFS {
|
|
cache, err := lru.New(1 << 24)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
return ExistsFS{
|
|
FS: fs,
|
|
cache: cache,
|
|
}
|
|
}
|
|
|
|
type ExistsFS struct {
|
|
fs.FS
|
|
UseCache bool
|
|
cache *lru.Cache
|
|
}
|
|
|
|
func (efs ExistsFS) WithCache(withCache bool) ExistsFS {
|
|
efs.UseCache = withCache
|
|
return efs
|
|
}
|
|
|
|
func (efs ExistsFS) Exists(name string) bool {
|
|
if efs.UseCache {
|
|
if result, ok := efs.cache.Get(name); ok {
|
|
return result.(bool)
|
|
}
|
|
}
|
|
_, err := fs.Stat(efs.FS, name)
|
|
result := err == nil
|
|
if efs.UseCache {
|
|
efs.cache.Add(name, result)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func (efs ExistsFS) Open(name string) (fs.File, error) {
|
|
return efs.FS.Open(name)
|
|
}
|
|
|
|
// ---
|
|
|
|
type ExistsHttpFS struct {
|
|
ExistsFS
|
|
httpFs http.FileSystem
|
|
}
|
|
|
|
func NewExistsHttpFS(fs ExistsFS) ExistsHttpFS {
|
|
return ExistsHttpFS{
|
|
ExistsFS: fs,
|
|
httpFs: http.FS(fs),
|
|
}
|
|
}
|
|
|
|
func (ehfs ExistsHttpFS) Exists(name string) bool {
|
|
if strings.HasPrefix(name, "/") {
|
|
name = name[1:]
|
|
}
|
|
return ehfs.ExistsFS.Exists(name)
|
|
}
|
|
|
|
func (ehfs ExistsHttpFS) Open(name string) (http.File, error) {
|
|
return ehfs.httpFs.Open(name)
|
|
}
|