[rget]fix error handling

Author: imura81gt

kadai3/imura81gt/rget/cmd/rget/main.go

@@ -27,5 +27,9 @@ func main() {
 
 	option.URL = urls[0]
 	fmt.Println(option)
-	rget.Run(option)
+	err := rget.Run(option)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "err: %s", err)
+		os.Exit(1)
+	}
 }

kadai3/imura81gt/rget/cmd/rget/main_test.go

@@ -0,0 +1,5 @@
+package main
+
+import "testing"
+
+func TestMain(t *testing.T) {}

kadai3/imura81gt/rget/rget.go

@@ -38,32 +38,33 @@ func (u *Unit) Write(data []byte) (int, error) {
 
 type Units []Unit
 
-func Run(option Option) {
+func Run(option Option) error {
 	fmt.Printf("%+v\n", option)
 	err := option.checkingHeaders()
 	if err != nil {
-		fmt.Errorf("%s", err)
+		return fmt.Errorf("%s", err)
 	}
 
 	option.divide()
 
 	tmpDir, err := ioutil.TempDir("", "rget")
 	if err != nil {
-		fmt.Errorf("%s", err)
+		return fmt.Errorf("%s", err)
 	}
 	defer os.RemoveAll(tmpDir)
 	fmt.Println(tmpDir)
 
 	err = option.parallelDownload(tmpDir)
 	if err != nil {
-		fmt.Errorf("%s", err)
+		return fmt.Errorf("%s", err)
 	}
 
 	err = option.combine(tmpDir)
 	if err != nil {
-		fmt.Errorf("%s", err)
+		return fmt.Errorf("%s", err)
 	}
 
+	return nil
 }
 
 func (o *Option) checkingHeaders() error {
@@ -173,6 +174,8 @@ func (o *Option) downloadWithContext(
 	req.Header.Set("Range", byteRange)
 
 	client := http.DefaultClient
+	// TODO: should check resp.StatusCode.
+	//   client.Do cannot seems to return the err when statusCode is 50x etc.
 	resp, err := client.Do(req)
 	if err != nil {
 		fmt.Printf("client err: %s", err)