From: Matt Joiner <anacrolix@gmail.com>
Date: Fri, 16 Dec 2022 04:38:30 +0000 (+1100)
Subject: Add --port flag to announce
X-Git-Tag: v1.49.0~7^2~14
X-Git-Url: http://www.git.stargrave.org/?a=commitdiff_plain;h=e554aa19a691615f349bf6dae0601706ea47dc03;p=btrtrc.git

Add --port flag to announce
---

diff --git a/cmd/torrent/announce.go b/cmd/torrent/announce.go
index 94500d96..f48dfd2a 100644
--- a/cmd/torrent/announce.go
+++ b/cmd/torrent/announce.go
@@ -12,19 +12,24 @@ import (
 
 type AnnounceCmd struct {
 	Event    udp.AnnounceEvent
+	Port     *uint16
 	Tracker  string           `arg:"positional"`
 	InfoHash torrent.InfoHash `arg:"positional"`
 }
 
 func announceErr(flags AnnounceCmd) error {
+	req := tracker.AnnounceRequest{
+		InfoHash: flags.InfoHash,
+		Port:     uint16(torrent.NewDefaultClientConfig().ListenPort),
+		NumWant:  -1,
+		Event:    flags.Event,
+	}
+	if flags.Port != nil {
+		req.Port = *flags.Port
+	}
 	response, err := tracker.Announce{
 		TrackerUrl: flags.Tracker,
-		Request: tracker.AnnounceRequest{
-			InfoHash: flags.InfoHash,
-			Port:     uint16(torrent.NewDefaultClientConfig().ListenPort),
-			NumWant:  -1,
-			Event:    flags.Event,
-		},
+		Request:    req,
 	}.Do()
 	if err != nil {
 		return fmt.Errorf("doing announce: %w", err)