tstor/torrent/node/torrent.go

package node

import (
	"context"
	"io"
	"syscall"

	"github.com/ajnavarro/distribyted/iio"
	"github.com/anacrolix/torrent"
	"github.com/hanwen/go-fuse/v2/fs"
	"github.com/hanwen/go-fuse/v2/fuse"
	"github.com/panjf2000/ants/v2"
)

var _ fs.NodeOnAdder = &Torrent{}
var _ fs.NodeGetattrer = &Torrent{}

type Torrent struct {
	fs.Inode
	t *torrent.Torrent

	pool *ants.Pool
}

func (folder *Torrent) OnAdd(ctx context.Context) {
	for _, file := range folder.t.Files() {
		file := file
		LoadNodeByPath(
			ctx,
			folder.pool,
			file.Path(),
			func() (io.ReaderAt, error) { return iio.NewReadAtWrapper(file.NewReader()), nil },
			&folder.Inode,
			file.Length(),
			int32(file.Torrent().Info().PieceLength),
			int64(file.Torrent().Info().NumPieces()),
		)
	}
}

func (folder *Torrent) Getattr(ctx context.Context, f fs.FileHandle, out *fuse.AttrOut) syscall.Errno {
	out.Mode = syscall.S_IFDIR & 07777

	return fs.OK
}