embedfs.go 1003 B

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455
  1. package util
  2. import (
  3. "embed"
  4. "errors"
  5. "io"
  6. "io/fs"
  7. "time"
  8. )
  9. type CachingEmbedFS struct {
  10. ModTime time.Time
  11. FS embed.FS
  12. }
  13. func (f CachingEmbedFS) Open(name string) (fs.File, error) {
  14. file, err := f.FS.Open(name)
  15. if err != nil {
  16. return nil, err
  17. }
  18. stat, err := file.Stat()
  19. if err != nil {
  20. return nil, err
  21. }
  22. return &cachingEmbedFile{file, f.ModTime, stat}, nil
  23. }
  24. type cachingEmbedFile struct {
  25. file fs.File
  26. modTime time.Time
  27. fs.FileInfo
  28. }
  29. func (f cachingEmbedFile) Stat() (fs.FileInfo, error) {
  30. return f, nil
  31. }
  32. func (f cachingEmbedFile) Read(bytes []byte) (int, error) {
  33. return f.file.Read(bytes)
  34. }
  35. func (f *cachingEmbedFile) Seek(offset int64, whence int) (int64, error) {
  36. if seeker, ok := f.file.(io.Seeker); ok {
  37. return seeker.Seek(offset, whence)
  38. }
  39. return 0, errors.New("io.Seeker not implemented")
  40. }
  41. func (f cachingEmbedFile) ModTime() time.Time {
  42. return f.modTime // We override this!
  43. }
  44. func (f cachingEmbedFile) Close() error {
  45. return f.file.Close()
  46. }