Specify the database connection in CLI argsHEADmaster

1 files changed, 6 insertions, 1 deletions

diff --git a/main.go b/main.go
index 5639ea4..9088cd4 100644
--- a/main.go
+++ b/main.go
@@ -1,6 +1,7 @@
 package main
 
 import (
+	"fmt"
 	"log"
 	"os"
 
@@ -13,7 +14,11 @@ import (
 var app = tview.NewApplication()
 
 func main() {
-	db, err := postgres.Open("postgres://postgres:password@localhost:5432/postgres?sslmode=disable")
+	if len(os.Args) < 2 {
+		fmt.Fprintf(os.Stderr, "usage: %s <conn>\n", os.Args[0])
+		os.Exit(1)
+	}
+	db, err := postgres.Open(os.Args[1])
 	if err != nil {
 		log.Printf("could not connect to db: %v", err)
 		os.Exit(1)