From 58a41bea55fbae032b47a65a3fe0f5a5d85d9a13 Mon Sep 17 00:00:00 2001
From: Fishcake <128653975+fishcakeday@users.noreply.github.com>
Date: Thu, 6 Apr 2023 06:49:27 +0000
Subject: [PATCH] feat(param): add db directory parameter

---
 main.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index ecee14d..5681981 100644
--- a/main.go
+++ b/main.go
@@ -8,6 +8,7 @@ import (
 	"os"
 	"strings"
 	"time"
+	"path"
 
 	"github.com/cockroachdb/pebble"
 	"github.com/fiatjaf/makeinvoice"
@@ -23,6 +24,7 @@ type Settings struct {
 	Host   string `envconfig:"HOST" default:"0.0.0.0"`
 	Port   string `envconfig:"PORT" required:"true"`
 	Domain string `envconfig:"DOMAIN" required:"true"`
+	DBDirectory                 string `envconfig:"DB_DIR" required:"false" default:""`
 	// GlobalUsers means that user@ part is globally unique across all domains
 	// WARNING: if you toggle this existing users won't work anymore for safety reasons!
 	GlobalUsers                 bool   `envconfig:"GLOBAL_USERS" default:"false"`
@@ -74,7 +76,7 @@ func main() {
 		makeinvoice.TorProxyURL = s.TorProxyURL
 	}
 
-	dbName := fmt.Sprintf("%v-multiple.db", s.SiteName)
+	dbName := path.Join(s.DBDirectory,fmt.Sprintf("%v-multiple.db", s.SiteName))
 	if _, err := os.Stat(dbName); os.IsNotExist(err) || s.ForceMigrate {
 		for _, one := range getDomains(s.Domain) {
 			tryMigrate(one, dbName)