Add interactive password input

go/cmd/ccql/main.go

@@ -6,10 +6,12 @@ import (
 	"log"
 	"os"
 	"os/user"
+	"syscall"
 
 	"github.com/github/ccql/go/logic"
 	"github.com/github/ccql/go/sql"
 	"github.com/github/ccql/go/text"
+	"golang.org/x/crypto/ssh/terminal"
 
 	golib_log "github.com/outbrain/golib/log"
 	"gopkg.in/gcfg.v1"
@@ -35,6 +37,7 @@ func main() {
 	help := flag.Bool("help", false, "Display usage")
 	user := flag.String("u", osUser, "MySQL username")
 	password := flag.String("p", "", "MySQL password")
+	askPassword := flag.Bool("ask-pass", false, "prompt for MySQL password")
 	credentialsFile := flag.String("C", "", "Credentials file, expecting [client] scope, with 'user', 'password' fields. Overrides -u and -p")
 	defaultSchema := flag.String("d", "information_schema", "Default schema to use")
 	hostsList := flag.String("h", "", "Comma or space delimited list of hosts in hostname[:port] format. If not given, hosts read from stdin")
@@ -105,6 +108,15 @@ func main() {
 		}
 	}
 
+	if *askPassword {
+		fmt.Print("Mysql password: ")
+		passwd, err := terminal.ReadPassword(int(syscall.Stdin))
+		if err != nil {
+			log.Fatalf("\nError while get password:", err)
+		}
+		*password = string(passwd)
+	}
+
 	if err := logic.QueryHosts(hosts, *user, *password, *defaultSchema, queries, *maxConcurrency, *timeout); err != nil {
 		os.Exit(1)
 	}