Merge pull request go-shiori#97 from sascha-andres/feature/database-name

feature/database name: allow setting a different database name
evvil · Jun 8, 2018 · e981017 · e981017
2 parents e63a9d0 + 4a1c364
commit e981017
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git 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
@@ -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"
+}