fs.go 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. package view
  2. import (
  3. "fmt"
  4. "io/fs"
  5. "path/filepath"
  6. "github.com/kataras/iris/v12/context"
  7. )
  8. // walk recursively in "fileSystem" descends "root" path, calling "walkFn".
  9. func walk(fileSystem fs.FS, root string, walkFn filepath.WalkFunc) error {
  10. if root != "" && root != "/" && root != "." {
  11. sub, err := fs.Sub(fileSystem, root)
  12. if err != nil {
  13. return err
  14. }
  15. fileSystem = sub
  16. }
  17. if root == "" {
  18. root = "."
  19. }
  20. return fs.WalkDir(fileSystem, root, func(path string, d fs.DirEntry, err error) error {
  21. if err != nil {
  22. return fmt.Errorf("walk: %s: %w", path, err)
  23. }
  24. info, err := d.Info()
  25. if err != nil {
  26. if err != filepath.SkipDir {
  27. return fmt.Errorf("walk stat: %s: %w", path, err)
  28. }
  29. return nil
  30. }
  31. if info.IsDir() {
  32. return nil
  33. }
  34. walkFnErr := walkFn(path, info, err)
  35. if walkFnErr != nil {
  36. return fmt.Errorf("walk: walkFn: %w", walkFnErr)
  37. }
  38. return nil
  39. })
  40. }
  41. func asset(fileSystem fs.FS, name string) ([]byte, error) {
  42. data, err := fs.ReadFile(fileSystem, name)
  43. if err != nil {
  44. return nil, fmt.Errorf("asset: read file: %w", err)
  45. }
  46. return data, nil
  47. }
  48. func getFS(fsOrDir interface{}) fs.FS {
  49. return context.ResolveFS(fsOrDir)
  50. }
  51. func getRootDirName(fileSystem fs.FS) string {
  52. rootDirFile, err := fileSystem.Open(".")
  53. if err == nil {
  54. rootDirStat, err := rootDirFile.Stat()
  55. if err == nil {
  56. return rootDirStat.Name()
  57. }
  58. }
  59. return ""
  60. }