bitwarden_generators/username_forwarders/
simplelogin.rsuse reqwest::{header::CONTENT_TYPE, StatusCode};
use crate::username::UsernameError;
pub async fn generate(
http: &reqwest::Client,
api_key: String,
website: Option<String>,
) -> Result<String, UsernameError> {
generate_with_api_url(http, api_key, website, "https://app.simplelogin.io".into()).await
}
async fn generate_with_api_url(
http: &reqwest::Client,
api_key: String,
website: Option<String>,
api_url: String,
) -> Result<String, UsernameError> {
let query = website
.as_ref()
.map(|w| format!("?hostname={}", w))
.unwrap_or_default();
let note = super::format_description(&website);
#[derive(serde::Serialize)]
struct Request {
note: String,
}
let response = http
.post(format!("{api_url}/api/alias/random/new{query}"))
.header(CONTENT_TYPE, "application/json")
.header("Authentication", api_key)
.json(&Request { note })
.send()
.await?;
if response.status() == StatusCode::UNAUTHORIZED {
return Err(UsernameError::InvalidApiKey);
}
response.error_for_status_ref()?;
#[derive(serde::Deserialize)]
struct Response {
alias: String,
}
let response: Response = response.json().await?;
Ok(response.alias)
}
#[cfg(test)]
mod tests {
use serde_json::json;
use crate::username::UsernameError;
#[tokio::test]
async fn test_mock_server() {
use wiremock::{matchers, Mock, ResponseTemplate};
let server = wiremock::MockServer::start().await;
server
.register(
Mock::given(matchers::path("/api/alias/random/new"))
.and(matchers::method("POST"))
.and(matchers::query_param("hostname", "example.com"))
.and(matchers::header("Content-Type", "application/json"))
.and(matchers::header("Authentication", "MY_TOKEN"))
.and(matchers::body_json(json!({
"note": "Website: example.com. Generated by Bitwarden."
})))
.respond_with(ResponseTemplate::new(201).set_body_json(json!({
"alias": "[email protected]",
})))
.expect(1),
)
.await;
server
.register(
Mock::given(matchers::path("/api/alias/random/new"))
.and(matchers::method("POST"))
.and(matchers::query_param("hostname", "example.com"))
.and(matchers::header("Content-Type", "application/json"))
.and(matchers::header("Authentication", "MY_FAKE_TOKEN"))
.and(matchers::body_json(json!({
"note": "Website: example.com. Generated by Bitwarden."
})))
.respond_with(ResponseTemplate::new(401))
.expect(1),
)
.await;
let address = super::generate_with_api_url(
&reqwest::Client::new(),
"MY_TOKEN".into(),
Some("example.com".into()),
format!("http://{}", server.address()),
)
.await
.unwrap();
assert_eq!(address, "[email protected]");
let fake_token_error = super::generate_with_api_url(
&reqwest::Client::new(),
"MY_FAKE_TOKEN".into(),
Some("example.com".into()),
format!("http://{}", server.address()),
)
.await
.unwrap_err();
assert_eq!(
fake_token_error.to_string(),
UsernameError::InvalidApiKey.to_string()
);
server.verify().await;
}
}