-
Notifications
You must be signed in to change notification settings - Fork 289
/
Copy pathuser.go
142 lines (123 loc) · 3.69 KB
/
user.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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
package iam
import (
"context"
"github.com/hashicorp/boundary/internal/db"
"github.com/hashicorp/boundary/internal/errors"
"github.com/hashicorp/boundary/internal/iam/store"
"github.com/hashicorp/boundary/internal/types/action"
"github.com/hashicorp/boundary/internal/types/resource"
"github.com/hashicorp/boundary/internal/types/scope"
"google.golang.org/protobuf/proto"
)
const (
defaultUserTableName = "iam_user"
defaultUserAccountInfoTableName = "iam_user_acct_info"
)
// User defines boundary users which are scoped to an Org
type User struct {
*store.User
tableName string `gorm:"-"`
}
// ensure that User implements the interfaces of: Resource, Cloneable and db.VetForWriter
var (
_ Resource = (*User)(nil)
_ Cloneable = (*User)(nil)
_ db.VetForWriter = (*User)(nil)
)
// NewUser creates a new in memory user and allows options:
// WithName - to specify the user's friendly name and WithDescription - to
// specify a user description
func NewUser(scopeId string, opt ...Option) (*User, error) {
const op = "iam.NewUser"
opts := getOpts(opt...)
if scopeId == "" {
return nil, errors.NewDeprecated(errors.InvalidParameter, op, "missing scope id")
}
u := &User{
User: &store.User{
Name: opts.withName,
Description: opts.withDescription,
ScopeId: scopeId,
},
}
return u, nil
}
// AllocUser will allocate an empty user
func AllocUser() User {
return User{
User: &store.User{},
}
}
// Clone creates a clone of the User
func (u *User) Clone() any {
cp := proto.Clone(u.User)
return &User{
User: cp.(*store.User),
}
}
// VetForWrite implements db.VetForWrite() interface and validates the user
// before it's written.
func (u *User) VetForWrite(ctx context.Context, r db.Reader, opType db.OpType, opt ...db.Option) error {
const op = "iam.(User).VetForWrite"
if u.PublicId == "" {
return errors.New(ctx, errors.InvalidParameter, op, "missing public id")
}
if err := validateScopeForWrite(ctx, r, u, opType, opt...); err != nil {
return errors.Wrap(ctx, err, op)
}
return nil
}
func (u *User) validScopeTypes() []scope.Type {
return []scope.Type{scope.Global, scope.Org}
}
// GetScope returns the scope for the User
func (u *User) GetScope(ctx context.Context, r db.Reader) (*Scope, error) {
return LookupScope(ctx, r, u)
}
// ResourceType returns the type of the User
func (*User) ResourceType() resource.Type { return resource.User }
// Actions returns the available actions for Users
func (*User) Actions() map[string]action.Type {
return CrudActions()
}
// TableName returns the tablename to override the default gorm table name
func (u *User) TableName() string {
if u.tableName != "" {
return u.tableName
}
return defaultUserTableName
}
// SetTableName sets the tablename and satisfies the ReplayableMessage
// interface. If the caller attempts to set the name to "" the name will be
// reset to the default name.
func (u *User) SetTableName(n string) {
u.tableName = n
}
// userAccountInfo provides a way to represent a user along with the user's
// account info from the scope's primary auth method
type userAccountInfo struct {
*store.User
tableName string `gorm:"-"`
}
func (u *userAccountInfo) shallowConversion() *User {
return &User{
User: u.User,
}
}
// TableName provides an overridden gorm table name..
func (u *userAccountInfo) TableName() string {
if u.tableName != "" {
return u.tableName
}
return defaultUserAccountInfoTableName
}
// SetTableName sets the table name for the resource. If the caller attempts to
// set the name to "" the name will be reset to the default name.
func (u *userAccountInfo) SetTableName(n string) {
switch n {
case "":
u.tableName = defaultUserAccountInfoTableName
default:
u.tableName = n
}
}