Version 3 starter

backend/internal/database/helpers.go

@@ -0,0 +1,46 @@
+package database
+
+import (
+	"fmt"
+	"strings"
+
+	"npm/internal/errors"
+	"npm/internal/model"
+	"npm/internal/util"
+)
+
+const (
+	// DateFormat for DateFormat
+	DateFormat = "2006-01-02"
+	// DateTimeFormat for DateTimeFormat
+	DateTimeFormat = "2006-01-02T15:04:05"
+)
+
+// GetByQuery returns a row given a query, populating the model given
+func GetByQuery(model interface{}, query string, params []interface{}) error {
+	db := GetInstance()
+	if db != nil {
+		err := db.Get(model, query, params...)
+		return err
+	}
+
+	return errors.ErrDatabaseUnavailable
+}
+
+// BuildOrderBySQL takes a `Sort` slice and constructs a query fragment
+func BuildOrderBySQL(columns []string, sort *[]model.Sort) (string, []model.Sort) {
+	var sortStrings []string
+	var newSort []model.Sort
+	for _, sortItem := range *sort {
+		if util.SliceContainsItem(columns, sortItem.Field) {
+			sortStrings = append(sortStrings, fmt.Sprintf("`%s` %s", sortItem.Field, sortItem.Direction))
+			newSort = append(newSort, sortItem)
+		}
+	}
+
+	if len(sortStrings) > 0 {
+		return fmt.Sprintf("ORDER BY %s", strings.Join(sortStrings, ", ")), newSort
+	}
+
+	return "", newSort
+}

backend/internal/database/setup.go

@@ -0,0 +1,38 @@
+package database
+
+import (
+	"database/sql"
+
+	"npm/internal/config"
+	"npm/internal/errors"
+	"npm/internal/logger"
+)
+
+// CheckSetup Quick check by counting the number of users in the database
+func CheckSetup() {
+	query := `SELECT COUNT(*) FROM "user" WHERE is_deleted = $1 and is_disabled = $2`
+	db := GetInstance()
+
+	if db != nil {
+		row := db.QueryRowx(query, false, false)
+		var totalRows int
+		queryErr := row.Scan(&totalRows)
+		if queryErr != nil && queryErr != sql.ErrNoRows {
+			logger.Error("SetupError", queryErr)
+			return
+		}
+		if totalRows == 0 {
+			logger.Warn("No users found, starting in Setup Mode")
+		} else {
+			config.IsSetup = true
+			logger.Info("Application is setup")
+		}
+
+		if config.ErrorReporting {
+			logger.Warn("Error reporting is enabled - Application Errors WILL be sent to Sentry, you can disable this in the Settings interface")
+		}
+
+	} else {
+		logger.Error("DatabaseError", errors.ErrDatabaseUnavailable)
+	}
+}

backend/internal/database/sqlite.go

@@ -0,0 +1,73 @@
+package database
+
+import (
+	"fmt"
+	"os"
+
+	"npm/internal/config"
+	"npm/internal/logger"
+
+	"github.com/jmoiron/sqlx"
+
+	// Blank import for Sqlite
+	_ "github.com/mattn/go-sqlite3"
+)
+
+var dbInstance *sqlx.DB
+
+// NewDB creates a new connection
+func NewDB() {
+	logger.Info("Creating new DB instance")
+	db := SqliteDB()
+	if db != nil {
+		dbInstance = db
+	}
+}
+
+// GetInstance returns an existing or new instance
+func GetInstance() *sqlx.DB {
+	if dbInstance == nil {
+		NewDB()
+	} else if err := dbInstance.Ping(); err != nil {
+		NewDB()
+	}
+
+	return dbInstance
+}
+
+// SqliteDB Create sqlite client
+func SqliteDB() *sqlx.DB {
+	dbFile := fmt.Sprintf("%s/nginxproxymanager.db", config.Configuration.DataFolder)
+	autocreate(dbFile)
+	db, err := sqlx.Open("sqlite3", dbFile)
+	if err != nil {
+		logger.Error("SqliteError", err)
+		return nil
+	}
+
+	return db
+}
+
+// Commit will close and reopen the db file
+func Commit() *sqlx.DB {
+	if dbInstance != nil {
+		err := dbInstance.Close()
+		if err != nil {
+			logger.Error("DatabaseCloseError", err)
+		}
+	}
+	NewDB()
+	return dbInstance
+}
+
+func autocreate(dbFile string) {
+	if _, err := os.Stat(dbFile); os.IsNotExist(err) {
+		// Create it
+		logger.Info("Creating Sqlite DB: %s", dbFile)
+		_, err = os.Create(dbFile)
+		if err != nil {
+			logger.Error("FileCreateError", err)
+		}
+		Commit()
+	}
+}