From 0e4546b738e43b1bc0132450c8fff5a2a8434d58 Mon Sep 17 00:00:00 2001
From: Matt Joiner <anacrolix@gmail.com>
Date: Sun, 20 Feb 2022 16:16:57 +1100
Subject: [PATCH] Add scrape subcommand to cmd/torrent

---
 cmd/torrent/main.go   | 10 ++++++++++
 cmd/torrent/scrape.go | 44 +++++++++++++++++++++++++++++++++++++++++++
 go.mod                |  2 +-
 go.sum                |  3 ++-
 4 files changed, 57 insertions(+), 2 deletions(-)
 create mode 100644 cmd/torrent/scrape.go

diff --git a/cmd/torrent/main.go b/cmd/torrent/main.go
index 4f83762a..b8318abf 100644
--- a/cmd/torrent/main.go
+++ b/cmd/torrent/main.go
@@ -41,6 +41,16 @@ func mainErr() error {
 			}
 			return announceErr(cmd)
 		}),
+		args.Subcommand("scrape", func(p args.SubCmdCtx) error {
+			var cmd ScrapeCmd
+			err := p.NewParser().AddParams(
+				args.Pos("tracker", &cmd.Tracker),
+				args.Pos("infohash", &cmd.InfoHashes, args.Arity('+'))).Parse()
+			if err != nil {
+				return err
+			}
+			return scrape(cmd)
+		}),
 		args.Subcommand("download", func(p args.SubCmdCtx) error {
 			var dlc DownloadCmd
 			err := p.NewParser().AddParams(
diff --git a/cmd/torrent/scrape.go b/cmd/torrent/scrape.go
new file mode 100644
index 00000000..300f6282
--- /dev/null
+++ b/cmd/torrent/scrape.go
@@ -0,0 +1,44 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"net/url"
+
+	"github.com/anacrolix/torrent/tracker/udp"
+	"github.com/davecgh/go-spew/spew"
+
+	"github.com/anacrolix/torrent"
+)
+
+type ScrapeCmd struct {
+	Tracker    string `arg:"positional"`
+	InfoHashes []torrent.InfoHash
+}
+
+func scrape(flags ScrapeCmd) error {
+	trackerUrl, err := url.Parse(flags.Tracker)
+	if err != nil {
+		return fmt.Errorf("parsing tracker url: %w", err)
+	}
+	cc, err := udp.NewConnClient(udp.NewConnClientOpts{
+		Network: trackerUrl.Scheme,
+		Host:    trackerUrl.Host,
+		//Ipv6:    nil,
+		//Logger:  log.Logger{},
+	})
+	if err != nil {
+		return fmt.Errorf("creaing new udp tracker conn client: %w", err)
+	}
+	defer cc.Close()
+	var ihs []udp.InfoHash
+	for _, ih := range flags.InfoHashes {
+		ihs = append(ihs, ih)
+	}
+	scrapeOut, err := cc.Client.Scrape(context.TODO(), ihs)
+	if err != nil {
+		return fmt.Errorf("scraping: %w", err)
+	}
+	spew.Dump(scrapeOut)
+	return nil
+}
diff --git a/go.mod b/go.mod
index d3289d06..0ad647b5 100644
--- a/go.mod
+++ b/go.mod
@@ -5,7 +5,7 @@ go 1.16
 require (
 	github.com/RoaringBitmap/roaring v0.9.4
 	github.com/alexflint/go-arg v1.4.2
-	github.com/anacrolix/args v0.4.1-0.20211104085705-59f0fe94eb8f
+	github.com/anacrolix/args v0.5.0
 	github.com/anacrolix/chansync v0.3.0
 	github.com/anacrolix/dht/v2 v2.15.2-0.20220123034220-0538803801cb
 	github.com/anacrolix/envpprof v1.1.1
diff --git a/go.sum b/go.sum
index 82f1c13f..07d64747 100644
--- a/go.sum
+++ b/go.sum
@@ -77,8 +77,9 @@ github.com/alexflint/go-scalar v1.0.0/go.mod h1:GpHzbCOZXEKMEcygYQ5n/aa4Aq84zbxj
 github.com/alexflint/go-scalar v1.1.0 h1:aaAouLLzI9TChcPXotr6gUhq+Scr8rl0P9P4PnltbhM=
 github.com/alexflint/go-scalar v1.1.0/go.mod h1:LoFvNMqS1CPrMVltza4LvnGKhaSpc3oyLEBUZVhhS2o=
 github.com/anacrolix/args v0.3.0/go.mod h1:41JBnF8sKExNVLHPkCdL74jkZc3dSxAkGsk1TuKOUFI=
-github.com/anacrolix/args v0.4.1-0.20211104085705-59f0fe94eb8f h1:NEHQxPja+2pvaFtFMUC4k14X1kwInDtX2DSZUNcxJKQ=
 github.com/anacrolix/args v0.4.1-0.20211104085705-59f0fe94eb8f/go.mod h1:41JBnF8sKExNVLHPkCdL74jkZc3dSxAkGsk1TuKOUFI=
+github.com/anacrolix/args v0.5.0 h1:bZCkbtn4QMB4ow2g34R/oqTFwrI6IojysKop6+ZcOSs=
+github.com/anacrolix/args v0.5.0/go.mod h1:Fj/N2PehEwTBE5t/V/9xgTcxDkuYQ+5IBoFw/8gkldI=
 github.com/anacrolix/chansync v0.0.0-20210524073341-a336ebc2de92/go.mod h1:DZsatdsdXxD0WiwcGl0nJVwyjCKMDv+knl1q2iBjA2k=
 github.com/anacrolix/chansync v0.1.0/go.mod h1:DZsatdsdXxD0WiwcGl0nJVwyjCKMDv+knl1q2iBjA2k=
 github.com/anacrolix/chansync v0.2.1-0.20210910114620-14955c95ded9/go.mod h1:DZsatdsdXxD0WiwcGl0nJVwyjCKMDv+knl1q2iBjA2k=
-- 
2.48.1