1 // Downloads torrents from the command-line.
17 _ "github.com/anacrolix/envpprof"
18 "github.com/dustin/go-humanize"
19 "github.com/jessevdk/go-flags"
21 "github.com/anacrolix/torrent"
22 "github.com/anacrolix/torrent/metainfo"
25 // fmt.Fprintf(os.Stderr, "Usage: %s \n", os.Args[0])
27 func resolvedPeerAddrs(ss []string) (ret []torrent.Peer, err error) {
28 for _, s := range ss {
30 addr, err = net.ResolveTCPAddr("tcp", s)
34 ret = append(ret, torrent.Peer{
42 func bytesCompleted(tc *torrent.Client) (ret int64) {
43 for _, t := range tc.Torrents() {
45 ret += t.BytesCompleted()
51 // Returns an estimate of the total bytes for all torrents.
52 func totalBytesEstimate(tc *torrent.Client) (ret int64) {
53 var noInfo, hadInfo int64
54 for _, t := range tc.Torrents() {
60 ret += info.TotalLength()
64 // Treat each torrent without info as the average of those with,
66 ret += (noInfo*ret + hadInfo - 1) / hadInfo
71 func progressLine(tc *torrent.Client) string {
72 return fmt.Sprintf("\033[K%s / %s\r", humanize.Bytes(uint64(bytesCompleted(tc))), humanize.Bytes(uint64(totalBytesEstimate(tc))))
75 func dstFileName(picked string) string {
76 parts := strings.Split(picked, "/")
77 return parts[len(parts)-1]
81 log.SetFlags(log.LstdFlags | log.Lshortfile)
82 var rootGroup struct {
83 Client torrent.Config `group:"Client Options"`
84 TestPeers []string `long:"test-peer" description:"address of peer to inject to every torrent"`
85 Pick string `long:"pick" description:"filename to pick"`
87 // Don't pass flags.PrintError because it's inconsistent with printing.
88 // https://github.com/jessevdk/go-flags/issues/132
89 parser := flags.NewParser(&rootGroup, flags.HelpFlag|flags.PassDoubleDash)
90 parser.Usage = "[OPTIONS] (magnet URI or .torrent file path)..."
91 posArgs, err := parser.Parse()
93 fmt.Fprintf(os.Stderr, "%s", "Download from the BitTorrent network.\n\n")
97 log.Printf("File to pick: %s", rootGroup.Pick)
99 testPeers, err := resolvedPeerAddrs(rootGroup.TestPeers)
104 if len(posArgs) == 0 {
105 fmt.Fprintln(os.Stderr, "no torrents specified")
109 tmpdir, err := ioutil.TempDir("", "torrent-pick-")
114 defer os.RemoveAll(tmpdir)
116 rootGroup.Client.DataDir = tmpdir
118 client, err := torrent.NewClient(&rootGroup.Client)
120 log.Fatalf("error creating client: %s", err)
122 http.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
123 client.WriteStatus(w)
127 dstName := dstFileName(rootGroup.Pick)
129 f, err := os.Create(dstName)
133 dstWriter := bufio.NewWriter(f)
135 done := make(chan struct{})
136 for _, arg := range posArgs {
137 t := func() *torrent.Torrent {
138 if strings.HasPrefix(arg, "magnet:") {
139 t, err := client.AddMagnet(arg)
141 log.Fatalf("error adding magnet: %s", err)
145 metaInfo, err := metainfo.LoadFromFile(arg)
149 t, err := client.AddTorrent(metaInfo)
156 t.AddPeers(testPeers)
161 for _, file := range t.Files() {
162 if file.DisplayPath() != rootGroup.Pick {
166 srcReader := file.NewReader()
167 defer srcReader.Close()
168 io.Copy(dstWriter, srcReader)
171 log.Print("file not found")
175 ticker := time.NewTicker(time.Second)
182 os.Stdout.WriteString(progressLine(client))
185 if rootGroup.Client.Seed {