Introduce SSL flag

lukasmartinelli · Jan 20, 2016 · e982dd1 · e982dd1
1 parent 8977995
commit e982dd1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/pgfutter.go b/pgfutter.go
@@ -57,6 +57,10 @@ func main() {
 			Usage:  "username",
 			EnvVar: "DB_USER",
 		},
+		cli.BoolFlag{
+			Name:  "ssl",
+			Usage: "require ssl mode",
+		},
 		cli.StringFlag{
 			Name:   "pass, pw",
 			Value:  "",

diff --git a/postgres.go b/postgres.go
@@ -85,6 +85,9 @@ func postgresify(identifier string) string {
 //parse sql connection string from cli flags
 func parseConnStr(c *cli.Context) string {
 	otherParams := "sslmode=disable connect_timeout=5"
+	if c.GlobalBool("ssl") {
+		otherParams = "sslmode=require connect_timeout=5"
+	}
 	return fmt.Sprintf("user=%s dbname=%s password='%s' host=%s port=%s %s",
 		c.GlobalString("username"),
 		c.GlobalString("dbname"),