feat: allow to change database file name

restore complete customization #93

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)
 

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"
+}