Separate torrent sources source file

[btrtrc.git] / sources.go

diff --git a/sources.go b/sources.go
new file mode 100644 (file)
index 0000000..b2d5b94
--- /dev/null
+++ b/sources.go
@@ -0,0 +1,71 @@
+package torrent
+
+import (
+       "context"
+       "net/http"
+
+       "github.com/anacrolix/log"
+       "github.com/anacrolix/torrent/bencode"
+       "github.com/anacrolix/torrent/metainfo"
+)
+
+func (t *Torrent) useSources(sources []string) {
+       select {
+       case <-t.Closed():
+               return
+       case <-t.GotInfo():
+               return
+       default:
+       }
+       for _, s := range sources {
+               _, loaded := t.activeSources.LoadOrStore(s, struct{}{})
+               if loaded {
+                       continue
+               }
+               s := s
+               go func() {
+                       err := t.useActiveTorrentSource(s)
+                       _, loaded := t.activeSources.LoadAndDelete(s)
+                       if !loaded {
+                               panic(s)
+                       }
+                       level := log.Debug
+                       if err != nil {
+                               level = log.Warning
+                       }
+                       t.logger.Levelf(level, "used torrent source %q [err=%q]", s, err)
+               }()
+       }
+}
+
+func (t *Torrent) useActiveTorrentSource(source string) error {
+       ctx, cancel := context.WithCancel(context.Background())
+       defer cancel()
+       go func() {
+               select {
+               case <-t.GotInfo():
+               case <-t.Closed():
+               case <-ctx.Done():
+               }
+               cancel()
+       }()
+       mi, err := getTorrentSource(ctx, source, t.cl.httpClient)
+       if err != nil {
+               return err
+       }
+       return t.MergeSpec(TorrentSpecFromMetaInfo(&mi))
+}
+
+func getTorrentSource(ctx context.Context, source string, hc *http.Client) (mi metainfo.MetaInfo, err error) {
+       var req *http.Request
+       if req, err = http.NewRequestWithContext(ctx, http.MethodGet, source, nil); err != nil {
+               return
+       }
+       var resp *http.Response
+       if resp, err = hc.Do(req); err != nil {
+               return
+       }
+       defer resp.Body.Close()
+       err = bencode.NewDecoder(resp.Body).Decode(&mi)
+       return
+}