-
Notifications
You must be signed in to change notification settings - Fork 928
/
user.go
162 lines (136 loc) · 3.61 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
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
package uaa
import (
"bytes"
"encoding/json"
"fmt"
"net/http"
"net/url"
"code.cloudfoundry.org/cli/api/uaa/internal"
)
// User represents an UAA user account.
type User struct {
ID string
Origin string
}
// newUserRequestBody represents the body of the request.
type newUserRequestBody struct {
Username string `json:"userName"`
Password string `json:"password"`
Origin string `json:"origin"`
Name userName `json:"name"`
Emails []email `json:"emails"`
}
type userName struct {
FamilyName string `json:"familyName"`
GivenName string `json:"givenName"`
}
type email struct {
Value string `json:"value"`
Primary bool `json:"primary"`
}
// newUserResponse represents the HTTP JSON response.
type newUserResponse struct {
ID string `json:"id"`
Origin string `json:"origin"`
}
type paginatedUsersResponse struct {
Resources []newUserResponse `json:"resources"`
}
// CreateUser creates a new UAA user account with the provided password.
func (client *Client) CreateUser(user string, password string, origin string) (User, error) {
userRequest := newUserRequestBody{
Username: user,
Password: password,
Origin: origin,
Name: userName{
FamilyName: user,
GivenName: user,
},
Emails: []email{
{
Value: user,
Primary: true,
},
},
}
bodyBytes, err := json.Marshal(userRequest)
if err != nil {
return User{}, err
}
request, err := client.newRequest(requestOptions{
RequestName: internal.PostUserRequest,
Header: http.Header{
"Content-Type": {"application/json"},
},
Body: bytes.NewBuffer(bodyBytes),
})
if err != nil {
return User{}, err
}
var userResponse newUserResponse
response := Response{
Result: &userResponse,
}
err = client.connection.Make(request, &response)
if err != nil {
return User{}, err
}
return User(userResponse), nil
}
func (client *Client) DeleteUser(userGuid string) (User, error) {
deleteRequest, err := client.newRequest(requestOptions{
RequestName: internal.DeleteUserRequest,
Header: http.Header{
"Content-Type": {"application/json"},
},
URIParams: map[string]string{"user_guid": userGuid},
})
if err != nil {
return User{}, err
}
var deleteUserResponse newUserResponse
deleteResponse := Response{
Result: &deleteUserResponse,
}
err = client.connection.Make(deleteRequest, &deleteResponse)
if err != nil {
return User{}, err
}
return User(deleteUserResponse), nil
}
// ListUsers gets a list of users from UAA with the given username and (if provided) origin.
// NOTE: that this is a paginated response and we are only currently returning the first page
// of users. This will mean, if no origin is passed and there are more than 100 users with
// the given username, only the first 100 will be returned. For our current purposes, this is
// more than enough, but it would be a problem if we ever need to get all users with a username.
func (client Client) ListUsers(userName, origin string) ([]User, error) {
filter := fmt.Sprintf(`userName eq "%s"`, userName)
if origin != "" {
filter = fmt.Sprintf(`%s and origin eq "%s"`, filter, origin)
}
request, err := client.newRequest(requestOptions{
RequestName: internal.ListUsersRequest,
Header: http.Header{
"Content-Type": {"application/json"},
},
Query: url.Values{
"filter": {filter},
},
})
if err != nil {
return nil, err
}
var usersResponse paginatedUsersResponse
response := Response{
Result: &usersResponse,
}
err = client.connection.Make(request, &response)
if err != nil {
return nil, err
}
var users []User
for _, user := range usersResponse.Resources {
users = append(users, User(user))
}
return users, nil
}