84 lines
2.1 KiB
Rust
84 lines
2.1 KiB
Rust
use reqwest::Client;
|
|
use serde::{Deserialize, Serialize};
|
|
use url::Url;
|
|
|
|
pub struct MCaptcha {
|
|
client: Client,
|
|
sitekey: String,
|
|
account_secret: String,
|
|
verify_path: Url,
|
|
}
|
|
|
|
impl MCaptcha {
|
|
pub fn new(sitekey: String, account_secret: String, instance_url: Url) -> Self {
|
|
let mut verify_path = instance_url.clone();
|
|
verify_path.set_path("/api/v1/pow/siteverify");
|
|
Self {
|
|
sitekey,
|
|
account_secret,
|
|
verify_path,
|
|
client: Client::new(),
|
|
}
|
|
}
|
|
|
|
pub async fn verify(&self, token: &str) -> Result<bool, reqwest::Error> {
|
|
let payload = CaptchaVerfiyPayload::from_ctx(self, token);
|
|
let res: CaptchaVerifyResp = self
|
|
.client
|
|
.post(self.verify_path.clone())
|
|
.json(&payload)
|
|
.send()
|
|
.await
|
|
.unwrap()
|
|
.json()
|
|
.await
|
|
.unwrap();
|
|
|
|
Ok(res.valid)
|
|
}
|
|
}
|
|
|
|
#[derive(Default, Debug, PartialEq, Clone, Serialize, Deserialize)]
|
|
struct CaptchaVerifyResp {
|
|
valid: bool,
|
|
}
|
|
|
|
#[derive(Default, Debug, PartialEq, Clone, Serialize, Deserialize)]
|
|
struct CaptchaVerfiyPayload<'a> {
|
|
token: &'a str,
|
|
key: &'a str,
|
|
secret: &'a str,
|
|
}
|
|
|
|
impl<'a> CaptchaVerfiyPayload<'a> {
|
|
fn from_ctx(m: &'a MCaptcha, token: &'a str) -> Self {
|
|
Self {
|
|
key: &m.sitekey,
|
|
token,
|
|
secret: &m.account_secret,
|
|
}
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
use std::env;
|
|
|
|
#[actix_rt::test]
|
|
async fn verify_works() {
|
|
let instance_url = env::var("INSTANCE_URL").expect("instance url not set");
|
|
let sitekey = env::var("SITEKEY").expect("sitekey not set");
|
|
let secret = env::var("SECRET").expect("secret not set");
|
|
let token = env::var("TOKEN").expect("token not set");
|
|
|
|
let mcaptcha = MCaptcha::new(
|
|
sitekey,
|
|
secret,
|
|
Url::parse(&instance_url).expect("instance_url is not URL"),
|
|
);
|
|
assert!(mcaptcha.verify(&token).await.unwrap());
|
|
assert!(!mcaptcha.verify("foo").await.unwrap());
|
|
}
|
|
}
|