Added verbose Mode, Makefile, Travis CI and Improved CLI

Author: ishanjain28

.travis.yml

@@ -0,0 +1,7 @@
+language: go
+
+go:
+  - 1.7.x
+  - 1.8.x
+  - 1.9.x
+  - master

Makefile

@@ -0,0 +1,11 @@
+build:
+	GOOS=linux GOARCH=amd64 go build -o=pluto.linux.amd64
+	GOOS=linux GOARCH=386 go build -o=pluto.linux.i386
+	GOOS=windows GOARCH=amd64 go build -o=pluto.amd64.exe
+	GOOS=windows GOARCH=386 go build -o=pluto.i386.exe
+
+clean:
+	rm pluto.linux.i386
+	rm pluto.linux.amd64
+	rm pluto.i386.exe
+	rm pluto.amd64.exe

pluto/pluto.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"io"
 	"io/ioutil"
+	"log"
 	"net/http"
 	"net/url"
 	"os"
@@ -28,7 +29,7 @@ type FileMeta struct {
 // then downloads the file by dividing it into given number of parts and downloading all parts concurrently.
 // If any error occurs in the downloading stage of any part, It'll wait for 2 seconds, Discard the existing part and restart it.
 // Discarding whatever bytes were downloaded isn't exactly a smart, So, I'll also be implementing a feature where it can skip over what is already downloaded.
-func Download(linkp *url.URL, parts int) (*os.File, error) {
+func Download(linkp *url.URL, parts int, verbose bool) (*os.File, error) {
 
 	if linkp == nil {
 		return nil, fmt.Errorf("No URL Provided")
@@ -63,10 +64,10 @@ func Download(linkp *url.URL, parts int) (*os.File, error) {
 		}
 	}
 
-	return startDownload(workers)
+	return startDownload(workers, verbose)
 }
 
-func startDownload(w []*worker) (*os.File, error) {
+func startDownload(w []*worker, verbose bool) (*os.File, error) {
 
 	var wg sync.WaitGroup
 	wg.Add(len(w))
@@ -91,6 +92,10 @@ func startDownload(w []*worker) (*os.File, error) {
 						cerr <- err
 						return
 					}
+
+					if verbose {
+						log.Printf("error in downloading a part: %v", err)
+					}
 					continue
 				}
 

pluto_cli.go

@@ -27,8 +27,8 @@ func main() {
 		os.Exit(0)
 	}()
 
-	parts := flag.Int("part", 16, "Number of Download parts")
-
+	parts := flag.Int("part", 32, "Number of Download parts")
+	verbose := flag.Bool("verbose", false, "Enable Verbose Mode")
 	flag.Parse()
 	urls := []string{}
 	u := ""
@@ -40,7 +40,7 @@ func main() {
 			log.Fatalln("No URL Provided")
 		}
 
-		download(u, *parts)
+		download(u, *parts, *verbose)
 	} else {
 
 		if strings.Contains(os.Args[1], "--part") {
@@ -57,33 +57,39 @@ func main() {
 			}
 		}
 		for _, v := range urls {
-			download(v, *parts)
+			download(v, *parts, *verbose)
 		}
 	}
 
 }
 
-func download(u string, parts int) {
+func download(u string, parts int, verbose bool) {
 	a := time.Now()
 
+	defer func() { select {} }()
+
 	up, err := url.Parse(u)
 	if err != nil {
-		log.Fatalln("Invalid URL")
+		log.Println("Invalid URL")
+		return
 	}
 
-	fname := filepath.Base(up.String())
+	fname := strings.Split(filepath.Base(up.String()), "?")[0]
+	fmt.Printf("Starting Download with %d parts\n", parts)
 
 	fmt.Printf("Downloading %s\n", up.String())
 
-	f, err := pluto.Download(up, parts)
+	f, err := pluto.Download(up, parts, verbose)
 	if err != nil {
-		log.Fatalln(err)
+		log.Println(err)
+		return
 	}
 	defer f.Close()
 
 	file, err := os.Create(fname)
 	if err != nil {
-		log.Fatalln(err.Error())
+		log.Println(err.Error())
+		return
 	}
 
 	defer file.Close()
@@ -92,6 +98,6 @@ func download(u string, parts int) {
 	fmt.Printf("Downloaded %s in %s\n", up.String(), time.Since(a))
 	_, err = io.Copy(file, f)
 	if err != nil {
-		log.Fatalln(err.Error())
+		log.Println(err.Error())
 	}
 }