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
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
#![allow(clippy::missing_errors_doc)]
use crate::emojis::*;
use crate::models::{User, UserInfo};
use chrono::prelude::*;
use console::style;
use std::process::Command;
pub struct UserReader {
pub aws_pool_id: String,
pub aws_region: String,
}
pub struct UserReaderOptions<'a> {
pub limit_of_users: Option<u32>,
pub show_unconfirmed_users: bool,
pub filtered_user_ids: &'a Option<Vec<String>>,
pub include_user_ids: bool,
pub filtered_user_emails: &'a Option<Vec<String>>,
pub include_user_emails: bool,
pub created_at: Option<DateTime<Utc>>,
}
impl UserReader {
#[must_use]
pub const fn new(aws_pool_id: String, aws_region: String) -> Self {
Self {
aws_pool_id,
aws_region,
}
}
pub fn get_users(
&self,
options: &UserReaderOptions,
show_messages: bool,
) -> Result<Vec<User>, std::io::Error> {
let mut users: Vec<User> = Vec::new();
let mut pending_users: u32 = 0;
let mut pagination_token: Option<String> = None;
let mut limit: Option<u32> = None;
if let Some(max_users) = options.limit_of_users {
if max_users <= 60 {
limit = Some(max_users);
} else {
limit = Some(60);
pending_users = max_users - 60;
}
}
loop {
match get_users_from_cognito_idp(
&self.aws_pool_id,
&self.aws_region,
&pagination_token,
limit,
options.show_unconfirmed_users,
) {
Ok(mut info) => {
if show_messages {
println!(
"{} {} {}",
ROCKET,
style(format!("We got a batch of {} users", info.users.len()))
.bold()
.green(),
ROCKET
);
}
pagination_token = info.pagination_token;
users.append(&mut info.users);
}
Err(e) => {
if show_messages {
println!(
"{} {} {}\n{}",
ERROR,
style("SOMETHING WENT WRONG!").bold().red(),
ERROR,
style(e).red(),
);
}
}
}
if pending_users == 0 && limit.is_some() {
break;
}
if pending_users <= 60 {
limit = Some(pending_users);
pending_users = 0;
} else {
limit = Some(60);
pending_users -= 60;
}
if pagination_token.is_none() {
break;
}
}
users.sort_by(|a, b| a.user_create_date.partial_cmp(&b.user_create_date).unwrap());
users = users
.into_iter()
.filter(|u| {
if let Some(ref avoid) = options.filtered_user_ids {
let is_in = avoid.contains(&u.username);
return if options.include_user_ids {
is_in
} else {
!is_in
};
}
true
})
.filter(|u| {
if let Some(ref avoid) = options.filtered_user_emails {
let is_in = avoid.contains(&u.get_email());
return if options.include_user_emails {
is_in
} else {
!is_in
};
}
true
})
.filter(|u| {
if let Some(created_at) = options.created_at {
let duration = u.creation_date().signed_duration_since(created_at);
return duration.num_days() >= 0;
}
true
})
.collect();
Ok(users)
}
}
fn get_users_from_cognito_idp(
pool_id: &str,
region: &str,
pagination_token: &Option<String>,
limit: Option<u32>,
show_unconfirmed_users: bool,
) -> Result<UserInfo, String> {
let mut cmd = Command::new("aws");
cmd.arg("cognito-idp")
.arg("list-users")
.arg("--user-pool-id")
.arg(pool_id)
.arg("--attributes-to-get")
.arg("email")
.arg("--region")
.arg(region);
if !show_unconfirmed_users {
cmd.arg("--filter").arg("cognito:user_status = 'CONFIRMED'");
}
if let Some(pt) = pagination_token {
cmd.arg("--pagination-token").arg(pt);
}
if let Some(l) = limit {
if l > 60 {
return Err("The limit cannot be greater than 60".to_string());
}
cmd.arg("--limit").arg(l.to_string());
}
let result = cmd
.output()
.map_err(|e| format!("Error calling aws {}", e))?;
if result.status.success() {
serde_json::from_slice(&result.stdout)
.map_err(|e| format!("Error deserializing users: {}", e))
} else {
Err(std::str::from_utf8(&result.stderr).unwrap().into())
}
}