From 3af77b6beab43b6e43e76ff2ebc90e8ae631dc52 Mon Sep 17 00:00:00 2001 From: Atridad Lahiji Date: Fri, 22 Nov 2024 17:34:41 -0600 Subject: [PATCH] no --- docker-compose.yml | 3 +-- lib/db.go | 11 ++++++++--- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/docker-compose.yml b/docker-compose.yml index ade1ef3..9ce7178 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -1,6 +1,5 @@ services: - himbot: - container_name: himbot + server: image: ${IMAGE} ports: - "3117:3000" diff --git a/lib/db.go b/lib/db.go index 968941d..e2b3b59 100644 --- a/lib/db.go +++ b/lib/db.go @@ -18,11 +18,16 @@ var DBConnector *libsql.Connector func InitDB() error { // Determine DB path based on /data directory existence - dbName := "file:./himbot.db" + var dbPath string + if _, err := os.Stat("/data"); os.IsNotExist(err) { + dbPath = "file:./himbot.db" + } else { + dbPath = "file:/data/himbot.db" + } - db, err := sql.Open("libsql", dbName) + db, err := sql.Open("libsql", dbPath) if err != nil { - fmt.Fprintf(os.Stderr, "failed to open db %s", err) + fmt.Fprintf(os.Stderr, "failed to open db %s: %v", dbPath, err) os.Exit(1) }