AppFlowy/rust-lib/flowy-user/src/services/server/server_api.rs

use crate::{
    entities::{SignInParams, SignInResponse, SignUpParams, SignUpResponse, UserProfile},
    errors::UserError,
};

use crate::{entities::UpdateUserParams, services::server::UserServerAPI};
use flowy_infra::future::ResultFuture;
use flowy_net::{
    config::*,
    request::{HttpRequestBuilder, ResponseMiddleware},
};

pub struct UserServer {
    config: ServerConfig,
}
impl UserServer {
    pub fn new(config: ServerConfig) -> Self { Self { config } }
}

impl UserServerAPI for UserServer {
    fn sign_up(&self, params: SignUpParams) -> ResultFuture<SignUpResponse, UserError> {
        let url = self.config.sign_up_url();
        ResultFuture::new(async move { user_sign_up_request(params, &url).await })
    }

    fn sign_in(&self, params: SignInParams) -> ResultFuture<SignInResponse, UserError> {
        let url = self.config.sign_in_url();
        ResultFuture::new(async move { user_sign_in_request(params, &url).await })
    }

    fn sign_out(&self, token: &str) -> ResultFuture<(), UserError> {
        let token = token.to_owned();
        let url = self.config.sign_out_url();
        ResultFuture::new(async move {
            let _ = user_sign_out_request(&token, &url).await;
            Ok(())
        })
    }

    fn update_user(&self, token: &str, params: UpdateUserParams) -> ResultFuture<(), UserError> {
        let token = token.to_owned();
        let url = self.config.user_profile_url();
        ResultFuture::new(async move { update_user_profile_request(&token, params, &url).await })
    }

    fn get_user(&self, token: &str) -> ResultFuture<UserProfile, UserError> {
        let token = token.to_owned();
        let url = self.config.user_profile_url();
        ResultFuture::new(async move { get_user_profile_request(&token, &url).await })
    }

    fn ws_addr(&self) -> String { self.config.ws_addr() }
}

use crate::notify::*;
use flowy_net::response::FlowyResponse;
use flowy_user_infra::errors::ErrorCode;
use lazy_static::lazy_static;
use std::sync::Arc;
lazy_static! {
    static ref MIDDLEWARE: Arc<Middleware> = Arc::new(Middleware {});
}

struct Middleware {}
impl ResponseMiddleware for Middleware {
    fn receive_response(&self, token: &Option<String>, response: &FlowyResponse) {
        if let Some(error) = &response.error {
            if error.is_unauthorized() {
                log::error!("user unauthorized");
                match token {
                    None => {},
                    Some(token) => {
                        let error = UserError::new(ErrorCode::UserUnauthorized, "");
                        dart_notify(token, UserNotification::UserUnauthorized)
                            .error(error)
                            .send()
                    },
                }
            }
        }
    }
}

pub(crate) fn request_builder() -> HttpRequestBuilder { HttpRequestBuilder::new().middleware(MIDDLEWARE.clone()) }

pub async fn user_sign_up_request(params: SignUpParams, url: &str) -> Result<SignUpResponse, UserError> {
    let response = request_builder()
        .post(&url.to_owned())
        .protobuf(params)?
        .response()
        .await?;
    Ok(response)
}

pub async fn user_sign_in_request(params: SignInParams, url: &str) -> Result<SignInResponse, UserError> {
    let response = request_builder()
        .post(&url.to_owned())
        .protobuf(params)?
        .response()
        .await?;
    Ok(response)
}

pub async fn user_sign_out_request(token: &str, url: &str) -> Result<(), UserError> {
    let _ = request_builder()
        .delete(&url.to_owned())
        .header(HEADER_TOKEN, token)
        .send()
        .await?;
    Ok(())
}

pub async fn get_user_profile_request(token: &str, url: &str) -> Result<UserProfile, UserError> {
    let user_profile = request_builder()
        .get(&url.to_owned())
        .header(HEADER_TOKEN, token)
        .response()
        .await?;
    Ok(user_profile)
}

pub async fn update_user_profile_request(token: &str, params: UpdateUserParams, url: &str) -> Result<(), UserError> {
    let _ = request_builder()
        .patch(&url.to_owned())
        .header(HEADER_TOKEN, token)
        .protobuf(params)?
        .send()
        .await?;
    Ok(())
}