AppFlowy/frontend/appflowy_tauri/src-tauri/src/request.rs

use flowy_core::FlowySDK;
use lib_dispatch::prelude::{
    AFPluginDispatcher, AFPluginEventResponse, AFPluginRequest, StatusCode,
};
use tauri::{AppHandle, Manager, State, Wry};

#[derive(Clone, Debug, serde::Deserialize)]
pub struct AFTauriRequest {
    ty: String,
    payload: Vec<u8>,
}

impl std::convert::From<AFTauriRequest> for AFPluginRequest {
    fn from(event: AFTauriRequest) -> Self {
        AFPluginRequest::new(event.ty).payload(event.payload)
    }
}

#[derive(Clone, serde::Serialize)]
pub struct AFTauriResponse {
    code: StatusCode,
    payload: Vec<u8>,
}

impl std::convert::From<AFPluginEventResponse> for AFTauriResponse {
    fn from(response: AFPluginEventResponse) -> Self {
        Self {
            code: response.status_code,
            payload: response.payload.to_vec(),
        }
    }
}

// Learn more about Tauri commands at https://tauri.app/v1/guides/features/command
#[tracing::instrument(level = "trace", skip(app_handler))]
#[tauri::command]
pub async fn invoke_request(
    request: AFTauriRequest,
    app_handler: AppHandle<Wry>,
) -> AFTauriResponse {
    let request: AFPluginRequest = request.into();
    let state: State<FlowySDK> = app_handler.state();
    let dispatcher = state.inner().dispatcher();
    let response = AFPluginDispatcher::async_send(dispatcher, request).await;
    response.into()
}