56 lines
841 B
Go
56 lines
841 B
Go
|
package loader
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/fs"
|
||
|
|
"path"
|
||
|
|
"path/filepath"
|
||
|
|
|
||
|
|
"github.com/distribyted/distribyted/config"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ Loader = &Folder{}
|
||
|
|
|
||
|
|
type Folder struct {
|
||
|
|
c []*config.Route
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewFolder(r []*config.Route) *Folder {
|
||
|
|
return &Folder{
|
||
|
|
c: r,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *Folder) ListMagnets() (map[string][]string, error) {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *Folder) ListTorrentPaths() (map[string][]string, error) {
|
||
|
|
out := make(map[string][]string)
|
||
|
|
for _, r := range f.c {
|
||
|
|
if r.TorrentFolder == "" {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
err := filepath.WalkDir(r.TorrentFolder, func(p string, d fs.DirEntry, err error) error {
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if d.IsDir() {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
if path.Ext(p) == ".torrent" {
|
||
|
|
out[r.Name] = append(out[r.Name], p)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return out, nil
|
||
|
|
}
|