bugfix: error instead of panic for invalid connection string

cmd/mgmt/mgmt.go

@@ -143,15 +143,7 @@ func main() {
 		return
 	}
 
-	connStr := dbURL
+	store, err = authstore.NewStore(dbURL, mgmt.InitSQL)
-	// TODO url.Parse
-	if strings.Contains(connStr, "@localhost/") || strings.Contains(connStr, "@localhost:") {
-		connStr += "?sslmode=disable"
-	} else {
-		connStr += "?sslmode=required"
-	}
-
-	store, err = authstore.NewStore(connStr, mgmt.InitSQL)
 	if nil != err {
 		log.Fatal("connection error", err)
 		return

cmd/sqlstore/sqlstore.go

@@ -10,10 +10,18 @@ import (
 
 func main() {
 	connStr := "postgres://postgres:postgres@localhost:5432/postgres"
-	if strings.Contains(connStr, "@localhost/") || strings.Contains(connStr, "@localhost:") {
+
-		connStr += "?sslmode=disable"
+	if !strings.Contains(connStr, "sslmode=") {
+		sep := "?"
+		if strings.Contains(connStr, sep) {
+			sep = "&"
+		}
+		if strings.Contains(connStr, "@localhost/") ||
+			strings.Contains(connStr, "@localhost:") {
+			connStr += sep + "sslmode=disable"
 		} else {
-		connStr += "?sslmode=required"
+			connStr += sep + "sslmode=required"
+		}
 	}
 
 	store, err := authstore.NewStore(connStr, initSQL)

internal/mgmt/authstore/authstore_postgres_test.go

@@ -1,14 +1,6 @@
 package authstore
 
-import "strings"
-
 var connStr = "postgres://postgres:postgres@localhost/postgres"
 
 func init() {
-	// TODO url.Parse
-	if strings.Contains(connStr, "@localhost/") || strings.Contains(connStr, "@localhost:") {
-		connStr += "?sslmode=disable"
-	} else {
-		connStr += "?sslmode=required"
-	}
 }

internal/mgmt/authstore/postgresql.go

@@ -5,6 +5,7 @@ import (
 	"database/sql"
 	"fmt"
 	"io/ioutil"
+	"strings"
 	"time"
 
 	"git.rootprojects.org/root/telebit/assets/files"
@@ -16,9 +17,22 @@ import (
 
 var initSQL = "./postgres.init.sql"
 
-func NewStore(pgURL, initSQL string) (Store, error) {
+func NewStore(dbURL, initSQL string) (Store, error) {
 	// https://godoc.org/github.com/lib/pq
 
+	// TODO url.Parse
+	if !strings.Contains(dbURL, "sslmode=") {
+		sep := "?"
+		if strings.Contains(connStr, sep) {
+			sep = "&"
+		}
+		if strings.Contains(connStr, "@localhost/") || strings.Contains(connStr, "@localhost:") {
+			connStr += sep + "sslmode=disable"
+		} else {
+			connStr += sep + "sslmode=required"
+		}
+	}
+
 	f, err := files.Open(initSQL)
 	if nil != err {
 		return nil, err