43 lines
927 B
Go
43 lines
927 B
Go
package node
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"syscall"
|
|
|
|
"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 = &Root{}
|
|
var _ fs.NodeGetattrer = &Root{}
|
|
|
|
type Root struct {
|
|
fs.Inode
|
|
pool *ants.Pool
|
|
torrents []*torrent.Torrent
|
|
}
|
|
|
|
func NewRoot(torrents []*torrent.Torrent, pool *ants.Pool) *Root {
|
|
return &Root{torrents: torrents, pool: pool}
|
|
}
|
|
|
|
func (root *Root) OnAdd(ctx context.Context) {
|
|
for _, torrent := range root.torrents {
|
|
root.pool.Submit(func() {
|
|
root.AddChild(
|
|
filepath.Clean(torrent.Name()),
|
|
root.NewPersistentInode(ctx, &Torrent{t: torrent, pool: root.pool}, fs.StableAttr{
|
|
Mode: syscall.S_IFDIR,
|
|
}), true)
|
|
})
|
|
}
|
|
}
|
|
|
|
func (root *Root) Getattr(ctx context.Context, f fs.FileHandle, out *fuse.AttrOut) syscall.Errno {
|
|
out.Mode = syscall.S_IFDIR & 07777
|
|
|
|
return fs.OK
|
|
}
|