Restructure project layout

internal/database/user.go

@@ -0,0 +1,91 @@
+package database
+
+import (
+	"errors"
+
+	"github.com/pushbits/server/internal/assert"
+	"github.com/pushbits/server/internal/model"
+
+	"gorm.io/gorm"
+)
+
+// CreateUser creates a user.
+func (d *Database) CreateUser(createUser model.CreateUser) (*model.User, error) {
+	user, err := createUser.IntoInternalUser(d.credentialsManager)
+	if err != nil {
+		return nil, err
+	}
+
+	return user, d.gormdb.Create(user).Error
+}
+
+// DeleteUser deletes a user.
+func (d *Database) DeleteUser(user *model.User) error {
+	if err := d.gormdb.Where("user_id = ?", user.ID).Delete(model.Application{}).Error; err != nil {
+		return err
+	}
+
+	return d.gormdb.Delete(user).Error
+}
+
+// UpdateUser updates a user.
+func (d *Database) UpdateUser(user *model.User) error {
+	return d.gormdb.Save(user).Error
+}
+
+// GetUserByID returns the user with the given ID or nil.
+func (d *Database) GetUserByID(ID uint) (*model.User, error) {
+	var user model.User
+
+	err := d.gormdb.First(&user, ID).Error
+
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+
+	assert.Assert(user.ID == ID)
+
+	return &user, err
+}
+
+// GetUserByName returns the user with the given name or nil.
+func (d *Database) GetUserByName(name string) (*model.User, error) {
+	var user model.User
+
+	err := d.gormdb.Where("name = ?", name).First(&user).Error
+
+	if errors.Is(err, gorm.ErrRecordNotFound) {
+		return nil, err
+	}
+
+	assert.Assert(user.Name == name)
+
+	return &user, err
+}
+
+// GetApplications returns the applications associated with a given user.
+func (d *Database) GetApplications(user *model.User) ([]model.Application, error) {
+	var applications []model.Application
+
+	err := d.gormdb.Model(user).Association("Applications").Find(&applications)
+
+	return applications, err
+}
+
+// GetUsers returns all users.
+func (d *Database) GetUsers() ([]model.User, error) {
+	var users []model.User
+
+	err := d.gormdb.Find(&users).Error
+
+	return users, err
+}
+
+// AdminUserCount returns the number of admins or an error.
+func (d *Database) AdminUserCount() (int64, error) {
+	var users []model.User
+
+	query := d.gormdb.Where("is_admin = ?", true).Find(&users)
+
+	return query.RowsAffected, query.Error
+}