/
user_repo.go
100 lines (88 loc) · 2.29 KB
/
user_repo.go
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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package repo
import (
"context"
"database/sql"
"fmt"
"github.com/amin1024/xtelbot/core/e"
"github.com/amin1024/xtelbot/core/repo/models"
"github.com/friendsofgo/errors"
"github.com/volatiletech/sqlboiler/v4/boil"
"github.com/volatiletech/sqlboiler/v4/queries/qm"
)
func SaveOrUpdateUser(u *models.Tuser) error {
err := u.Upsert(
context.Background(), db, true,
[]string{models.TuserColumns.Tid},
boil.Whitelist(models.TuserColumns.Username),
boil.Infer(),
)
return err
}
func UpdateUser(u *models.Tuser) error {
rowsAffected, err := u.Update(context.Background(), db, boil.Infer())
if err != nil {
//TODO: wrap error
return err
}
if rowsAffected != 1 {
return fmt.Errorf("!wtf")
}
return nil
}
func GetUserByTid(uid uint64) (*models.Tuser, error) {
u, err := models.Tusers(
qm.Load(models.TuserRels.Package),
qm.Where(models.TuserColumns.Tid+"=?", uid),
).One(context.Background(), db)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return u, e.UserNotFound
}
return u, fmt.Errorf("%s: %w", err, e.BaseError)
}
//if !u.Active {
// return u, e.UserIsNotActive
//}
return u, nil
}
func GetUser(userId int64) (*models.Tuser, error) {
u, err := models.Tusers(
qm.Load(models.TuserRels.Package),
qm.Where(models.TuserColumns.ID+"=?", userId),
qm.And("active=?", true),
).One(context.Background(), db)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return u, e.UserNotFound
}
return u, fmt.Errorf("%s: %w", err, e.BaseError)
}
return u, nil
}
func GetUserByToken(token string) (*models.Tuser, error) {
u, err := models.Tusers(
models.TuserWhere.Token.EQ(token),
models.TuserWhere.Active.EQ(true),
).One(context.Background(), db)
if err != nil {
if errors.Is(err, sql.ErrNoRows) {
return u, e.UserNotFound
}
return u, fmt.Errorf("%s: %w", err, e.BaseError)
}
return u, nil
}
func GetAllUsers() ([]*models.Tuser, error) {
// TODO: potential bug when the number of users grow, work around it later on
users, err := models.Tusers(
models.TuserWhere.Active.EQ(true),
).All(context.Background(), db)
return users, err
}
func GetAllUsersWithPackages() ([]*models.Tuser, error) {
users, err := models.Tusers(
models.TuserWhere.Active.EQ(true),
qm.Load(models.TuserRels.Package),
).All(context.Background(), db)
return users, err
}