implemented request login flows

This commit is contained in:
Edward Shen 2019-08-21 01:05:26 -04:00
parent a7085d3334
commit 7d89ed4a44
Signed by: edward
GPG key ID: F350507060ED6C90
3 changed files with 79 additions and 20 deletions

View file

@ -28,7 +28,7 @@
- [ ] 5.3.6.2 Third-party ID - [ ] 5.3.6.2 Third-party ID
- [ ] 5.3.6.3 Phone number - [ ] 5.3.6.3 Phone number
- [ ] 5.4 Login - [ ] 5.4 Login
- [ ] 5.4.1 GET /_matrix/client/r0/login - [x] 5.4.1 ~~GET /_matrix/client/r0/login~~
- [ ] 5.4.2 POST /_matrix/client/r0/login - [ ] 5.4.2 POST /_matrix/client/r0/login
- [ ] 5.4.3 POST /_matrix/client/r0/logout - [ ] 5.4.3 POST /_matrix/client/r0/logout
- [ ] 5.4.4 POST /_matrix/client/r0/logout/all - [ ] 5.4.4 POST /_matrix/client/r0/logout/all

View file

@ -1,16 +1,25 @@
use crate::api::methods::sync::SyncResponse; use crate::api::methods::{login::ValidLoginFlows, sync::SyncResponse};
use reqwest::{ use reqwest::{
header::{HeaderMap, HeaderValue, CONTENT_TYPE, USER_AGENT}, header::{HeaderMap, HeaderValue, CONTENT_TYPE, USER_AGENT},
Client as reqwest_client, Response, StatusCode, Client as reqwest_client, Response, StatusCode,
}; };
use serde::Deserialize; use serde::Deserialize;
use std::{collections::HashMap, error::Error, fmt, time}; use serde_json::Value;
use std::{
collections::HashMap,
error::Error,
fmt,
sync::{
atomic::{AtomicBool, Ordering},
Arc,
},
thread, time,
};
use url::{ParseError, Url}; use url::{ParseError, Url};
#[cfg(test)] #[cfg(test)]
use mockito; use mockito;
const V2_API_PATH: &str = "/_matrix/client/r0";
const SUPPORTED_VERSION: &str = "r0.5.0"; const SUPPORTED_VERSION: &str = "r0.5.0";
#[derive(Debug)] #[derive(Debug)]
@ -71,18 +80,18 @@ pub struct SupportedSpecs {
#[derive(Deserialize)] #[derive(Deserialize)]
pub struct DiscoveryInformation { pub struct DiscoveryInformation {
#[serde(rename = "m.homeserver")] #[serde(rename = "m.homeserver")]
homeserver: HomeserverInformation, pub homeserver: HomeserverInformation,
identity_server: Option<IdentityServerInformation>, pub identity_server: Option<IdentityServerInformation>,
} }
#[derive(Deserialize)] #[derive(Deserialize)]
pub struct HomeserverInformation { pub struct HomeserverInformation {
base_url: String, pub base_url: String,
} }
#[derive(Deserialize)] #[derive(Deserialize)]
pub struct IdentityServerInformation { pub struct IdentityServerInformation {
base_url: String, pub base_url: String,
} }
impl Client { impl Client {
@ -175,6 +184,7 @@ impl Client {
#[cfg(test)] #[cfg(test)]
let url = &mockito::server_url(); let url = &mockito::server_url();
#[cfg(not(test))] #[cfg(not(test))]
let url = &self.homeserver_url; let url = &self.homeserver_url;
@ -218,16 +228,17 @@ impl Client {
if res.status().is_success() { if res.status().is_success() {
return Ok(res); return Ok(res);
} else if res.status() == StatusCode::TOO_MANY_REQUESTS { } else if res.status() == StatusCode::TOO_MANY_REQUESTS {
let mut body: HashMap<String, String> = res.json()?; let mut body: &Value = &res.json()?;
if let Some(value) = body.get("error") { if let Some(value) = body.get("error") {
body = serde_json::from_str(value)?; body = value;
} }
if let Some(value) = body.get("retry_after_ms") { std::thread::sleep(time::Duration::from_millis(
std::thread::sleep(time::Duration::from_millis(value.parse::<u64>()?)); body.get("retry_after_ms").map_or_else(
} else { || self.default_492_wait_ms,
std::thread::sleep(time::Duration::from_millis(self.default_492_wait_ms)); |v| v.as_u64().unwrap_or_else(|| self.default_492_wait_ms),
} ),
));
} else { } else {
return Err(Box::from(ResponseError { return Err(Box::from(ResponseError {
code: res.status(), code: res.status(),
@ -289,6 +300,21 @@ impl Client {
} }
} }
// Holds implementation for Section 5.4, Login
impl Client {
pub fn login_flows(&self) -> Result<ValidLoginFlows, Box<dyn Error>> {
Ok(self
.send(
MatrixHTTPMethod::Get,
"/_matrix/client/r0/login",
None,
None,
None,
)?
.json()?)
}
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
@ -340,6 +366,42 @@ mod tests {
assert!(resp.unstable_features.is_none()); assert!(resp.unstable_features.is_none());
} }
#[test]
fn send_handles_too_many_requests() {
let _mock_version_check = mock("GET", "/_matrix/client/versions")
.with_body(r#"{"versions": ["r0.5.0"]}"#)
.create();
let _m2 = mock("GET", "/hello")
.with_body(
r#"{
"errcode": "M_LIMIT_EXCEEDED",
"error": "Too many requests",
"retry_after_ms": 2000
}"#,
)
.with_status(429)
.create();
// Run request in separate thread. Once async lands use async/await
thread::spawn(|| {
let resp: Value = Client::new("http://dummy.website", None, None, None)
.expect("failed to get valid client")
.send(MatrixHTTPMethod::Get, "/hello", None, None, None)
.expect("failed to unwrap Response")
.json()
.unwrap();
assert_eq!(dbg!(resp)["hello"], "world");
});
thread::sleep(time::Duration::from_secs(1));
// Override 429 response with valid response.
let _m = mock("GET", "/hello")
.with_body(r#"{"hello": "world"}"#)
.create();
}
} }
#[derive(Default)] #[derive(Default)]
pub struct ApiError {} pub struct ApiError {}

View file

@ -12,9 +12,7 @@ pub struct LoginFlow {
} }
/// Request helper for 5.4.2: POST /_matrix/client/r0/login /// Request helper for 5.4.2: POST /_matrix/client/r0/login
pub struct LoginRequest { pub struct LoginRequest {}
}
/// Response object for a valid login from 5.4.2: POST /_matrix/client/r0/login /// Response object for a valid login from 5.4.2: POST /_matrix/client/r0/login
pub struct LoginResponse { pub struct LoginResponse {
@ -25,5 +23,4 @@ pub struct LoginResponse {
pub well_known: Option<DiscoveryInformation>, pub well_known: Option<DiscoveryInformation>,
} }
pub struct DiscoveryInformation { pub struct DiscoveryInformation {}
}