feat: support custom tracker url (#59)

internal/protocol/bt/config.go

@@ -0,0 +1,5 @@
+package bt
+
+type config struct {
+	Trackers []string `json:"trackers"`
+}

internal/protocol/bt/fetcher.go

@@ -17,7 +17,7 @@ import (
 var client *torrent.Client
 
 type Fetcher struct {
-	Ctl controller.Controller
+	Ctl *controller.Controller
 
 	torrent *torrent.Torrent
 	res     *base.Resource
@@ -29,7 +29,11 @@ type Fetcher struct {
 	torrentPaths map[string]string
 }
 
-func (f *Fetcher) Setup(ctl controller.Controller) (err error) {
+func (f *Fetcher) Name() string {
+	return "bt"
+}
+
+func (f *Fetcher) Setup(ctl *controller.Controller) (err error) {
 	f.Ctl = ctl
 	var once sync.Once
 	once.Do(func() {
@@ -105,9 +109,7 @@ func (f *Fetcher) Resolve(req *base.Request) (res *base.Resource, err error) {
 		Name:  f.torrent.Name(),
 		Range: true,
 		Files: make([]*base.FileInfo, len(f.torrent.Files())),
-		Extra: map[string]any{
-			"infoHash": f.torrent.InfoHash().String(),
-		},
+		Hash:  f.torrent.InfoHash().String(),
 	}
 	for i, file := range f.torrent.Files() {
 		res.Files[i] = &base.FileInfo{
@@ -130,10 +132,9 @@ func (f *Fetcher) Create(res *base.Resource, opts *base.Options) (err error) {
 		}
 	}
 	if f.opts.Path != "" {
-		if extra, ok := res.Extra.(map[string]any); ok {
-			f.torrentPaths[extra["infoHash"].(string)] = f.opts.Path
-		}
+		f.torrentPaths[res.Hash] = f.opts.Path
 	}
+
 	f.progress = make(fetcher.Progress, len(f.opts.SelectFiles))
 	f.ready.Store(false)
 	return nil
@@ -192,6 +193,18 @@ func (f *Fetcher) addTorrent(url string, resolve bool) (err error) {
 	if err != nil {
 		return
 	}
+	var cfg config
+	exist, err := f.Ctl.GetConfig(&cfg)
+	if err != nil {
+		return
+	}
+	if exist && len(cfg.Trackers) > 0 {
+		announceList := make([][]string, 0)
+		for _, tracker := range cfg.Trackers {
+			announceList = append(announceList, []string{tracker})
+		}
+		f.torrent.AddTrackers(announceList)
+	}
 	<-f.torrent.GotInfo()
 	if resolve {
 		return

internal/protocol/bt/fetcher_test.go

@@ -1,15 +1,24 @@
 package bt
 
 import (
+	"encoding/json"
 	"github.com/monkeyWie/gopeed/internal/controller"
 	"github.com/monkeyWie/gopeed/internal/fetcher"
+	"github.com/monkeyWie/gopeed/internal/test"
 	"github.com/monkeyWie/gopeed/pkg/base"
 	"reflect"
 	"testing"
 )
 
 func TestFetcher_Resolve_Torrent(t *testing.T) {
-	fetcher := buildFetcher()
+	doResolve(t, buildFetcher())
+}
+
+func TestFetcher_Config(t *testing.T) {
+	doResolve(t, buildConfigFetcher())
+}
+
+func doResolve(t *testing.T, fetcher fetcher.Fetcher) {
 	res, err := fetcher.Resolve(&base.Request{
 		URL: "./testdata/ubuntu-22.04-live-server-amd64.iso.torrent",
 	})
@@ -28,9 +37,7 @@ func TestFetcher_Resolve_Torrent(t *testing.T) {
 				Size: 1466714112,
 			},
 		},
-		Extra: map[string]any{
-			"infoHash": "8a55cfbd5ca5d11507364765936c4f9e55b253ed",
-		},
+		Hash: "8a55cfbd5ca5d11507364765936c4f9e55b253ed",
 	}
 	if !reflect.DeepEqual(want, res) {
 		t.Errorf("Resolve() got = %v, want %v", res, want)
@@ -42,3 +49,21 @@ func buildFetcher() fetcher.Fetcher {
 	fetcher.Setup(controller.NewController())
 	return fetcher
 }
+
+func buildConfigFetcher() fetcher.Fetcher {
+	fetcher := new(FetcherBuilder).Build()
+	newController := controller.NewController()
+	mockCfg := config{
+		Trackers: []string{
+			"udp://tracker.birkenwald.de:6969/announce",
+			"udp://tracker.bitsearch.to:1337/announce",
+		}}
+	newController.GetConfig = func(v any) (bool, error) {
+		if err := json.Unmarshal([]byte(test.ToJson(mockCfg)), v); err != nil {
+			return false, err
+		}
+		return true, nil
+	}
+	fetcher.Setup(newController)
+	return fetcher
+}