mirror of
https://github.com/mountain-loop/yaak.git
synced 2026-04-22 16:48:30 +02:00
Reload plugins on change
This commit is contained in:
4
.github/workflows/release.yml
vendored
4
.github/workflows/release.yml
vendored
@@ -81,7 +81,9 @@ jobs:
|
|||||||
repo-token: ${{ secrets.GITHUB_TOKEN }}
|
repo-token: ${{ secrets.GITHUB_TOKEN }}
|
||||||
|
|
||||||
- name: Install yaak CLI
|
- name: Install yaak CLI
|
||||||
run: go install github.com/yaakapp/cli/cmd/yaakcli@latest
|
run: |
|
||||||
|
npm install -g @yaakapp/cli
|
||||||
|
yaakcli --version
|
||||||
|
|
||||||
- name: Run lint
|
- name: Run lint
|
||||||
run: npm run lint
|
run: npm run lint
|
||||||
|
|||||||
@@ -19,10 +19,12 @@ import type { ImportRequest } from "./ImportRequest";
|
|||||||
import type { ImportResponse } from "./ImportResponse";
|
import type { ImportResponse } from "./ImportResponse";
|
||||||
import type { PluginBootRequest } from "./PluginBootRequest";
|
import type { PluginBootRequest } from "./PluginBootRequest";
|
||||||
import type { PluginBootResponse } from "./PluginBootResponse";
|
import type { PluginBootResponse } from "./PluginBootResponse";
|
||||||
|
import type { PluginReloadRequest } from "./PluginReloadRequest";
|
||||||
|
import type { PluginReloadResponse } from "./PluginReloadResponse";
|
||||||
import type { RenderHttpRequestRequest } from "./RenderHttpRequestRequest";
|
import type { RenderHttpRequestRequest } from "./RenderHttpRequestRequest";
|
||||||
import type { RenderHttpRequestResponse } from "./RenderHttpRequestResponse";
|
import type { RenderHttpRequestResponse } from "./RenderHttpRequestResponse";
|
||||||
import type { SendHttpRequestRequest } from "./SendHttpRequestRequest";
|
import type { SendHttpRequestRequest } from "./SendHttpRequestRequest";
|
||||||
import type { SendHttpRequestResponse } from "./SendHttpRequestResponse";
|
import type { SendHttpRequestResponse } from "./SendHttpRequestResponse";
|
||||||
import type { ShowToastRequest } from "./ShowToastRequest";
|
import type { ShowToastRequest } from "./ShowToastRequest";
|
||||||
|
|
||||||
export type InternalEventPayload = { "type": "boot_request" } & PluginBootRequest | { "type": "boot_response" } & PluginBootResponse | { "type": "import_request" } & ImportRequest | { "type": "import_response" } & ImportResponse | { "type": "filter_request" } & FilterRequest | { "type": "filter_response" } & FilterResponse | { "type": "export_http_request_request" } & ExportHttpRequestRequest | { "type": "export_http_request_response" } & ExportHttpRequestResponse | { "type": "send_http_request_request" } & SendHttpRequestRequest | { "type": "send_http_request_response" } & SendHttpRequestResponse | { "type": "get_http_request_actions_request" } & GetHttpRequestActionsRequest | { "type": "get_http_request_actions_response" } & GetHttpRequestActionsResponse | { "type": "call_http_request_action_request" } & CallHttpRequestActionRequest | { "type": "get_template_functions_request" } | { "type": "get_template_functions_response" } & GetTemplateFunctionsResponse | { "type": "call_template_function_request" } & CallTemplateFunctionRequest | { "type": "call_template_function_response" } & CallTemplateFunctionResponse | { "type": "copy_text_request" } & CopyTextRequest | { "type": "render_http_request_request" } & RenderHttpRequestRequest | { "type": "render_http_request_response" } & RenderHttpRequestResponse | { "type": "show_toast_request" } & ShowToastRequest | { "type": "get_http_request_by_id_request" } & GetHttpRequestByIdRequest | { "type": "get_http_request_by_id_response" } & GetHttpRequestByIdResponse | { "type": "find_http_responses_request" } & FindHttpResponsesRequest | { "type": "find_http_responses_response" } & FindHttpResponsesResponse | { "type": "empty_response" } & EmptyResponse;
|
export type InternalEventPayload = { "type": "boot_request" } & PluginBootRequest | { "type": "boot_response" } & PluginBootResponse | { "type": "reload_request" } & PluginReloadRequest | { "type": "reload_response" } & PluginReloadResponse | { "type": "import_request" } & ImportRequest | { "type": "import_response" } & ImportResponse | { "type": "filter_request" } & FilterRequest | { "type": "filter_response" } & FilterResponse | { "type": "export_http_request_request" } & ExportHttpRequestRequest | { "type": "export_http_request_response" } & ExportHttpRequestResponse | { "type": "send_http_request_request" } & SendHttpRequestRequest | { "type": "send_http_request_response" } & SendHttpRequestResponse | { "type": "get_http_request_actions_request" } & GetHttpRequestActionsRequest | { "type": "get_http_request_actions_response" } & GetHttpRequestActionsResponse | { "type": "call_http_request_action_request" } & CallHttpRequestActionRequest | { "type": "get_template_functions_request" } | { "type": "get_template_functions_response" } & GetTemplateFunctionsResponse | { "type": "call_template_function_request" } & CallTemplateFunctionRequest | { "type": "call_template_function_response" } & CallTemplateFunctionResponse | { "type": "copy_text_request" } & CopyTextRequest | { "type": "render_http_request_request" } & RenderHttpRequestRequest | { "type": "render_http_request_response" } & RenderHttpRequestResponse | { "type": "show_toast_request" } & ShowToastRequest | { "type": "get_http_request_by_id_request" } & GetHttpRequestByIdRequest | { "type": "get_http_request_by_id_response" } & GetHttpRequestByIdResponse | { "type": "find_http_responses_request" } & FindHttpResponsesRequest | { "type": "find_http_responses_response" } & FindHttpResponsesResponse | { "type": "empty_response" } & EmptyResponse;
|
||||||
|
|||||||
@@ -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 Plugin = { id: string, model: "plugin", createdAt: string, updatedAt: string, checkedAt: string | null, name: string, version: string, uri: string, enabled: boolean, };
|
export type Plugin = { id: string, model: "plugin", createdAt: string, updatedAt: string, checkedAt: string | null, directory: string, url: string | null, enabled: boolean, };
|
||||||
|
|||||||
3
plugin-runtime-types/src/gen/PluginReloadRequest.ts
Normal file
3
plugin-runtime-types/src/gen/PluginReloadRequest.ts
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
export type PluginReloadRequest = {};
|
||||||
3
plugin-runtime-types/src/gen/PluginReloadResponse.ts
Normal file
3
plugin-runtime-types/src/gen/PluginReloadResponse.ts
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
export type PluginReloadResponse = {};
|
||||||
@@ -45,6 +45,8 @@ export * from './gen/KeyValue';
|
|||||||
export * from './gen/Model';
|
export * from './gen/Model';
|
||||||
export * from './gen/PluginBootRequest';
|
export * from './gen/PluginBootRequest';
|
||||||
export * from './gen/PluginBootResponse';
|
export * from './gen/PluginBootResponse';
|
||||||
|
export * from './gen/PluginReloadRequest';
|
||||||
|
export * from './gen/PluginReloadResponse';
|
||||||
export * from './gen/RenderHttpRequestRequest';
|
export * from './gen/RenderHttpRequestRequest';
|
||||||
export * from './gen/RenderHttpRequestResponse';
|
export * from './gen/RenderHttpRequestResponse';
|
||||||
export * from './gen/RenderPurpose';
|
export * from './gen/RenderPurpose';
|
||||||
|
|||||||
@@ -4,30 +4,35 @@ import { Worker } from 'node:worker_threads';
|
|||||||
import { EventChannel } from './EventChannel';
|
import { EventChannel } from './EventChannel';
|
||||||
|
|
||||||
export class PluginHandle {
|
export class PluginHandle {
|
||||||
readonly #worker: Worker;
|
#worker: Worker;
|
||||||
|
|
||||||
constructor(
|
constructor(
|
||||||
readonly pluginDir: string,
|
readonly pluginDir: string,
|
||||||
readonly pluginRefId: string,
|
readonly pluginRefId: string,
|
||||||
readonly events: EventChannel,
|
readonly events: EventChannel,
|
||||||
) {
|
) {
|
||||||
const workerPath = process.env.YAAK_WORKER_PATH ?? path.join(__dirname, 'index.worker.cjs');
|
this.#worker = this.#createWorker();
|
||||||
this.#worker = new Worker(workerPath, {
|
|
||||||
workerData: {
|
|
||||||
pluginDir,
|
|
||||||
pluginRefId,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
this.#worker.on('message', (e) => this.events.emit(e));
|
|
||||||
this.#worker.on('error', this.#handleError.bind(this));
|
|
||||||
this.#worker.on('exit', this.#handleExit.bind(this));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
sendToWorker(event: InternalEvent) {
|
sendToWorker(event: InternalEvent) {
|
||||||
this.#worker.postMessage(event);
|
this.#worker.postMessage(event);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#createWorker(): Worker {
|
||||||
|
const workerPath = process.env.YAAK_WORKER_PATH ?? path.join(__dirname, 'index.worker.cjs');
|
||||||
|
const worker = new Worker(workerPath, {
|
||||||
|
workerData: { pluginDir: this.pluginDir, pluginRefId: this.pluginRefId },
|
||||||
|
});
|
||||||
|
|
||||||
|
worker.on('message', (e) => this.events.emit(e));
|
||||||
|
worker.on('error', this.#handleError.bind(this));
|
||||||
|
worker.on('exit', this.#handleExit.bind(this));
|
||||||
|
|
||||||
|
console.log('Created plugin worker for ', this.pluginDir);
|
||||||
|
|
||||||
|
return worker;
|
||||||
|
}
|
||||||
|
|
||||||
async #handleError(err: Error) {
|
async #handleError(err: Error) {
|
||||||
console.error('Plugin errored', this.pluginDir, err);
|
console.error('Plugin errored', this.pluginDir, err);
|
||||||
}
|
}
|
||||||
@@ -36,7 +41,7 @@ export class PluginHandle {
|
|||||||
if (code === 0) {
|
if (code === 0) {
|
||||||
console.log('Plugin exited successfully', this.pluginDir);
|
console.log('Plugin exited successfully', this.pluginDir);
|
||||||
} else {
|
} else {
|
||||||
console.log('Plugin exited with error', code, this.pluginDir);
|
console.log('Plugin exited with status', code, this.pluginDir);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -10,34 +10,31 @@ import {
|
|||||||
SendHttpRequestResponse,
|
SendHttpRequestResponse,
|
||||||
TemplateFunction,
|
TemplateFunction,
|
||||||
} from '@yaakapp/api';
|
} from '@yaakapp/api';
|
||||||
import { HttpRequestActionPlugin } from '@yaakapp/api/lib/plugins/httpRequestAction';
|
import { HttpRequestActionPlugin } from '@yaakapp/api/lib/plugins/HttpRequestActionPlugin';
|
||||||
import { TemplateFunctionPlugin } from '@yaakapp/api/lib/plugins/TemplateFunctionPlugin';
|
import { TemplateFunctionPlugin } from '@yaakapp/api/lib/plugins/TemplateFunctionPlugin';
|
||||||
import interceptStdout from 'intercept-stdout';
|
import interceptStdout from 'intercept-stdout';
|
||||||
import * as console from 'node:console';
|
import * as console from 'node:console';
|
||||||
import { readFileSync } from 'node:fs';
|
import { readFileSync, watch } from 'node:fs';
|
||||||
import path from 'node:path';
|
import path from 'node:path';
|
||||||
import * as util from 'node:util';
|
import * as util from 'node:util';
|
||||||
import { parentPort, workerData } from 'node:worker_threads';
|
import { parentPort, workerData } from 'node:worker_threads';
|
||||||
|
|
||||||
new Promise<void>(async (resolve, reject) => {
|
async function initialize() {
|
||||||
const { pluginDir, pluginRefId } = workerData;
|
const { pluginDir, pluginRefId } = workerData;
|
||||||
const pathPkg = path.join(pluginDir, 'package.json');
|
const pathPkg = path.join(pluginDir, 'package.json');
|
||||||
|
|
||||||
// NOTE: Use POSIX join because require() needs forward slash
|
|
||||||
const pathMod = path.posix.join(pluginDir, 'build', 'index.js');
|
const pathMod = path.posix.join(pluginDir, 'build', 'index.js');
|
||||||
|
async function importModule() {
|
||||||
let pkg: { [x: string]: any };
|
const id = require.resolve(pathMod);
|
||||||
try {
|
delete require.cache[id];
|
||||||
pkg = JSON.parse(readFileSync(pathPkg, 'utf8'));
|
return require(id);
|
||||||
} catch (err) {
|
|
||||||
// TODO: Do something better here
|
|
||||||
reject(err);
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const pkg = JSON.parse(readFileSync(pathPkg, 'utf8'));
|
||||||
|
|
||||||
prefixStdout(`[plugin][${pkg.name}] %s`);
|
prefixStdout(`[plugin][${pkg.name}] %s`);
|
||||||
|
|
||||||
const mod = (await import(pathMod)).default ?? {};
|
let mod = await importModule();
|
||||||
|
|
||||||
const capabilities: string[] = [];
|
const capabilities: string[] = [];
|
||||||
if (typeof mod.pluginHookExport === 'function') capabilities.push('export');
|
if (typeof mod.pluginHookExport === 'function') capabilities.push('export');
|
||||||
@@ -94,6 +91,18 @@ new Promise<void>(async (resolve, reject) => {
|
|||||||
return promise as unknown as Promise<T>;
|
return promise as unknown as Promise<T>;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async function reloadModule() {
|
||||||
|
mod = await importModule();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reload plugin if JS or package.json changes
|
||||||
|
const cb = async () => {
|
||||||
|
await reloadModule();
|
||||||
|
return sendPayload({ type: 'reload_response' }, null);
|
||||||
|
};
|
||||||
|
watch(path.join(pathMod), cb);
|
||||||
|
watch(path.join(pathPkg), cb);
|
||||||
|
|
||||||
const ctx: Context = {
|
const ctx: Context = {
|
||||||
clipboard: {
|
clipboard: {
|
||||||
async copyText(text) {
|
async copyText(text) {
|
||||||
@@ -248,6 +257,10 @@ new Promise<void>(async (resolve, reject) => {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (payload.type === 'reload_request') {
|
||||||
|
await reloadModule();
|
||||||
|
}
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
console.log('Plugin call threw exception', payload.type, err);
|
console.log('Plugin call threw exception', payload.type, err);
|
||||||
// TODO: Return errors to server
|
// TODO: Return errors to server
|
||||||
@@ -256,9 +269,9 @@ new Promise<void>(async (resolve, reject) => {
|
|||||||
// No matches, so send back an empty response so the caller doesn't block forever
|
// No matches, so send back an empty response so the caller doesn't block forever
|
||||||
sendEmpty(replyId);
|
sendEmpty(replyId);
|
||||||
});
|
});
|
||||||
|
}
|
||||||
|
|
||||||
resolve();
|
initialize().catch((err) => {
|
||||||
}).catch((err) => {
|
|
||||||
console.log('failed to boot plugin', err);
|
console.log('failed to boot plugin', err);
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
@@ -11,6 +11,7 @@ use std::str::FromStr;
|
|||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
use base64::Engine;
|
use base64::Engine;
|
||||||
|
use chrono::Utc;
|
||||||
use fern::colors::ColoredLevelConfig;
|
use fern::colors::ColoredLevelConfig;
|
||||||
use log::{debug, error, info, warn};
|
use log::{debug, error, info, warn};
|
||||||
use rand::random;
|
use rand::random;
|
||||||
@@ -61,6 +62,7 @@ use yaak_plugin_runtime::events::{
|
|||||||
InternalEvent, InternalEventPayload, PluginBootResponse, RenderHttpRequestResponse,
|
InternalEvent, InternalEventPayload, PluginBootResponse, RenderHttpRequestResponse,
|
||||||
SendHttpRequestResponse,
|
SendHttpRequestResponse,
|
||||||
};
|
};
|
||||||
|
use yaak_plugin_runtime::handle::PluginHandle;
|
||||||
use yaak_templates::{Parser, Tokens};
|
use yaak_templates::{Parser, Tokens};
|
||||||
|
|
||||||
mod analytics;
|
mod analytics;
|
||||||
@@ -1170,7 +1172,11 @@ async fn cmd_create_workspace(name: &str, w: WebviewWindow) -> Result<Workspace,
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
async fn cmd_create_plugin(directory: &str, url: Option<String>, w: WebviewWindow) -> Result<Plugin, String> {
|
async fn cmd_create_plugin(
|
||||||
|
directory: &str,
|
||||||
|
url: Option<String>,
|
||||||
|
w: WebviewWindow,
|
||||||
|
) -> Result<Plugin, String> {
|
||||||
upsert_plugin(
|
upsert_plugin(
|
||||||
&w,
|
&w,
|
||||||
Plugin {
|
Plugin {
|
||||||
@@ -1456,6 +1462,12 @@ async fn cmd_list_plugins(w: WebviewWindow) -> Result<Vec<Plugin>, String> {
|
|||||||
list_plugins(&w).await.map_err(|e| e.to_string())
|
list_plugins(&w).await.map_err(|e| e.to_string())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
async fn cmd_reload_plugins(plugin_manager: State<'_, PluginManager>) -> Result<(), String> {
|
||||||
|
plugin_manager.reload_all().await;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
async fn cmd_plugin_info(
|
async fn cmd_plugin_info(
|
||||||
id: &str,
|
id: &str,
|
||||||
@@ -1734,9 +1746,6 @@ pub fn run() {
|
|||||||
cmd_delete_http_response,
|
cmd_delete_http_response,
|
||||||
cmd_delete_workspace,
|
cmd_delete_workspace,
|
||||||
cmd_dismiss_notification,
|
cmd_dismiss_notification,
|
||||||
cmd_parse_template,
|
|
||||||
cmd_template_tokens_to_string,
|
|
||||||
cmd_render_template,
|
|
||||||
cmd_duplicate_grpc_request,
|
cmd_duplicate_grpc_request,
|
||||||
cmd_duplicate_http_request,
|
cmd_duplicate_http_request,
|
||||||
cmd_export_data,
|
cmd_export_data,
|
||||||
@@ -1752,7 +1761,6 @@ pub fn run() {
|
|||||||
cmd_grpc_go,
|
cmd_grpc_go,
|
||||||
cmd_grpc_reflect,
|
cmd_grpc_reflect,
|
||||||
cmd_http_request_actions,
|
cmd_http_request_actions,
|
||||||
cmd_template_functions,
|
|
||||||
cmd_import_data,
|
cmd_import_data,
|
||||||
cmd_list_cookie_jars,
|
cmd_list_cookie_jars,
|
||||||
cmd_list_environments,
|
cmd_list_environments,
|
||||||
@@ -1763,16 +1771,21 @@ pub fn run() {
|
|||||||
cmd_list_http_requests,
|
cmd_list_http_requests,
|
||||||
cmd_list_http_responses,
|
cmd_list_http_responses,
|
||||||
cmd_list_plugins,
|
cmd_list_plugins,
|
||||||
cmd_plugin_info,
|
|
||||||
cmd_list_workspaces,
|
cmd_list_workspaces,
|
||||||
cmd_metadata,
|
cmd_metadata,
|
||||||
cmd_new_nested_window,
|
cmd_new_nested_window,
|
||||||
cmd_new_window,
|
cmd_new_window,
|
||||||
|
cmd_parse_template,
|
||||||
|
cmd_plugin_info,
|
||||||
|
cmd_reload_plugins,
|
||||||
|
cmd_render_template,
|
||||||
cmd_save_response,
|
cmd_save_response,
|
||||||
cmd_send_ephemeral_request,
|
cmd_send_ephemeral_request,
|
||||||
cmd_send_http_request,
|
cmd_send_http_request,
|
||||||
cmd_set_key_value,
|
cmd_set_key_value,
|
||||||
cmd_set_update_mode,
|
cmd_set_update_mode,
|
||||||
|
cmd_template_functions,
|
||||||
|
cmd_template_tokens_to_string,
|
||||||
cmd_track_event,
|
cmd_track_event,
|
||||||
cmd_update_cookie_jar,
|
cmd_update_cookie_jar,
|
||||||
cmd_update_environment,
|
cmd_update_environment,
|
||||||
@@ -1996,17 +2009,25 @@ fn monitor_plugin_events<R: Runtime>(app_handle: &AppHandle<R>) {
|
|||||||
|
|
||||||
while let Some(event) = rx.recv().await {
|
while let Some(event) = rx.recv().await {
|
||||||
let app_handle = app_handle.clone();
|
let app_handle = app_handle.clone();
|
||||||
|
let plugin = plugin_manager
|
||||||
|
.get_plugin(event.plugin_ref_id.as_str())
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
// We might have recursive back-and-forth calls between app and plugin, so we don't
|
// We might have recursive back-and-forth calls between app and plugin, so we don't
|
||||||
// want to block here
|
// want to block here
|
||||||
tauri::async_runtime::spawn(async move {
|
tauri::async_runtime::spawn(async move {
|
||||||
handle_plugin_event(&app_handle, &event).await;
|
handle_plugin_event(&app_handle, &event, &plugin).await;
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_plugin_event<R: Runtime>(app_handle: &AppHandle<R>, event: &InternalEvent) {
|
async fn handle_plugin_event<R: Runtime>(
|
||||||
|
app_handle: &AppHandle<R>,
|
||||||
|
event: &InternalEvent,
|
||||||
|
plugin_handle: &PluginHandle,
|
||||||
|
) {
|
||||||
// info!("Got event to app {}", event.id);
|
// info!("Got event to app {}", event.id);
|
||||||
let response_event: Option<InternalEventPayload> = match event.clone().payload {
|
let response_event: Option<InternalEventPayload> = match event.clone().payload {
|
||||||
InternalEventPayload::CopyTextRequest(req) => {
|
InternalEventPayload::CopyTextRequest(req) => {
|
||||||
@@ -2064,6 +2085,27 @@ async fn handle_plugin_event<R: Runtime>(app_handle: &AppHandle<R>, event: &Inte
|
|||||||
},
|
},
|
||||||
))
|
))
|
||||||
}
|
}
|
||||||
|
InternalEventPayload::ReloadResponse(_) => {
|
||||||
|
let w = get_focused_window_no_lock(app_handle).expect("No focused window");
|
||||||
|
let plugins = list_plugins(&w).await.unwrap();
|
||||||
|
for plugin in plugins {
|
||||||
|
if plugin.directory != plugin_handle.dir {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
upsert_plugin(
|
||||||
|
&w,
|
||||||
|
Plugin {
|
||||||
|
// TODO: Add reloaded_at field to use instead
|
||||||
|
updated_at: Utc::now().naive_utc(),
|
||||||
|
..plugin
|
||||||
|
},
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
InternalEventPayload::SendHttpRequestRequest(req) => {
|
InternalEventPayload::SendHttpRequestRequest(req) => {
|
||||||
let w = get_focused_window_no_lock(app_handle).expect("No focused window");
|
let w = get_focused_window_no_lock(app_handle).expect("No focused window");
|
||||||
let url = w.url().unwrap();
|
let url = w.url().unwrap();
|
||||||
@@ -2124,7 +2166,6 @@ fn get_focused_window_no_lock<R: Runtime>(app_handle: &AppHandle<R>) -> Option<W
|
|||||||
// TODO: Getting the focused window doesn't seem to work on Windows, so
|
// TODO: Getting the focused window doesn't seem to work on Windows, so
|
||||||
// we'll need to pass the window label into plugin events instead.
|
// we'll need to pass the window label into plugin events instead.
|
||||||
if app_handle.webview_windows().len() == 1 {
|
if app_handle.webview_windows().len() == 1 {
|
||||||
debug!("Returning only webview window");
|
|
||||||
let w = app_handle
|
let w = app_handle
|
||||||
.webview_windows()
|
.webview_windows()
|
||||||
.iter()
|
.iter()
|
||||||
|
|||||||
@@ -2,7 +2,10 @@ use serde::{Deserialize, Serialize};
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use ts_rs::TS;
|
use ts_rs::TS;
|
||||||
|
|
||||||
use yaak_models::models::{CookieJar, Environment, Folder, GrpcConnection, GrpcEvent, GrpcRequest, HttpRequest, HttpResponse, KeyValue, Plugin, Settings, Workspace};
|
use yaak_models::models::{
|
||||||
|
CookieJar, Environment, Folder, GrpcConnection, GrpcEvent, GrpcRequest, HttpRequest,
|
||||||
|
HttpResponse, KeyValue, Plugin, Settings, Workspace,
|
||||||
|
};
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize, TS)]
|
#[derive(Debug, Clone, Serialize, Deserialize, TS)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
@@ -21,6 +24,9 @@ pub enum InternalEventPayload {
|
|||||||
BootRequest(PluginBootRequest),
|
BootRequest(PluginBootRequest),
|
||||||
BootResponse(PluginBootResponse),
|
BootResponse(PluginBootResponse),
|
||||||
|
|
||||||
|
ReloadRequest(EmptyResponse),
|
||||||
|
ReloadResponse(EmptyResponse),
|
||||||
|
|
||||||
ImportRequest(ImportRequest),
|
ImportRequest(ImportRequest),
|
||||||
ImportResponse(ImportResponse),
|
ImportResponse(ImportResponse),
|
||||||
|
|
||||||
|
|||||||
66
src-tauri/yaak_plugin_runtime/src/handle.rs
Normal file
66
src-tauri/yaak_plugin_runtime/src/handle.rs
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
use crate::events::{EmptyResponse, InternalEvent, InternalEventPayload, PluginBootResponse};
|
||||||
|
use crate::server::plugin_runtime::EventStreamEvent;
|
||||||
|
use crate::util::gen_id;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::sync::{mpsc, Mutex};
|
||||||
|
|
||||||
|
#[derive(Clone)]
|
||||||
|
pub struct PluginHandle {
|
||||||
|
pub ref_id: String,
|
||||||
|
pub dir: String,
|
||||||
|
pub(crate) to_plugin_tx: Arc<Mutex<mpsc::Sender<tonic::Result<EventStreamEvent>>>>,
|
||||||
|
pub(crate) boot_resp: Arc<Mutex<Option<PluginBootResponse>>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl PluginHandle {
|
||||||
|
pub async fn name(&self) -> String {
|
||||||
|
match &*self.boot_resp.lock().await {
|
||||||
|
None => "__NOT_BOOTED__".to_string(),
|
||||||
|
Some(r) => r.name.to_owned(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn info(&self) -> Option<PluginBootResponse> {
|
||||||
|
let resp = &*self.boot_resp.lock().await;
|
||||||
|
resp.clone()
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn build_event_to_send(
|
||||||
|
&self,
|
||||||
|
payload: &InternalEventPayload,
|
||||||
|
reply_id: Option<String>,
|
||||||
|
) -> InternalEvent {
|
||||||
|
InternalEvent {
|
||||||
|
id: gen_id(),
|
||||||
|
plugin_ref_id: self.ref_id.clone(),
|
||||||
|
reply_id,
|
||||||
|
payload: payload.clone(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn reload(&self) -> crate::error::Result<()> {
|
||||||
|
let event = self.build_event_to_send(&InternalEventPayload::ReloadRequest(EmptyResponse{}), None);
|
||||||
|
self.send(&event).await
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn send(&self, event: &InternalEvent) -> crate::error::Result<()> {
|
||||||
|
// info!(
|
||||||
|
// "Sending event to plugin {} {:?}",
|
||||||
|
// event.id,
|
||||||
|
// self.name().await
|
||||||
|
// );
|
||||||
|
self.to_plugin_tx
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.send(Ok(EventStreamEvent {
|
||||||
|
event: serde_json::to_string(&event)?,
|
||||||
|
}))
|
||||||
|
.await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
pub async fn boot(&self, resp: &PluginBootResponse) {
|
||||||
|
let mut boot_resp = self.boot_resp.lock().await;
|
||||||
|
*boot_resp = Some(resp.clone());
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -4,3 +4,5 @@ pub mod manager;
|
|||||||
mod nodejs;
|
mod nodejs;
|
||||||
pub mod plugin;
|
pub mod plugin;
|
||||||
mod server;
|
mod server;
|
||||||
|
pub mod handle;
|
||||||
|
mod util;
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ use std::time::Duration;
|
|||||||
use tauri::{AppHandle, Runtime};
|
use tauri::{AppHandle, Runtime};
|
||||||
use tokio::sync::mpsc;
|
use tokio::sync::mpsc;
|
||||||
use tokio::sync::watch::Sender;
|
use tokio::sync::watch::Sender;
|
||||||
|
use crate::handle::PluginHandle;
|
||||||
|
|
||||||
pub struct PluginManager {
|
pub struct PluginManager {
|
||||||
kill_tx: Sender<bool>,
|
kill_tx: Sender<bool>,
|
||||||
@@ -38,6 +39,10 @@ impl PluginManager {
|
|||||||
PluginManager { kill_tx, server }
|
PluginManager { kill_tx, server }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn reload_all(&self) {
|
||||||
|
self.server.reload_plugins().await
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn subscribe(&self) -> (String, mpsc::Receiver<InternalEvent>) {
|
pub async fn subscribe(&self) -> (String, mpsc::Receiver<InternalEvent>) {
|
||||||
self.server.subscribe().await
|
self.server.subscribe().await
|
||||||
}
|
}
|
||||||
@@ -68,6 +73,10 @@ impl PluginManager {
|
|||||||
self.server.plugin_by_dir(dir).await.ok()?.info().await
|
self.server.plugin_by_dir(dir).await.ok()?.info().await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn get_plugin(&self, ref_id: &str) -> Result<PluginHandle> {
|
||||||
|
self.server.plugin_by_ref_id(ref_id).await
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn get_http_request_actions(&self) -> Result<Vec<GetHttpRequestActionsResponse>> {
|
pub async fn get_http_request_actions(&self) -> Result<Vec<GetHttpRequestActionsResponse>> {
|
||||||
let reply_events = self
|
let reply_events = self
|
||||||
.server
|
.server
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
use rand::distributions::{Alphanumeric, DistString};
|
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::pin::Pin;
|
use std::pin::Pin;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
|
use log::warn;
|
||||||
use tokio::sync::mpsc::Receiver;
|
use tokio::sync::mpsc::Receiver;
|
||||||
use tokio::sync::{mpsc, Mutex};
|
use tokio::sync::{mpsc, Mutex};
|
||||||
use tonic::codegen::tokio_stream::wrappers::ReceiverStream;
|
use tonic::codegen::tokio_stream::wrappers::ReceiverStream;
|
||||||
@@ -10,8 +10,10 @@ use tonic::{Request, Response, Status, Streaming};
|
|||||||
|
|
||||||
use crate::error::Error::PluginNotFoundErr;
|
use crate::error::Error::PluginNotFoundErr;
|
||||||
use crate::error::Result;
|
use crate::error::Result;
|
||||||
use crate::events::{PluginBootRequest, PluginBootResponse, InternalEvent, InternalEventPayload};
|
use crate::events::{InternalEvent, InternalEventPayload, PluginBootRequest, PluginBootResponse};
|
||||||
|
use crate::handle::PluginHandle;
|
||||||
use crate::server::plugin_runtime::plugin_runtime_server::PluginRuntime;
|
use crate::server::plugin_runtime::plugin_runtime_server::PluginRuntime;
|
||||||
|
use crate::util::gen_id;
|
||||||
use plugin_runtime::EventStreamEvent;
|
use plugin_runtime::EventStreamEvent;
|
||||||
use yaak_models::queries::generate_id;
|
use yaak_models::queries::generate_id;
|
||||||
|
|
||||||
@@ -22,62 +24,6 @@ pub mod plugin_runtime {
|
|||||||
type ResponseStream =
|
type ResponseStream =
|
||||||
Pin<Box<dyn Stream<Item = std::result::Result<EventStreamEvent, Status>> + Send>>;
|
Pin<Box<dyn Stream<Item = std::result::Result<EventStreamEvent, Status>> + Send>>;
|
||||||
|
|
||||||
#[derive(Clone)]
|
|
||||||
pub struct PluginHandle {
|
|
||||||
dir: String,
|
|
||||||
to_plugin_tx: Arc<Mutex<mpsc::Sender<tonic::Result<EventStreamEvent>>>>,
|
|
||||||
ref_id: String,
|
|
||||||
boot_resp: Arc<Mutex<Option<PluginBootResponse>>>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl PluginHandle {
|
|
||||||
pub async fn name(&self) -> String {
|
|
||||||
match &*self.boot_resp.lock().await {
|
|
||||||
None => "__NOT_BOOTED__".to_string(),
|
|
||||||
Some(r) => r.name.to_owned(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn info(&self) -> Option<PluginBootResponse> {
|
|
||||||
let resp = &*self.boot_resp.lock().await;
|
|
||||||
resp.clone()
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn build_event_to_send(
|
|
||||||
&self,
|
|
||||||
payload: &InternalEventPayload,
|
|
||||||
reply_id: Option<String>,
|
|
||||||
) -> InternalEvent {
|
|
||||||
InternalEvent {
|
|
||||||
id: gen_id(),
|
|
||||||
plugin_ref_id: self.ref_id.clone(),
|
|
||||||
reply_id,
|
|
||||||
payload: payload.clone(),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn send(&self, event: &InternalEvent) -> Result<()> {
|
|
||||||
// info!(
|
|
||||||
// "Sending event to plugin {} {:?}",
|
|
||||||
// event.id,
|
|
||||||
// self.name().await
|
|
||||||
// );
|
|
||||||
self.to_plugin_tx
|
|
||||||
.lock()
|
|
||||||
.await
|
|
||||||
.send(Ok(EventStreamEvent {
|
|
||||||
event: serde_json::to_string(&event)?,
|
|
||||||
}))
|
|
||||||
.await?;
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn boot(&self, resp: &PluginBootResponse) {
|
|
||||||
let mut boot_resp = self.boot_resp.lock().await;
|
|
||||||
*boot_resp = Some(resp.clone());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
pub struct PluginRuntimeGrpcServer {
|
pub struct PluginRuntimeGrpcServer {
|
||||||
plugin_ref_to_plugin: Arc<Mutex<HashMap<String, PluginHandle>>>,
|
plugin_ref_to_plugin: Arc<Mutex<HashMap<String, PluginHandle>>>,
|
||||||
@@ -96,6 +42,15 @@ impl PluginRuntimeGrpcServer {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn plugins(&self) -> Vec<PluginHandle> {
|
||||||
|
self.plugin_ref_to_plugin
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.iter()
|
||||||
|
.map(|p| p.1.to_owned())
|
||||||
|
.collect::<Vec<PluginHandle>>()
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn subscribe(&self) -> (String, Receiver<InternalEvent>) {
|
pub async fn subscribe(&self) -> (String, Receiver<InternalEvent>) {
|
||||||
let (tx, rx) = mpsc::channel(128);
|
let (tx, rx) = mpsc::channel(128);
|
||||||
let rx_id = generate_id();
|
let rx_id = generate_id();
|
||||||
@@ -115,23 +70,15 @@ impl PluginRuntimeGrpcServer {
|
|||||||
|
|
||||||
pub async fn remove_plugin(&self, id: &str) {
|
pub async fn remove_plugin(&self, id: &str) {
|
||||||
match self.plugin_ref_to_plugin.lock().await.remove(id) {
|
match self.plugin_ref_to_plugin.lock().await.remove(id) {
|
||||||
None => {
|
None => println!("Tried to remove non-existing plugin {}", id),
|
||||||
println!("Tried to remove non-existing plugin {}", id);
|
Some(plugin) => println!("Removed plugin {} {}", id, plugin.name().await),
|
||||||
}
|
|
||||||
Some(plugin) => {
|
|
||||||
println!("Removed plugin {} {}", id, plugin.name().await);
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn boot_plugin(&self, id: &str, resp: &PluginBootResponse) {
|
pub async fn boot_plugin(&self, id: &str, resp: &PluginBootResponse) {
|
||||||
match self.plugin_ref_to_plugin.lock().await.get(id) {
|
match self.plugin_ref_to_plugin.lock().await.get(id) {
|
||||||
None => {
|
None => println!("Tried booting non-existing plugin {}", id),
|
||||||
println!("Tried booting non-existing plugin {}", id);
|
Some(plugin) => plugin.clone().boot(resp).await,
|
||||||
}
|
|
||||||
Some(plugin) => {
|
|
||||||
plugin.clone().boot(resp).await;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -162,7 +109,7 @@ impl PluginRuntimeGrpcServer {
|
|||||||
Some(p) => Ok(p.to_owned()),
|
Some(p) => Ok(p.to_owned()),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn plugin_by_dir(&self, dir: &str) -> Result<PluginHandle> {
|
pub async fn plugin_by_dir(&self, dir: &str) -> Result<PluginHandle> {
|
||||||
let plugins = self.plugin_ref_to_plugin.lock().await;
|
let plugins = self.plugin_ref_to_plugin.lock().await;
|
||||||
for p in plugins.values() {
|
for p in plugins.values() {
|
||||||
@@ -299,6 +246,14 @@ impl PluginRuntimeGrpcServer {
|
|||||||
Ok(events)
|
Ok(events)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub async fn reload_plugins(&self) {
|
||||||
|
for (_, plugin) in self.plugin_ref_to_plugin.lock().await.clone() {
|
||||||
|
if let Err(e) = plugin.reload().await {
|
||||||
|
warn!("Failed to reload plugin {} {}", plugin.dir, e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
async fn load_plugins(
|
async fn load_plugins(
|
||||||
&self,
|
&self,
|
||||||
to_plugin_tx: mpsc::Sender<tonic::Result<EventStreamEvent>>,
|
to_plugin_tx: mpsc::Sender<tonic::Result<EventStreamEvent>>,
|
||||||
@@ -396,7 +351,3 @@ impl PluginRuntime for PluginRuntimeGrpcServer {
|
|||||||
))
|
))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn gen_id() -> String {
|
|
||||||
Alphanumeric.sample_string(&mut rand::thread_rng(), 5)
|
|
||||||
}
|
|
||||||
|
|||||||
5
src-tauri/yaak_plugin_runtime/src/util.rs
Normal file
5
src-tauri/yaak_plugin_runtime/src/util.rs
Normal file
@@ -0,0 +1,5 @@
|
|||||||
|
use rand::distributions::{Alphanumeric, DistString};
|
||||||
|
|
||||||
|
pub fn gen_id() -> String {
|
||||||
|
Alphanumeric.sample_string(&mut rand::thread_rng(), 5)
|
||||||
|
}
|
||||||
@@ -5,10 +5,12 @@ import type {
|
|||||||
HttpRequest,
|
HttpRequest,
|
||||||
} from '@yaakapp/api';
|
} from '@yaakapp/api';
|
||||||
import { invokeCmd } from '../lib/tauri';
|
import { invokeCmd } from '../lib/tauri';
|
||||||
|
import { usePlugins } from './usePlugins';
|
||||||
|
|
||||||
export function useHttpRequestActions() {
|
export function useHttpRequestActions() {
|
||||||
|
const plugins = usePlugins();
|
||||||
const httpRequestActions = useQuery({
|
const httpRequestActions = useQuery({
|
||||||
queryKey: ['http_request_actions'],
|
queryKey: ['http_request_actions', plugins.map((p) => p.updatedAt)],
|
||||||
refetchOnWindowFocus: false,
|
refetchOnWindowFocus: false,
|
||||||
queryFn: async () => {
|
queryFn: async () => {
|
||||||
const responses = (await invokeCmd(
|
const responses = (await invokeCmd(
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ import type { Plugin } from '@yaakapp/api';
|
|||||||
import { atom, useAtomValue, useSetAtom } from 'jotai';
|
import { atom, useAtomValue, useSetAtom } from 'jotai';
|
||||||
import { minPromiseMillis } from '../lib/minPromiseMillis';
|
import { minPromiseMillis } from '../lib/minPromiseMillis';
|
||||||
import { listPlugins } from '../lib/store';
|
import { listPlugins } from '../lib/store';
|
||||||
|
import { invokeCmd } from '../lib/tauri';
|
||||||
|
|
||||||
const plugins = await listPlugins();
|
const plugins = await listPlugins();
|
||||||
export const pluginsAtom = atom<Plugin[]>(plugins);
|
export const pluginsAtom = atom<Plugin[]>(plugins);
|
||||||
@@ -11,12 +12,20 @@ export function usePlugins() {
|
|||||||
return useAtomValue(pluginsAtom);
|
return useAtomValue(pluginsAtom);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Reload all plugins and refresh the list of plugins
|
||||||
|
*/
|
||||||
export function useRefreshPlugins() {
|
export function useRefreshPlugins() {
|
||||||
const setPlugins = useSetAtom(pluginsAtom);
|
const setPlugins = useSetAtom(pluginsAtom);
|
||||||
return useMutation({
|
return useMutation({
|
||||||
mutationKey: ['refresh_plugins'],
|
mutationKey: ['refresh_plugins'],
|
||||||
mutationFn: async () => {
|
mutationFn: async () => {
|
||||||
const plugins = await minPromiseMillis(listPlugins());
|
const plugins = await minPromiseMillis(
|
||||||
|
(async function () {
|
||||||
|
await invokeCmd('cmd_reload_plugins');
|
||||||
|
return listPlugins();
|
||||||
|
})(),
|
||||||
|
);
|
||||||
setPlugins(plugins);
|
setPlugins(plugins);
|
||||||
},
|
},
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -56,6 +56,7 @@ type TauriCmd =
|
|||||||
| 'cmd_parse_template'
|
| 'cmd_parse_template'
|
||||||
| 'cmd_plugin_info'
|
| 'cmd_plugin_info'
|
||||||
| 'cmd_render_template'
|
| 'cmd_render_template'
|
||||||
|
| 'cmd_reload_plugins'
|
||||||
| 'cmd_save_response'
|
| 'cmd_save_response'
|
||||||
| 'cmd_send_ephemeral_request'
|
| 'cmd_send_ephemeral_request'
|
||||||
| 'cmd_send_http_request'
|
| 'cmd_send_http_request'
|
||||||
|
|||||||
Reference in New Issue
Block a user