From e4001dbccbe82add95f32450e4d082133a197e8e Mon Sep 17 00:00:00 2001
From: Julian Hurst <ark@mansus.space>
Date: Thu, 10 Aug 2023 15:33:37 +0200
Subject: Add support for a custom db path

---
 db.go   | 2 +-
 main.go | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/db.go b/db.go
index 9de5e2d..b70e885 100644
--- a/db.go
+++ b/db.go
@@ -17,7 +17,7 @@ type User struct {
 }
 
 func InitAndGetDB(driverName, dataSourceName string) (*sql.DB, error) {
-	db, err := sql.Open("sqlite3", "./db/test.db")
+	db, err := sql.Open("sqlite3", dataSourceName)
 	if err != nil {
 		return nil, err
 	}
diff --git a/main.go b/main.go
index 15e7281..04bcf7c 100644
--- a/main.go
+++ b/main.go
@@ -629,9 +629,11 @@ func upload(w http.ResponseWriter, r *http.Request) {
 
 func main() {
 	p := flag.Int("p", 8080, "the port to bind to")
+	dbPath := flag.String("d", "./db/test.db", "the db to connect to")
 	flag.Parse()
 	var err error
-	db, err = InitAndGetDB("sqlite3", "./db/test.db")
+	log.Printf("Connecting to db: %s\n", *dbPath)
+	db, err = InitAndGetDB("sqlite3", *dbPath)
 	if err != nil {
 		panic(err)
 	}
-- 
cgit v1.2.3