cmd/torrent: Add flags -disableTrackers and -seed

diff --git a/cmd/torrent/main.go b/cmd/torrent/main.go
index 93b02407fbd3a6f351e16a2a48e9a42e5bf38b10..59c1d801c4d3a034a33dc83f5a508750dc4b6197 100644 (file)
--- a/cmd/torrent/main.go
+++ b/cmd/torrent/main.go
@@ -19,7 +19,9 @@ var (
        testPeer    = flag.String("testPeer", "", "bootstrap peer address")
        profAddr    = flag.String("profAddr", "", "http serve address")
        // TODO: Check the default torrent listen port.
-       listenAddr = flag.String("listenAddr", ":6882", "incoming connection address")
+       listenAddr      = flag.String("listenAddr", ":6882", "incoming connection address")
+       disableTrackers = flag.Bool("disableTrackers", false, "disable trackers")
+       seed            = flag.Bool("seed", false, "seed after downloading")
 )
 
 func init() {
@@ -40,8 +42,9 @@ func main() {
                go http.ListenAndServe(*profAddr, nil)
        }
        client := torrent.Client{
-               DataDir:  *downloadDir,
-               Listener: makeListener(),
+               DataDir:         *downloadDir,
+               Listener:        makeListener(),
+               DisableTrackers: *disableTrackers,
        }
        client.Start()
        defer client.Stop()
@@ -76,6 +79,9 @@ func main() {
                        log.Fatal(err)
                }
        }
+       if *seed {
+               select {}
+       }
        if client.WaitAll() {
                log.Print("all torrents completed!")
        } else {