1 // Downloads torrents from the command-line.
14 "github.com/anacrolix/torrent/iplist"
16 "github.com/anacrolix/envpprof"
17 "github.com/anacrolix/tagflag"
18 "github.com/dustin/go-humanize"
19 "github.com/gosuri/uiprogress"
20 "golang.org/x/time/rate"
22 "github.com/anacrolix/torrent"
23 "github.com/anacrolix/torrent/metainfo"
24 "github.com/anacrolix/torrent/storage"
27 var progress = uiprogress.New()
29 func torrentBar(t *torrent.Torrent) {
30 bar := progress.AddBar(1)
32 bar.AppendFunc(func(*uiprogress.Bar) (ret string) {
40 } else if t.BytesCompleted() == t.Info().TotalLength() {
43 return fmt.Sprintf("downloading (%s/%s)", humanize.Bytes(uint64(t.BytesCompleted())), humanize.Bytes(uint64(t.Info().TotalLength())))
46 bar.PrependFunc(func(*uiprogress.Bar) string {
51 tl := int(t.Info().TotalLength())
58 bc := t.BytesCompleted()
60 time.Sleep(time.Second)
65 func addTorrents(client *torrent.Client) {
66 for _, arg := range flags.Torrent {
67 t := func() *torrent.Torrent {
68 if strings.HasPrefix(arg, "magnet:") {
69 t, err := client.AddMagnet(arg)
71 log.Fatalf("error adding magnet: %s", err)
74 } else if strings.HasPrefix(arg, "http://") || strings.HasPrefix(arg, "https://") {
75 response, err := http.Get(arg)
77 log.Fatalf("Error downloading torrent file: %s", err)
80 metaInfo, err := metainfo.Load(response.Body)
81 defer response.Body.Close()
83 fmt.Fprintf(os.Stderr, "error loading torrent file %q: %s\n", arg, err)
86 t, err := client.AddTorrent(metaInfo)
91 } else if strings.HasPrefix(arg, "infohash:") {
92 t, _ := client.AddTorrentInfoHash(metainfo.NewHashFromHex(strings.TrimPrefix(arg, "infohash:")))
95 metaInfo, err := metainfo.LoadFromFile(arg)
97 fmt.Fprintf(os.Stderr, "error loading torrent file %q: %s\n", arg, err)
100 t, err := client.AddTorrent(metaInfo)
108 t.AddPeers(func() (ret []torrent.Peer) {
109 for _, ta := range flags.TestPeer {
110 ret = append(ret, torrent.Peer{
125 Mmap bool `help:"memory-map torrent data"`
126 TestPeer []*net.TCPAddr `help:"addresses of some starting peers"`
127 Seed bool `help:"seed after download is complete"`
128 Addr *net.TCPAddr `help:"network listen addr"`
129 UploadRate tagflag.Bytes `help:"max piece bytes to send per second"`
130 DownloadRate tagflag.Bytes `help:"max bytes per second down from peers"`
132 PackedBlocklist string
134 Torrent []string `arity:"+" help:"torrent file path or magnet uri"`
140 func stdoutAndStderrAreSameFile() bool {
141 fi1, _ := os.Stdout.Stat()
142 fi2, _ := os.Stderr.Stat()
143 return os.SameFile(fi1, fi2)
147 log.SetFlags(log.LstdFlags | log.Lshortfile)
148 tagflag.Parse(&flags)
149 clientConfig := torrent.Config{
153 if flags.PackedBlocklist != "" {
154 blocklist, err := iplist.MMapPackedFile(flags.PackedBlocklist)
156 log.Fatalf("error loading blocklist: %s", err)
158 defer blocklist.Close()
159 clientConfig.IPBlocklist = blocklist
162 clientConfig.DefaultStorage = storage.NewMMap("")
164 if flags.Addr != nil {
165 clientConfig.SetListenAddr(flags.Addr.String())
167 if flags.UploadRate != -1 {
168 clientConfig.UploadRateLimiter = rate.NewLimiter(rate.Limit(flags.UploadRate), 256<<10)
170 if flags.DownloadRate != -1 {
171 clientConfig.DownloadRateLimiter = rate.NewLimiter(rate.Limit(flags.DownloadRate), 1<<20)
174 client, err := torrent.NewClient(&clientConfig)
176 log.Fatalf("error creating client: %s", err)
179 // Write status on the root path on the default HTTP muxer. This will be
180 // bound to localhost somewhere if GOPPROF is set, thanks to the envpprof
182 http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
183 client.WriteStatus(w)
185 if stdoutAndStderrAreSameFile() {
186 log.SetOutput(progress.Bypass())
190 if client.WaitAll() {
191 log.Print("downloaded ALL the torrents")
193 log.Fatal("y u no complete torrents?!")
198 expvar.Do(func(kv expvar.KeyValue) {
199 fmt.Printf("%s: %s\n", kv.Key, kv.Value)