Load torrent files from folder on start. (#149)

torrent/loader/config.go

@@ -17,7 +17,6 @@ func NewConfig(r []*config.Route) *Config {
 func (l *Config) ListMagnets() (map[string][]string, error) {
 	out := make(map[string][]string)
 	for _, r := range l.c {
-		out[r.Name] = make([]string, 0)
 		for _, t := range r.Torrents {
 			if t.MagnetURI == "" {
 				continue
@@ -33,7 +32,6 @@ func (l *Config) ListMagnets() (map[string][]string, error) {
 func (l *Config) ListTorrentPaths() (map[string][]string, error) {
 	out := make(map[string][]string)
 	for _, r := range l.c {
-		out[r.Name] = make([]string, 0)
 		for _, t := range r.Torrents {
 			if t.TorrentPath == "" {
 				continue

torrent/loader/folder.go

@@ -0,0 +1,55 @@
+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
+}

torrent/service.go

@@ -23,21 +23,21 @@ type Service struct {
 	mu  sync.Mutex
 	fss map[string]fs.Filesystem
 
-	cfgLoader loader.Loader
-	db        loader.LoaderAdder
+	loaders []loader.Loader
+	db      loader.LoaderAdder
 
 	log                     zerolog.Logger
 	addTimeout, readTimeout int
 }
 
-func NewService(cfg loader.Loader, db loader.LoaderAdder, stats *Stats, c *torrent.Client, addTimeout, readTimeout int) *Service {
+func NewService(loaders []loader.Loader, db loader.LoaderAdder, stats *Stats, c *torrent.Client, addTimeout, readTimeout int) *Service {
 	l := log.Logger.With().Str("component", "torrent-service").Logger()
 	return &Service{
 		log:         l,
 		s:           stats,
 		c:           c,
 		fss:         make(map[string]fs.Filesystem),
-		cfgLoader:   cfg,
+		loaders:     loaders,
 		db:          db,
 		addTimeout:  addTimeout,
 		readTimeout: readTimeout,
@@ -47,8 +47,10 @@ func NewService(cfg loader.Loader, db loader.LoaderAdder, stats *Stats, c *torre
 func (s *Service) Load() (map[string]fs.Filesystem, error) {
 	// Load from config
 	s.log.Info().Msg("adding torrents from configuration")
-	if err := s.load(s.cfgLoader); err != nil {
-		return nil, err
+	for _, loader := range s.loaders {
+		if err := s.load(loader); err != nil {
+			return nil, err
+		}
 	}
 
 	// Load from DB