62 lines
1.1 KiB
Rust
62 lines
1.1 KiB
Rust
use app_error::AppError;
|
|
use std::pin::Pin;
|
|
use std::task::{Context, Poll};
|
|
use tokio::io::{self, AsyncRead, ReadBuf};
|
|
|
|
pub struct CountingReader<R> {
|
|
reader: R,
|
|
count: usize,
|
|
}
|
|
|
|
impl<R> AsyncRead for CountingReader<R>
|
|
where
|
|
R: AsyncRead + Unpin,
|
|
{
|
|
fn poll_read(
|
|
mut self: Pin<&mut Self>,
|
|
cx: &mut Context<'_>,
|
|
buf: &mut ReadBuf<'_>,
|
|
) -> Poll<io::Result<()>> {
|
|
let before = buf.filled().len();
|
|
let poll = Pin::new(&mut self.reader).poll_read(cx, buf);
|
|
let after = buf.filled().len();
|
|
self.count += after - before;
|
|
poll
|
|
}
|
|
}
|
|
|
|
impl<R> CountingReader<R> {
|
|
pub fn new(reader: R) -> Self {
|
|
Self { reader, count: 0 }
|
|
}
|
|
|
|
pub fn count(&self) -> usize {
|
|
self.count
|
|
}
|
|
}
|
|
|
|
impl<R> AsRef<R> for CountingReader<R>
|
|
where
|
|
R: AsyncRead + Unpin,
|
|
{
|
|
fn as_ref(&self) -> &R {
|
|
&self.reader
|
|
}
|
|
}
|
|
|
|
pub async fn check_user_exists(
|
|
admin_token: &str,
|
|
gotrue_client: &gotrue::api::Client,
|
|
email: &str,
|
|
) -> Result<bool, AppError> {
|
|
let users = gotrue_client
|
|
.admin_list_user(admin_token, Some(email))
|
|
.await?;
|
|
for user in users.users {
|
|
if user.email == email {
|
|
return Ok(true);
|
|
}
|
|
}
|
|
Ok(false)
|
|
}
|