mirrors/mCaptcha
1
0
Fork 0
mirror of https://github.com/mCaptcha/mCaptcha.git synced 2025-02-17 08:59:46 +03:00
Code Issues Projects Releases Packages Wiki Activity

feat: implement captcha exists interface for sqlx postgres

Browse source
This commit is contained in:
realaravinth 2022-05-12 19:31:47 +05:30
parent 55518ef650
commit 0bb975a230
No known key found for this signature in database
GPG key ID: AD9F0F08E855ED88
1 changed files with 43 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
db/db-sqlx-postgres/src/lib.rs
View file

@ -331,6 +331,49 @@ impl MCDatabase for Database {
try_join_all(futs).await.map_err(map_register_err)?;
Ok(())
}
/// check if captcha exists
async fn captcha_exists(
&self,
username: Option<&str>,
captcha_key: &str,
) -> DBResult<bool> {
let mut exists = false;
match username {
Some(username) => {
let x = sqlx::query!(
"SELECT EXISTS (
SELECT 1 from mcaptcha_config WHERE key = $1
AND user_id = (SELECT ID FROM mcaptcha_users WHERE name = $2)
)",
captcha_key,
username
)
.fetch_one(&self.pool)
.await
.map_err(map_register_err)?;
if let Some(x) = x.exists {
exists = x;
};
}
None => {
let x = sqlx::query!(
"SELECT EXISTS (SELECT 1 from mcaptcha_config WHERE key = $1)",
&captcha_key,
)
.fetch_one(&self.pool)
.await
.map_err(map_register_err)?;
if let Some(x) = x.exists {
exists = x;
};
}
};
Ok(exists)
}
}
fn now_unix_time_stamp() -> i64 {