forked from firebase/firebase-admin-go
-
Notifications
You must be signed in to change notification settings - Fork 1
/
export_users.go
120 lines (105 loc) · 3.38 KB
/
export_users.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
// Copyright 2019 Google Inc. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package auth
import (
"context"
"encoding/json"
"fmt"
"net/http"
"net/url"
"strconv"
"google.golang.org/api/iterator"
)
const maxReturnedResults = 1000
// Users returns an iterator over Users.
//
// If nextPageToken is empty, the iterator will start at the beginning.
// If the nextPageToken is not empty, the iterator starts after the token.
func (c *userManagementClient) Users(ctx context.Context, nextPageToken string) *UserIterator {
it := &UserIterator{
ctx: ctx,
client: c,
}
it.pageInfo, it.nextFunc = iterator.NewPageInfo(
it.fetch,
func() int { return len(it.users) },
func() interface{} { b := it.users; it.users = nil; return b })
it.pageInfo.MaxSize = maxReturnedResults
it.pageInfo.Token = nextPageToken
return it
}
// UserIterator is an iterator over Users.
//
// Also see: https://github.com/GoogleCloudPlatform/google-cloud-go/wiki/Iterator-Guidelines
type UserIterator struct {
client *userManagementClient
ctx context.Context
nextFunc func() error
pageInfo *iterator.PageInfo
users []*ExportedUserRecord
}
// PageInfo supports pagination. See the google.golang.org/api/iterator package for details.
// Page size can be determined by the NewPager(...) function described there.
func (it *UserIterator) PageInfo() *iterator.PageInfo { return it.pageInfo }
// Next returns the next result. Its second return value is [iterator.Done] if
// there are no more results. Once Next returns [iterator.Done], all subsequent
// calls will return [iterator.Done].
func (it *UserIterator) Next() (*ExportedUserRecord, error) {
if err := it.nextFunc(); err != nil {
return nil, err
}
user := it.users[0]
it.users = it.users[1:]
return user, nil
}
func (it *UserIterator) fetch(pageSize int, pageToken string) (string, error) {
query := make(url.Values)
query.Set("maxResults", strconv.Itoa(pageSize))
if pageToken != "" {
query.Set("nextPageToken", pageToken)
}
req, err := it.client.newRequest(http.MethodGet, fmt.Sprintf("/accounts:batchGet?%s", query.Encode()))
if err != nil {
return "", err
}
resp, err := it.client.httpClient.Do(it.ctx, req)
if err != nil {
return "", err
}
if resp.Status != http.StatusOK {
return "", handleHTTPError(resp)
}
var parsed struct {
Users []userQueryResponse `json:"users"`
NextPageToken string `json:"nextPageToken"`
}
if err := json.Unmarshal(resp.Body, &parsed); err != nil {
return "", err
}
for _, u := range parsed.Users {
eu, err := u.makeExportedUserRecord()
if err != nil {
return "", err
}
it.users = append(it.users, eu)
}
it.pageInfo.Token = parsed.NextPageToken
return parsed.NextPageToken, nil
}
// ExportedUserRecord is the returned user value used when listing all the users.
type ExportedUserRecord struct {
*UserRecord
PasswordHash string
PasswordSalt string
}