Big refactor

internal/zfs/zfs.go

@@ -0,0 +1,185 @@
+package zfs
+
+import (
+	"strings"
+
+	"github.com/debridmediamanager.com/zurg/internal/config"
+	"github.com/debridmediamanager.com/zurg/internal/torrent"
+	"github.com/debridmediamanager.com/zurg/pkg/chunk"
+	"github.com/winfsp/cgofuse/fuse"
+	"go.uber.org/zap"
+)
+
+type ZurgFS struct {
+	fuse.FileSystemBase
+	TorrentManager *torrent.TorrentManager
+	Config         config.ConfigInterface
+	Chunk          *chunk.Manager
+	Log            *zap.SugaredLogger
+}
+
+func NewZurgFS(tm *torrent.TorrentManager, cfg config.ConfigInterface, chunk *chunk.Manager, log *zap.SugaredLogger) *ZurgFS {
+	return &ZurgFS{
+		TorrentManager: tm,
+		Config:         cfg,
+		Chunk:          chunk,
+		Log:            log,
+	}
+}
+
+func (fs *ZurgFS) Open(path string, flags int) (errc int, fh uint64) {
+	segments := splitIntoSegments(path)
+	switch len(segments) {
+	case 0:
+		return 0, 0
+	case 1:
+		if _, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT, ^uint64(0)
+		}
+		return 0, 0
+	case 2:
+		if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT, ^uint64(0)
+		} else if _, torFound := directory.Get(segments[1]); !torFound {
+			return -fuse.ENOENT, ^uint64(0)
+		}
+		return 0, 0
+	case 3:
+		if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT, ^uint64(0)
+		} else if torrent, torFound := directory.Get(segments[1]); !torFound {
+			return -fuse.ENOENT, ^uint64(0)
+		} else if file, fileFound := torrent.SelectedFiles.Get(segments[2]); !fileFound {
+			return -fuse.ENOENT, ^uint64(0)
+		} else {
+			return 0, file.ZurgFS
+		}
+	}
+
+	return -fuse.ENOENT, ^uint64(0)
+}
+
+func (fs *ZurgFS) Getattr(path string, stat *fuse.Stat_t, fh uint64) (errc int) {
+	segments := splitIntoSegments(path)
+	switch len(segments) {
+	case 0:
+		stat.Mode = fuse.S_IFDIR | 0555
+		return 0
+	case 1:
+		if _, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT
+		}
+		stat.Mode = fuse.S_IFDIR | 0555
+		return 0
+	case 2:
+		if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT
+		} else if _, torFound := directory.Get(segments[1]); !torFound {
+			return -fuse.ENOENT
+		}
+		stat.Mode = fuse.S_IFDIR | 0555
+		return 0
+	case 3:
+		if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT
+		} else if torrent, torFound := directory.Get(segments[1]); !torFound {
+			return -fuse.ENOENT
+		} else if file, fileFound := torrent.SelectedFiles.Get(segments[2]); !fileFound {
+			return -fuse.ENOENT
+		} else {
+			stat.Mode = fuse.S_IFREG | 0444
+			stat.Size = file.Bytes
+			return 0
+		}
+	}
+
+	return -fuse.ENOENT
+}
+
+func (fs *ZurgFS) Read(path string, buff []byte, ofst int64, fh uint64) (n int) {
+	segments := splitIntoSegments(path)
+	if len(segments) != 3 {
+		return -fuse.ENOENT
+	} else if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+		return -fuse.ENOENT
+	} else if torrent, torFound := directory.Get(segments[1]); !torFound {
+		return -fuse.ENOENT
+	} else if file, fileFound := torrent.SelectedFiles.Get(segments[2]); !fileFound {
+		return -fuse.ENOENT
+	} else {
+		size := int64(len(buff))
+		if size < int64(fs.Config.GetNetworkBufferSize()) {
+			size = int64(fs.Config.GetNetworkBufferSize())
+		}
+		endofst := ofst + size
+		if endofst > file.Bytes {
+			endofst = file.Bytes
+		}
+		if endofst < ofst {
+			return 0
+		}
+		response, err := fs.Chunk.GetChunk(file, ofst, size)
+		if err != nil {
+			return -fuse.ENOENT
+		}
+		// response := universal.GetFileReader(torrent, file, ofst, int(size), fs.TorrentManager, fs.Config, fs.Log)
+		// if response == nil {
+		// 	return -fuse.ENOENT
+		// }
+		n = copy(buff, response[:endofst-ofst])
+		return n
+	}
+}
+
+func (fs *ZurgFS) Readdir(path string,
+	fill func(name string, stat *fuse.Stat_t, ofst int64) bool,
+	ofst int64,
+	fh uint64) (errc int) {
+
+	segments := splitIntoSegments(path)
+	switch len(segments) {
+	case 0:
+		fill(".", nil, 0)
+		fill("..", nil, 0)
+		for el := fs.TorrentManager.DirectoryMap.Front(); el != nil; el = el.Next() {
+			fill(el.Key, nil, 0)
+		}
+	case 1:
+		fill(".", nil, 0)
+		fill("..", nil, 0)
+		if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT
+		} else {
+			for el := directory.Front(); el != nil; el = el.Next() {
+				fill(el.Key, nil, 0)
+			}
+		}
+	case 2:
+		fill(".", nil, 0)
+		fill("..", nil, 0)
+		if directory, dirFound := fs.TorrentManager.DirectoryMap.Get(segments[0]); !dirFound {
+			return -fuse.ENOENT
+		} else if torrent, torFound := directory.Get(segments[1]); !torFound {
+			return -fuse.ENOENT
+		} else {
+			for el := torrent.SelectedFiles.Front(); el != nil; el = el.Next() {
+				fill(el.Key, nil, 0)
+			}
+		}
+	default:
+		return -fuse.ENOENT
+	}
+	return 0
+}
+
+func splitIntoSegments(path string) []string {
+	segments := strings.Split(path, "/")
+	// remove empty segments
+	for i := 0; i < len(segments); i++ {
+		if segments[i] == "" {
+			segments = append(segments[:i], segments[i+1:]...)
+			i--
+		}
+	}
+	return segments
+}