Finally Framework

internal/auth/auth.go

@@ -3,16 +3,26 @@ package auth
 import (
     "errors"
 	"gorm.io/gorm"
-    "reichard.io/imagini/internal/query"
+    "reichard.io/imagini/internal/db"
     "reichard.io/imagini/internal/models"
     log "github.com/sirupsen/logrus"
 )
 
-func AuthenticateUser(db *gorm.DB, creds models.APICredentials) bool {
+type AuthManager struct {
+    DB  *db.DBManager
+}
+
+func NewMgr(db *db.DBManager) *AuthManager {
+    return &AuthManager{
+        DB: db,
+    }
+}
+
+func (auth *AuthManager) AuthenticateUser(creds models.APICredentials) bool {
     // By Username
-    foundUser, err := query.User(db, models.User{Username: creds.User})
+    foundUser, err := auth.DB.User(models.User{Username: creds.User})
     if errors.Is(err, gorm.ErrRecordNotFound) {
-        foundUser, err = query.User(db, models.User{Email: creds.User})
+        foundUser, err = auth.DB.User(models.User{Email: creds.User})
     }
 
     // Error Checking