mirror of
https://github.com/AppFlowy-IO/AppFlowy.git
synced 2024-08-30 18:12:39 +00:00
feat: folder web (#4580)
* chore: folder wasm * chore: folder wasm * chore: resolve deps * chore: fix trait * chore: try localset * chore: fix * chore: fix * chore: fix * chore: async init sdk * chore: fix test * chore: fix test
This commit is contained in:
@ -2,7 +2,6 @@ pub use server::*;
|
||||
|
||||
pub mod af_cloud;
|
||||
pub mod local_server;
|
||||
// mod request;
|
||||
mod response;
|
||||
mod server;
|
||||
|
||||
|
@ -1,198 +0,0 @@
|
||||
use std::{sync::Arc, time::Duration};
|
||||
|
||||
use bytes::Bytes;
|
||||
use hyper::http;
|
||||
use reqwest::{header::HeaderMap, Client, Method, Response};
|
||||
use tokio::sync::oneshot;
|
||||
|
||||
use flowy_error::{internal_error, FlowyError};
|
||||
|
||||
use crate::af_cloud::configuration::HEADER_TOKEN;
|
||||
use crate::response::HttpResponse;
|
||||
|
||||
pub trait ResponseMiddleware {
|
||||
fn receive_response(&self, token: &Option<String>, response: &HttpResponse);
|
||||
}
|
||||
|
||||
pub struct HttpRequestBuilder {
|
||||
url: String,
|
||||
body: Option<Bytes>,
|
||||
response: Option<Bytes>,
|
||||
headers: HeaderMap,
|
||||
method: Method,
|
||||
middleware: Vec<Arc<dyn ResponseMiddleware + Send + Sync>>,
|
||||
}
|
||||
|
||||
impl std::default::Default for HttpRequestBuilder {
|
||||
fn default() -> Self {
|
||||
Self {
|
||||
url: "".to_owned(),
|
||||
body: None,
|
||||
response: None,
|
||||
headers: HeaderMap::new(),
|
||||
method: Method::GET,
|
||||
middleware: Vec::new(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl HttpRequestBuilder {
|
||||
pub fn new() -> Self {
|
||||
HttpRequestBuilder::default()
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
pub fn middleware<T>(mut self, middleware: Arc<T>) -> Self
|
||||
where
|
||||
T: 'static + ResponseMiddleware + Send + Sync,
|
||||
{
|
||||
self.middleware.push(middleware);
|
||||
self
|
||||
}
|
||||
|
||||
pub fn get(mut self, url: &str) -> Self {
|
||||
self.url = url.to_owned();
|
||||
self.method = Method::GET;
|
||||
self
|
||||
}
|
||||
|
||||
pub fn post(mut self, url: &str) -> Self {
|
||||
self.url = url.to_owned();
|
||||
self.method = Method::POST;
|
||||
self
|
||||
}
|
||||
|
||||
pub fn patch(mut self, url: &str) -> Self {
|
||||
self.url = url.to_owned();
|
||||
self.method = Method::PATCH;
|
||||
self
|
||||
}
|
||||
|
||||
pub fn delete(mut self, url: &str) -> Self {
|
||||
self.url = url.to_owned();
|
||||
self.method = Method::DELETE;
|
||||
self
|
||||
}
|
||||
|
||||
pub fn header(mut self, key: &'static str, value: &str) -> Self {
|
||||
self.headers.insert(key, value.parse().unwrap());
|
||||
self
|
||||
}
|
||||
|
||||
pub fn json<T>(self, body: T) -> Result<Self, FlowyError>
|
||||
where
|
||||
T: serde::Serialize,
|
||||
{
|
||||
let bytes = Bytes::from(serde_json::to_vec(&body)?);
|
||||
self.bytes(bytes)
|
||||
}
|
||||
|
||||
pub fn bytes(mut self, body: Bytes) -> Result<Self, FlowyError> {
|
||||
self.body = Some(body);
|
||||
Ok(self)
|
||||
}
|
||||
|
||||
pub async fn send(self) -> Result<(), FlowyError> {
|
||||
let _ = self.inner_send().await?;
|
||||
Ok(())
|
||||
}
|
||||
|
||||
pub async fn response<T>(self) -> Result<T, FlowyError>
|
||||
where
|
||||
T: serde::de::DeserializeOwned,
|
||||
{
|
||||
let builder = self.inner_send().await?;
|
||||
match builder.response {
|
||||
None => Err(unexpected_empty_payload(&builder.url)),
|
||||
Some(data) => {
|
||||
let value = serde_json::from_slice(&data)?;
|
||||
Ok(value)
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
fn token(&self) -> Option<String> {
|
||||
match self.headers.get(HEADER_TOKEN) {
|
||||
None => None,
|
||||
Some(header) => match header.to_str() {
|
||||
Ok(val) => Some(val.to_owned()),
|
||||
Err(_) => None,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
async fn inner_send(mut self) -> Result<Self, FlowyError> {
|
||||
let (tx, rx) = oneshot::channel::<Result<Response, _>>();
|
||||
let url = self.url.clone();
|
||||
let body = self.body.take();
|
||||
let method = self.method.clone();
|
||||
let headers = self.headers.clone();
|
||||
|
||||
// reqwest client is not 'Sync' but channel is.
|
||||
tokio::spawn(async move {
|
||||
let client = default_client();
|
||||
let mut builder = client.request(method.clone(), url).headers(headers);
|
||||
if let Some(body) = body {
|
||||
builder = builder.body(body);
|
||||
}
|
||||
|
||||
let response = builder.send().await;
|
||||
let _ = tx.send(response);
|
||||
});
|
||||
|
||||
let response = rx.await.map_err(internal_error)?;
|
||||
tracing::trace!("Http Response: {:?}", response);
|
||||
let flowy_response = flowy_response_from(response?).await?;
|
||||
let token = self.token();
|
||||
self.middleware.iter().for_each(|middleware| {
|
||||
middleware.receive_response(&token, &flowy_response);
|
||||
});
|
||||
match flowy_response.error {
|
||||
None => {
|
||||
self.response = Some(flowy_response.data);
|
||||
Ok(self)
|
||||
},
|
||||
Some(error) => Err(FlowyError::new(error.code, &error.msg)),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn unexpected_empty_payload(url: &str) -> FlowyError {
|
||||
let msg = format!("Request: {} receives unexpected empty payload", url);
|
||||
FlowyError::payload_none().with_context(msg)
|
||||
}
|
||||
|
||||
async fn flowy_response_from(original: Response) -> Result<HttpResponse, FlowyError> {
|
||||
let bytes = original.bytes().await?;
|
||||
let response: HttpResponse = serde_json::from_slice(&bytes)?;
|
||||
Ok(response)
|
||||
}
|
||||
|
||||
#[allow(dead_code)]
|
||||
async fn get_response_data(original: Response) -> Result<Bytes, FlowyError> {
|
||||
if original.status() == http::StatusCode::OK {
|
||||
let bytes = original.bytes().await?;
|
||||
let response: HttpResponse = serde_json::from_slice(&bytes)?;
|
||||
match response.error {
|
||||
None => Ok(response.data),
|
||||
Some(error) => Err(FlowyError::new(error.code, &error.msg)),
|
||||
}
|
||||
} else {
|
||||
Err(FlowyError::http().with_context(original))
|
||||
}
|
||||
}
|
||||
|
||||
fn default_client() -> Client {
|
||||
let result = reqwest::Client::builder()
|
||||
.connect_timeout(Duration::from_millis(500))
|
||||
.timeout(Duration::from_secs(5))
|
||||
.build();
|
||||
|
||||
match result {
|
||||
Ok(client) => client,
|
||||
Err(e) => {
|
||||
tracing::error!("Create reqwest client failed: {}", e);
|
||||
reqwest::Client::new()
|
||||
},
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user