blob: 0a3da4026aac18df115d6be36267b2116fb6acb9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
package mail
import (
"dove/database"
"dove/models/mail"
"dove/utils/meta"
"gorm.io/gorm"
)
func FindUserByID(userID uint) *mail.User {
var user mail.User
result := database.DB.First(&user, userID)
if result.Error == gorm.ErrRecordNotFound {
return nil
}
return &user
}
func FindUserByUsername(username string) *mail.User {
var user mail.User
result := database.DB.Where("username = ?", username).First(&user)
if result.Error == gorm.ErrRecordNotFound {
return nil
}
return &user
}
func CreateUser(user *mail.User) error {
return database.DB.Create(user).Error
}
func ListUsers(pagination meta.Pagination, sorting meta.Sorting, search string) ([]mail.User, int64) {
var users []mail.User
var total int64
query := database.DB.Model(&mail.User{})
if search != "" {
like := "%" + search + "%"
query = query.Where("username LIKE ? OR display_name LIKE ?", like, like)
}
query.Count(&total)
pagination.Apply(sorting.Apply(query)).Preload("Mailboxes").Find(&users)
return users, total
}
func AllUsers() []mail.User {
var users []mail.User
database.DB.Order("username ASC").Find(&users)
return users
}
func UpdateUser(user *mail.User) error {
return database.DB.Save(user).Error
}
func DeleteUser(user *mail.User) error {
return database.DB.Delete(user).Error
}
func CountMailboxesByUserID(userID uint) int64 {
var count int64
database.DB.Model(&mail.Mailbox{}).Where("user_id = ?", userID).Count(&count)
return count
}
func CountUsers() int64 {
var count int64
database.DB.Model(&mail.User{}).Count(&count)
return count
}
|