77 lines
1.3 KiB
Go
77 lines
1.3 KiB
Go
package fs
|
|
|
|
import (
|
|
"github.com/anacrolix/torrent"
|
|
"github.com/distribyted/distribyted/iio"
|
|
)
|
|
|
|
var _ Filesystem = &Torrent{}
|
|
|
|
type Torrent struct {
|
|
t *torrent.Torrent
|
|
s *storage
|
|
}
|
|
|
|
func NewTorrent(t *torrent.Torrent) *Torrent {
|
|
return &Torrent{
|
|
t: t,
|
|
s: newStorage(SupportedFactories),
|
|
}
|
|
}
|
|
|
|
func (fs *Torrent) load() {
|
|
<-fs.t.GotInfo()
|
|
for _, file := range fs.t.Files() {
|
|
fs.s.Add(&torrentFile{readerFunc: file.NewReader, len: file.Length()}, file.Path())
|
|
}
|
|
}
|
|
|
|
func (fs *Torrent) Open(filename string) (File, error) {
|
|
fs.load()
|
|
return fs.s.Get(filename)
|
|
}
|
|
|
|
func (fs *Torrent) ReadDir(path string) (map[string]File, error) {
|
|
fs.load()
|
|
return fs.s.Children(path), nil
|
|
}
|
|
|
|
var _ File = &torrentFile{}
|
|
|
|
type torrentFile struct {
|
|
readerFunc func() torrent.Reader
|
|
reader iio.Reader
|
|
len int64
|
|
}
|
|
|
|
func (d *torrentFile) load() {
|
|
if d.reader != nil {
|
|
return
|
|
}
|
|
|
|
d.reader = iio.NewReadAtWrapper(d.readerFunc())
|
|
}
|
|
|
|
func (d *torrentFile) Size() int64 {
|
|
return d.len
|
|
}
|
|
|
|
func (d *torrentFile) IsDir() bool {
|
|
return false
|
|
}
|
|
|
|
func (d *torrentFile) Close() error {
|
|
err := d.reader.Close()
|
|
d.reader = nil
|
|
return err
|
|
}
|
|
|
|
func (d *torrentFile) Read(p []byte) (n int, err error) {
|
|
d.load()
|
|
return d.reader.Read(p)
|
|
}
|
|
|
|
func (d *torrentFile) ReadAt(p []byte, off int64) (n int, err error) {
|
|
d.load()
|
|
return d.reader.ReadAt(p, off)
|
|
}
|