Get password interactively if not provided.

shahinam · Jul 30, 2017 · cd99536 · cd99536
1 parent fc7c621
commit cd99536
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 3 deletions.
diff --git a/Gopkg.lock b/Gopkg.lock
diff --git a/client/client.go b/client/client.go
@@ -104,7 +104,7 @@ func (c *Client) DownloadCourse(co *Course) {
 		filePath := filepath.Join(co.SaveDir, coursePath, fileName)
 		i++
 
-		fmt.Printf("Downloading: %s", link.Title)
+		fmt.Printf("Downloading Video: %s", link.Title)
 		videoURL, err := c.GetVideoUrl(link.URL, co)
 		if err != nil {
 			fmt.Printf(" ERROR: Unable to grab video file\n")

diff --git a/main.go b/main.go
@@ -4,8 +4,11 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"syscall"
 
 	"github.com/shahinam/cloudac-dl/client"
+
+	"golang.org/x/crypto/ssh/terminal"
 )
 
 var version = "1.1.0"
@@ -74,11 +77,18 @@ func parseCommandLineArgs() *CommandLineOptions {
 
 	flag.Parse()
 
-	if flag.NArg() == 0 || args.userName == "" || args.passWord == "" {
+	if flag.NArg() == 0 || args.userName == "" {
 		flag.Usage()
 		os.Exit(1)
 	}
 
+	// If password is not provided - get it interactively.
+	if args.passWord == "" {
+		fmt.Print("Please enter password: ")
+		password, _ := terminal.ReadPassword(int(syscall.Stdin))
+		args.passWord = string(password)
+	}
+
 	args.courseURL = flag.Arg(0)
 
 	return args