find_mockfs.go 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121
  1. package mk2rbc
  2. import (
  3. "io/fs"
  4. "os"
  5. "path/filepath"
  6. "time"
  7. )
  8. // Mock FS. Maps a directory name to an array of entries.
  9. // An entry implements fs.DirEntry, fs.FIleInfo and fs.File interface
  10. type FindMockFS struct {
  11. dirs map[string][]myFileInfo
  12. }
  13. func (m FindMockFS) locate(name string) (myFileInfo, bool) {
  14. if name == "." {
  15. return myFileInfo{".", true}, true
  16. }
  17. dir := filepath.Dir(name)
  18. base := filepath.Base(name)
  19. if entries, ok := m.dirs[dir]; ok {
  20. for _, e := range entries {
  21. if e.name == base {
  22. return e, true
  23. }
  24. }
  25. }
  26. return myFileInfo{}, false
  27. }
  28. func (m FindMockFS) create(name string, isDir bool) {
  29. dir := filepath.Dir(name)
  30. m.dirs[dir] = append(m.dirs[dir], myFileInfo{filepath.Base(name), isDir})
  31. }
  32. func (m FindMockFS) Stat(name string) (fs.FileInfo, error) {
  33. if fi, ok := m.locate(name); ok {
  34. return fi, nil
  35. }
  36. return nil, os.ErrNotExist
  37. }
  38. type myFileInfo struct {
  39. name string
  40. isDir bool
  41. }
  42. func (m myFileInfo) Info() (fs.FileInfo, error) {
  43. panic("implement me")
  44. }
  45. func (m myFileInfo) Size() int64 {
  46. panic("implement me")
  47. }
  48. func (m myFileInfo) Mode() fs.FileMode {
  49. panic("implement me")
  50. }
  51. func (m myFileInfo) ModTime() time.Time {
  52. panic("implement me")
  53. }
  54. func (m myFileInfo) Sys() interface{} {
  55. return nil
  56. }
  57. func (m myFileInfo) Stat() (fs.FileInfo, error) {
  58. return m, nil
  59. }
  60. func (m myFileInfo) Read(bytes []byte) (int, error) {
  61. panic("implement me")
  62. }
  63. func (m myFileInfo) Close() error {
  64. panic("implement me")
  65. }
  66. func (m myFileInfo) Name() string {
  67. return m.name
  68. }
  69. func (m myFileInfo) IsDir() bool {
  70. return m.isDir
  71. }
  72. func (m myFileInfo) Type() fs.FileMode {
  73. return m.Mode()
  74. }
  75. func (m FindMockFS) Open(name string) (fs.File, error) {
  76. panic("implement me")
  77. }
  78. func (m FindMockFS) ReadDir(name string) ([]fs.DirEntry, error) {
  79. if d, ok := m.dirs[name]; ok {
  80. var res []fs.DirEntry
  81. for _, e := range d {
  82. res = append(res, e)
  83. }
  84. return res, nil
  85. }
  86. return nil, os.ErrNotExist
  87. }
  88. func NewFindMockFS(files []string) FindMockFS {
  89. myfs := FindMockFS{make(map[string][]myFileInfo)}
  90. for _, f := range files {
  91. isDir := false
  92. for f != "." {
  93. if _, ok := myfs.locate(f); !ok {
  94. myfs.create(f, isDir)
  95. }
  96. isDir = true
  97. f = filepath.Dir(f)
  98. }
  99. }
  100. return myfs
  101. }