Moved everything out of internal

database/database.go

@@ -0,0 +1,69 @@
+package database
+
+import (
+	"fmt"
+	"github.com/gobuffalo/packr/v2"
+	"github.com/golang-migrate/migrate/v4"
+	"github.com/golang-migrate/migrate/v4/source"
+	"github.com/jmoiron/sqlx"
+	_ "github.com/mattn/go-sqlite3"
+	"github.com/stashapp/stash/logger"
+	"runtime"
+	"strings"
+)
+
+var DB *sqlx.DB
+var appSchemaVersion uint = 1
+
+func Initialize(databasePath string) {
+	runMigrations(databasePath)
+
+	// https://github.com/mattn/go-sqlite3
+	conn, err := sqlx.Open("sqlite3", "file:"+databasePath+"?_fk=true")
+	conn.SetMaxOpenConns(10)
+	if err != nil {
+		logger.Fatalf("db.Open(): %q\n", err)
+	}
+	DB = conn
+}
+
+func Reset(databasePath string) {
+	_, _ = DB.Exec("PRAGMA writable_schema = 1;")
+	_, _ = DB.Exec("delete from sqlite_master where type in ('table', 'index', 'trigger');")
+	_, _ = DB.Exec("PRAGMA writable_schema = 0;")
+	_, _ = DB.Exec("VACUUM;")
+	_, _ = DB.Exec("PRAGMA INTEGRITY_CHECK;")
+	runMigrations(databasePath)
+}
+
+// Migrate the database
+func runMigrations(databasePath string) {
+	migrationsBox := packr.New("Migrations Box", "./migrations")
+	packrSource := &Packr2Source{
+		Box:        migrationsBox,
+		Migrations: source.NewMigrations(),
+	}
+
+	// Golang migrate has an issue where the \ isn't recognized as a valid on windows
+	if runtime.GOOS == "windows" {
+		databasePath = strings.Replace(databasePath, "\\", "/", -1)
+	}
+	s, _ := WithInstance(packrSource)
+	m, err := migrate.NewWithSourceInstance(
+		"packr2",
+		s,
+		fmt.Sprintf("sqlite3://%s", "file:"+databasePath),
+	)
+	if err != nil {
+		panic(err.Error())
+	}
+
+	databaseSchemaVersion, _, _ := m.Version()
+	stepNumber := appSchemaVersion - databaseSchemaVersion
+	if stepNumber != 0 {
+		err = m.Steps(int(stepNumber))
+		if err != nil {
+			panic(err.Error())
+		}
+	}
+}