From 4a1c36463bc3813788899d087e86521356dbf06c Mon Sep 17 00:00:00 2001 From: Sascha Andres Date: Wed, 6 Jun 2018 07:20:47 +0200 Subject: [PATCH] feat: allow to change database file name restore complete customization #93 --- main.go | 3 ++- path-generator.go | 12 ++++++++++++ 2 files changed, 14 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index cd82f9b9..d98dd553 100644 --- a/main.go +++ b/main.go @@ -12,10 +12,11 @@ import ( ) var dataDir = "." +var databaseName = "shiori.db" func main() { // Open database - dbPath := fp.Join(dataDir, "shiori.db") + dbPath := fp.Join(dataDir, databaseName) sqliteDB, err := dt.OpenSQLiteDatabase(dbPath) checkError(err) diff --git a/path-generator.go b/path-generator.go index 1662279d..c9b9fde8 100644 --- a/path-generator.go +++ b/path-generator.go @@ -11,6 +11,7 @@ import ( func init() { // Get data directory dataDir = getDataDirectory() + databaseName = getDatabaseName() // Make sure directory exist os.MkdirAll(dataDir, os.ModePerm) @@ -33,3 +34,14 @@ func getDataDirectory() string { // When all fail, use current working directory return "." } + +func getDatabaseName() string { + // Try to look at environment variables + databaseName, found := os.LookupEnv("ENV_SHIORI_DB") + if found { + return databaseName + } + + // When all fail, use current working directory + return "shiori.db" +}