AppFlowy-Cloud/libs/client-api-wasm/src/lib.rs

pub mod entities;

use crate::entities::*;

use client_api::notify::TokenState;
use client_api::{Client, ClientConfiguration};
use std::sync::Arc;
use uuid::Uuid;

use client_api::error::ErrorCode;

use database_entity::dto::QueryCollab;
use wasm_bindgen::prelude::*;

#[global_allocator]
static ALLOC: wee_alloc::WeeAlloc = wee_alloc::WeeAlloc::INIT;

#[wasm_bindgen]
extern "C" {
  #[wasm_bindgen(js_namespace = console)]
  fn log(msg: &str);

  #[wasm_bindgen(js_namespace = console)]
  fn error(msg: &str);

  #[wasm_bindgen(js_namespace = console)]
  fn info(msg: &str);

  #[wasm_bindgen(js_namespace = console)]
  fn debug(msg: &str);

  #[wasm_bindgen(js_namespace = console)]
  fn warn(msg: &str);

  #[wasm_bindgen(js_namespace = console)]
  fn trace(msg: &str);

  #[wasm_bindgen(js_namespace = window)]
  fn refresh_token(token: &str);

  #[wasm_bindgen(js_namespace = window)]
  fn invalid_token();
}

#[wasm_bindgen]
pub struct ClientAPI {
  client: Arc<Client>,
}

#[wasm_bindgen]
impl ClientAPI {
  pub fn new(config: ClientAPIConfig) -> ClientAPI {
    tracing_wasm::set_as_global_default();
    console_error_panic_hook::set_once();
    let configuration = ClientConfiguration::default();

    if let Some(compression) = &config.configuration {
      configuration
        .to_owned()
        .with_compression_buffer_size(compression.compression_buffer_size)
        .with_compression_quality(compression.compression_quality);
    }

    let client = Client::new(
      config.base_url.as_str(),
      config.ws_addr.as_str(),
      config.gotrue_url.as_str(),
      config.device_id.as_str(),
      configuration,
      config.client_id.as_str(),
    );

    tracing::debug!("Client API initialized, config: {:?}", config);
    ClientAPI {
      client: Arc::new(client),
    }
  }

  pub fn subscribe(&self) {
    let mut rx = self.client.subscribe_token_state();
    let client = self.client.clone();

    wasm_bindgen_futures::spawn_local(async move {
      while let Ok(state) = rx.recv().await {
        match state {
          TokenState::Refresh => {
            if let Ok(token) = client.get_token() {
              refresh_token(token.as_str());
            } else {
              invalid_token();
            }
          },
          TokenState::Invalid => {
            invalid_token();
          },
        }
      }
    });
  }
  pub async fn login(&self, email: &str, password: &str) -> Result<(), ClientResponse> {
    match self.client.sign_in_password(email, password).await {
      Ok(_) => Ok(()),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn sign_up(&self, email: &str, password: &str) -> Result<(), ClientResponse> {
    match self.client.sign_up(email, password).await {
      Ok(_) => Ok(()),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn logout(&self) -> Result<(), ClientResponse> {
    match self.client.sign_out().await {
      Ok(_) => Ok(()),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn sign_in_with_magic_link(
    &self,
    email: &str,
    redirect_to: &str,
  ) -> Result<(), ClientResponse> {
    match self
      .client
      .sign_in_with_magic_link(email, Some(redirect_to.to_string()))
      .await
    {
      Ok(_) => Ok(()),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn generate_oauth_url_with_provider(
    &self,
    provider: &str,
    redirect_to: &str,
  ) -> Result<OAuthURLResponse, ClientResponse> {
    if provider.is_empty() {
      return Err(ClientResponse {
        code: ErrorCode::OAuthError,
        message: "Provider is empty".to_string(),
      });
    }

    let provider = parse_provider(provider);

    match self
      .client
      .generate_url_with_provider_and_redirect_to(&provider, Some(redirect_to.to_string()))
      .await
    {
      Ok(url) => Ok(OAuthURLResponse { url }),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn sign_in_with_url(&self, url: &str) -> Result<(), ClientResponse> {
    match self.client.sign_in_with_url(url).await {
      Ok(_) => Ok(()),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn get_user(&self) -> Result<User, ClientResponse> {
    match self.client.get_profile().await {
      Ok(profile) => Ok(User::from(profile)),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub fn restore_token(&self, token: &str) -> Result<(), ClientResponse> {
    match self.client.restore_token(token) {
      Ok(_) => Ok(()),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn get_collab(
    &self,
    params: ClientQueryCollabParams,
  ) -> Result<ClientEncodeCollab, ClientResponse> {
    tracing::debug!("get_collab: {:?}", params);
    match self.client.get_collab(params.into()).await {
      Ok(data) => Ok(ClientEncodeCollab::from(data.encode_collab)),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn batch_get_collab(
    &self,
    workspace_id: String,
    params: BatchClientQueryCollab,
  ) -> Result<BatchClientEncodeCollab, ClientResponse> {
    tracing::debug!("batch_get_collab: {:?}", params);
    let workspace_id = workspace_id.as_str();
    let params: Vec<QueryCollab> = params.0.into_iter().map(|p| p.into()).collect();
    match self.client.batch_post_collab(workspace_id, params).await {
      Ok(data) => Ok(BatchClientEncodeCollab::from(data)),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn get_user_workspace(&self) -> Result<UserWorkspace, ClientResponse> {
    match self.client.get_user_workspace_info().await {
      Ok(workspace_info) => Ok(UserWorkspace::from(workspace_info)),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn get_publish_view_meta(
    &self,
    publish_namespace: String,
    publish_name: String,
  ) -> Result<PublishViewMeta, ClientResponse> {
    match self
      .client
      .get_published_collab::<serde_json::Value>(publish_namespace.as_str(), publish_name.as_str())
      .await
    {
      Ok(data) => Ok(PublishViewMeta {
        data: data.to_string(),
      }),

      Err(err) => Err(ClientResponse::from(err)),
    }
  }

  pub async fn get_publish_view(
    &self,
    publish_namespace: String,
    publish_name: String,
  ) -> Result<PublishViewPayload, ClientResponse> {
    let meta = self
      .get_publish_view_meta(publish_namespace.clone(), publish_name.clone())
      .await?;

    let data = self
      .client
      .get_published_collab_blob(publish_namespace.as_str(), publish_name.as_str())
      .await
      .map_err(ClientResponse::from)?;

    Ok(PublishViewPayload {
      meta,
      data: data.to_vec(),
    })
  }

  pub async fn get_publish_info(&self, view_id: String) -> Result<PublishInfo, ClientResponse> {
    let view_id = Uuid::parse_str(view_id.as_str()).map_err(|err| ClientResponse {
      code: ErrorCode::UuidError,
      message: format!("Failed to parse view_id: {}", err),
    })?;
    match self.client.get_published_collab_info(&view_id).await {
      Ok(info) => Ok(PublishInfo {
        namespace: info.namespace,
        publish_name: info.publish_name,
      }),
      Err(err) => Err(ClientResponse::from(err)),
    }
  }
}