/
default-handler.go
91 lines (81 loc) · 3.04 KB
/
default-handler.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
package authentication
import (
"context"
"github.com/neuronlabs/neuron/auth"
"github.com/neuronlabs/neuron/core"
"github.com/neuronlabs/neuron/database"
"github.com/neuronlabs/neuron/errors"
"github.com/neuronlabs/neuron/mapping"
"github.com/neuronlabs/neuron/query/filter"
)
// CheckUsernameHandler is an interface that allows to check the username in a custom way.
type CheckUsernameHandler interface {
HandleCheckUsername(ctx context.Context, db database.DB, account auth.Account) error
}
// AccountGetHandler is an interface that handles getting account.
type AccountGetterHandler interface {
HandleGetAccount(ctx context.Context, db database.DB, account auth.Account) error
}
// DefaultHandler is the default model handler for the accounts.
type DefaultHandler struct {
Account auth.Account
Model *mapping.ModelStruct
UsernameField *mapping.StructField
PasswordField *mapping.StructField
}
// HandleCheckUsername implements CheckUsernameHandler interface.
func (d *DefaultHandler) HandleCheckUsername(ctx context.Context, db database.DB, account auth.Account) error {
cnt, err := db.QueryCtx(ctx, d.Model).
Filter(filter.New(d.UsernameField, filter.OpEqual, account.GetUsername())).
Count()
if err != nil {
return err
}
if cnt > 0 {
return errors.WrapDet(auth.ErrAccountAlreadyExists, "account already exists").
WithDetail("An account with provided username already exists")
}
return nil
}
// HandleRegisterAccount implements InsertAccountHandler interface.
func (d *DefaultHandler) HandleRegisterAccount(ctx context.Context, db database.DB, options *RegisterAccountOptions) error {
return db.Insert(ctx, d.Model, options.Account)
}
// HandleLoginAccountRefresh implements LoginAccountRefreshHandler.
func (d *DefaultHandler) HandleLoginAccountRefresh(ctx context.Context, db database.DB, options *LoginOptions) error {
// Find the account with provided username.
model, err := db.QueryCtx(ctx, d.Model).
Filter(filter.New(d.UsernameField, filter.OpEqual, options.Account.GetUsername())).Get()
if err != nil {
return err
}
// Try to set the fields into already created account.
fromSetter, ok := options.Account.(mapping.FromSetter)
if ok {
if err := fromSetter.SetFrom(model); err != nil {
return err
}
return nil
}
options.Account = model.(auth.Account)
return nil
}
// Initialize implements core.Initializer.
func (d *DefaultHandler) Initialize(c *core.Controller) error {
// Find the username field.
var err error
d.Model, err = c.ModelStruct(d.Account)
if err != nil {
return err
}
var ok bool
d.UsernameField, ok = d.Model.FieldByName(d.Account.UsernameField())
if !ok {
return errors.Wrapf(auth.ErrInitialization, "provided invalid account model - no username field: '%s' found in the model: %s", d.Account.UsernameField(), d.Model)
}
d.PasswordField, ok = d.Model.FieldByName(d.Account.PasswordHashField())
if !ok {
return errors.Wrapf(auth.ErrInitialization, "provided invalid account model - no password hash field: '%s' found in the model: %s", d.Account.PasswordHashField(), d.Model)
}
return nil
}