Function get_extension_state

src-tauri/src/extensions/mod.rs:48–84

(app: AppHandle)

#[tracing::instrument(level = "debug", skip(app))]
pub fn get_extension_state(app: AppHandle) -> Result<ExtensionHandler> {
    let ext_path = app.path().app_data_dir().unwrap().join("extensions");
    let tmp_dir = app.path().temp_dir().unwrap();
    let cache_dir = app.path().cache_dir().unwrap();
    let (ext_handler, mut ui_request_rx, ui_reply_tx) =
        ExtensionHandler::new(ext_path, cache_dir, tmp_dir);

    let app_clone = app.clone();
    async_runtime::spawn(async move {
        let app_handle = app.clone();
        let reply_handler = ReplyHandler::new(app_handle);
        loop {
            tracing::trace!("Waiting for extension UI requests");
            if let Some(resp) = ui_request_rx.recv().await {
                if let Some(data) = resp.data {
                    tracing::debug!("Got main command {:?}", data);
                    match reply_handler.handle_request(data).await {
                        Ok(reply) => {
                            ui_reply_tx
                                .send(GenericExtensionHostRequest {
                                    channel: resp.channel,
                                    data: Some(reply),
                                })
                                .unwrap();
                        }
                        Err(e) => {
                            tracing::error!("Failed to handle extension command {:?}", e)
                        }
                    }
                }
            }
        }
    });
    async_runtime::spawn(extension_runner_connected(app_clone));

    Ok(ext_handler)
}

#[tracing::instrument(level = "debug", skip(app))]
pub fn get_extension_handler(app: &AppHandle) -> State<'_, ExtensionHandler> {