Basic Auth Context

internal/db/db.go

@@ -1,14 +1,14 @@
 package db
 
 import (
-    "fmt"
-    "path"
     "errors"
+    "path"
+    "fmt"
 
-    "gorm.io/gorm"
+    log "github.com/sirupsen/logrus"
     // "gorm.io/gorm/logger"
     "gorm.io/driver/sqlite"
-    log "github.com/sirupsen/logrus"
+    "gorm.io/gorm"
 
     "reichard.io/imagini/internal/config"
     "reichard.io/imagini/graph/model"
@@ -59,6 +59,7 @@ func (dbm *DBManager) bootstrapDatabase() {
         Username: "admin",
         AuthType: "Local",
         Password: &password,
+        Role: model.RoleAdmin,
     }
 
     err := dbm.CreateUser(user)
@@ -68,8 +69,6 @@ func (dbm *DBManager) bootstrapDatabase() {
     }
 }
 
-// func (dmb *DBManager) {}
-
 func (dbm *DBManager) QueryBuilder(dest interface{}, params []byte) (int64, error) {
     // TODO:
     //  - Where Filters

internal/db/users.go

@@ -7,7 +7,7 @@ import (
     "reichard.io/imagini/graph/model"
 )
 
-func (dbm *DBManager) CreateUser(user *model.User) error {
+func (dbm *DBManager) CreateUser (user *model.User) error {
     log.Info("[db] Creating user: ", user.Username)
     hashedPassword, err := bcrypt.GenerateFromPassword([]byte(*user.Password), bcrypt.DefaultCost)
     if err != nil {
@@ -27,6 +27,13 @@ func (dbm *DBManager) User (user *model.User) (model.User, error) {
     return foundUser, err
 }
 
+func (dbm *DBManager) Users () ([]*model.User, int64, error) {
+    var foundUsers []*model.User
+    var count int64
+    err := dbm.db.Find(&foundUsers).Count(&count).Error
+    return foundUsers, count, err
+}
+
 func (dbm *DBManager) DeleteUser (user model.User) error {
     return nil
 }