Pass render purpose to render

This commit is contained in:
Gregory Schier
2024-08-22 11:27:55 -07:00
parent e5ec86bfcf
commit 9cdc13b632
12 changed files with 38 additions and 25 deletions

8
package-lock.json generated
View File

@@ -27,7 +27,7 @@
"@tauri-apps/plugin-log": "^2.0.0-rc.0", "@tauri-apps/plugin-log": "^2.0.0-rc.0",
"@tauri-apps/plugin-os": "^2.0.0-rc.0", "@tauri-apps/plugin-os": "^2.0.0-rc.0",
"@tauri-apps/plugin-shell": "^2.0.0-rc.0", "@tauri-apps/plugin-shell": "^2.0.0-rc.0",
"@yaakapp/api": "^0.1.10", "@yaakapp/api": "^0.1.11",
"buffer": "^6.0.3", "buffer": "^6.0.3",
"classnames": "^2.3.2", "classnames": "^2.3.2",
"cm6-graphql": "^0.0.9", "cm6-graphql": "^0.0.9",
@@ -2990,9 +2990,9 @@
"integrity": "sha512-N8tkAACJx2ww8vFMneJmaAgmjAG1tnVBZJRLRcx061tmsLRZHSEZSLuGWnwPtunsSLvSqXQ2wfp7Mgqg1I+2dQ==" "integrity": "sha512-N8tkAACJx2ww8vFMneJmaAgmjAG1tnVBZJRLRcx061tmsLRZHSEZSLuGWnwPtunsSLvSqXQ2wfp7Mgqg1I+2dQ=="
}, },
"node_modules/@yaakapp/api": { "node_modules/@yaakapp/api": {
"version": "0.1.10", "version": "0.1.11",
"resolved": "https://registry.npmjs.org/@yaakapp/api/-/api-0.1.10.tgz", "resolved": "https://registry.npmjs.org/@yaakapp/api/-/api-0.1.11.tgz",
"integrity": "sha512-K4iV7Et+5MmE+4Eui0mu/e8Y5vlvfSyykEy6Wl2sBOg6vRqr87N+OyUs/gkct7TU07QM8zew/hbM/OTESVhQWQ==", "integrity": "sha512-dRZAXrQpftWygy9nJXiIYPzLA9om6reO/RiEacMe9RKqMjNG6FRF8cGmj7BcdyreizJOkH/DmcOpxn09kDD0XA==",
"dependencies": { "dependencies": {
"@types/node": "^22.0.0" "@types/node": "^22.0.0"
} }

View File

@@ -42,7 +42,7 @@
"@tauri-apps/plugin-log": "^2.0.0-rc.0", "@tauri-apps/plugin-log": "^2.0.0-rc.0",
"@tauri-apps/plugin-os": "^2.0.0-rc.0", "@tauri-apps/plugin-os": "^2.0.0-rc.0",
"@tauri-apps/plugin-shell": "^2.0.0-rc.0", "@tauri-apps/plugin-shell": "^2.0.0-rc.0",
"@yaakapp/api": "^0.1.10", "@yaakapp/api": "^0.1.11",
"buffer": "^6.0.3", "buffer": "^6.0.3",
"classnames": "^2.3.2", "classnames": "^2.3.2",
"cm6-graphql": "^0.0.9", "cm6-graphql": "^0.0.9",

View File

@@ -1,12 +1,12 @@
{ {
"name": "@yaakapp/api", "name": "@yaakapp/api",
"version": "0.1.7", "version": "0.1.11",
"lockfileVersion": 3, "lockfileVersion": 3,
"requires": true, "requires": true,
"packages": { "packages": {
"": { "": {
"name": "@yaakapp/api", "name": "@yaakapp/api",
"version": "0.1.7", "version": "0.1.11",
"dependencies": { "dependencies": {
"@types/node": "^22.0.0" "@types/node": "^22.0.0"
}, },

View File

@@ -1,6 +1,6 @@
{ {
"name": "@yaakapp/api", "name": "@yaakapp/api",
"version": "0.1.10", "version": "0.1.11",
"main": "lib/index.js", "main": "lib/index.js",
"typings": "./lib/index.d.ts", "typings": "./lib/index.d.ts",
"files": [ "files": [

View File

@@ -1,3 +1,3 @@
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually. // This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
export type CallTemplateFunctionPurpose = { "type": "send" } | { "type": "preview" }; export type CallTemplateFunctionPurpose = "send" | "preview";

View File

@@ -1,16 +1,16 @@
const {readdirSync, cpSync} = require("node:fs"); const { readdirSync, cpSync } = require('node:fs');
const path = require("node:path"); const path = require('node:path');
const {execSync} = require("node:child_process"); const { execSync } = require('node:child_process');
const pluginsDir = process.env.YAAK_PLUGINS_DIR; const pluginsDir = process.env.YAAK_PLUGINS_DIR;
if (!pluginsDir) { if (!pluginsDir) {
console.log("YAAK_PLUGINS_DIR is not set"); console.log('YAAK_PLUGINS_DIR is not set');
process.exit(1); process.exit(1);
} }
console.log('Installing Yaak plugins dependencies', pluginsDir); console.log('Installing Yaak plugins dependencies', pluginsDir);
execSync('npm ci', {cwd: pluginsDir}); execSync('npm ci', { cwd: pluginsDir });
console.log('Building Yaak plugins', pluginsDir); console.log('Building Yaak plugins', pluginsDir);
execSync('npm run build', {cwd: pluginsDir}); execSync('npm run build', { cwd: pluginsDir });
console.log('Copying Yaak plugins to', pluginsDir); console.log('Copying Yaak plugins to', pluginsDir);

View File

@@ -9,6 +9,7 @@ use std::time::Duration;
use crate::render::render_http_request; use crate::render::render_http_request;
use crate::response_err; use crate::response_err;
use crate::template_callback::PluginTemplateCallback;
use base64::Engine; use base64::Engine;
use http::header::{ACCEPT, USER_AGENT}; use http::header::{ACCEPT, USER_AGENT};
use http::{HeaderMap, HeaderName, HeaderValue}; use http::{HeaderMap, HeaderName, HeaderValue};
@@ -37,11 +38,13 @@ pub async fn send_http_request<R: Runtime>(
let workspace = get_workspace(window, &request.workspace_id) let workspace = get_workspace(window, &request.workspace_id)
.await .await
.expect("Failed to get Workspace"); .expect("Failed to get Workspace");
let cb = &*window.app_handle().state::<PluginTemplateCallback>();
let cb = cb.for_send();
let rendered_request = render_http_request( let rendered_request = render_http_request(
window.app_handle(),
&request, &request,
&workspace, &workspace,
environment.as_ref(), environment.as_ref(),
&cb,
) )
.await; .await;

View File

@@ -2006,11 +2006,12 @@ async fn handle_plugin_event<R: Runtime>(
None => None, None => None,
Some(id) => get_environment(&w, id.as_str()).await.ok(), Some(id) => get_environment(&w, id.as_str()).await.ok(),
}; };
let cb = &*app_handle.state::<PluginTemplateCallback>();
let rendered_http_request = render_http_request( let rendered_http_request = render_http_request(
app_handle,
&req.http_request, &req.http_request,
&workspace, &workspace,
environment.as_ref(), environment.as_ref(),
cb,
) )
.await; .await;
Some(InternalEventPayload::RenderHttpRequestResponse( Some(InternalEventPayload::RenderHttpRequestResponse(

View File

@@ -52,13 +52,12 @@ pub async fn render_grpc_request<R: Runtime>(
} }
} }
pub async fn render_http_request<R: Runtime>( pub async fn render_http_request(
app_handle: &AppHandle<R>,
r: &HttpRequest, r: &HttpRequest,
w: &Workspace, w: &Workspace,
e: Option<&Environment>, e: Option<&Environment>,
cb: &PluginTemplateCallback,
) -> HttpRequest { ) -> HttpRequest {
let cb = &*app_handle.state::<PluginTemplateCallback>();
let vars = &variables_from_environment(w, e, cb).await; let vars = &variables_from_environment(w, e, cb).await;
let mut url_parameters = Vec::new(); let mut url_parameters = Vec::new();

View File

@@ -1,15 +1,24 @@
use std::collections::HashMap; use std::collections::HashMap;
use tauri::{AppHandle, Manager}; use tauri::{AppHandle, Manager};
use yaak_plugin_runtime::events::CallTemplateFunctionPurpose;
use yaak_plugin_runtime::manager::PluginManager; use yaak_plugin_runtime::manager::PluginManager;
use yaak_templates::TemplateCallback; use yaak_templates::TemplateCallback;
#[derive(Clone)]
pub struct PluginTemplateCallback { pub struct PluginTemplateCallback {
app_handle: AppHandle, app_handle: AppHandle,
purpose: CallTemplateFunctionPurpose,
} }
impl PluginTemplateCallback { impl PluginTemplateCallback {
pub fn new(app_handle: AppHandle) -> PluginTemplateCallback { pub fn new(app_handle: AppHandle) -> PluginTemplateCallback {
PluginTemplateCallback { app_handle } PluginTemplateCallback { app_handle, purpose: CallTemplateFunctionPurpose::Preview }
}
pub fn for_send(&self) -> PluginTemplateCallback {
let mut v = self.clone();
v.purpose = CallTemplateFunctionPurpose::Send;
v
} }
} }
@@ -17,7 +26,7 @@ impl TemplateCallback for PluginTemplateCallback {
async fn run(&self, fn_name: &str, args: HashMap<String, String>) -> Result<String, String> { async fn run(&self, fn_name: &str, args: HashMap<String, String>) -> Result<String, String> {
let plugin_manager = self.app_handle.state::<PluginManager>(); let plugin_manager = self.app_handle.state::<PluginManager>();
let resp = plugin_manager let resp = plugin_manager
.call_template_function(fn_name, args) .call_template_function(fn_name, args, self.purpose.clone())
.await .await
.map_err(|e| e.to_string())?; .map_err(|e| e.to_string())?;
Ok(resp.unwrap_or_default()) Ok(resp.unwrap_or_default())

View File

@@ -171,7 +171,7 @@ pub struct ShowToastRequest {
} }
#[derive(Debug, Clone, Serialize, Deserialize, TS)] #[derive(Debug, Clone, Serialize, Deserialize, TS)]
#[serde(rename_all = "camelCase")] #[serde(rename_all = "snake_case")]
#[ts(export)] #[ts(export)]
pub enum ToastVariant { pub enum ToastVariant {
Custom, Custom,
@@ -294,7 +294,7 @@ pub struct CallTemplateFunctionArgs {
} }
#[derive(Debug, Clone, Serialize, Deserialize, TS)] #[derive(Debug, Clone, Serialize, Deserialize, TS)]
#[serde(rename_all = "snake_case", tag = "type")] #[serde(rename_all = "snake_case")]
#[ts(export)] #[ts(export)]
pub enum CallTemplateFunctionPurpose { pub enum CallTemplateFunctionPurpose {
Send, Send,

View File

@@ -115,11 +115,12 @@ impl PluginManager {
&self, &self,
fn_name: &str, fn_name: &str,
args: HashMap<String, String>, args: HashMap<String, String>,
purpose: CallTemplateFunctionPurpose,
) -> Result<Option<String>> { ) -> Result<Option<String>> {
let req = CallTemplateFunctionRequest { let req = CallTemplateFunctionRequest {
name: fn_name.to_string(), name: fn_name.to_string(),
args: CallTemplateFunctionArgs { args: CallTemplateFunctionArgs {
purpose: CallTemplateFunctionPurpose::Preview, purpose,
values: args, values: args,
}, },
}; };