mirror of
https://github.com/mountain-loop/yaak.git
synced 2026-06-10 08:32:50 +02:00
Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 986143c4ae | |||
| 50b0e23d53 |
Generated
+47
-18
@@ -689,9 +689,9 @@ checksum = "8f1fe948ff07f4bd06c30984e69f5b4899c516a3ef74f34df92a2df2ab535495"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "bytes"
|
name = "bytes"
|
||||||
version = "1.11.1"
|
version = "1.10.1"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "1e748733b7cbc798e1434b6ac524f0c1ff2ab456fe201501e6497c8417a4fc33"
|
checksum = "d71b6127be86fdcfddb610f7182ac57211d4b18a3e9c82eb2d17662f2227ad6a"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"serde",
|
"serde",
|
||||||
]
|
]
|
||||||
@@ -1316,12 +1316,12 @@ checksum = "da692b8d1080ea3045efaab14434d40468c3d8657e42abddfffca87b428f4c1b"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "deranged"
|
name = "deranged"
|
||||||
version = "0.5.5"
|
version = "0.4.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "ececcb659e7ba858fb4f10388c250a7252eb0a27373f1a72b8748afdd248e587"
|
checksum = "9c9e6a11ca8224451684bc0d7d5a7adbf8f2fd6887261a1cfc3c0432f9d4068e"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"powerfmt",
|
"powerfmt",
|
||||||
"serde_core",
|
"serde",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
@@ -2136,9 +2136,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "git2"
|
name = "git2"
|
||||||
version = "0.20.4"
|
version = "0.20.2"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "7b88256088d75a56f8ecfa070513a775dd9107f6530ef14919dac831af9cfe2b"
|
checksum = "2deb07a133b1520dc1a5690e9bd08950108873d7ed5de38dcc74d3b5ebffa110"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"bitflags 2.9.1",
|
"bitflags 2.9.1",
|
||||||
"libc",
|
"libc",
|
||||||
@@ -3036,9 +3036,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "libgit2-sys"
|
name = "libgit2-sys"
|
||||||
version = "0.18.3+1.9.2"
|
version = "0.18.1+1.9.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "c9b3acc4b91781bb0b3386669d325163746af5f6e4f73e6d2d630e09a35f3487"
|
checksum = "e1dcb20f84ffcdd825c7a311ae347cce604a6f084a767dec4a4929829645290e"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"cc",
|
"cc",
|
||||||
"libc",
|
"libc",
|
||||||
@@ -3446,9 +3446,9 @@ checksum = "5e0826a989adedc2a244799e823aece04662b66609d96af8dff7ac6df9a8925d"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "num-conv"
|
name = "num-conv"
|
||||||
version = "0.2.0"
|
version = "0.1.0"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "cf97ec579c3c42f953ef76dbf8d55ac91fb219dde70e49aa4a6b7d74e9919050"
|
checksum = "51d515d32fb182ee37cda2ccdcb92950d6a3c2893aa280e540671c2cd0f3b1d9"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "num-traits"
|
name = "num-traits"
|
||||||
@@ -6341,9 +6341,9 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "time"
|
name = "time"
|
||||||
version = "0.3.47"
|
version = "0.3.41"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "743bd48c283afc0388f9b8827b976905fb217ad9e647fae3a379a9283c4def2c"
|
checksum = "8a7619e19bc266e0f9c5e6686659d394bc57973859340060a69221e57dbc0c40"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"deranged",
|
"deranged",
|
||||||
"itoa",
|
"itoa",
|
||||||
@@ -6351,22 +6351,22 @@ dependencies = [
|
|||||||
"num-conv",
|
"num-conv",
|
||||||
"num_threads",
|
"num_threads",
|
||||||
"powerfmt",
|
"powerfmt",
|
||||||
"serde_core",
|
"serde",
|
||||||
"time-core",
|
"time-core",
|
||||||
"time-macros",
|
"time-macros",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "time-core"
|
name = "time-core"
|
||||||
version = "0.1.8"
|
version = "0.1.4"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "7694e1cfe791f8d31026952abf09c69ca6f6fa4e1a1229e18988f06a04a12dca"
|
checksum = "c9e9a38711f559d9e3ce1cdb06dd7c5b8ea546bc90052da6d06bb76da74bb07c"
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "time-macros"
|
name = "time-macros"
|
||||||
version = "0.2.27"
|
version = "0.2.22"
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "2e70e4c5a0e0a8a4823ad65dfe1a6930e4f4d756dcd9dd7939022b5e8c501215"
|
checksum = "3526739392ec93fd8b359c8e98514cb3e8e021beb4e5f597b00a0221f8ed8a49"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"num-conv",
|
"num-conv",
|
||||||
"time-core",
|
"time-core",
|
||||||
@@ -7994,6 +7994,33 @@ dependencies = [
|
|||||||
"rustix 1.0.7",
|
"rustix 1.0.7",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "yaak-actions"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"thiserror 2.0.17",
|
||||||
|
"tokio",
|
||||||
|
"ts-rs",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "yaak-actions-builtin"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = [
|
||||||
|
"log",
|
||||||
|
"serde",
|
||||||
|
"serde_json",
|
||||||
|
"tokio",
|
||||||
|
"yaak-actions",
|
||||||
|
"yaak-crypto",
|
||||||
|
"yaak-http",
|
||||||
|
"yaak-models",
|
||||||
|
"yaak-plugins",
|
||||||
|
"yaak-templates",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "yaak-app"
|
name = "yaak-app"
|
||||||
version = "0.0.0"
|
version = "0.0.0"
|
||||||
@@ -8063,6 +8090,8 @@ dependencies = [
|
|||||||
"log",
|
"log",
|
||||||
"serde_json",
|
"serde_json",
|
||||||
"tokio",
|
"tokio",
|
||||||
|
"yaak-actions",
|
||||||
|
"yaak-actions-builtin",
|
||||||
"yaak-crypto",
|
"yaak-crypto",
|
||||||
"yaak-http",
|
"yaak-http",
|
||||||
"yaak-models",
|
"yaak-models",
|
||||||
|
|||||||
@@ -2,6 +2,8 @@
|
|||||||
resolver = "2"
|
resolver = "2"
|
||||||
members = [
|
members = [
|
||||||
# Shared crates (no Tauri dependency)
|
# Shared crates (no Tauri dependency)
|
||||||
|
"crates/yaak-actions",
|
||||||
|
"crates/yaak-actions-builtin",
|
||||||
"crates/yaak-core",
|
"crates/yaak-core",
|
||||||
"crates/yaak-common",
|
"crates/yaak-common",
|
||||||
"crates/yaak-crypto",
|
"crates/yaak-crypto",
|
||||||
@@ -45,6 +47,8 @@ tokio = "1.48.0"
|
|||||||
ts-rs = "11.1.0"
|
ts-rs = "11.1.0"
|
||||||
|
|
||||||
# Internal crates - shared
|
# Internal crates - shared
|
||||||
|
yaak-actions = { path = "crates/yaak-actions" }
|
||||||
|
yaak-actions-builtin = { path = "crates/yaak-actions-builtin" }
|
||||||
yaak-core = { path = "crates/yaak-core" }
|
yaak-core = { path = "crates/yaak-core" }
|
||||||
yaak-common = { path = "crates/yaak-common" }
|
yaak-common = { path = "crates/yaak-common" }
|
||||||
yaak-crypto = { path = "crates/yaak-crypto" }
|
yaak-crypto = { path = "crates/yaak-crypto" }
|
||||||
|
|||||||
@@ -22,7 +22,7 @@
|
|||||||
<!-- sponsors-premium --><a href="https://github.com/MVST-Solutions"><img src="https://github.com/MVST-Solutions.png" width="80px" alt="User avatar: MVST-Solutions" /></a> <a href="https://github.com/dharsanb"><img src="https://github.com/dharsanb.png" width="80px" alt="User avatar: dharsanb" /></a> <a href="https://github.com/railwayapp"><img src="https://github.com/railwayapp.png" width="80px" alt="User avatar: railwayapp" /></a> <a href="https://github.com/caseyamcl"><img src="https://github.com/caseyamcl.png" width="80px" alt="User avatar: caseyamcl" /></a> <a href="https://github.com/bytebase"><img src="https://github.com/bytebase.png" width="80px" alt="User avatar: bytebase" /></a> <a href="https://github.com/"><img src="https://raw.githubusercontent.com/JamesIves/github-sponsors-readme-action/dev/.github/assets/placeholder.png" width="80px" alt="User avatar: " /></a> <!-- sponsors-premium -->
|
<!-- sponsors-premium --><a href="https://github.com/MVST-Solutions"><img src="https://github.com/MVST-Solutions.png" width="80px" alt="User avatar: MVST-Solutions" /></a> <a href="https://github.com/dharsanb"><img src="https://github.com/dharsanb.png" width="80px" alt="User avatar: dharsanb" /></a> <a href="https://github.com/railwayapp"><img src="https://github.com/railwayapp.png" width="80px" alt="User avatar: railwayapp" /></a> <a href="https://github.com/caseyamcl"><img src="https://github.com/caseyamcl.png" width="80px" alt="User avatar: caseyamcl" /></a> <a href="https://github.com/bytebase"><img src="https://github.com/bytebase.png" width="80px" alt="User avatar: bytebase" /></a> <a href="https://github.com/"><img src="https://raw.githubusercontent.com/JamesIves/github-sponsors-readme-action/dev/.github/assets/placeholder.png" width="80px" alt="User avatar: " /></a> <!-- sponsors-premium -->
|
||||||
</p>
|
</p>
|
||||||
<p align="center">
|
<p align="center">
|
||||||
<!-- sponsors-base --><a href="https://github.com/seanwash"><img src="https://github.com/seanwash.png" width="50px" alt="User avatar: seanwash" /></a> <a href="https://github.com/jerath"><img src="https://github.com/jerath.png" width="50px" alt="User avatar: jerath" /></a> <a href="https://github.com/itsa-sh"><img src="https://github.com/itsa-sh.png" width="50px" alt="User avatar: itsa-sh" /></a> <a href="https://github.com/dmmulroy"><img src="https://github.com/dmmulroy.png" width="50px" alt="User avatar: dmmulroy" /></a> <a href="https://github.com/timcole"><img src="https://github.com/timcole.png" width="50px" alt="User avatar: timcole" /></a> <a href="https://github.com/VLZH"><img src="https://github.com/VLZH.png" width="50px" alt="User avatar: VLZH" /></a> <a href="https://github.com/terasaka2k"><img src="https://github.com/terasaka2k.png" width="50px" alt="User avatar: terasaka2k" /></a> <a href="https://github.com/andriyor"><img src="https://github.com/andriyor.png" width="50px" alt="User avatar: andriyor" /></a> <a href="https://github.com/majudhu"><img src="https://github.com/majudhu.png" width="50px" alt="User avatar: majudhu" /></a> <a href="https://github.com/axelrindle"><img src="https://github.com/axelrindle.png" width="50px" alt="User avatar: axelrindle" /></a> <a href="https://github.com/jirizverina"><img src="https://github.com/jirizverina.png" width="50px" alt="User avatar: jirizverina" /></a> <a href="https://github.com/chip-well"><img src="https://github.com/chip-well.png" width="50px" alt="User avatar: chip-well" /></a> <a href="https://github.com/GRAYAH"><img src="https://github.com/GRAYAH.png" width="50px" alt="User avatar: GRAYAH" /></a> <a href="https://github.com/flashblaze"><img src="https://github.com/flashblaze.png" width="50px" alt="User avatar: flashblaze" /></a> <!-- sponsors-base -->
|
<!-- sponsors-base --><a href="https://github.com/seanwash"><img src="https://github.com/seanwash.png" width="50px" alt="User avatar: seanwash" /></a> <a href="https://github.com/jerath"><img src="https://github.com/jerath.png" width="50px" alt="User avatar: jerath" /></a> <a href="https://github.com/itsa-sh"><img src="https://github.com/itsa-sh.png" width="50px" alt="User avatar: itsa-sh" /></a> <a href="https://github.com/dmmulroy"><img src="https://github.com/dmmulroy.png" width="50px" alt="User avatar: dmmulroy" /></a> <a href="https://github.com/timcole"><img src="https://github.com/timcole.png" width="50px" alt="User avatar: timcole" /></a> <a href="https://github.com/VLZH"><img src="https://github.com/VLZH.png" width="50px" alt="User avatar: VLZH" /></a> <a href="https://github.com/terasaka2k"><img src="https://github.com/terasaka2k.png" width="50px" alt="User avatar: terasaka2k" /></a> <a href="https://github.com/andriyor"><img src="https://github.com/andriyor.png" width="50px" alt="User avatar: andriyor" /></a> <a href="https://github.com/majudhu"><img src="https://github.com/majudhu.png" width="50px" alt="User avatar: majudhu" /></a> <a href="https://github.com/axelrindle"><img src="https://github.com/axelrindle.png" width="50px" alt="User avatar: axelrindle" /></a> <a href="https://github.com/jirizverina"><img src="https://github.com/jirizverina.png" width="50px" alt="User avatar: jirizverina" /></a> <a href="https://github.com/chip-well"><img src="https://github.com/chip-well.png" width="50px" alt="User avatar: chip-well" /></a> <a href="https://github.com/GRAYAH"><img src="https://github.com/GRAYAH.png" width="50px" alt="User avatar: GRAYAH" /></a> <!-- sponsors-base -->
|
||||||
</p>
|
</p>
|
||||||
|
|
||||||

|

|
||||||
|
|||||||
@@ -15,6 +15,8 @@ env_logger = "0.11"
|
|||||||
log = { workspace = true }
|
log = { workspace = true }
|
||||||
serde_json = { workspace = true }
|
serde_json = { workspace = true }
|
||||||
tokio = { workspace = true, features = ["rt-multi-thread", "macros"] }
|
tokio = { workspace = true, features = ["rt-multi-thread", "macros"] }
|
||||||
|
yaak-actions = { workspace = true }
|
||||||
|
yaak-actions-builtin = { workspace = true }
|
||||||
yaak-crypto = { workspace = true }
|
yaak-crypto = { workspace = true }
|
||||||
yaak-http = { workspace = true }
|
yaak-http = { workspace = true }
|
||||||
yaak-models = { workspace = true }
|
yaak-models = { workspace = true }
|
||||||
|
|||||||
+55
-174
@@ -1,21 +1,13 @@
|
|||||||
use clap::{Parser, Subcommand};
|
use clap::{Parser, Subcommand};
|
||||||
use log::info;
|
|
||||||
use serde_json::Value;
|
|
||||||
use std::collections::BTreeMap;
|
|
||||||
use std::path::PathBuf;
|
use std::path::PathBuf;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use tokio::sync::mpsc;
|
use tokio::sync::mpsc;
|
||||||
use yaak_crypto::manager::EncryptionManager;
|
|
||||||
use yaak_http::path_placeholders::apply_path_placeholders;
|
|
||||||
use yaak_http::sender::{HttpSender, ReqwestSender};
|
use yaak_http::sender::{HttpSender, ReqwestSender};
|
||||||
use yaak_http::types::{SendableHttpRequest, SendableHttpRequestOptions};
|
use yaak_http::types::{SendableHttpRequest, SendableHttpRequestOptions};
|
||||||
use yaak_models::models::{HttpRequest, HttpRequestHeader, HttpUrlParameter};
|
use yaak_models::models::HttpRequest;
|
||||||
use yaak_models::render::make_vars_hashmap;
|
|
||||||
use yaak_models::util::UpdateSource;
|
use yaak_models::util::UpdateSource;
|
||||||
use yaak_plugins::events::{PluginContext, RenderPurpose};
|
use yaak_plugins::events::PluginContext;
|
||||||
use yaak_plugins::manager::PluginManager;
|
use yaak_plugins::manager::PluginManager;
|
||||||
use yaak_plugins::template_callback::PluginTemplateCallback;
|
|
||||||
use yaak_templates::{RenderOptions, parse_and_render, render_json_value_raw};
|
|
||||||
|
|
||||||
#[derive(Parser)]
|
#[derive(Parser)]
|
||||||
#[command(name = "yaakcli")]
|
#[command(name = "yaakcli")]
|
||||||
@@ -72,86 +64,6 @@ enum Commands {
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Render an HTTP request with template variables and plugin functions
|
|
||||||
async fn render_http_request(
|
|
||||||
r: &HttpRequest,
|
|
||||||
environment_chain: Vec<yaak_models::models::Environment>,
|
|
||||||
cb: &PluginTemplateCallback,
|
|
||||||
opt: &RenderOptions,
|
|
||||||
) -> yaak_templates::error::Result<HttpRequest> {
|
|
||||||
let vars = &make_vars_hashmap(environment_chain);
|
|
||||||
|
|
||||||
let mut url_parameters = Vec::new();
|
|
||||||
for p in r.url_parameters.clone() {
|
|
||||||
if !p.enabled {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
url_parameters.push(HttpUrlParameter {
|
|
||||||
enabled: p.enabled,
|
|
||||||
name: parse_and_render(p.name.as_str(), vars, cb, opt).await?,
|
|
||||||
value: parse_and_render(p.value.as_str(), vars, cb, opt).await?,
|
|
||||||
id: p.id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut headers = Vec::new();
|
|
||||||
for p in r.headers.clone() {
|
|
||||||
if !p.enabled {
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
headers.push(HttpRequestHeader {
|
|
||||||
enabled: p.enabled,
|
|
||||||
name: parse_and_render(p.name.as_str(), vars, cb, opt).await?,
|
|
||||||
value: parse_and_render(p.value.as_str(), vars, cb, opt).await?,
|
|
||||||
id: p.id,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut body = BTreeMap::new();
|
|
||||||
for (k, v) in r.body.clone() {
|
|
||||||
body.insert(k, render_json_value_raw(v, vars, cb, opt).await?);
|
|
||||||
}
|
|
||||||
|
|
||||||
let authentication = {
|
|
||||||
let mut disabled = false;
|
|
||||||
let mut auth = BTreeMap::new();
|
|
||||||
match r.authentication.get("disabled") {
|
|
||||||
Some(Value::Bool(true)) => {
|
|
||||||
disabled = true;
|
|
||||||
}
|
|
||||||
Some(Value::String(tmpl)) => {
|
|
||||||
disabled = parse_and_render(tmpl.as_str(), vars, cb, opt)
|
|
||||||
.await
|
|
||||||
.unwrap_or_default()
|
|
||||||
.is_empty();
|
|
||||||
info!(
|
|
||||||
"Rendering authentication.disabled as a template: {disabled} from \"{tmpl}\""
|
|
||||||
);
|
|
||||||
}
|
|
||||||
_ => {}
|
|
||||||
}
|
|
||||||
if disabled {
|
|
||||||
auth.insert("disabled".to_string(), Value::Bool(true));
|
|
||||||
} else {
|
|
||||||
for (k, v) in r.authentication.clone() {
|
|
||||||
if k == "disabled" {
|
|
||||||
auth.insert(k, Value::Bool(false));
|
|
||||||
} else {
|
|
||||||
auth.insert(k, render_json_value_raw(v, vars, cb, opt).await?);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
auth
|
|
||||||
};
|
|
||||||
|
|
||||||
let url = parse_and_render(r.url.clone().as_str(), vars, cb, opt).await?;
|
|
||||||
|
|
||||||
// Apply path placeholders (e.g., /users/:id -> /users/123)
|
|
||||||
let (url, url_parameters) = apply_path_placeholders(&url, &url_parameters);
|
|
||||||
|
|
||||||
Ok(HttpRequest { url, url_parameters, headers, body, authentication, ..r.to_owned() })
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() {
|
async fn main() {
|
||||||
let cli = Cli::parse();
|
let cli = Cli::parse();
|
||||||
@@ -176,10 +88,6 @@ async fn main() {
|
|||||||
|
|
||||||
let db = query_manager.connect();
|
let db = query_manager.connect();
|
||||||
|
|
||||||
// Initialize encryption manager for secure() template function
|
|
||||||
// Use the same app_id as the Tauri app for keyring access
|
|
||||||
let encryption_manager = Arc::new(EncryptionManager::new(query_manager.clone(), app_id));
|
|
||||||
|
|
||||||
// Initialize plugin manager for template functions
|
// Initialize plugin manager for template functions
|
||||||
let vendored_plugin_dir = data_dir.join("vendored-plugins");
|
let vendored_plugin_dir = data_dir.join("vendored-plugins");
|
||||||
let installed_plugin_dir = data_dir.join("installed-plugins");
|
let installed_plugin_dir = data_dir.join("installed-plugins");
|
||||||
@@ -198,9 +106,9 @@ async fn main() {
|
|||||||
// Create plugin manager (plugins may not be available in CLI context)
|
// Create plugin manager (plugins may not be available in CLI context)
|
||||||
let plugin_manager = Arc::new(
|
let plugin_manager = Arc::new(
|
||||||
PluginManager::new(
|
PluginManager::new(
|
||||||
vendored_plugin_dir,
|
vendored_plugin_dir.clone(),
|
||||||
installed_plugin_dir,
|
installed_plugin_dir.clone(),
|
||||||
node_bin_path,
|
node_bin_path.clone(),
|
||||||
plugin_runtime_main,
|
plugin_runtime_main,
|
||||||
false,
|
false,
|
||||||
)
|
)
|
||||||
@@ -239,94 +147,67 @@ async fn main() {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
Commands::Send { request_id } => {
|
Commands::Send { request_id } => {
|
||||||
let request = db.get_http_request(&request_id).expect("Failed to get request");
|
use yaak_actions::{
|
||||||
|
ActionExecutor, ActionId, ActionParams, ActionResult, ActionTarget, CurrentContext,
|
||||||
|
};
|
||||||
|
use yaak_actions_builtin::{BuiltinActionDependencies, register_http_actions};
|
||||||
|
|
||||||
// Resolve environment chain for variable substitution
|
// Create dependencies
|
||||||
let environment_chain = db
|
let deps = BuiltinActionDependencies::new_standalone(
|
||||||
.resolve_environments(
|
&db_path,
|
||||||
&request.workspace_id,
|
&blob_path,
|
||||||
request.folder_id.as_deref(),
|
&app_id,
|
||||||
cli.environment.as_deref(),
|
vendored_plugin_dir.clone(),
|
||||||
)
|
installed_plugin_dir.clone(),
|
||||||
.unwrap_or_default();
|
node_bin_path.clone(),
|
||||||
|
|
||||||
// Create template callback with plugin support
|
|
||||||
let plugin_context = PluginContext::new(None, Some(request.workspace_id.clone()));
|
|
||||||
let template_callback = PluginTemplateCallback::new(
|
|
||||||
plugin_manager.clone(),
|
|
||||||
encryption_manager.clone(),
|
|
||||||
&plugin_context,
|
|
||||||
RenderPurpose::Send,
|
|
||||||
);
|
|
||||||
|
|
||||||
// Render templates in the request
|
|
||||||
let rendered_request = render_http_request(
|
|
||||||
&request,
|
|
||||||
environment_chain,
|
|
||||||
&template_callback,
|
|
||||||
&RenderOptions::throw(),
|
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.expect("Failed to render request templates");
|
.expect("Failed to initialize dependencies");
|
||||||
|
|
||||||
if cli.verbose {
|
// Create executor and register actions
|
||||||
println!("> {} {}", rendered_request.method, rendered_request.url);
|
let executor = ActionExecutor::new();
|
||||||
}
|
executor.register_builtin_groups().await.expect("Failed to register groups");
|
||||||
|
register_http_actions(&executor, &deps).await.expect("Failed to register HTTP actions");
|
||||||
|
|
||||||
// Convert to sendable request
|
// Prepare context
|
||||||
let sendable = SendableHttpRequest::from_http_request(
|
let context = CurrentContext {
|
||||||
&rendered_request,
|
target: Some(ActionTarget::HttpRequest { id: request_id.clone() }),
|
||||||
SendableHttpRequestOptions::default(),
|
environment_id: cli.environment.clone(),
|
||||||
)
|
workspace_id: None,
|
||||||
.await
|
has_window: false,
|
||||||
.expect("Failed to build request");
|
can_prompt: false,
|
||||||
|
|
||||||
// Create event channel for progress
|
|
||||||
let (event_tx, mut event_rx) = mpsc::channel(100);
|
|
||||||
|
|
||||||
// Spawn task to print events if verbose
|
|
||||||
let verbose = cli.verbose;
|
|
||||||
let verbose_handle = if verbose {
|
|
||||||
Some(tokio::spawn(async move {
|
|
||||||
while let Some(event) = event_rx.recv().await {
|
|
||||||
println!("{}", event);
|
|
||||||
}
|
|
||||||
}))
|
|
||||||
} else {
|
|
||||||
// Drain events silently
|
|
||||||
tokio::spawn(async move { while event_rx.recv().await.is_some() {} });
|
|
||||||
None
|
|
||||||
};
|
};
|
||||||
|
|
||||||
// Send the request
|
// Prepare params
|
||||||
let sender = ReqwestSender::new().expect("Failed to create HTTP client");
|
let params = ActionParams {
|
||||||
let response = sender.send(sendable, event_tx).await.expect("Failed to send request");
|
data: serde_json::json!({
|
||||||
|
"render": true,
|
||||||
|
"follow_redirects": false,
|
||||||
|
"timeout_ms": 30000,
|
||||||
|
}),
|
||||||
|
};
|
||||||
|
|
||||||
// Wait for event handler to finish
|
// Invoke action
|
||||||
if let Some(handle) = verbose_handle {
|
let action_id = ActionId::builtin("http", "send-request");
|
||||||
let _ = handle.await;
|
let result = executor.invoke(&action_id, context, params).await.expect("Action failed");
|
||||||
}
|
|
||||||
|
|
||||||
// Print response
|
// Handle result
|
||||||
if verbose {
|
match result {
|
||||||
println!();
|
ActionResult::Success { data, message } => {
|
||||||
}
|
if let Some(msg) = message {
|
||||||
println!(
|
println!("{}", msg);
|
||||||
"HTTP {} {}",
|
}
|
||||||
response.status,
|
if let Some(data) = data {
|
||||||
response.status_reason.as_deref().unwrap_or("")
|
println!("{}", serde_json::to_string_pretty(&data).unwrap());
|
||||||
);
|
}
|
||||||
|
}
|
||||||
if verbose {
|
ActionResult::RequiresInput { .. } => {
|
||||||
for (name, value) in &response.headers {
|
eprintln!("Action requires input (not supported in CLI)");
|
||||||
println!("{}: {}", name, value);
|
}
|
||||||
|
ActionResult::Cancelled => {
|
||||||
|
eprintln!("Action cancelled");
|
||||||
}
|
}
|
||||||
println!();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Print body
|
|
||||||
let (body, _stats) = response.text().await.expect("Failed to read response body");
|
|
||||||
println!("{}", body);
|
|
||||||
}
|
}
|
||||||
Commands::Get { url } => {
|
Commands::Get { url } => {
|
||||||
if cli.verbose {
|
if cli.verbose {
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ use crate::PluginContextExt;
|
|||||||
use crate::error::Result;
|
use crate::error::Result;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use tauri::{AppHandle, Manager, Runtime, State, WebviewWindow, command};
|
use tauri::{AppHandle, Manager, Runtime, State, WebviewWindow, command};
|
||||||
|
use tauri_plugin_dialog::{DialogExt, MessageDialogKind};
|
||||||
use yaak_crypto::manager::EncryptionManager;
|
use yaak_crypto::manager::EncryptionManager;
|
||||||
use yaak_models::models::HttpRequestHeader;
|
use yaak_models::models::HttpRequestHeader;
|
||||||
use yaak_models::queries::workspaces::default_headers;
|
use yaak_models::queries::workspaces::default_headers;
|
||||||
@@ -22,6 +23,20 @@ impl<'a, R: Runtime, M: Manager<R>> EncryptionManagerExt<'a, R> for M {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub(crate) async fn cmd_show_workspace_key<R: Runtime>(
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
workspace_id: &str,
|
||||||
|
) -> Result<()> {
|
||||||
|
let key = window.crypto().reveal_workspace_key(workspace_id)?;
|
||||||
|
window
|
||||||
|
.dialog()
|
||||||
|
.message(format!("Your workspace key is \n\n{}", key))
|
||||||
|
.kind(MessageDialogKind::Info)
|
||||||
|
.show(|_v| {});
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
pub(crate) async fn cmd_decrypt_template<R: Runtime>(
|
pub(crate) async fn cmd_decrypt_template<R: Runtime>(
|
||||||
window: WebviewWindow<R>,
|
window: WebviewWindow<R>,
|
||||||
|
|||||||
@@ -9,8 +9,8 @@ use yaak_git::{
|
|||||||
BranchDeleteResult, CloneResult, GitCommit, GitRemote, GitStatusSummary, PullResult,
|
BranchDeleteResult, CloneResult, GitCommit, GitRemote, GitStatusSummary, PullResult,
|
||||||
PushResult, git_add, git_add_credential, git_add_remote, git_checkout_branch, git_clone,
|
PushResult, git_add, git_add_credential, git_add_remote, git_checkout_branch, git_clone,
|
||||||
git_commit, git_create_branch, git_delete_branch, git_delete_remote_branch, git_fetch_all,
|
git_commit, git_create_branch, git_delete_branch, git_delete_remote_branch, git_fetch_all,
|
||||||
git_init, git_log, git_merge_branch, git_pull, git_pull_force_reset, git_pull_merge, git_push,
|
git_init, git_log, git_merge_branch, git_pull, git_push, git_remotes, git_rename_branch,
|
||||||
git_remotes, git_rename_branch, git_reset_changes, git_rm_remote, git_status, git_unstage,
|
git_rm_remote, git_status, git_unstage,
|
||||||
};
|
};
|
||||||
|
|
||||||
// NOTE: All of these commands are async to prevent blocking work from locking up the UI
|
// NOTE: All of these commands are async to prevent blocking work from locking up the UI
|
||||||
@@ -89,20 +89,6 @@ pub async fn cmd_git_pull(dir: &Path) -> Result<PullResult> {
|
|||||||
Ok(git_pull(dir).await?)
|
Ok(git_pull(dir).await?)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[command]
|
|
||||||
pub async fn cmd_git_pull_force_reset(
|
|
||||||
dir: &Path,
|
|
||||||
remote: &str,
|
|
||||||
branch: &str,
|
|
||||||
) -> Result<PullResult> {
|
|
||||||
Ok(git_pull_force_reset(dir, remote, branch).await?)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command]
|
|
||||||
pub async fn cmd_git_pull_merge(dir: &Path, remote: &str, branch: &str) -> Result<PullResult> {
|
|
||||||
Ok(git_pull_merge(dir, remote, branch).await?)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
pub async fn cmd_git_add(dir: &Path, rela_paths: Vec<PathBuf>) -> Result<()> {
|
pub async fn cmd_git_add(dir: &Path, rela_paths: Vec<PathBuf>) -> Result<()> {
|
||||||
for path in rela_paths {
|
for path in rela_paths {
|
||||||
@@ -119,11 +105,6 @@ pub async fn cmd_git_unstage(dir: &Path, rela_paths: Vec<PathBuf>) -> Result<()>
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
#[command]
|
|
||||||
pub async fn cmd_git_reset_changes(dir: &Path) -> Result<()> {
|
|
||||||
Ok(git_reset_changes(dir).await?)
|
|
||||||
}
|
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
pub async fn cmd_git_add_credential(
|
pub async fn cmd_git_add_credential(
|
||||||
remote_url: &str,
|
remote_url: &str,
|
||||||
|
|||||||
@@ -182,14 +182,7 @@ async fn send_http_request_inner<R: Runtime>(
|
|||||||
);
|
);
|
||||||
let env_chain =
|
let env_chain =
|
||||||
window.db().resolve_environments(&workspace.id, folder_id, environment_id.as_deref())?;
|
window.db().resolve_environments(&workspace.id, folder_id, environment_id.as_deref())?;
|
||||||
let mut cancel_rx = cancelled_rx.clone();
|
let request = render_http_request(&resolved, env_chain, &cb, &RenderOptions::throw()).await?;
|
||||||
let render_options = RenderOptions::throw();
|
|
||||||
let request = tokio::select! {
|
|
||||||
result = render_http_request(&resolved, env_chain, &cb, &render_options) => result?,
|
|
||||||
_ = cancel_rx.changed() => {
|
|
||||||
return Err(GenericError("Request canceled".to_string()));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
// Build the sendable request using the new SendableHttpRequest type
|
// Build the sendable request using the new SendableHttpRequest type
|
||||||
let options = SendableHttpRequestOptions {
|
let options = SendableHttpRequestOptions {
|
||||||
@@ -251,22 +244,16 @@ async fn send_http_request_inner<R: Runtime>(
|
|||||||
})
|
})
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
// Apply authentication to the request, racing against cancellation since
|
// Apply authentication to the request
|
||||||
// auth plugins (e.g. OAuth2) can block indefinitely waiting for user action.
|
apply_authentication(
|
||||||
let mut cancel_rx = cancelled_rx.clone();
|
&window,
|
||||||
tokio::select! {
|
&mut sendable_request,
|
||||||
result = apply_authentication(
|
&request,
|
||||||
&window,
|
auth_context_id,
|
||||||
&mut sendable_request,
|
&plugin_manager,
|
||||||
&request,
|
plugin_context,
|
||||||
auth_context_id,
|
)
|
||||||
&plugin_manager,
|
.await?;
|
||||||
plugin_context,
|
|
||||||
) => result?,
|
|
||||||
_ = cancel_rx.changed() => {
|
|
||||||
return Err(GenericError("Request canceled".to_string()));
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
let cookie_store = maybe_cookie_store.as_ref().map(|(cs, _)| cs.clone());
|
let cookie_store = maybe_cookie_store.as_ref().map(|(cs, _)| cs.clone());
|
||||||
let result = execute_transaction(
|
let result = execute_transaction(
|
||||||
|
|||||||
@@ -37,8 +37,8 @@ use yaak_grpc::{Code, ServiceDefinition, serialize_message};
|
|||||||
use yaak_mac_window::AppHandleMacWindowExt;
|
use yaak_mac_window::AppHandleMacWindowExt;
|
||||||
use yaak_models::models::{
|
use yaak_models::models::{
|
||||||
AnyModel, CookieJar, Environment, GrpcConnection, GrpcConnectionState, GrpcEvent,
|
AnyModel, CookieJar, Environment, GrpcConnection, GrpcConnectionState, GrpcEvent,
|
||||||
GrpcEventType, HttpRequest, HttpResponse, HttpResponseEvent, HttpResponseState, Plugin,
|
GrpcEventType, GrpcRequest, HttpRequest, HttpResponse, HttpResponseEvent, HttpResponseState,
|
||||||
Workspace, WorkspaceMeta,
|
Plugin, Workspace, WorkspaceMeta,
|
||||||
};
|
};
|
||||||
use yaak_models::util::{BatchUpsertResult, UpdateSource, get_workspace_export_resources};
|
use yaak_models::util::{BatchUpsertResult, UpdateSource, get_workspace_export_resources};
|
||||||
use yaak_plugins::events::{
|
use yaak_plugins::events::{
|
||||||
@@ -1271,6 +1271,35 @@ async fn cmd_save_response<R: Runtime>(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
async fn cmd_send_folder<R: Runtime>(
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
environment_id: Option<String>,
|
||||||
|
cookie_jar_id: Option<String>,
|
||||||
|
folder_id: &str,
|
||||||
|
) -> YaakResult<()> {
|
||||||
|
let requests = app_handle.db().list_http_requests_for_folder_recursive(folder_id)?;
|
||||||
|
for request in requests {
|
||||||
|
let app_handle = app_handle.clone();
|
||||||
|
let window = window.clone();
|
||||||
|
let environment_id = environment_id.clone();
|
||||||
|
let cookie_jar_id = cookie_jar_id.clone();
|
||||||
|
tokio::spawn(async move {
|
||||||
|
let _ = cmd_send_http_request(
|
||||||
|
app_handle,
|
||||||
|
window,
|
||||||
|
environment_id.as_deref(),
|
||||||
|
cookie_jar_id.as_deref(),
|
||||||
|
request,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
async fn cmd_send_http_request<R: Runtime>(
|
async fn cmd_send_http_request<R: Runtime>(
|
||||||
app_handle: AppHandle<R>,
|
app_handle: AppHandle<R>,
|
||||||
@@ -1367,6 +1396,27 @@ async fn cmd_install_plugin<R: Runtime>(
|
|||||||
Ok(plugin)
|
Ok(plugin)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tauri::command]
|
||||||
|
async fn cmd_create_grpc_request<R: Runtime>(
|
||||||
|
workspace_id: &str,
|
||||||
|
name: &str,
|
||||||
|
sort_priority: f64,
|
||||||
|
folder_id: Option<&str>,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
) -> YaakResult<GrpcRequest> {
|
||||||
|
Ok(app_handle.db().upsert_grpc_request(
|
||||||
|
&GrpcRequest {
|
||||||
|
workspace_id: workspace_id.to_string(),
|
||||||
|
name: name.to_string(),
|
||||||
|
folder_id: folder_id.map(|s| s.to_string()),
|
||||||
|
sort_priority,
|
||||||
|
..Default::default()
|
||||||
|
},
|
||||||
|
&UpdateSource::from_window_label(window.label()),
|
||||||
|
)?)
|
||||||
|
}
|
||||||
|
|
||||||
#[tauri::command]
|
#[tauri::command]
|
||||||
async fn cmd_reload_plugins<R: Runtime>(
|
async fn cmd_reload_plugins<R: Runtime>(
|
||||||
app_handle: AppHandle<R>,
|
app_handle: AppHandle<R>,
|
||||||
@@ -1629,6 +1679,7 @@ pub fn run() {
|
|||||||
cmd_call_folder_action,
|
cmd_call_folder_action,
|
||||||
cmd_call_grpc_request_action,
|
cmd_call_grpc_request_action,
|
||||||
cmd_check_for_updates,
|
cmd_check_for_updates,
|
||||||
|
cmd_create_grpc_request,
|
||||||
cmd_curl_to_request,
|
cmd_curl_to_request,
|
||||||
cmd_delete_all_grpc_connections,
|
cmd_delete_all_grpc_connections,
|
||||||
cmd_delete_all_http_responses,
|
cmd_delete_all_http_responses,
|
||||||
@@ -1662,6 +1713,7 @@ pub fn run() {
|
|||||||
cmd_save_response,
|
cmd_save_response,
|
||||||
cmd_send_ephemeral_request,
|
cmd_send_ephemeral_request,
|
||||||
cmd_send_http_request,
|
cmd_send_http_request,
|
||||||
|
cmd_send_folder,
|
||||||
cmd_template_function_config,
|
cmd_template_function_config,
|
||||||
cmd_template_function_summaries,
|
cmd_template_function_summaries,
|
||||||
cmd_template_tokens_to_string,
|
cmd_template_tokens_to_string,
|
||||||
@@ -1676,6 +1728,7 @@ pub fn run() {
|
|||||||
crate::commands::cmd_reveal_workspace_key,
|
crate::commands::cmd_reveal_workspace_key,
|
||||||
crate::commands::cmd_secure_template,
|
crate::commands::cmd_secure_template,
|
||||||
crate::commands::cmd_set_workspace_key,
|
crate::commands::cmd_set_workspace_key,
|
||||||
|
crate::commands::cmd_show_workspace_key,
|
||||||
//
|
//
|
||||||
// Models commands
|
// Models commands
|
||||||
models_ext::models_delete,
|
models_ext::models_delete,
|
||||||
@@ -1709,11 +1762,8 @@ pub fn run() {
|
|||||||
git_ext::cmd_git_fetch_all,
|
git_ext::cmd_git_fetch_all,
|
||||||
git_ext::cmd_git_push,
|
git_ext::cmd_git_push,
|
||||||
git_ext::cmd_git_pull,
|
git_ext::cmd_git_pull,
|
||||||
git_ext::cmd_git_pull_force_reset,
|
|
||||||
git_ext::cmd_git_pull_merge,
|
|
||||||
git_ext::cmd_git_add,
|
git_ext::cmd_git_add,
|
||||||
git_ext::cmd_git_unstage,
|
git_ext::cmd_git_unstage,
|
||||||
git_ext::cmd_git_reset_changes,
|
|
||||||
git_ext::cmd_git_add_credential,
|
git_ext::cmd_git_add_credential,
|
||||||
git_ext::cmd_git_remotes,
|
git_ext::cmd_git_remotes,
|
||||||
git_ext::cmd_git_add_remote,
|
git_ext::cmd_git_add_remote,
|
||||||
@@ -1727,7 +1777,14 @@ pub fn run() {
|
|||||||
plugins_ext::cmd_plugins_update_all,
|
plugins_ext::cmd_plugins_update_all,
|
||||||
//
|
//
|
||||||
// WebSocket commands
|
// WebSocket commands
|
||||||
|
ws_ext::cmd_ws_upsert_request,
|
||||||
|
ws_ext::cmd_ws_duplicate_request,
|
||||||
|
ws_ext::cmd_ws_delete_request,
|
||||||
|
ws_ext::cmd_ws_delete_connection,
|
||||||
ws_ext::cmd_ws_delete_connections,
|
ws_ext::cmd_ws_delete_connections,
|
||||||
|
ws_ext::cmd_ws_list_events,
|
||||||
|
ws_ext::cmd_ws_list_requests,
|
||||||
|
ws_ext::cmd_ws_list_connections,
|
||||||
ws_ext::cmd_ws_send,
|
ws_ext::cmd_ws_send,
|
||||||
ws_ext::cmd_ws_close,
|
ws_ext::cmd_ws_close,
|
||||||
ws_ext::cmd_ws_connect,
|
ws_ext::cmd_ws_connect,
|
||||||
|
|||||||
@@ -12,7 +12,7 @@ use chrono::Utc;
|
|||||||
use cookie::Cookie;
|
use cookie::Cookie;
|
||||||
use log::error;
|
use log::error;
|
||||||
use std::sync::Arc;
|
use std::sync::Arc;
|
||||||
use tauri::{AppHandle, Emitter, Listener, Manager, Runtime};
|
use tauri::{AppHandle, Emitter, Manager, Runtime};
|
||||||
use tauri_plugin_clipboard_manager::ClipboardExt;
|
use tauri_plugin_clipboard_manager::ClipboardExt;
|
||||||
use tauri_plugin_opener::OpenerExt;
|
use tauri_plugin_opener::OpenerExt;
|
||||||
use yaak_crypto::manager::EncryptionManager;
|
use yaak_crypto::manager::EncryptionManager;
|
||||||
@@ -59,55 +59,7 @@ pub(crate) async fn handle_plugin_event<R: Runtime>(
|
|||||||
}
|
}
|
||||||
InternalEventPayload::PromptFormRequest(_) => {
|
InternalEventPayload::PromptFormRequest(_) => {
|
||||||
let window = get_window_from_plugin_context(app_handle, &plugin_context)?;
|
let window = get_window_from_plugin_context(app_handle, &plugin_context)?;
|
||||||
if event.reply_id.is_some() {
|
Ok(call_frontend(&window, event).await)
|
||||||
// Follow-up update from plugin runtime with resolved inputs — forward to frontend
|
|
||||||
window.emit_to(window.label(), "plugin_event", event.clone())?;
|
|
||||||
Ok(None)
|
|
||||||
} else {
|
|
||||||
// Initial request — set up bidirectional communication
|
|
||||||
window.emit_to(window.label(), "plugin_event", event.clone()).unwrap();
|
|
||||||
|
|
||||||
let event_id = event.id.clone();
|
|
||||||
let plugin_handle = plugin_handle.clone();
|
|
||||||
let plugin_context = plugin_context.clone();
|
|
||||||
let window = window.clone();
|
|
||||||
|
|
||||||
// Spawn async task to handle bidirectional form communication
|
|
||||||
tauri::async_runtime::spawn(async move {
|
|
||||||
let (tx, mut rx) = tokio::sync::mpsc::channel::<InternalEvent>(128);
|
|
||||||
|
|
||||||
// Listen for replies from the frontend
|
|
||||||
let listener_id = window.listen(event_id, move |ev: tauri::Event| {
|
|
||||||
let resp: InternalEvent = serde_json::from_str(ev.payload()).unwrap();
|
|
||||||
let _ = tx.try_send(resp);
|
|
||||||
});
|
|
||||||
|
|
||||||
// Forward each reply to the plugin runtime
|
|
||||||
while let Some(resp) = rx.recv().await {
|
|
||||||
let is_done = matches!(
|
|
||||||
&resp.payload,
|
|
||||||
InternalEventPayload::PromptFormResponse(r) if r.done.unwrap_or(false)
|
|
||||||
);
|
|
||||||
|
|
||||||
let event_to_send = plugin_handle.build_event_to_send(
|
|
||||||
&plugin_context,
|
|
||||||
&resp.payload,
|
|
||||||
Some(resp.reply_id.unwrap_or_default()),
|
|
||||||
);
|
|
||||||
if let Err(e) = plugin_handle.send(&event_to_send).await {
|
|
||||||
log::warn!("Failed to forward form response to plugin: {:?}", e);
|
|
||||||
}
|
|
||||||
|
|
||||||
if is_done {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
window.unlisten(listener_id);
|
|
||||||
});
|
|
||||||
|
|
||||||
Ok(None)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
InternalEventPayload::FindHttpResponsesRequest(req) => {
|
InternalEventPayload::FindHttpResponsesRequest(req) => {
|
||||||
let http_responses = app_handle
|
let http_responses = app_handle
|
||||||
|
|||||||
@@ -162,16 +162,11 @@ pub(crate) fn create_window<R: Runtime>(
|
|||||||
"dev.reset_size" => webview_window
|
"dev.reset_size" => webview_window
|
||||||
.set_size(LogicalSize::new(DEFAULT_WINDOW_WIDTH, DEFAULT_WINDOW_HEIGHT))
|
.set_size(LogicalSize::new(DEFAULT_WINDOW_WIDTH, DEFAULT_WINDOW_HEIGHT))
|
||||||
.unwrap(),
|
.unwrap(),
|
||||||
"dev.reset_size_16x9" => {
|
"dev.reset_size_record" => {
|
||||||
let width = webview_window.outer_size().unwrap().width;
|
let width = webview_window.outer_size().unwrap().width;
|
||||||
let height = width * 9 / 16;
|
let height = width * 9 / 16;
|
||||||
webview_window.set_size(PhysicalSize::new(width, height)).unwrap()
|
webview_window.set_size(PhysicalSize::new(width, height)).unwrap()
|
||||||
}
|
}
|
||||||
"dev.reset_size_16x10" => {
|
|
||||||
let width = webview_window.outer_size().unwrap().width;
|
|
||||||
let height = width * 10 / 16;
|
|
||||||
webview_window.set_size(PhysicalSize::new(width, height)).unwrap()
|
|
||||||
}
|
|
||||||
"dev.refresh" => webview_window.eval("location.reload()").unwrap(),
|
"dev.refresh" => webview_window.eval("location.reload()").unwrap(),
|
||||||
"dev.generate_theme_css" => {
|
"dev.generate_theme_css" => {
|
||||||
w.emit("generate_theme_css", true).unwrap();
|
w.emit("generate_theme_css", true).unwrap();
|
||||||
|
|||||||
@@ -154,13 +154,8 @@ pub fn app_menu<R: Runtime>(app_handle: &AppHandle<R>) -> tauri::Result<Menu<R>>
|
|||||||
&MenuItemBuilder::with_id("dev.reset_size".to_string(), "Reset Size")
|
&MenuItemBuilder::with_id("dev.reset_size".to_string(), "Reset Size")
|
||||||
.build(app_handle)?,
|
.build(app_handle)?,
|
||||||
&MenuItemBuilder::with_id(
|
&MenuItemBuilder::with_id(
|
||||||
"dev.reset_size_16x9".to_string(),
|
"dev.reset_size_record".to_string(),
|
||||||
"Resize to 16x9",
|
"Reset Size 16x9",
|
||||||
)
|
|
||||||
.build(app_handle)?,
|
|
||||||
&MenuItemBuilder::with_id(
|
|
||||||
"dev.reset_size_16x10".to_string(),
|
|
||||||
"Resize to 16x10",
|
|
||||||
)
|
)
|
||||||
.build(app_handle)?,
|
.build(app_handle)?,
|
||||||
&MenuItemBuilder::with_id(
|
&MenuItemBuilder::with_id(
|
||||||
|
|||||||
@@ -28,6 +28,52 @@ use yaak_templates::{RenderErrorBehavior, RenderOptions};
|
|||||||
use yaak_tls::find_client_certificate;
|
use yaak_tls::find_client_certificate;
|
||||||
use yaak_ws::{WebsocketManager, render_websocket_request};
|
use yaak_ws::{WebsocketManager, render_websocket_request};
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_upsert_request<R: Runtime>(
|
||||||
|
request: WebsocketRequest,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
) -> Result<WebsocketRequest> {
|
||||||
|
Ok(app_handle
|
||||||
|
.db()
|
||||||
|
.upsert_websocket_request(&request, &UpdateSource::from_window_label(window.label()))?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_duplicate_request<R: Runtime>(
|
||||||
|
request_id: &str,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
) -> Result<WebsocketRequest> {
|
||||||
|
let db = app_handle.db();
|
||||||
|
let request = db.get_websocket_request(request_id)?;
|
||||||
|
Ok(db.duplicate_websocket_request(&request, &UpdateSource::from_window_label(window.label()))?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_delete_request<R: Runtime>(
|
||||||
|
request_id: &str,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
) -> Result<WebsocketRequest> {
|
||||||
|
Ok(app_handle.db().delete_websocket_request_by_id(
|
||||||
|
request_id,
|
||||||
|
&UpdateSource::from_window_label(window.label()),
|
||||||
|
)?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_delete_connection<R: Runtime>(
|
||||||
|
connection_id: &str,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
window: WebviewWindow<R>,
|
||||||
|
) -> Result<WebsocketConnection> {
|
||||||
|
Ok(app_handle.db().delete_websocket_connection_by_id(
|
||||||
|
connection_id,
|
||||||
|
&UpdateSource::from_window_label(window.label()),
|
||||||
|
)?)
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
pub async fn cmd_ws_delete_connections<R: Runtime>(
|
pub async fn cmd_ws_delete_connections<R: Runtime>(
|
||||||
request_id: &str,
|
request_id: &str,
|
||||||
@@ -40,6 +86,30 @@ pub async fn cmd_ws_delete_connections<R: Runtime>(
|
|||||||
)?)
|
)?)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_list_events<R: Runtime>(
|
||||||
|
connection_id: &str,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
) -> Result<Vec<WebsocketEvent>> {
|
||||||
|
Ok(app_handle.db().list_websocket_events(connection_id)?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_list_requests<R: Runtime>(
|
||||||
|
workspace_id: &str,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
) -> Result<Vec<WebsocketRequest>> {
|
||||||
|
Ok(app_handle.db().list_websocket_requests(workspace_id)?)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[command]
|
||||||
|
pub async fn cmd_ws_list_connections<R: Runtime>(
|
||||||
|
workspace_id: &str,
|
||||||
|
app_handle: AppHandle<R>,
|
||||||
|
) -> Result<Vec<WebsocketConnection>> {
|
||||||
|
Ok(app_handle.db().list_websocket_connections(workspace_id)?)
|
||||||
|
}
|
||||||
|
|
||||||
#[command]
|
#[command]
|
||||||
pub async fn cmd_ws_send<R: Runtime>(
|
pub async fn cmd_ws_send<R: Runtime>(
|
||||||
connection_id: &str,
|
connection_id: &str,
|
||||||
|
|||||||
@@ -0,0 +1,18 @@
|
|||||||
|
[package]
|
||||||
|
name = "yaak-actions-builtin"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2024"
|
||||||
|
authors = ["Gregory Schier"]
|
||||||
|
publish = false
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
yaak-actions = { workspace = true }
|
||||||
|
yaak-http = { workspace = true }
|
||||||
|
yaak-models = { workspace = true }
|
||||||
|
yaak-templates = { workspace = true }
|
||||||
|
yaak-plugins = { workspace = true }
|
||||||
|
yaak-crypto = { workspace = true }
|
||||||
|
serde = { workspace = true }
|
||||||
|
serde_json = { workspace = true }
|
||||||
|
tokio = { workspace = true, features = ["sync", "rt-multi-thread"] }
|
||||||
|
log = { workspace = true }
|
||||||
@@ -0,0 +1,88 @@
|
|||||||
|
//! Dependency injection for built-in actions.
|
||||||
|
|
||||||
|
use std::path::{Path, PathBuf};
|
||||||
|
use std::sync::Arc;
|
||||||
|
use yaak_crypto::manager::EncryptionManager;
|
||||||
|
use yaak_models::query_manager::QueryManager;
|
||||||
|
use yaak_plugins::events::PluginContext;
|
||||||
|
use yaak_plugins::manager::PluginManager;
|
||||||
|
|
||||||
|
/// Dependencies needed by built-in action implementations.
|
||||||
|
///
|
||||||
|
/// This struct bundles all the dependencies that action handlers need,
|
||||||
|
/// providing a clean way to initialize them in different contexts
|
||||||
|
/// (CLI, Tauri app, MCP server, etc.).
|
||||||
|
pub struct BuiltinActionDependencies {
|
||||||
|
pub query_manager: Arc<QueryManager>,
|
||||||
|
pub plugin_manager: Arc<PluginManager>,
|
||||||
|
pub encryption_manager: Arc<EncryptionManager>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl BuiltinActionDependencies {
|
||||||
|
/// Create dependencies for standalone usage (CLI, MCP server, etc.)
|
||||||
|
///
|
||||||
|
/// This initializes all the necessary managers following the same pattern
|
||||||
|
/// as the yaak-cli implementation.
|
||||||
|
pub async fn new_standalone(
|
||||||
|
db_path: &Path,
|
||||||
|
blob_path: &Path,
|
||||||
|
app_id: &str,
|
||||||
|
plugin_vendored_dir: PathBuf,
|
||||||
|
plugin_installed_dir: PathBuf,
|
||||||
|
node_path: PathBuf,
|
||||||
|
) -> Result<Self, Box<dyn std::error::Error>> {
|
||||||
|
// Initialize database
|
||||||
|
let (query_manager, _, _) = yaak_models::init_standalone(db_path, blob_path)?;
|
||||||
|
|
||||||
|
// Initialize encryption manager (takes QueryManager by value)
|
||||||
|
let encryption_manager = Arc::new(EncryptionManager::new(
|
||||||
|
query_manager.clone(),
|
||||||
|
app_id.to_string(),
|
||||||
|
));
|
||||||
|
|
||||||
|
let query_manager = Arc::new(query_manager);
|
||||||
|
|
||||||
|
// Find plugin runtime
|
||||||
|
let plugin_runtime_main = std::env::var("YAAK_PLUGIN_RUNTIME")
|
||||||
|
.map(PathBuf::from)
|
||||||
|
.unwrap_or_else(|_| {
|
||||||
|
// Development fallback
|
||||||
|
PathBuf::from(env!("CARGO_MANIFEST_DIR"))
|
||||||
|
.join("../../crates-tauri/yaak-app/vendored/plugin-runtime/index.cjs")
|
||||||
|
});
|
||||||
|
|
||||||
|
// Initialize plugin manager
|
||||||
|
let plugin_manager = Arc::new(
|
||||||
|
PluginManager::new(
|
||||||
|
plugin_vendored_dir,
|
||||||
|
plugin_installed_dir,
|
||||||
|
node_path,
|
||||||
|
plugin_runtime_main,
|
||||||
|
false, // not sandboxed in CLI
|
||||||
|
)
|
||||||
|
.await,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Initialize plugins from database
|
||||||
|
let db = query_manager.connect();
|
||||||
|
let plugins = db.list_plugins().unwrap_or_default();
|
||||||
|
if !plugins.is_empty() {
|
||||||
|
let errors = plugin_manager
|
||||||
|
.initialize_all_plugins(plugins, &PluginContext::new_empty())
|
||||||
|
.await;
|
||||||
|
for (plugin_dir, error_msg) in errors {
|
||||||
|
log::warn!(
|
||||||
|
"Failed to initialize plugin '{}': {}",
|
||||||
|
plugin_dir,
|
||||||
|
error_msg
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(Self {
|
||||||
|
query_manager,
|
||||||
|
plugin_manager,
|
||||||
|
encryption_manager,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,24 @@
|
|||||||
|
//! HTTP action implementations.
|
||||||
|
|
||||||
|
pub mod send;
|
||||||
|
|
||||||
|
use crate::BuiltinActionDependencies;
|
||||||
|
use yaak_actions::{ActionError, ActionExecutor, ActionSource};
|
||||||
|
|
||||||
|
/// Register all HTTP-related actions with the executor.
|
||||||
|
pub async fn register_http_actions(
|
||||||
|
executor: &ActionExecutor,
|
||||||
|
deps: &BuiltinActionDependencies,
|
||||||
|
) -> Result<(), ActionError> {
|
||||||
|
let handler = send::HttpSendActionHandler {
|
||||||
|
query_manager: deps.query_manager.clone(),
|
||||||
|
plugin_manager: deps.plugin_manager.clone(),
|
||||||
|
encryption_manager: deps.encryption_manager.clone(),
|
||||||
|
};
|
||||||
|
|
||||||
|
executor
|
||||||
|
.register(send::metadata(), ActionSource::Builtin, handler)
|
||||||
|
.await?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@@ -0,0 +1,293 @@
|
|||||||
|
//! HTTP send action implementation.
|
||||||
|
|
||||||
|
use std::collections::BTreeMap;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use serde_json::{json, Value};
|
||||||
|
use tokio::sync::mpsc;
|
||||||
|
use yaak_actions::{
|
||||||
|
ActionError, ActionGroupId, ActionHandler, ActionId, ActionMetadata,
|
||||||
|
ActionParams, ActionResult, ActionScope, CurrentContext,
|
||||||
|
RequiredContext,
|
||||||
|
};
|
||||||
|
use yaak_crypto::manager::EncryptionManager;
|
||||||
|
use yaak_http::path_placeholders::apply_path_placeholders;
|
||||||
|
use yaak_http::sender::{HttpSender, ReqwestSender};
|
||||||
|
use yaak_http::types::{SendableHttpRequest, SendableHttpRequestOptions};
|
||||||
|
use yaak_models::models::{HttpRequest, HttpRequestHeader, HttpUrlParameter};
|
||||||
|
use yaak_models::query_manager::QueryManager;
|
||||||
|
use yaak_models::render::make_vars_hashmap;
|
||||||
|
use yaak_plugins::events::{PluginContext, RenderPurpose};
|
||||||
|
use yaak_plugins::manager::PluginManager;
|
||||||
|
use yaak_plugins::template_callback::PluginTemplateCallback;
|
||||||
|
use yaak_templates::{parse_and_render, render_json_value_raw, RenderOptions};
|
||||||
|
|
||||||
|
/// Handler for HTTP send action.
|
||||||
|
pub struct HttpSendActionHandler {
|
||||||
|
pub query_manager: Arc<QueryManager>,
|
||||||
|
pub plugin_manager: Arc<PluginManager>,
|
||||||
|
pub encryption_manager: Arc<EncryptionManager>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Metadata for the HTTP send action.
|
||||||
|
pub fn metadata() -> ActionMetadata {
|
||||||
|
ActionMetadata {
|
||||||
|
id: ActionId::builtin("http", "send-request"),
|
||||||
|
label: "Send HTTP Request".to_string(),
|
||||||
|
description: Some("Execute an HTTP request and return the response".to_string()),
|
||||||
|
icon: Some("play".to_string()),
|
||||||
|
scope: ActionScope::HttpRequest,
|
||||||
|
keyboard_shortcut: None,
|
||||||
|
requires_selection: true,
|
||||||
|
enabled_condition: None,
|
||||||
|
group_id: Some(ActionGroupId::builtin("send")),
|
||||||
|
order: 10,
|
||||||
|
required_context: RequiredContext::requires_target(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionHandler for HttpSendActionHandler {
|
||||||
|
fn handle(
|
||||||
|
&self,
|
||||||
|
context: CurrentContext,
|
||||||
|
params: ActionParams,
|
||||||
|
) -> std::pin::Pin<
|
||||||
|
Box<dyn std::future::Future<Output = Result<ActionResult, ActionError>> + Send + 'static>,
|
||||||
|
> {
|
||||||
|
let query_manager = self.query_manager.clone();
|
||||||
|
let plugin_manager = self.plugin_manager.clone();
|
||||||
|
let encryption_manager = self.encryption_manager.clone();
|
||||||
|
|
||||||
|
Box::pin(async move {
|
||||||
|
// Extract request_id from context
|
||||||
|
let request_id = context
|
||||||
|
.target
|
||||||
|
.as_ref()
|
||||||
|
.ok_or_else(|| {
|
||||||
|
ActionError::ContextMissing {
|
||||||
|
missing_fields: vec!["target".to_string()],
|
||||||
|
}
|
||||||
|
})?
|
||||||
|
.id()
|
||||||
|
.ok_or_else(|| {
|
||||||
|
ActionError::ContextMissing {
|
||||||
|
missing_fields: vec!["target.id".to_string()],
|
||||||
|
}
|
||||||
|
})?
|
||||||
|
.to_string();
|
||||||
|
|
||||||
|
// Fetch request and environment from database (synchronous)
|
||||||
|
let (request, environment_chain) = {
|
||||||
|
let db = query_manager.connect();
|
||||||
|
|
||||||
|
// Fetch HTTP request from database
|
||||||
|
let request = db.get_http_request(&request_id).map_err(|e| {
|
||||||
|
ActionError::Internal(format!("Failed to fetch request {}: {}", request_id, e))
|
||||||
|
})?;
|
||||||
|
|
||||||
|
// Resolve environment chain for variable substitution
|
||||||
|
let environment_chain = if let Some(env_id) = &context.environment_id {
|
||||||
|
db.resolve_environments(
|
||||||
|
&request.workspace_id,
|
||||||
|
request.folder_id.as_deref(),
|
||||||
|
Some(env_id),
|
||||||
|
)
|
||||||
|
.unwrap_or_default()
|
||||||
|
} else {
|
||||||
|
db.resolve_environments(
|
||||||
|
&request.workspace_id,
|
||||||
|
request.folder_id.as_deref(),
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.unwrap_or_default()
|
||||||
|
};
|
||||||
|
|
||||||
|
(request, environment_chain)
|
||||||
|
}; // db is dropped here
|
||||||
|
|
||||||
|
// Create template callback with plugin support
|
||||||
|
let plugin_context = PluginContext::new(None, Some(request.workspace_id.clone()));
|
||||||
|
let template_callback = PluginTemplateCallback::new(
|
||||||
|
plugin_manager,
|
||||||
|
encryption_manager,
|
||||||
|
&plugin_context,
|
||||||
|
RenderPurpose::Send,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Render templates in the request
|
||||||
|
let rendered_request = render_http_request(
|
||||||
|
&request,
|
||||||
|
environment_chain,
|
||||||
|
&template_callback,
|
||||||
|
&RenderOptions::throw(),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.map_err(|e| ActionError::Internal(format!("Failed to render request: {}", e)))?;
|
||||||
|
|
||||||
|
// Build sendable request
|
||||||
|
let options = SendableHttpRequestOptions {
|
||||||
|
timeout: params
|
||||||
|
.data
|
||||||
|
.get("timeout_ms")
|
||||||
|
.and_then(|v| v.as_u64())
|
||||||
|
.map(|ms| std::time::Duration::from_millis(ms)),
|
||||||
|
follow_redirects: params
|
||||||
|
.data
|
||||||
|
.get("follow_redirects")
|
||||||
|
.and_then(|v| v.as_bool())
|
||||||
|
.unwrap_or(false),
|
||||||
|
};
|
||||||
|
|
||||||
|
let sendable = SendableHttpRequest::from_http_request(&rendered_request, options)
|
||||||
|
.await
|
||||||
|
.map_err(|e| ActionError::Internal(format!("Failed to build request: {}", e)))?;
|
||||||
|
|
||||||
|
// Create event channel
|
||||||
|
let (event_tx, mut event_rx) = mpsc::channel(100);
|
||||||
|
|
||||||
|
// Spawn task to drain events
|
||||||
|
let _event_handle = tokio::spawn(async move {
|
||||||
|
while event_rx.recv().await.is_some() {
|
||||||
|
// For now, just drain events
|
||||||
|
// In the future, we could log them or emit them to UI
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// Send the request
|
||||||
|
let sender = ReqwestSender::new()
|
||||||
|
.map_err(|e| ActionError::Internal(format!("Failed to create HTTP client: {}", e)))?;
|
||||||
|
let response = sender
|
||||||
|
.send(sendable, event_tx)
|
||||||
|
.await
|
||||||
|
.map_err(|e| ActionError::Internal(format!("Failed to send request: {}", e)))?;
|
||||||
|
|
||||||
|
// Consume response body
|
||||||
|
let status = response.status;
|
||||||
|
let status_reason = response.status_reason.clone();
|
||||||
|
let headers = response.headers.clone();
|
||||||
|
let url = response.url.clone();
|
||||||
|
|
||||||
|
let (body_text, stats) = response
|
||||||
|
.text()
|
||||||
|
.await
|
||||||
|
.map_err(|e| ActionError::Internal(format!("Failed to read response body: {}", e)))?;
|
||||||
|
|
||||||
|
// Return success result with response data
|
||||||
|
Ok(ActionResult::Success {
|
||||||
|
data: Some(json!({
|
||||||
|
"status": status,
|
||||||
|
"statusReason": status_reason,
|
||||||
|
"headers": headers,
|
||||||
|
"body": body_text,
|
||||||
|
"contentLength": stats.size_decompressed,
|
||||||
|
"url": url,
|
||||||
|
})),
|
||||||
|
message: Some(format!("HTTP {}", status)),
|
||||||
|
})
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Helper function to render templates in an HTTP request.
|
||||||
|
/// Copied from yaak-cli implementation.
|
||||||
|
async fn render_http_request(
|
||||||
|
r: &HttpRequest,
|
||||||
|
environment_chain: Vec<yaak_models::models::Environment>,
|
||||||
|
cb: &PluginTemplateCallback,
|
||||||
|
opt: &RenderOptions,
|
||||||
|
) -> Result<HttpRequest, String> {
|
||||||
|
let vars = &make_vars_hashmap(environment_chain);
|
||||||
|
|
||||||
|
let mut url_parameters = Vec::new();
|
||||||
|
for p in r.url_parameters.clone() {
|
||||||
|
if !p.enabled {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
url_parameters.push(HttpUrlParameter {
|
||||||
|
enabled: p.enabled,
|
||||||
|
name: parse_and_render(p.name.as_str(), vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?,
|
||||||
|
value: parse_and_render(p.value.as_str(), vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?,
|
||||||
|
id: p.id,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut headers = Vec::new();
|
||||||
|
for p in r.headers.clone() {
|
||||||
|
if !p.enabled {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
headers.push(HttpRequestHeader {
|
||||||
|
enabled: p.enabled,
|
||||||
|
name: parse_and_render(p.name.as_str(), vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?,
|
||||||
|
value: parse_and_render(p.value.as_str(), vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?,
|
||||||
|
id: p.id,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut body = BTreeMap::new();
|
||||||
|
for (k, v) in r.body.clone() {
|
||||||
|
body.insert(
|
||||||
|
k,
|
||||||
|
render_json_value_raw(v, vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
let authentication = {
|
||||||
|
let mut disabled = false;
|
||||||
|
let mut auth = BTreeMap::new();
|
||||||
|
match r.authentication.get("disabled") {
|
||||||
|
Some(Value::Bool(true)) => {
|
||||||
|
disabled = true;
|
||||||
|
}
|
||||||
|
Some(Value::String(tmpl)) => {
|
||||||
|
disabled = parse_and_render(tmpl.as_str(), vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.unwrap_or_default()
|
||||||
|
.is_empty();
|
||||||
|
}
|
||||||
|
_ => {}
|
||||||
|
}
|
||||||
|
if disabled {
|
||||||
|
auth.insert("disabled".to_string(), Value::Bool(true));
|
||||||
|
} else {
|
||||||
|
for (k, v) in r.authentication.clone() {
|
||||||
|
if k == "disabled" {
|
||||||
|
auth.insert(k, Value::Bool(false));
|
||||||
|
} else {
|
||||||
|
auth.insert(
|
||||||
|
k,
|
||||||
|
render_json_value_raw(v, vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
auth
|
||||||
|
};
|
||||||
|
|
||||||
|
let url = parse_and_render(r.url.clone().as_str(), vars, cb, opt)
|
||||||
|
.await
|
||||||
|
.map_err(|e| e.to_string())?;
|
||||||
|
|
||||||
|
// Apply path placeholders (e.g., /users/:id -> /users/123)
|
||||||
|
let (url, url_parameters) = apply_path_placeholders(&url, &url_parameters);
|
||||||
|
|
||||||
|
Ok(HttpRequest {
|
||||||
|
url,
|
||||||
|
url_parameters,
|
||||||
|
headers,
|
||||||
|
body,
|
||||||
|
authentication,
|
||||||
|
..r.to_owned()
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -0,0 +1,11 @@
|
|||||||
|
//! Built-in action implementations for Yaak.
|
||||||
|
//!
|
||||||
|
//! This crate provides concrete implementations of built-in actions using
|
||||||
|
//! the yaak-actions framework. It depends on domain-specific crates like
|
||||||
|
//! yaak-http, yaak-models, yaak-plugins, etc.
|
||||||
|
|
||||||
|
pub mod dependencies;
|
||||||
|
pub mod http;
|
||||||
|
|
||||||
|
pub use dependencies::BuiltinActionDependencies;
|
||||||
|
pub use http::register_http_actions;
|
||||||
@@ -0,0 +1,15 @@
|
|||||||
|
[package]
|
||||||
|
name = "yaak-actions"
|
||||||
|
version = "0.1.0"
|
||||||
|
edition = "2021"
|
||||||
|
description = "Centralized action system for Yaak"
|
||||||
|
|
||||||
|
[dependencies]
|
||||||
|
serde = { workspace = true, features = ["derive"] }
|
||||||
|
serde_json = { workspace = true }
|
||||||
|
thiserror = { workspace = true }
|
||||||
|
tokio = { workspace = true, features = ["sync"] }
|
||||||
|
ts-rs = { workspace = true }
|
||||||
|
|
||||||
|
[dev-dependencies]
|
||||||
|
tokio = { workspace = true, features = ["rt-multi-thread", "macros"] }
|
||||||
+14
@@ -0,0 +1,14 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Availability status for an action.
|
||||||
|
*/
|
||||||
|
export type ActionAvailability = { "status": "available" } | { "status": "available-with-prompt",
|
||||||
|
/**
|
||||||
|
* Fields that will require prompting.
|
||||||
|
*/
|
||||||
|
prompt_fields: Array<string>, } | { "status": "unavailable",
|
||||||
|
/**
|
||||||
|
* Fields that are missing.
|
||||||
|
*/
|
||||||
|
missing_fields: Array<string>, } | { "status": "not-found" };
|
||||||
+13
@@ -0,0 +1,13 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { ActionGroupId } from "./ActionGroupId";
|
||||||
|
import type { ActionId } from "./ActionId";
|
||||||
|
import type { ActionScope } from "./ActionScope";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Errors that can occur during action operations.
|
||||||
|
*/
|
||||||
|
export type ActionError = { "type": "not-found" } & ActionId | { "type": "disabled", action_id: ActionId, reason: string, } | { "type": "invalid-scope", expected: ActionScope, actual: ActionScope, } | { "type": "timeout" } & ActionId | { "type": "plugin-error" } & string | { "type": "validation-error" } & string | { "type": "permission-denied" } & string | { "type": "cancelled" } | { "type": "internal" } & string | { "type": "context-missing",
|
||||||
|
/**
|
||||||
|
* The context fields that are missing.
|
||||||
|
*/
|
||||||
|
missing_fields: Array<string>, } | { "type": "group-not-found" } & ActionGroupId | { "type": "group-already-exists" } & ActionGroupId;
|
||||||
+10
@@ -0,0 +1,10 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unique identifier for an action group.
|
||||||
|
*
|
||||||
|
* Format: `namespace:group-name`
|
||||||
|
* - Built-in: `yaak:export`
|
||||||
|
* - Plugin: `plugin.my-plugin:utilities`
|
||||||
|
*/
|
||||||
|
export type ActionGroupId = string;
|
||||||
+32
@@ -0,0 +1,32 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { ActionGroupId } from "./ActionGroupId";
|
||||||
|
import type { ActionScope } from "./ActionScope";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Metadata about an action group.
|
||||||
|
*/
|
||||||
|
export type ActionGroupMetadata = {
|
||||||
|
/**
|
||||||
|
* Unique identifier for this group.
|
||||||
|
*/
|
||||||
|
id: ActionGroupId,
|
||||||
|
/**
|
||||||
|
* Display name for the group.
|
||||||
|
*/
|
||||||
|
name: string,
|
||||||
|
/**
|
||||||
|
* Optional description of the group's purpose.
|
||||||
|
*/
|
||||||
|
description: string | null,
|
||||||
|
/**
|
||||||
|
* Icon to display for the group.
|
||||||
|
*/
|
||||||
|
icon: string | null,
|
||||||
|
/**
|
||||||
|
* Sort order for displaying groups (lower = earlier).
|
||||||
|
*/
|
||||||
|
order: number,
|
||||||
|
/**
|
||||||
|
* Optional scope restriction (if set, group only appears in this scope).
|
||||||
|
*/
|
||||||
|
scope: ActionScope | null, };
|
||||||
+18
@@ -0,0 +1,18 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Where an action group was registered from.
|
||||||
|
*/
|
||||||
|
export type ActionGroupSource = { "type": "builtin" } | { "type": "plugin",
|
||||||
|
/**
|
||||||
|
* Plugin reference ID.
|
||||||
|
*/
|
||||||
|
ref_id: string,
|
||||||
|
/**
|
||||||
|
* Plugin name.
|
||||||
|
*/
|
||||||
|
name: string, } | { "type": "dynamic",
|
||||||
|
/**
|
||||||
|
* Source identifier.
|
||||||
|
*/
|
||||||
|
source_id: string, };
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { ActionGroupMetadata } from "./ActionGroupMetadata";
|
||||||
|
import type { ActionMetadata } from "./ActionMetadata";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A group with its actions for UI rendering.
|
||||||
|
*/
|
||||||
|
export type ActionGroupWithActions = {
|
||||||
|
/**
|
||||||
|
* Group metadata.
|
||||||
|
*/
|
||||||
|
group: ActionGroupMetadata,
|
||||||
|
/**
|
||||||
|
* Actions in this group.
|
||||||
|
*/
|
||||||
|
actions: Array<ActionMetadata>, };
|
||||||
Generated
+10
@@ -0,0 +1,10 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Unique identifier for an action.
|
||||||
|
*
|
||||||
|
* Format: `namespace:category:name`
|
||||||
|
* - Built-in: `yaak:http-request:send`
|
||||||
|
* - Plugin: `plugin.copy-curl:http-request:copy`
|
||||||
|
*/
|
||||||
|
export type ActionId = string;
|
||||||
+54
@@ -0,0 +1,54 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { ActionGroupId } from "./ActionGroupId";
|
||||||
|
import type { ActionId } from "./ActionId";
|
||||||
|
import type { ActionScope } from "./ActionScope";
|
||||||
|
import type { RequiredContext } from "./RequiredContext";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Metadata about an action for discovery.
|
||||||
|
*/
|
||||||
|
export type ActionMetadata = {
|
||||||
|
/**
|
||||||
|
* Unique identifier for this action.
|
||||||
|
*/
|
||||||
|
id: ActionId,
|
||||||
|
/**
|
||||||
|
* Display label for the action.
|
||||||
|
*/
|
||||||
|
label: string,
|
||||||
|
/**
|
||||||
|
* Optional description of what the action does.
|
||||||
|
*/
|
||||||
|
description: string | null,
|
||||||
|
/**
|
||||||
|
* Icon name to display.
|
||||||
|
*/
|
||||||
|
icon: string | null,
|
||||||
|
/**
|
||||||
|
* The scope this action applies to.
|
||||||
|
*/
|
||||||
|
scope: ActionScope,
|
||||||
|
/**
|
||||||
|
* Keyboard shortcut (e.g., "Cmd+Enter").
|
||||||
|
*/
|
||||||
|
keyboardShortcut: string | null,
|
||||||
|
/**
|
||||||
|
* Whether the action requires a selection/target.
|
||||||
|
*/
|
||||||
|
requiresSelection: boolean,
|
||||||
|
/**
|
||||||
|
* Optional condition expression for when action is enabled.
|
||||||
|
*/
|
||||||
|
enabledCondition: string | null,
|
||||||
|
/**
|
||||||
|
* Optional group this action belongs to.
|
||||||
|
*/
|
||||||
|
groupId: ActionGroupId | null,
|
||||||
|
/**
|
||||||
|
* Sort order within a group (lower = earlier).
|
||||||
|
*/
|
||||||
|
order: number,
|
||||||
|
/**
|
||||||
|
* Context requirements for this action.
|
||||||
|
*/
|
||||||
|
requiredContext: RequiredContext, };
|
||||||
+10
@@ -0,0 +1,10 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Parameters passed to action handlers.
|
||||||
|
*/
|
||||||
|
export type ActionParams = {
|
||||||
|
/**
|
||||||
|
* Arbitrary JSON parameters.
|
||||||
|
*/
|
||||||
|
data: unknown, };
|
||||||
+23
@@ -0,0 +1,23 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { InputPrompt } from "./InputPrompt";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Result of action execution.
|
||||||
|
*/
|
||||||
|
export type ActionResult = { "type": "success",
|
||||||
|
/**
|
||||||
|
* Optional data to return.
|
||||||
|
*/
|
||||||
|
data: unknown,
|
||||||
|
/**
|
||||||
|
* Optional message to display.
|
||||||
|
*/
|
||||||
|
message: string | null, } | { "type": "requires-input",
|
||||||
|
/**
|
||||||
|
* Prompt to show user.
|
||||||
|
*/
|
||||||
|
prompt: InputPrompt,
|
||||||
|
/**
|
||||||
|
* Continuation token.
|
||||||
|
*/
|
||||||
|
continuation_id: string, } | { "type": "cancelled" };
|
||||||
+6
@@ -0,0 +1,6 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The scope in which an action can be invoked.
|
||||||
|
*/
|
||||||
|
export type ActionScope = "global" | "http-request" | "websocket-request" | "grpc-request" | "workspace" | "folder" | "environment" | "cookie-jar";
|
||||||
+18
@@ -0,0 +1,18 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Where an action was registered from.
|
||||||
|
*/
|
||||||
|
export type ActionSource = { "type": "builtin" } | { "type": "plugin",
|
||||||
|
/**
|
||||||
|
* Plugin reference ID.
|
||||||
|
*/
|
||||||
|
ref_id: string,
|
||||||
|
/**
|
||||||
|
* Plugin name.
|
||||||
|
*/
|
||||||
|
name: string, } | { "type": "dynamic",
|
||||||
|
/**
|
||||||
|
* Source identifier.
|
||||||
|
*/
|
||||||
|
source_id: string, };
|
||||||
+6
@@ -0,0 +1,6 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* The target entity for an action.
|
||||||
|
*/
|
||||||
|
export type ActionTarget = { "type": "none" } | { "type": "http-request", id: string, } | { "type": "websocket-request", id: string, } | { "type": "grpc-request", id: string, } | { "type": "workspace", id: string, } | { "type": "folder", id: string, } | { "type": "environment", id: string, } | { "type": "multiple", targets: Array<ActionTarget>, };
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* How strictly a context field is required.
|
||||||
|
*/
|
||||||
|
export type ContextRequirement = "not-required" | "optional" | "required" | "required-with-prompt";
|
||||||
+27
@@ -0,0 +1,27 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { ActionTarget } from "./ActionTarget";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Current context state from the application.
|
||||||
|
*/
|
||||||
|
export type CurrentContext = {
|
||||||
|
/**
|
||||||
|
* Current workspace ID (if any).
|
||||||
|
*/
|
||||||
|
workspaceId: string | null,
|
||||||
|
/**
|
||||||
|
* Current environment ID (if any).
|
||||||
|
*/
|
||||||
|
environmentId: string | null,
|
||||||
|
/**
|
||||||
|
* Currently selected target (if any).
|
||||||
|
*/
|
||||||
|
target: ActionTarget | null,
|
||||||
|
/**
|
||||||
|
* Whether a window context is available.
|
||||||
|
*/
|
||||||
|
hasWindow: boolean,
|
||||||
|
/**
|
||||||
|
* Whether the context provider can prompt for missing fields.
|
||||||
|
*/
|
||||||
|
canPrompt: boolean, };
|
||||||
+7
@@ -0,0 +1,7 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { SelectOption } from "./SelectOption";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A prompt for user input.
|
||||||
|
*/
|
||||||
|
export type InputPrompt = { "type": "text", label: string, placeholder: string | null, default_value: string | null, } | { "type": "select", label: string, options: Array<SelectOption>, } | { "type": "confirm", label: string, };
|
||||||
+23
@@ -0,0 +1,23 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
import type { ContextRequirement } from "./ContextRequirement";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Specifies what context fields an action requires.
|
||||||
|
*/
|
||||||
|
export type RequiredContext = {
|
||||||
|
/**
|
||||||
|
* Action requires a workspace to be active.
|
||||||
|
*/
|
||||||
|
workspace: ContextRequirement,
|
||||||
|
/**
|
||||||
|
* Action requires an environment to be selected.
|
||||||
|
*/
|
||||||
|
environment: ContextRequirement,
|
||||||
|
/**
|
||||||
|
* Action requires a specific target entity (request, folder, etc.).
|
||||||
|
*/
|
||||||
|
target: ContextRequirement,
|
||||||
|
/**
|
||||||
|
* Action requires a window context (for UI operations).
|
||||||
|
*/
|
||||||
|
window: ContextRequirement, };
|
||||||
+6
@@ -0,0 +1,6 @@
|
|||||||
|
// This file was generated by [ts-rs](https://github.com/Aleph-Alpha/ts-rs). Do not edit this file manually.
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An option in a select prompt.
|
||||||
|
*/
|
||||||
|
export type SelectOption = { label: string, value: string, };
|
||||||
@@ -0,0 +1,331 @@
|
|||||||
|
//! Action context types and context-aware filtering.
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use ts_rs::TS;
|
||||||
|
|
||||||
|
use crate::ActionScope;
|
||||||
|
|
||||||
|
/// Specifies what context fields an action requires.
|
||||||
|
#[derive(Clone, Debug, Default, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct RequiredContext {
|
||||||
|
/// Action requires a workspace to be active.
|
||||||
|
#[serde(default)]
|
||||||
|
pub workspace: ContextRequirement,
|
||||||
|
|
||||||
|
/// Action requires an environment to be selected.
|
||||||
|
#[serde(default)]
|
||||||
|
pub environment: ContextRequirement,
|
||||||
|
|
||||||
|
/// Action requires a specific target entity (request, folder, etc.).
|
||||||
|
#[serde(default)]
|
||||||
|
pub target: ContextRequirement,
|
||||||
|
|
||||||
|
/// Action requires a window context (for UI operations).
|
||||||
|
#[serde(default)]
|
||||||
|
pub window: ContextRequirement,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl RequiredContext {
|
||||||
|
/// Action requires a target entity.
|
||||||
|
pub fn requires_target() -> Self {
|
||||||
|
Self {
|
||||||
|
target: ContextRequirement::Required,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Action requires workspace and target.
|
||||||
|
pub fn requires_workspace_and_target() -> Self {
|
||||||
|
Self {
|
||||||
|
workspace: ContextRequirement::Required,
|
||||||
|
target: ContextRequirement::Required,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Action works globally, no specific context needed.
|
||||||
|
pub fn global() -> Self {
|
||||||
|
Self::default()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Action requires target with prompt if missing.
|
||||||
|
pub fn requires_target_with_prompt() -> Self {
|
||||||
|
Self {
|
||||||
|
target: ContextRequirement::RequiredWithPrompt,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Action requires environment with prompt if missing.
|
||||||
|
pub fn requires_environment_with_prompt() -> Self {
|
||||||
|
Self {
|
||||||
|
environment: ContextRequirement::RequiredWithPrompt,
|
||||||
|
..Default::default()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// How strictly a context field is required.
|
||||||
|
#[derive(Clone, Debug, Default, PartialEq, Eq, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "kebab-case")]
|
||||||
|
pub enum ContextRequirement {
|
||||||
|
/// Field is not needed.
|
||||||
|
#[default]
|
||||||
|
NotRequired,
|
||||||
|
|
||||||
|
/// Field is optional but will be used if available.
|
||||||
|
Optional,
|
||||||
|
|
||||||
|
/// Field must be present; action will fail without it.
|
||||||
|
Required,
|
||||||
|
|
||||||
|
/// Field must be present; prompt user to select if missing.
|
||||||
|
RequiredWithPrompt,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Current context state from the application.
|
||||||
|
#[derive(Clone, Debug, Default, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct CurrentContext {
|
||||||
|
/// Current workspace ID (if any).
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub workspace_id: Option<String>,
|
||||||
|
|
||||||
|
/// Current environment ID (if any).
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub environment_id: Option<String>,
|
||||||
|
|
||||||
|
/// Currently selected target (if any).
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub target: Option<ActionTarget>,
|
||||||
|
|
||||||
|
/// Whether a window context is available.
|
||||||
|
#[serde(default)]
|
||||||
|
pub has_window: bool,
|
||||||
|
|
||||||
|
/// Whether the context provider can prompt for missing fields.
|
||||||
|
#[serde(default)]
|
||||||
|
pub can_prompt: bool,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// The target entity for an action.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "type", rename_all = "kebab-case")]
|
||||||
|
pub enum ActionTarget {
|
||||||
|
/// No target.
|
||||||
|
None,
|
||||||
|
/// HTTP request target.
|
||||||
|
HttpRequest { id: String },
|
||||||
|
/// WebSocket request target.
|
||||||
|
WebsocketRequest { id: String },
|
||||||
|
/// gRPC request target.
|
||||||
|
GrpcRequest { id: String },
|
||||||
|
/// Workspace target.
|
||||||
|
Workspace { id: String },
|
||||||
|
/// Folder target.
|
||||||
|
Folder { id: String },
|
||||||
|
/// Environment target.
|
||||||
|
Environment { id: String },
|
||||||
|
/// Multiple targets.
|
||||||
|
Multiple { targets: Vec<ActionTarget> },
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionTarget {
|
||||||
|
/// Get the scope this target corresponds to.
|
||||||
|
pub fn scope(&self) -> Option<ActionScope> {
|
||||||
|
match self {
|
||||||
|
Self::None => None,
|
||||||
|
Self::HttpRequest { .. } => Some(ActionScope::HttpRequest),
|
||||||
|
Self::WebsocketRequest { .. } => Some(ActionScope::WebsocketRequest),
|
||||||
|
Self::GrpcRequest { .. } => Some(ActionScope::GrpcRequest),
|
||||||
|
Self::Workspace { .. } => Some(ActionScope::Workspace),
|
||||||
|
Self::Folder { .. } => Some(ActionScope::Folder),
|
||||||
|
Self::Environment { .. } => Some(ActionScope::Environment),
|
||||||
|
Self::Multiple { .. } => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the ID of the target (if single target).
|
||||||
|
pub fn id(&self) -> Option<&str> {
|
||||||
|
match self {
|
||||||
|
Self::HttpRequest { id }
|
||||||
|
| Self::WebsocketRequest { id }
|
||||||
|
| Self::GrpcRequest { id }
|
||||||
|
| Self::Workspace { id }
|
||||||
|
| Self::Folder { id }
|
||||||
|
| Self::Environment { id } => Some(id),
|
||||||
|
Self::None | Self::Multiple { .. } => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Availability status for an action.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "status", rename_all = "kebab-case")]
|
||||||
|
pub enum ActionAvailability {
|
||||||
|
/// Action is ready to execute.
|
||||||
|
Available,
|
||||||
|
|
||||||
|
/// Action can execute but will prompt for missing context.
|
||||||
|
AvailableWithPrompt {
|
||||||
|
/// Fields that will require prompting.
|
||||||
|
prompt_fields: Vec<String>,
|
||||||
|
},
|
||||||
|
|
||||||
|
/// Action cannot execute due to missing context.
|
||||||
|
Unavailable {
|
||||||
|
/// Fields that are missing.
|
||||||
|
missing_fields: Vec<String>,
|
||||||
|
},
|
||||||
|
|
||||||
|
/// Action not found in registry.
|
||||||
|
NotFound,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionAvailability {
|
||||||
|
/// Check if the action is available (possibly with prompts).
|
||||||
|
pub fn is_available(&self) -> bool {
|
||||||
|
matches!(self, Self::Available | Self::AvailableWithPrompt { .. })
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Check if the action is immediately available without prompts.
|
||||||
|
pub fn is_immediately_available(&self) -> bool {
|
||||||
|
matches!(self, Self::Available)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Check if required context is satisfied by current context.
|
||||||
|
pub fn check_context_availability(
|
||||||
|
required: &RequiredContext,
|
||||||
|
current: &CurrentContext,
|
||||||
|
) -> ActionAvailability {
|
||||||
|
let mut missing_fields = Vec::new();
|
||||||
|
let mut prompt_fields = Vec::new();
|
||||||
|
|
||||||
|
// Check workspace
|
||||||
|
check_field(
|
||||||
|
"workspace",
|
||||||
|
current.workspace_id.is_some(),
|
||||||
|
&required.workspace,
|
||||||
|
current.can_prompt,
|
||||||
|
&mut missing_fields,
|
||||||
|
&mut prompt_fields,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Check environment
|
||||||
|
check_field(
|
||||||
|
"environment",
|
||||||
|
current.environment_id.is_some(),
|
||||||
|
&required.environment,
|
||||||
|
current.can_prompt,
|
||||||
|
&mut missing_fields,
|
||||||
|
&mut prompt_fields,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Check target
|
||||||
|
check_field(
|
||||||
|
"target",
|
||||||
|
current.target.is_some(),
|
||||||
|
&required.target,
|
||||||
|
current.can_prompt,
|
||||||
|
&mut missing_fields,
|
||||||
|
&mut prompt_fields,
|
||||||
|
);
|
||||||
|
|
||||||
|
// Check window
|
||||||
|
check_field(
|
||||||
|
"window",
|
||||||
|
current.has_window,
|
||||||
|
&required.window,
|
||||||
|
false, // Can't prompt for window
|
||||||
|
&mut missing_fields,
|
||||||
|
&mut prompt_fields,
|
||||||
|
);
|
||||||
|
|
||||||
|
if !missing_fields.is_empty() {
|
||||||
|
ActionAvailability::Unavailable { missing_fields }
|
||||||
|
} else if !prompt_fields.is_empty() {
|
||||||
|
ActionAvailability::AvailableWithPrompt { prompt_fields }
|
||||||
|
} else {
|
||||||
|
ActionAvailability::Available
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn check_field(
|
||||||
|
name: &str,
|
||||||
|
has_value: bool,
|
||||||
|
requirement: &ContextRequirement,
|
||||||
|
can_prompt: bool,
|
||||||
|
missing: &mut Vec<String>,
|
||||||
|
promptable: &mut Vec<String>,
|
||||||
|
) {
|
||||||
|
match requirement {
|
||||||
|
ContextRequirement::NotRequired | ContextRequirement::Optional => {}
|
||||||
|
ContextRequirement::Required => {
|
||||||
|
if !has_value {
|
||||||
|
missing.push(name.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ContextRequirement::RequiredWithPrompt => {
|
||||||
|
if !has_value {
|
||||||
|
if can_prompt {
|
||||||
|
promptable.push(name.to_string());
|
||||||
|
} else {
|
||||||
|
missing.push(name.to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_context_available() {
|
||||||
|
let required = RequiredContext::requires_target();
|
||||||
|
let current = CurrentContext {
|
||||||
|
target: Some(ActionTarget::HttpRequest {
|
||||||
|
id: "123".to_string(),
|
||||||
|
}),
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let availability = check_context_availability(&required, ¤t);
|
||||||
|
assert!(matches!(availability, ActionAvailability::Available));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_context_missing() {
|
||||||
|
let required = RequiredContext::requires_target();
|
||||||
|
let current = CurrentContext::default();
|
||||||
|
|
||||||
|
let availability = check_context_availability(&required, ¤t);
|
||||||
|
assert!(matches!(
|
||||||
|
availability,
|
||||||
|
ActionAvailability::Unavailable { missing_fields } if missing_fields == vec!["target"]
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_context_promptable() {
|
||||||
|
let required = RequiredContext::requires_target_with_prompt();
|
||||||
|
let current = CurrentContext {
|
||||||
|
can_prompt: true,
|
||||||
|
..Default::default()
|
||||||
|
};
|
||||||
|
|
||||||
|
let availability = check_context_availability(&required, ¤t);
|
||||||
|
assert!(matches!(
|
||||||
|
availability,
|
||||||
|
ActionAvailability::AvailableWithPrompt { prompt_fields } if prompt_fields == vec!["target"]
|
||||||
|
));
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,131 @@
|
|||||||
|
//! Error types for the action system.
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use thiserror::Error;
|
||||||
|
use ts_rs::TS;
|
||||||
|
|
||||||
|
use crate::{ActionGroupId, ActionId};
|
||||||
|
|
||||||
|
/// Errors that can occur during action operations.
|
||||||
|
#[derive(Debug, Error, Clone, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "type", rename_all = "kebab-case")]
|
||||||
|
pub enum ActionError {
|
||||||
|
/// Action not found in registry.
|
||||||
|
#[error("Action not found: {0}")]
|
||||||
|
NotFound(ActionId),
|
||||||
|
|
||||||
|
/// Action is disabled in current context.
|
||||||
|
#[error("Action is disabled: {action_id} - {reason}")]
|
||||||
|
Disabled { action_id: ActionId, reason: String },
|
||||||
|
|
||||||
|
/// Invalid scope for the action.
|
||||||
|
#[error("Invalid scope: expected {expected:?}, got {actual:?}")]
|
||||||
|
InvalidScope {
|
||||||
|
expected: crate::ActionScope,
|
||||||
|
actual: crate::ActionScope,
|
||||||
|
},
|
||||||
|
|
||||||
|
/// Action execution timed out.
|
||||||
|
#[error("Action timed out: {0}")]
|
||||||
|
Timeout(ActionId),
|
||||||
|
|
||||||
|
/// Error from plugin execution.
|
||||||
|
#[error("Plugin error: {0}")]
|
||||||
|
PluginError(String),
|
||||||
|
|
||||||
|
/// Validation error in action parameters.
|
||||||
|
#[error("Validation error: {0}")]
|
||||||
|
ValidationError(String),
|
||||||
|
|
||||||
|
/// Permission denied for action.
|
||||||
|
#[error("Permission denied: {0}")]
|
||||||
|
PermissionDenied(String),
|
||||||
|
|
||||||
|
/// Action was cancelled by user.
|
||||||
|
#[error("Action cancelled by user")]
|
||||||
|
Cancelled,
|
||||||
|
|
||||||
|
/// Internal error.
|
||||||
|
#[error("Internal error: {0}")]
|
||||||
|
Internal(String),
|
||||||
|
|
||||||
|
/// Required context is missing.
|
||||||
|
#[error("Required context missing: {missing_fields:?}")]
|
||||||
|
ContextMissing {
|
||||||
|
/// The context fields that are missing.
|
||||||
|
missing_fields: Vec<String>,
|
||||||
|
},
|
||||||
|
|
||||||
|
/// Action group not found.
|
||||||
|
#[error("Group not found: {0}")]
|
||||||
|
GroupNotFound(ActionGroupId),
|
||||||
|
|
||||||
|
/// Action group already exists.
|
||||||
|
#[error("Group already exists: {0}")]
|
||||||
|
GroupAlreadyExists(ActionGroupId),
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionError {
|
||||||
|
/// Get a user-friendly error message.
|
||||||
|
pub fn user_message(&self) -> String {
|
||||||
|
match self {
|
||||||
|
Self::NotFound(id) => format!("Action '{}' is not available", id),
|
||||||
|
Self::Disabled { reason, .. } => reason.clone(),
|
||||||
|
Self::InvalidScope { expected, actual } => {
|
||||||
|
format!("Action requires {:?} scope, but got {:?}", expected, actual)
|
||||||
|
}
|
||||||
|
Self::Timeout(_) => "The operation took too long and was cancelled".into(),
|
||||||
|
Self::PluginError(msg) => format!("Plugin error: {}", msg),
|
||||||
|
Self::ValidationError(msg) => format!("Invalid input: {}", msg),
|
||||||
|
Self::PermissionDenied(resource) => format!("Permission denied for {}", resource),
|
||||||
|
Self::Cancelled => "Operation was cancelled".into(),
|
||||||
|
Self::Internal(_) => "An unexpected error occurred".into(),
|
||||||
|
Self::ContextMissing { missing_fields } => {
|
||||||
|
format!("Missing required context: {}", missing_fields.join(", "))
|
||||||
|
}
|
||||||
|
Self::GroupNotFound(id) => format!("Action group '{}' not found", id),
|
||||||
|
Self::GroupAlreadyExists(id) => format!("Action group '{}' already exists", id),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Whether this error should be reported to telemetry.
|
||||||
|
pub fn is_reportable(&self) -> bool {
|
||||||
|
matches!(self, Self::Internal(_) | Self::PluginError(_))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Whether this error can potentially be resolved by user interaction.
|
||||||
|
pub fn is_promptable(&self) -> bool {
|
||||||
|
matches!(self, Self::ContextMissing { .. })
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Whether this is a user-initiated cancellation.
|
||||||
|
pub fn is_cancelled(&self) -> bool {
|
||||||
|
matches!(self, Self::Cancelled)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_error_messages() {
|
||||||
|
let err = ActionError::ContextMissing {
|
||||||
|
missing_fields: vec!["workspace".into()],
|
||||||
|
};
|
||||||
|
assert_eq!(err.user_message(), "Missing required context: workspace");
|
||||||
|
assert!(err.is_promptable());
|
||||||
|
assert!(!err.is_cancelled());
|
||||||
|
|
||||||
|
let cancelled = ActionError::Cancelled;
|
||||||
|
assert!(cancelled.is_cancelled());
|
||||||
|
assert!(!cancelled.is_promptable());
|
||||||
|
|
||||||
|
let not_found = ActionError::NotFound(ActionId::builtin("test", "action"));
|
||||||
|
assert_eq!(
|
||||||
|
not_found.user_message(),
|
||||||
|
"Action 'yaak:test:action' is not available"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,606 @@
|
|||||||
|
//! Action executor - central hub for action registration and invocation.
|
||||||
|
|
||||||
|
use std::collections::HashMap;
|
||||||
|
use std::sync::Arc;
|
||||||
|
use tokio::sync::RwLock;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
check_context_availability, ActionAvailability, ActionError, ActionGroupId,
|
||||||
|
ActionGroupMetadata, ActionGroupSource, ActionGroupWithActions, ActionHandler, ActionId,
|
||||||
|
ActionMetadata, ActionParams, ActionResult, ActionScope, ActionSource, CurrentContext,
|
||||||
|
RegisteredActionGroup,
|
||||||
|
};
|
||||||
|
|
||||||
|
/// Options for listing actions.
|
||||||
|
#[derive(Clone, Debug, Default)]
|
||||||
|
pub struct ListActionsOptions {
|
||||||
|
/// Filter by scope.
|
||||||
|
pub scope: Option<ActionScope>,
|
||||||
|
/// Filter by group.
|
||||||
|
pub group_id: Option<ActionGroupId>,
|
||||||
|
/// Search term for label/description.
|
||||||
|
pub search: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A registered action with its handler.
|
||||||
|
struct RegisteredAction {
|
||||||
|
/// Action metadata.
|
||||||
|
metadata: ActionMetadata,
|
||||||
|
/// Where the action was registered from.
|
||||||
|
source: ActionSource,
|
||||||
|
/// The handler for this action.
|
||||||
|
handler: Arc<dyn ActionHandler>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Central hub for action registration and invocation.
|
||||||
|
///
|
||||||
|
/// The executor owns all action metadata and handlers, ensuring every
|
||||||
|
/// registered action has a handler by construction.
|
||||||
|
pub struct ActionExecutor {
|
||||||
|
/// All registered actions indexed by ID.
|
||||||
|
actions: RwLock<HashMap<ActionId, RegisteredAction>>,
|
||||||
|
|
||||||
|
/// Actions indexed by scope for efficient filtering.
|
||||||
|
scope_index: RwLock<HashMap<ActionScope, Vec<ActionId>>>,
|
||||||
|
|
||||||
|
/// All registered groups indexed by ID.
|
||||||
|
groups: RwLock<HashMap<ActionGroupId, RegisteredActionGroup>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Default for ActionExecutor {
|
||||||
|
fn default() -> Self {
|
||||||
|
Self::new()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionExecutor {
|
||||||
|
/// Create a new empty executor.
|
||||||
|
pub fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
actions: RwLock::new(HashMap::new()),
|
||||||
|
scope_index: RwLock::new(HashMap::new()),
|
||||||
|
groups: RwLock::new(HashMap::new()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
// Action Registration
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
/// Register an action with its handler.
|
||||||
|
///
|
||||||
|
/// Every action must have a handler - this is enforced by the API.
|
||||||
|
pub async fn register<H: ActionHandler + 'static>(
|
||||||
|
&self,
|
||||||
|
metadata: ActionMetadata,
|
||||||
|
source: ActionSource,
|
||||||
|
handler: H,
|
||||||
|
) -> Result<ActionId, ActionError> {
|
||||||
|
let id = metadata.id.clone();
|
||||||
|
let scope = metadata.scope.clone();
|
||||||
|
|
||||||
|
let action = RegisteredAction {
|
||||||
|
metadata,
|
||||||
|
source,
|
||||||
|
handler: Arc::new(handler),
|
||||||
|
};
|
||||||
|
|
||||||
|
// Insert action
|
||||||
|
{
|
||||||
|
let mut actions = self.actions.write().await;
|
||||||
|
actions.insert(id.clone(), action);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update scope index
|
||||||
|
{
|
||||||
|
let mut index = self.scope_index.write().await;
|
||||||
|
index.entry(scope).or_default().push(id.clone());
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Unregister an action.
|
||||||
|
pub async fn unregister(&self, id: &ActionId) -> Result<(), ActionError> {
|
||||||
|
let mut actions = self.actions.write().await;
|
||||||
|
|
||||||
|
let action = actions
|
||||||
|
.remove(id)
|
||||||
|
.ok_or_else(|| ActionError::NotFound(id.clone()))?;
|
||||||
|
|
||||||
|
// Update scope index
|
||||||
|
{
|
||||||
|
let mut index = self.scope_index.write().await;
|
||||||
|
if let Some(ids) = index.get_mut(&action.metadata.scope) {
|
||||||
|
ids.retain(|i| i != id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remove from group if assigned
|
||||||
|
if let Some(group_id) = &action.metadata.group_id {
|
||||||
|
let mut groups = self.groups.write().await;
|
||||||
|
if let Some(group) = groups.get_mut(group_id) {
|
||||||
|
group.action_ids.retain(|i| i != id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Unregister all actions from a specific source.
|
||||||
|
pub async fn unregister_source(&self, source_id: &str) -> Vec<ActionId> {
|
||||||
|
let actions_to_remove: Vec<ActionId> = {
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
actions
|
||||||
|
.iter()
|
||||||
|
.filter(|(_, a)| match &a.source {
|
||||||
|
ActionSource::Plugin { ref_id, .. } => ref_id == source_id,
|
||||||
|
ActionSource::Dynamic {
|
||||||
|
source_id: sid, ..
|
||||||
|
} => sid == source_id,
|
||||||
|
ActionSource::Builtin => false,
|
||||||
|
})
|
||||||
|
.map(|(id, _)| id.clone())
|
||||||
|
.collect()
|
||||||
|
};
|
||||||
|
|
||||||
|
for id in &actions_to_remove {
|
||||||
|
let _ = self.unregister(id).await;
|
||||||
|
}
|
||||||
|
|
||||||
|
actions_to_remove
|
||||||
|
}
|
||||||
|
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
// Action Invocation
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
/// Invoke an action with the given context and parameters.
|
||||||
|
///
|
||||||
|
/// This will:
|
||||||
|
/// 1. Look up the action metadata
|
||||||
|
/// 2. Check context availability
|
||||||
|
/// 3. Execute the handler
|
||||||
|
pub async fn invoke(
|
||||||
|
&self,
|
||||||
|
action_id: &ActionId,
|
||||||
|
context: CurrentContext,
|
||||||
|
params: ActionParams,
|
||||||
|
) -> Result<ActionResult, ActionError> {
|
||||||
|
// Get action and handler
|
||||||
|
let (metadata, handler) = {
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
let action = actions
|
||||||
|
.get(action_id)
|
||||||
|
.ok_or_else(|| ActionError::NotFound(action_id.clone()))?;
|
||||||
|
(action.metadata.clone(), action.handler.clone())
|
||||||
|
};
|
||||||
|
|
||||||
|
// Check context availability
|
||||||
|
let availability = check_context_availability(&metadata.required_context, &context);
|
||||||
|
|
||||||
|
match availability {
|
||||||
|
ActionAvailability::Available | ActionAvailability::AvailableWithPrompt { .. } => {
|
||||||
|
// Context is satisfied, proceed with execution
|
||||||
|
}
|
||||||
|
ActionAvailability::Unavailable { missing_fields } => {
|
||||||
|
return Err(ActionError::ContextMissing { missing_fields });
|
||||||
|
}
|
||||||
|
ActionAvailability::NotFound => {
|
||||||
|
return Err(ActionError::NotFound(action_id.clone()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Execute handler
|
||||||
|
handler.handle(context, params).await
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Invoke an action, skipping context validation.
|
||||||
|
///
|
||||||
|
/// Use this when you've already validated the context externally.
|
||||||
|
pub async fn invoke_unchecked(
|
||||||
|
&self,
|
||||||
|
action_id: &ActionId,
|
||||||
|
context: CurrentContext,
|
||||||
|
params: ActionParams,
|
||||||
|
) -> Result<ActionResult, ActionError> {
|
||||||
|
// Get handler
|
||||||
|
let handler = {
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
let action = actions
|
||||||
|
.get(action_id)
|
||||||
|
.ok_or_else(|| ActionError::NotFound(action_id.clone()))?;
|
||||||
|
action.handler.clone()
|
||||||
|
};
|
||||||
|
|
||||||
|
// Execute handler
|
||||||
|
handler.handle(context, params).await
|
||||||
|
}
|
||||||
|
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
// Action Queries
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
/// Get action metadata by ID.
|
||||||
|
pub async fn get(&self, id: &ActionId) -> Option<ActionMetadata> {
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
actions.get(id).map(|a| a.metadata.clone())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// List all actions, optionally filtered.
|
||||||
|
pub async fn list(&self, options: ListActionsOptions) -> Vec<ActionMetadata> {
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
|
||||||
|
let mut result: Vec<_> = actions
|
||||||
|
.values()
|
||||||
|
.filter(|a| {
|
||||||
|
// Scope filter
|
||||||
|
if let Some(scope) = &options.scope {
|
||||||
|
if &a.metadata.scope != scope {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Group filter
|
||||||
|
if let Some(group_id) = &options.group_id {
|
||||||
|
if a.metadata.group_id.as_ref() != Some(group_id) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Search filter
|
||||||
|
if let Some(search) = &options.search {
|
||||||
|
let search = search.to_lowercase();
|
||||||
|
let matches_label = a.metadata.label.to_lowercase().contains(&search);
|
||||||
|
let matches_desc = a
|
||||||
|
.metadata
|
||||||
|
.description
|
||||||
|
.as_ref()
|
||||||
|
.map(|d| d.to_lowercase().contains(&search))
|
||||||
|
.unwrap_or(false);
|
||||||
|
if !matches_label && !matches_desc {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
true
|
||||||
|
})
|
||||||
|
.map(|a| a.metadata.clone())
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
// Sort by order then label
|
||||||
|
result.sort_by(|a, b| a.order.cmp(&b.order).then_with(|| a.label.cmp(&b.label)));
|
||||||
|
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
/// List actions available in the given context.
|
||||||
|
pub async fn list_available(
|
||||||
|
&self,
|
||||||
|
context: &CurrentContext,
|
||||||
|
options: ListActionsOptions,
|
||||||
|
) -> Vec<(ActionMetadata, ActionAvailability)> {
|
||||||
|
let all_actions = self.list(options).await;
|
||||||
|
|
||||||
|
all_actions
|
||||||
|
.into_iter()
|
||||||
|
.map(|action| {
|
||||||
|
let availability =
|
||||||
|
check_context_availability(&action.required_context, context);
|
||||||
|
(action, availability)
|
||||||
|
})
|
||||||
|
.filter(|(_, availability)| availability.is_available())
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get availability status for a specific action.
|
||||||
|
pub async fn get_availability(
|
||||||
|
&self,
|
||||||
|
id: &ActionId,
|
||||||
|
context: &CurrentContext,
|
||||||
|
) -> ActionAvailability {
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
|
||||||
|
match actions.get(id) {
|
||||||
|
Some(action) => {
|
||||||
|
check_context_availability(&action.metadata.required_context, context)
|
||||||
|
}
|
||||||
|
None => ActionAvailability::NotFound,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
// Group Registration
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
/// Register an action group.
|
||||||
|
pub async fn register_group(
|
||||||
|
&self,
|
||||||
|
metadata: ActionGroupMetadata,
|
||||||
|
source: ActionGroupSource,
|
||||||
|
) -> Result<ActionGroupId, ActionError> {
|
||||||
|
let id = metadata.id.clone();
|
||||||
|
|
||||||
|
let mut groups = self.groups.write().await;
|
||||||
|
if groups.contains_key(&id) {
|
||||||
|
return Err(ActionError::GroupAlreadyExists(id));
|
||||||
|
}
|
||||||
|
|
||||||
|
groups.insert(
|
||||||
|
id.clone(),
|
||||||
|
RegisteredActionGroup {
|
||||||
|
metadata,
|
||||||
|
action_ids: Vec::new(),
|
||||||
|
source,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
|
||||||
|
Ok(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Unregister a group (does not unregister its actions).
|
||||||
|
pub async fn unregister_group(&self, id: &ActionGroupId) -> Result<(), ActionError> {
|
||||||
|
let mut groups = self.groups.write().await;
|
||||||
|
groups
|
||||||
|
.remove(id)
|
||||||
|
.ok_or_else(|| ActionError::GroupNotFound(id.clone()))?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Add an action to a group.
|
||||||
|
pub async fn add_to_group(
|
||||||
|
&self,
|
||||||
|
action_id: &ActionId,
|
||||||
|
group_id: &ActionGroupId,
|
||||||
|
) -> Result<(), ActionError> {
|
||||||
|
// Update action's group_id
|
||||||
|
{
|
||||||
|
let mut actions = self.actions.write().await;
|
||||||
|
let action = actions
|
||||||
|
.get_mut(action_id)
|
||||||
|
.ok_or_else(|| ActionError::NotFound(action_id.clone()))?;
|
||||||
|
action.metadata.group_id = Some(group_id.clone());
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add to group's action list
|
||||||
|
{
|
||||||
|
let mut groups = self.groups.write().await;
|
||||||
|
let group = groups
|
||||||
|
.get_mut(group_id)
|
||||||
|
.ok_or_else(|| ActionError::GroupNotFound(group_id.clone()))?;
|
||||||
|
|
||||||
|
if !group.action_ids.contains(action_id) {
|
||||||
|
group.action_ids.push(action_id.clone());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
// Group Queries
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
/// Get a group by ID.
|
||||||
|
pub async fn get_group(&self, id: &ActionGroupId) -> Option<ActionGroupMetadata> {
|
||||||
|
let groups = self.groups.read().await;
|
||||||
|
groups.get(id).map(|g| g.metadata.clone())
|
||||||
|
}
|
||||||
|
|
||||||
|
/// List all groups, optionally filtered by scope.
|
||||||
|
pub async fn list_groups(&self, scope: Option<ActionScope>) -> Vec<ActionGroupMetadata> {
|
||||||
|
let groups = self.groups.read().await;
|
||||||
|
|
||||||
|
let mut result: Vec<_> = groups
|
||||||
|
.values()
|
||||||
|
.filter(|g| {
|
||||||
|
scope.as_ref().map_or(true, |s| {
|
||||||
|
g.metadata.scope.as_ref().map_or(true, |gs| gs == s)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.map(|g| g.metadata.clone())
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
result.sort_by_key(|g| g.order);
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
/// List all actions in a specific group.
|
||||||
|
pub async fn list_by_group(&self, group_id: &ActionGroupId) -> Vec<ActionMetadata> {
|
||||||
|
let groups = self.groups.read().await;
|
||||||
|
let actions = self.actions.read().await;
|
||||||
|
|
||||||
|
groups
|
||||||
|
.get(group_id)
|
||||||
|
.map(|group| {
|
||||||
|
let mut result: Vec<_> = group
|
||||||
|
.action_ids
|
||||||
|
.iter()
|
||||||
|
.filter_map(|id| actions.get(id).map(|a| a.metadata.clone()))
|
||||||
|
.collect();
|
||||||
|
result.sort_by_key(|a| a.order);
|
||||||
|
result
|
||||||
|
})
|
||||||
|
.unwrap_or_default()
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get actions organized by their groups.
|
||||||
|
pub async fn list_grouped(&self, scope: Option<ActionScope>) -> Vec<ActionGroupWithActions> {
|
||||||
|
let group_list = self.list_groups(scope).await;
|
||||||
|
let mut result = Vec::new();
|
||||||
|
|
||||||
|
for group in group_list {
|
||||||
|
let actions = self.list_by_group(&group.id).await;
|
||||||
|
result.push(ActionGroupWithActions { group, actions });
|
||||||
|
}
|
||||||
|
|
||||||
|
result
|
||||||
|
}
|
||||||
|
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
// Built-in Registration
|
||||||
|
// ─────────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
/// Register all built-in groups.
|
||||||
|
pub async fn register_builtin_groups(&self) -> Result<(), ActionError> {
|
||||||
|
for group in crate::groups::builtin::all() {
|
||||||
|
self.register_group(group, ActionGroupSource::Builtin).await?;
|
||||||
|
}
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
use crate::{handler_fn, RequiredContext};
|
||||||
|
|
||||||
|
async fn create_test_executor() -> ActionExecutor {
|
||||||
|
let executor = ActionExecutor::new();
|
||||||
|
executor
|
||||||
|
.register(
|
||||||
|
ActionMetadata {
|
||||||
|
id: ActionId::builtin("test", "echo"),
|
||||||
|
label: "Echo".to_string(),
|
||||||
|
description: None,
|
||||||
|
icon: None,
|
||||||
|
scope: ActionScope::Global,
|
||||||
|
keyboard_shortcut: None,
|
||||||
|
requires_selection: false,
|
||||||
|
enabled_condition: None,
|
||||||
|
group_id: None,
|
||||||
|
order: 0,
|
||||||
|
required_context: RequiredContext::default(),
|
||||||
|
},
|
||||||
|
ActionSource::Builtin,
|
||||||
|
handler_fn(|_ctx, params| async move {
|
||||||
|
let msg: String = params.get("message").unwrap_or_default();
|
||||||
|
Ok(ActionResult::with_message(msg))
|
||||||
|
}),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
executor
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_register_and_invoke() {
|
||||||
|
let executor = create_test_executor().await;
|
||||||
|
let action_id = ActionId::builtin("test", "echo");
|
||||||
|
|
||||||
|
let params = ActionParams::from_json(serde_json::json!({
|
||||||
|
"message": "Hello, World!"
|
||||||
|
}));
|
||||||
|
|
||||||
|
let result = executor
|
||||||
|
.invoke(&action_id, CurrentContext::default(), params)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
match result {
|
||||||
|
ActionResult::Success { message, .. } => {
|
||||||
|
assert_eq!(message, Some("Hello, World!".to_string()));
|
||||||
|
}
|
||||||
|
_ => panic!("Expected Success result"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_invoke_not_found() {
|
||||||
|
let executor = ActionExecutor::new();
|
||||||
|
let action_id = ActionId::builtin("test", "unknown");
|
||||||
|
|
||||||
|
let result = executor
|
||||||
|
.invoke(&action_id, CurrentContext::default(), ActionParams::empty())
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(matches!(result, Err(ActionError::NotFound(_))));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_list_by_scope() {
|
||||||
|
let executor = ActionExecutor::new();
|
||||||
|
|
||||||
|
executor
|
||||||
|
.register(
|
||||||
|
ActionMetadata {
|
||||||
|
id: ActionId::builtin("global", "one"),
|
||||||
|
label: "Global One".to_string(),
|
||||||
|
description: None,
|
||||||
|
icon: None,
|
||||||
|
scope: ActionScope::Global,
|
||||||
|
keyboard_shortcut: None,
|
||||||
|
requires_selection: false,
|
||||||
|
enabled_condition: None,
|
||||||
|
group_id: None,
|
||||||
|
order: 0,
|
||||||
|
required_context: RequiredContext::default(),
|
||||||
|
},
|
||||||
|
ActionSource::Builtin,
|
||||||
|
handler_fn(|_ctx, _params| async move { Ok(ActionResult::ok()) }),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
executor
|
||||||
|
.register(
|
||||||
|
ActionMetadata {
|
||||||
|
id: ActionId::builtin("http", "one"),
|
||||||
|
label: "HTTP One".to_string(),
|
||||||
|
description: None,
|
||||||
|
icon: None,
|
||||||
|
scope: ActionScope::HttpRequest,
|
||||||
|
keyboard_shortcut: None,
|
||||||
|
requires_selection: false,
|
||||||
|
enabled_condition: None,
|
||||||
|
group_id: None,
|
||||||
|
order: 0,
|
||||||
|
required_context: RequiredContext::default(),
|
||||||
|
},
|
||||||
|
ActionSource::Builtin,
|
||||||
|
handler_fn(|_ctx, _params| async move { Ok(ActionResult::ok()) }),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let global_actions = executor
|
||||||
|
.list(ListActionsOptions {
|
||||||
|
scope: Some(ActionScope::Global),
|
||||||
|
..Default::default()
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
assert_eq!(global_actions.len(), 1);
|
||||||
|
|
||||||
|
let http_actions = executor
|
||||||
|
.list(ListActionsOptions {
|
||||||
|
scope: Some(ActionScope::HttpRequest),
|
||||||
|
..Default::default()
|
||||||
|
})
|
||||||
|
.await;
|
||||||
|
assert_eq!(http_actions.len(), 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_groups() {
|
||||||
|
let executor = ActionExecutor::new();
|
||||||
|
executor.register_builtin_groups().await.unwrap();
|
||||||
|
|
||||||
|
let groups = executor.list_groups(None).await;
|
||||||
|
assert!(!groups.is_empty());
|
||||||
|
|
||||||
|
let export_group = executor.get_group(&ActionGroupId::builtin("export")).await;
|
||||||
|
assert!(export_group.is_some());
|
||||||
|
assert_eq!(export_group.unwrap().name, "Export");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_unregister() {
|
||||||
|
let executor = create_test_executor().await;
|
||||||
|
let action_id = ActionId::builtin("test", "echo");
|
||||||
|
|
||||||
|
assert!(executor.get(&action_id).await.is_some());
|
||||||
|
|
||||||
|
executor.unregister(&action_id).await.unwrap();
|
||||||
|
assert!(executor.get(&action_id).await.is_none());
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,208 @@
|
|||||||
|
//! Action group types and management.
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use ts_rs::TS;
|
||||||
|
|
||||||
|
use crate::{ActionId, ActionMetadata, ActionScope};
|
||||||
|
|
||||||
|
/// Unique identifier for an action group.
|
||||||
|
///
|
||||||
|
/// Format: `namespace:group-name`
|
||||||
|
/// - Built-in: `yaak:export`
|
||||||
|
/// - Plugin: `plugin.my-plugin:utilities`
|
||||||
|
#[derive(Clone, Debug, Hash, Eq, PartialEq, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
pub struct ActionGroupId(pub String);
|
||||||
|
|
||||||
|
impl ActionGroupId {
|
||||||
|
/// Create a namespaced group ID.
|
||||||
|
pub fn new(namespace: &str, name: &str) -> Self {
|
||||||
|
Self(format!("{}:{}", namespace, name))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create ID for built-in groups.
|
||||||
|
pub fn builtin(name: &str) -> Self {
|
||||||
|
Self::new("yaak", name)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create ID for plugin groups.
|
||||||
|
pub fn plugin(plugin_ref_id: &str, name: &str) -> Self {
|
||||||
|
Self::new(&format!("plugin.{}", plugin_ref_id), name)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the raw string value.
|
||||||
|
pub fn as_str(&self) -> &str {
|
||||||
|
&self.0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::fmt::Display for ActionGroupId {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
write!(f, "{}", self.0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Metadata about an action group.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct ActionGroupMetadata {
|
||||||
|
/// Unique identifier for this group.
|
||||||
|
pub id: ActionGroupId,
|
||||||
|
|
||||||
|
/// Display name for the group.
|
||||||
|
pub name: String,
|
||||||
|
|
||||||
|
/// Optional description of the group's purpose.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub description: Option<String>,
|
||||||
|
|
||||||
|
/// Icon to display for the group.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub icon: Option<String>,
|
||||||
|
|
||||||
|
/// Sort order for displaying groups (lower = earlier).
|
||||||
|
#[serde(default)]
|
||||||
|
pub order: i32,
|
||||||
|
|
||||||
|
/// Optional scope restriction (if set, group only appears in this scope).
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub scope: Option<ActionScope>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Where an action group was registered from.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "type", rename_all = "kebab-case")]
|
||||||
|
pub enum ActionGroupSource {
|
||||||
|
/// Built into Yaak core.
|
||||||
|
Builtin,
|
||||||
|
/// Registered by a plugin.
|
||||||
|
Plugin {
|
||||||
|
/// Plugin reference ID.
|
||||||
|
ref_id: String,
|
||||||
|
/// Plugin name.
|
||||||
|
name: String,
|
||||||
|
},
|
||||||
|
/// Registered at runtime.
|
||||||
|
Dynamic {
|
||||||
|
/// Source identifier.
|
||||||
|
source_id: String,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A registered action group with its actions.
|
||||||
|
#[derive(Clone, Debug)]
|
||||||
|
pub struct RegisteredActionGroup {
|
||||||
|
/// Group metadata.
|
||||||
|
pub metadata: ActionGroupMetadata,
|
||||||
|
|
||||||
|
/// IDs of actions in this group (ordered by action's order field).
|
||||||
|
pub action_ids: Vec<ActionId>,
|
||||||
|
|
||||||
|
/// Where the group was registered from.
|
||||||
|
pub source: ActionGroupSource,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A group with its actions for UI rendering.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct ActionGroupWithActions {
|
||||||
|
/// Group metadata.
|
||||||
|
pub group: ActionGroupMetadata,
|
||||||
|
|
||||||
|
/// Actions in this group.
|
||||||
|
pub actions: Vec<ActionMetadata>,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Built-in action group definitions.
|
||||||
|
pub mod builtin {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
/// Export group - export and copy actions.
|
||||||
|
pub fn export() -> ActionGroupMetadata {
|
||||||
|
ActionGroupMetadata {
|
||||||
|
id: ActionGroupId::builtin("export"),
|
||||||
|
name: "Export".into(),
|
||||||
|
description: Some("Export and copy actions".into()),
|
||||||
|
icon: Some("download".into()),
|
||||||
|
order: 100,
|
||||||
|
scope: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Code generation group.
|
||||||
|
pub fn code_generation() -> ActionGroupMetadata {
|
||||||
|
ActionGroupMetadata {
|
||||||
|
id: ActionGroupId::builtin("code-generation"),
|
||||||
|
name: "Code Generation".into(),
|
||||||
|
description: Some("Generate code snippets from requests".into()),
|
||||||
|
icon: Some("code".into()),
|
||||||
|
order: 200,
|
||||||
|
scope: Some(ActionScope::HttpRequest),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Send group - request sending actions.
|
||||||
|
pub fn send() -> ActionGroupMetadata {
|
||||||
|
ActionGroupMetadata {
|
||||||
|
id: ActionGroupId::builtin("send"),
|
||||||
|
name: "Send".into(),
|
||||||
|
description: Some("Actions for sending requests".into()),
|
||||||
|
icon: Some("play".into()),
|
||||||
|
order: 50,
|
||||||
|
scope: Some(ActionScope::HttpRequest),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Import group.
|
||||||
|
pub fn import() -> ActionGroupMetadata {
|
||||||
|
ActionGroupMetadata {
|
||||||
|
id: ActionGroupId::builtin("import"),
|
||||||
|
name: "Import".into(),
|
||||||
|
description: Some("Import data from files".into()),
|
||||||
|
icon: Some("upload".into()),
|
||||||
|
order: 150,
|
||||||
|
scope: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Workspace management group.
|
||||||
|
pub fn workspace() -> ActionGroupMetadata {
|
||||||
|
ActionGroupMetadata {
|
||||||
|
id: ActionGroupId::builtin("workspace"),
|
||||||
|
name: "Workspace".into(),
|
||||||
|
description: Some("Workspace management actions".into()),
|
||||||
|
icon: Some("folder".into()),
|
||||||
|
order: 300,
|
||||||
|
scope: Some(ActionScope::Workspace),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get all built-in group definitions.
|
||||||
|
pub fn all() -> Vec<ActionGroupMetadata> {
|
||||||
|
vec![send(), export(), import(), code_generation(), workspace()]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_group_id_creation() {
|
||||||
|
let id = ActionGroupId::builtin("export");
|
||||||
|
assert_eq!(id.as_str(), "yaak:export");
|
||||||
|
|
||||||
|
let plugin_id = ActionGroupId::plugin("my-plugin", "utilities");
|
||||||
|
assert_eq!(plugin_id.as_str(), "plugin.my-plugin:utilities");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_builtin_groups() {
|
||||||
|
let groups = builtin::all();
|
||||||
|
assert!(!groups.is_empty());
|
||||||
|
assert!(groups.iter().any(|g| g.id == ActionGroupId::builtin("export")));
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,103 @@
|
|||||||
|
//! Action handler types and execution.
|
||||||
|
|
||||||
|
use std::future::Future;
|
||||||
|
use std::pin::Pin;
|
||||||
|
use std::sync::Arc;
|
||||||
|
|
||||||
|
use crate::{ActionError, ActionParams, ActionResult, CurrentContext};
|
||||||
|
|
||||||
|
/// A boxed future for async action handlers.
|
||||||
|
pub type BoxFuture<'a, T> = Pin<Box<dyn Future<Output = T> + Send + 'a>>;
|
||||||
|
|
||||||
|
/// Function signature for action handlers.
|
||||||
|
pub type ActionHandlerFn = Arc<
|
||||||
|
dyn Fn(CurrentContext, ActionParams) -> BoxFuture<'static, Result<ActionResult, ActionError>>
|
||||||
|
+ Send
|
||||||
|
+ Sync,
|
||||||
|
>;
|
||||||
|
|
||||||
|
/// Trait for types that can handle action invocations.
|
||||||
|
pub trait ActionHandler: Send + Sync {
|
||||||
|
/// Execute the action with the given context and parameters.
|
||||||
|
fn handle(
|
||||||
|
&self,
|
||||||
|
context: CurrentContext,
|
||||||
|
params: ActionParams,
|
||||||
|
) -> BoxFuture<'static, Result<ActionResult, ActionError>>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Wrapper to create an ActionHandler from a function.
|
||||||
|
pub struct FnHandler<F>(pub F);
|
||||||
|
|
||||||
|
impl<F, Fut> ActionHandler for FnHandler<F>
|
||||||
|
where
|
||||||
|
F: Fn(CurrentContext, ActionParams) -> Fut + Send + Sync,
|
||||||
|
Fut: Future<Output = Result<ActionResult, ActionError>> + Send + 'static,
|
||||||
|
{
|
||||||
|
fn handle(
|
||||||
|
&self,
|
||||||
|
context: CurrentContext,
|
||||||
|
params: ActionParams,
|
||||||
|
) -> BoxFuture<'static, Result<ActionResult, ActionError>> {
|
||||||
|
Box::pin((self.0)(context, params))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create an action handler from an async function.
|
||||||
|
///
|
||||||
|
/// # Example
|
||||||
|
/// ```ignore
|
||||||
|
/// let handler = handler_fn(|ctx, params| async move {
|
||||||
|
/// Ok(ActionResult::ok())
|
||||||
|
/// });
|
||||||
|
/// ```
|
||||||
|
pub fn handler_fn<F, Fut>(f: F) -> FnHandler<F>
|
||||||
|
where
|
||||||
|
F: Fn(CurrentContext, ActionParams) -> Fut + Send + Sync,
|
||||||
|
Fut: Future<Output = Result<ActionResult, ActionError>> + Send + 'static,
|
||||||
|
{
|
||||||
|
FnHandler(f)
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handler_fn() {
|
||||||
|
let handler = handler_fn(|_ctx, _params| async move { Ok(ActionResult::ok()) });
|
||||||
|
|
||||||
|
let result = handler
|
||||||
|
.handle(CurrentContext::default(), ActionParams::empty())
|
||||||
|
.await;
|
||||||
|
|
||||||
|
assert!(result.is_ok());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handler_with_params() {
|
||||||
|
let handler = handler_fn(|_ctx, params| async move {
|
||||||
|
let name: Option<String> = params.get("name");
|
||||||
|
Ok(ActionResult::with_message(format!(
|
||||||
|
"Hello, {}!",
|
||||||
|
name.unwrap_or_else(|| "World".to_string())
|
||||||
|
)))
|
||||||
|
});
|
||||||
|
|
||||||
|
let params = ActionParams::from_json(serde_json::json!({
|
||||||
|
"name": "Yaak"
|
||||||
|
}));
|
||||||
|
|
||||||
|
let result = handler
|
||||||
|
.handle(CurrentContext::default(), params)
|
||||||
|
.await
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
match result {
|
||||||
|
ActionResult::Success { message, .. } => {
|
||||||
|
assert_eq!(message, Some("Hello, Yaak!".to_string()));
|
||||||
|
}
|
||||||
|
_ => panic!("Expected Success result"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
//! Centralized action system for Yaak.
|
||||||
|
//!
|
||||||
|
//! This crate provides a unified hub for registering and invoking actions
|
||||||
|
//! across all entry points: plugins, Tauri desktop app, CLI, deep links, and MCP server.
|
||||||
|
|
||||||
|
mod context;
|
||||||
|
mod error;
|
||||||
|
mod executor;
|
||||||
|
mod groups;
|
||||||
|
mod handler;
|
||||||
|
mod types;
|
||||||
|
|
||||||
|
pub use context::*;
|
||||||
|
pub use error::*;
|
||||||
|
pub use executor::*;
|
||||||
|
pub use groups::*;
|
||||||
|
pub use handler::*;
|
||||||
|
pub use types::*;
|
||||||
@@ -0,0 +1,273 @@
|
|||||||
|
//! Core types for the action system.
|
||||||
|
|
||||||
|
use serde::{Deserialize, Serialize};
|
||||||
|
use ts_rs::TS;
|
||||||
|
|
||||||
|
use crate::{ActionGroupId, RequiredContext};
|
||||||
|
|
||||||
|
/// Unique identifier for an action.
|
||||||
|
///
|
||||||
|
/// Format: `namespace:category:name`
|
||||||
|
/// - Built-in: `yaak:http-request:send`
|
||||||
|
/// - Plugin: `plugin.copy-curl:http-request:copy`
|
||||||
|
#[derive(Clone, Debug, Hash, Eq, PartialEq, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
pub struct ActionId(pub String);
|
||||||
|
|
||||||
|
impl ActionId {
|
||||||
|
/// Create a namespaced action ID.
|
||||||
|
pub fn new(namespace: &str, category: &str, name: &str) -> Self {
|
||||||
|
Self(format!("{}:{}:{}", namespace, category, name))
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create ID for built-in actions.
|
||||||
|
pub fn builtin(category: &str, name: &str) -> Self {
|
||||||
|
Self::new("yaak", category, name)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create ID for plugin actions.
|
||||||
|
pub fn plugin(plugin_ref_id: &str, category: &str, name: &str) -> Self {
|
||||||
|
Self::new(&format!("plugin.{}", plugin_ref_id), category, name)
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get the raw string value.
|
||||||
|
pub fn as_str(&self) -> &str {
|
||||||
|
&self.0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl std::fmt::Display for ActionId {
|
||||||
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||||
|
write!(f, "{}", self.0)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// The scope in which an action can be invoked.
|
||||||
|
#[derive(Clone, Debug, Hash, Eq, PartialEq, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "kebab-case")]
|
||||||
|
pub enum ActionScope {
|
||||||
|
/// Global actions available everywhere.
|
||||||
|
Global,
|
||||||
|
/// Actions on HTTP requests.
|
||||||
|
HttpRequest,
|
||||||
|
/// Actions on WebSocket requests.
|
||||||
|
WebsocketRequest,
|
||||||
|
/// Actions on gRPC requests.
|
||||||
|
GrpcRequest,
|
||||||
|
/// Actions on workspaces.
|
||||||
|
Workspace,
|
||||||
|
/// Actions on folders.
|
||||||
|
Folder,
|
||||||
|
/// Actions on environments.
|
||||||
|
Environment,
|
||||||
|
/// Actions on cookie jars.
|
||||||
|
CookieJar,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Metadata about an action for discovery.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct ActionMetadata {
|
||||||
|
/// Unique identifier for this action.
|
||||||
|
pub id: ActionId,
|
||||||
|
|
||||||
|
/// Display label for the action.
|
||||||
|
pub label: String,
|
||||||
|
|
||||||
|
/// Optional description of what the action does.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub description: Option<String>,
|
||||||
|
|
||||||
|
/// Icon name to display.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub icon: Option<String>,
|
||||||
|
|
||||||
|
/// The scope this action applies to.
|
||||||
|
pub scope: ActionScope,
|
||||||
|
|
||||||
|
/// Keyboard shortcut (e.g., "Cmd+Enter").
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub keyboard_shortcut: Option<String>,
|
||||||
|
|
||||||
|
/// Whether the action requires a selection/target.
|
||||||
|
#[serde(default)]
|
||||||
|
pub requires_selection: bool,
|
||||||
|
|
||||||
|
/// Optional condition expression for when action is enabled.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub enabled_condition: Option<String>,
|
||||||
|
|
||||||
|
/// Optional group this action belongs to.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
pub group_id: Option<ActionGroupId>,
|
||||||
|
|
||||||
|
/// Sort order within a group (lower = earlier).
|
||||||
|
#[serde(default)]
|
||||||
|
pub order: i32,
|
||||||
|
|
||||||
|
/// Context requirements for this action.
|
||||||
|
#[serde(default)]
|
||||||
|
pub required_context: RequiredContext,
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Where an action was registered from.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "type", rename_all = "kebab-case")]
|
||||||
|
pub enum ActionSource {
|
||||||
|
/// Built into Yaak core.
|
||||||
|
Builtin,
|
||||||
|
/// Registered by a plugin.
|
||||||
|
Plugin {
|
||||||
|
/// Plugin reference ID.
|
||||||
|
ref_id: String,
|
||||||
|
/// Plugin name.
|
||||||
|
name: String,
|
||||||
|
},
|
||||||
|
/// Registered at runtime (e.g., by MCP tools).
|
||||||
|
Dynamic {
|
||||||
|
/// Source identifier.
|
||||||
|
source_id: String,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Parameters passed to action handlers.
|
||||||
|
#[derive(Clone, Debug, Default, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
pub struct ActionParams {
|
||||||
|
/// Arbitrary JSON parameters.
|
||||||
|
#[serde(default)]
|
||||||
|
#[ts(type = "unknown")]
|
||||||
|
pub data: serde_json::Value,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionParams {
|
||||||
|
/// Create empty params.
|
||||||
|
pub fn empty() -> Self {
|
||||||
|
Self {
|
||||||
|
data: serde_json::Value::Null,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create params from a JSON value.
|
||||||
|
pub fn from_json(data: serde_json::Value) -> Self {
|
||||||
|
Self { data }
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Get a typed value from the params.
|
||||||
|
pub fn get<T: serde::de::DeserializeOwned>(&self, key: &str) -> Option<T> {
|
||||||
|
self.data
|
||||||
|
.get(key)
|
||||||
|
.and_then(|v| serde_json::from_value(v.clone()).ok())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Result of action execution.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "type", rename_all = "kebab-case")]
|
||||||
|
pub enum ActionResult {
|
||||||
|
/// Action completed successfully.
|
||||||
|
Success {
|
||||||
|
/// Optional data to return.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
#[ts(type = "unknown")]
|
||||||
|
data: Option<serde_json::Value>,
|
||||||
|
/// Optional message to display.
|
||||||
|
#[serde(skip_serializing_if = "Option::is_none")]
|
||||||
|
message: Option<String>,
|
||||||
|
},
|
||||||
|
|
||||||
|
/// Action requires user input to continue.
|
||||||
|
RequiresInput {
|
||||||
|
/// Prompt to show user.
|
||||||
|
prompt: InputPrompt,
|
||||||
|
/// Continuation token.
|
||||||
|
continuation_id: String,
|
||||||
|
},
|
||||||
|
|
||||||
|
/// Action was cancelled by the user.
|
||||||
|
Cancelled,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ActionResult {
|
||||||
|
/// Create a success result with no data.
|
||||||
|
pub fn ok() -> Self {
|
||||||
|
Self::Success {
|
||||||
|
data: None,
|
||||||
|
message: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a success result with a message.
|
||||||
|
pub fn with_message(message: impl Into<String>) -> Self {
|
||||||
|
Self::Success {
|
||||||
|
data: None,
|
||||||
|
message: Some(message.into()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Create a success result with data.
|
||||||
|
pub fn with_data(data: serde_json::Value) -> Self {
|
||||||
|
Self::Success {
|
||||||
|
data: Some(data),
|
||||||
|
message: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// A prompt for user input.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
#[serde(tag = "type", rename_all = "kebab-case")]
|
||||||
|
pub enum InputPrompt {
|
||||||
|
/// Text input prompt.
|
||||||
|
Text {
|
||||||
|
label: String,
|
||||||
|
placeholder: Option<String>,
|
||||||
|
default_value: Option<String>,
|
||||||
|
},
|
||||||
|
/// Selection prompt.
|
||||||
|
Select {
|
||||||
|
label: String,
|
||||||
|
options: Vec<SelectOption>,
|
||||||
|
},
|
||||||
|
/// Confirmation prompt.
|
||||||
|
Confirm { label: String },
|
||||||
|
}
|
||||||
|
|
||||||
|
/// An option in a select prompt.
|
||||||
|
#[derive(Clone, Debug, Serialize, Deserialize, TS)]
|
||||||
|
#[ts(export)]
|
||||||
|
pub struct SelectOption {
|
||||||
|
pub label: String,
|
||||||
|
pub value: String,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_action_id_creation() {
|
||||||
|
let id = ActionId::builtin("http-request", "send");
|
||||||
|
assert_eq!(id.as_str(), "yaak:http-request:send");
|
||||||
|
|
||||||
|
let plugin_id = ActionId::plugin("copy-curl", "http-request", "copy");
|
||||||
|
assert_eq!(plugin_id.as_str(), "plugin.copy-curl:http-request:copy");
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_action_params() {
|
||||||
|
let params = ActionParams::from_json(serde_json::json!({
|
||||||
|
"name": "test",
|
||||||
|
"count": 42
|
||||||
|
}));
|
||||||
|
|
||||||
|
assert_eq!(params.get::<String>("name"), Some("test".to_string()));
|
||||||
|
assert_eq!(params.get::<i32>("count"), Some(42));
|
||||||
|
assert_eq!(params.get::<String>("missing"), None);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -6,7 +6,7 @@ publish = false
|
|||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
chrono = { workspace = true, features = ["serde"] }
|
chrono = { workspace = true, features = ["serde"] }
|
||||||
git2 = { version = "0.20.4", features = ["vendored-libgit2", "vendored-openssl"] }
|
git2 = { version = "0.20.0", features = ["vendored-libgit2", "vendored-openssl"] }
|
||||||
log = { workspace = true }
|
log = { workspace = true }
|
||||||
serde = { workspace = true, features = ["derive"] }
|
serde = { workspace = true, features = ["derive"] }
|
||||||
serde_json = { workspace = true }
|
serde_json = { workspace = true }
|
||||||
|
|||||||
Generated
+2
-2
@@ -15,8 +15,8 @@ export type GitStatus = "untracked" | "conflict" | "current" | "modified" | "rem
|
|||||||
|
|
||||||
export type GitStatusEntry = { relaPath: string, status: GitStatus, staged: boolean, prev: SyncModel | null, next: SyncModel | null, };
|
export type GitStatusEntry = { relaPath: string, status: GitStatus, staged: boolean, prev: SyncModel | null, next: SyncModel | null, };
|
||||||
|
|
||||||
export type GitStatusSummary = { path: string, headRef: string | null, headRefShorthand: string | null, entries: Array<GitStatusEntry>, origins: Array<string>, localBranches: Array<string>, remoteBranches: Array<string>, ahead: number, behind: number, };
|
export type GitStatusSummary = { path: string, headRef: string | null, headRefShorthand: string | null, entries: Array<GitStatusEntry>, origins: Array<string>, localBranches: Array<string>, remoteBranches: Array<string>, };
|
||||||
|
|
||||||
export type PullResult = { "type": "success", message: string, } | { "type": "up_to_date" } | { "type": "needs_credentials", url: string, error: string | null, } | { "type": "diverged", remote: string, branch: string, } | { "type": "uncommitted_changes" };
|
export type PullResult = { "type": "success", message: string, } | { "type": "up_to_date" } | { "type": "needs_credentials", url: string, error: string | null, };
|
||||||
|
|
||||||
export type PushResult = { "type": "success", message: string, } | { "type": "up_to_date" } | { "type": "needs_credentials", url: string, error: string | null, };
|
export type PushResult = { "type": "success", message: string, } | { "type": "up_to_date" } | { "type": "needs_credentials", url: string, error: string | null, };
|
||||||
|
|||||||
+14
-68
@@ -4,7 +4,6 @@ import { createFastMutation } from '@yaakapp/app/hooks/useFastMutation';
|
|||||||
import { queryClient } from '@yaakapp/app/lib/queryClient';
|
import { queryClient } from '@yaakapp/app/lib/queryClient';
|
||||||
import { useMemo } from 'react';
|
import { useMemo } from 'react';
|
||||||
import { BranchDeleteResult, CloneResult, GitCommit, GitRemote, GitStatusSummary, PullResult, PushResult } from './bindings/gen_git';
|
import { BranchDeleteResult, CloneResult, GitCommit, GitRemote, GitStatusSummary, PullResult, PushResult } from './bindings/gen_git';
|
||||||
import { showToast } from '@yaakapp/app/lib/toast';
|
|
||||||
|
|
||||||
export * from './bindings/gen_git';
|
export * from './bindings/gen_git';
|
||||||
export * from './bindings/gen_models';
|
export * from './bindings/gen_models';
|
||||||
@@ -14,20 +13,11 @@ export interface GitCredentials {
|
|||||||
password: string;
|
password: string;
|
||||||
}
|
}
|
||||||
|
|
||||||
export type DivergedStrategy = 'force_reset' | 'merge' | 'cancel';
|
|
||||||
|
|
||||||
export type UncommittedChangesStrategy = 'reset' | 'cancel';
|
|
||||||
|
|
||||||
export interface GitCallbacks {
|
export interface GitCallbacks {
|
||||||
addRemote: () => Promise<GitRemote | null>;
|
addRemote: () => Promise<GitRemote | null>;
|
||||||
promptCredentials: (
|
promptCredentials: (
|
||||||
result: Extract<PushResult, { type: 'needs_credentials' }>,
|
result: Extract<PushResult, { type: 'needs_credentials' }>,
|
||||||
) => Promise<GitCredentials | null>;
|
) => Promise<GitCredentials | null>;
|
||||||
promptDiverged: (
|
|
||||||
result: Extract<PullResult, { type: 'diverged' }>,
|
|
||||||
) => Promise<DivergedStrategy>;
|
|
||||||
promptUncommittedChanges: () => Promise<UncommittedChangesStrategy>;
|
|
||||||
forceSync: () => Promise<void>;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const onSuccess = () => queryClient.invalidateQueries({ queryKey: ['git'] });
|
const onSuccess = () => queryClient.invalidateQueries({ queryKey: ['git'] });
|
||||||
@@ -79,15 +69,6 @@ export const gitMutations = (dir: string, callbacks: GitCallbacks) => {
|
|||||||
return invoke<PushResult>('cmd_git_push', { dir });
|
return invoke<PushResult>('cmd_git_push', { dir });
|
||||||
};
|
};
|
||||||
|
|
||||||
const handleError = (err: unknown) => {
|
|
||||||
showToast({
|
|
||||||
id: `${err}`,
|
|
||||||
message: `${err}`,
|
|
||||||
color: 'danger',
|
|
||||||
timeout: 5000,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return {
|
return {
|
||||||
init: createFastMutation<void, string, void>({
|
init: createFastMutation<void, string, void>({
|
||||||
mutationKey: ['git', 'init'],
|
mutationKey: ['git', 'init'],
|
||||||
@@ -152,9 +133,10 @@ export const gitMutations = (dir: string, callbacks: GitCallbacks) => {
|
|||||||
},
|
},
|
||||||
onSuccess,
|
onSuccess,
|
||||||
}),
|
}),
|
||||||
fetchAll: createFastMutation<void, string, void>({
|
fetchAll: createFastMutation<string, string, void>({
|
||||||
mutationKey: ['git', 'fetch_all', dir],
|
mutationKey: ['git', 'checkout', dir],
|
||||||
mutationFn: () => invoke('cmd_git_fetch_all', { dir }),
|
mutationFn: () => invoke('cmd_git_fetch_all', { dir }),
|
||||||
|
onSuccess,
|
||||||
}),
|
}),
|
||||||
push: createFastMutation<PushResult, string, void>({
|
push: createFastMutation<PushResult, string, void>({
|
||||||
mutationKey: ['git', 'push', dir],
|
mutationKey: ['git', 'push', dir],
|
||||||
@@ -165,51 +147,20 @@ export const gitMutations = (dir: string, callbacks: GitCallbacks) => {
|
|||||||
mutationKey: ['git', 'pull', dir],
|
mutationKey: ['git', 'pull', dir],
|
||||||
async mutationFn() {
|
async mutationFn() {
|
||||||
const result = await invoke<PullResult>('cmd_git_pull', { dir });
|
const result = await invoke<PullResult>('cmd_git_pull', { dir });
|
||||||
|
if (result.type !== 'needs_credentials') return result;
|
||||||
|
|
||||||
if (result.type === 'needs_credentials') {
|
// Needs credentials, prompt for them
|
||||||
const creds = await callbacks.promptCredentials(result);
|
const creds = await callbacks.promptCredentials(result);
|
||||||
if (creds == null) throw new Error('Canceled');
|
if (creds == null) throw new Error('Canceled');
|
||||||
|
|
||||||
await invoke('cmd_git_add_credential', {
|
await invoke('cmd_git_add_credential', {
|
||||||
remoteUrl: result.url,
|
remoteUrl: result.url,
|
||||||
username: creds.username,
|
username: creds.username,
|
||||||
password: creds.password,
|
password: creds.password,
|
||||||
});
|
});
|
||||||
|
|
||||||
// Pull again after credentials
|
// Pull again
|
||||||
return invoke<PullResult>('cmd_git_pull', { dir });
|
return invoke<PullResult>('cmd_git_pull', { dir });
|
||||||
}
|
|
||||||
|
|
||||||
if (result.type === 'uncommitted_changes') {
|
|
||||||
callbacks.promptUncommittedChanges().then(async (strategy) => {
|
|
||||||
if (strategy === 'cancel') return;
|
|
||||||
|
|
||||||
await invoke('cmd_git_reset_changes', { dir });
|
|
||||||
return invoke<PullResult>('cmd_git_pull', { dir });
|
|
||||||
}).then(async () => { onSuccess(); await callbacks.forceSync(); }, handleError);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (result.type === 'diverged') {
|
|
||||||
callbacks.promptDiverged(result).then((strategy) => {
|
|
||||||
if (strategy === 'cancel') return;
|
|
||||||
|
|
||||||
if (strategy === 'force_reset') {
|
|
||||||
return invoke<PullResult>('cmd_git_pull_force_reset', {
|
|
||||||
dir,
|
|
||||||
remote: result.remote,
|
|
||||||
branch: result.branch,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
return invoke<PullResult>('cmd_git_pull_merge', {
|
|
||||||
dir,
|
|
||||||
remote: result.remote,
|
|
||||||
branch: result.branch,
|
|
||||||
});
|
|
||||||
}).then(async () => { onSuccess(); await callbacks.forceSync(); }, handleError);
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
},
|
},
|
||||||
onSuccess,
|
onSuccess,
|
||||||
}),
|
}),
|
||||||
@@ -218,11 +169,6 @@ export const gitMutations = (dir: string, callbacks: GitCallbacks) => {
|
|||||||
mutationFn: (args) => invoke('cmd_git_unstage', { dir, ...args }),
|
mutationFn: (args) => invoke('cmd_git_unstage', { dir, ...args }),
|
||||||
onSuccess,
|
onSuccess,
|
||||||
}),
|
}),
|
||||||
resetChanges: createFastMutation<void, string, void>({
|
|
||||||
mutationKey: ['git', 'reset-changes', dir],
|
|
||||||
mutationFn: () => invoke('cmd_git_reset_changes', { dir }),
|
|
||||||
onSuccess,
|
|
||||||
}),
|
|
||||||
} as const;
|
} as const;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -13,7 +13,6 @@ mod pull;
|
|||||||
mod push;
|
mod push;
|
||||||
mod remotes;
|
mod remotes;
|
||||||
mod repository;
|
mod repository;
|
||||||
mod reset;
|
|
||||||
mod status;
|
mod status;
|
||||||
mod unstage;
|
mod unstage;
|
||||||
mod util;
|
mod util;
|
||||||
@@ -30,9 +29,8 @@ pub use credential::git_add_credential;
|
|||||||
pub use fetch::git_fetch_all;
|
pub use fetch::git_fetch_all;
|
||||||
pub use init::git_init;
|
pub use init::git_init;
|
||||||
pub use log::{GitCommit, git_log};
|
pub use log::{GitCommit, git_log};
|
||||||
pub use pull::{PullResult, git_pull, git_pull_force_reset, git_pull_merge};
|
pub use pull::{PullResult, git_pull};
|
||||||
pub use push::{PushResult, git_push};
|
pub use push::{PushResult, git_push};
|
||||||
pub use remotes::{GitRemote, git_add_remote, git_remotes, git_rm_remote};
|
pub use remotes::{GitRemote, git_add_remote, git_remotes, git_rm_remote};
|
||||||
pub use reset::git_reset_changes;
|
|
||||||
pub use status::{GitStatusSummary, git_status};
|
pub use status::{GitStatusSummary, git_status};
|
||||||
pub use unstage::git_unstage;
|
pub use unstage::git_unstage;
|
||||||
|
|||||||
@@ -15,23 +15,9 @@ pub enum PullResult {
|
|||||||
Success { message: String },
|
Success { message: String },
|
||||||
UpToDate,
|
UpToDate,
|
||||||
NeedsCredentials { url: String, error: Option<String> },
|
NeedsCredentials { url: String, error: Option<String> },
|
||||||
Diverged { remote: String, branch: String },
|
|
||||||
UncommittedChanges,
|
|
||||||
}
|
|
||||||
|
|
||||||
fn has_uncommitted_changes(dir: &Path) -> Result<bool> {
|
|
||||||
let repo = open_repo(dir)?;
|
|
||||||
let mut opts = git2::StatusOptions::new();
|
|
||||||
opts.include_ignored(false).include_untracked(false);
|
|
||||||
let statuses = repo.statuses(Some(&mut opts))?;
|
|
||||||
Ok(statuses.iter().any(|e| e.status() != git2::Status::CURRENT))
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn git_pull(dir: &Path) -> Result<PullResult> {
|
pub async fn git_pull(dir: &Path) -> Result<PullResult> {
|
||||||
if has_uncommitted_changes(dir)? {
|
|
||||||
return Ok(PullResult::UncommittedChanges);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Extract all git2 data before any await points (git2 types are not Send)
|
// Extract all git2 data before any await points (git2 types are not Send)
|
||||||
let (branch_name, remote_name, remote_url) = {
|
let (branch_name, remote_name, remote_url) = {
|
||||||
let repo = open_repo(dir)?;
|
let repo = open_repo(dir)?;
|
||||||
@@ -70,13 +56,6 @@ pub async fn git_pull(dir: &Path) -> Result<PullResult> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if !out.status.success() {
|
if !out.status.success() {
|
||||||
let combined_lower = combined.to_lowercase();
|
|
||||||
if combined_lower.contains("cannot fast-forward")
|
|
||||||
|| combined_lower.contains("not possible to fast-forward")
|
|
||||||
|| combined_lower.contains("diverged")
|
|
||||||
{
|
|
||||||
return Ok(PullResult::Diverged { remote: remote_name, branch: branch_name });
|
|
||||||
}
|
|
||||||
return Err(GenericError(format!("Failed to pull {combined}")));
|
return Err(GenericError(format!("Failed to pull {combined}")));
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -87,65 +66,6 @@ pub async fn git_pull(dir: &Path) -> Result<PullResult> {
|
|||||||
Ok(PullResult::Success { message: format!("Pulled from {}/{}", remote_name, branch_name) })
|
Ok(PullResult::Success { message: format!("Pulled from {}/{}", remote_name, branch_name) })
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn git_pull_force_reset(dir: &Path, remote: &str, branch: &str) -> Result<PullResult> {
|
|
||||||
// Step 1: fetch the remote
|
|
||||||
let fetch_out = new_binary_command(dir)
|
|
||||||
.await?
|
|
||||||
.args(["fetch", remote])
|
|
||||||
.env("GIT_TERMINAL_PROMPT", "0")
|
|
||||||
.output()
|
|
||||||
.await
|
|
||||||
.map_err(|e| GenericError(format!("failed to run git fetch: {e}")))?;
|
|
||||||
|
|
||||||
if !fetch_out.status.success() {
|
|
||||||
let stderr = String::from_utf8_lossy(&fetch_out.stderr);
|
|
||||||
return Err(GenericError(format!("Failed to fetch: {stderr}")));
|
|
||||||
}
|
|
||||||
|
|
||||||
// Step 2: reset --hard to remote/branch
|
|
||||||
let ref_name = format!("{}/{}", remote, branch);
|
|
||||||
let reset_out = new_binary_command(dir)
|
|
||||||
.await?
|
|
||||||
.args(["reset", "--hard", &ref_name])
|
|
||||||
.output()
|
|
||||||
.await
|
|
||||||
.map_err(|e| GenericError(format!("failed to run git reset: {e}")))?;
|
|
||||||
|
|
||||||
if !reset_out.status.success() {
|
|
||||||
let stderr = String::from_utf8_lossy(&reset_out.stderr);
|
|
||||||
return Err(GenericError(format!("Failed to reset: {}", stderr.trim())));
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PullResult::Success { message: format!("Reset to {}/{}", remote, branch) })
|
|
||||||
}
|
|
||||||
|
|
||||||
pub async fn git_pull_merge(dir: &Path, remote: &str, branch: &str) -> Result<PullResult> {
|
|
||||||
let out = new_binary_command(dir)
|
|
||||||
.await?
|
|
||||||
.args(["pull", "--no-rebase", remote, branch])
|
|
||||||
.env("GIT_TERMINAL_PROMPT", "0")
|
|
||||||
.output()
|
|
||||||
.await
|
|
||||||
.map_err(|e| GenericError(format!("failed to run git pull --no-rebase: {e}")))?;
|
|
||||||
|
|
||||||
let stdout = String::from_utf8_lossy(&out.stdout);
|
|
||||||
let stderr = String::from_utf8_lossy(&out.stderr);
|
|
||||||
let combined = format!("{}{}", stdout, stderr);
|
|
||||||
|
|
||||||
info!("Pull merge status={} {combined}", out.status);
|
|
||||||
|
|
||||||
if !out.status.success() {
|
|
||||||
if combined.to_lowercase().contains("conflict") {
|
|
||||||
return Err(GenericError(
|
|
||||||
"Merge conflicts detected. Please resolve them manually.".to_string(),
|
|
||||||
));
|
|
||||||
}
|
|
||||||
return Err(GenericError(format!("Failed to merge pull: {}", combined.trim())));
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(PullResult::Success { message: format!("Merged from {}/{}", remote, branch) })
|
|
||||||
}
|
|
||||||
|
|
||||||
// pub(crate) fn git_pull_old(dir: &Path) -> Result<PullResult> {
|
// pub(crate) fn git_pull_old(dir: &Path) -> Result<PullResult> {
|
||||||
// let repo = open_repo(dir)?;
|
// let repo = open_repo(dir)?;
|
||||||
//
|
//
|
||||||
|
|||||||
@@ -1,20 +0,0 @@
|
|||||||
use crate::binary::new_binary_command;
|
|
||||||
use crate::error::Error::GenericError;
|
|
||||||
use crate::error::Result;
|
|
||||||
use std::path::Path;
|
|
||||||
|
|
||||||
pub async fn git_reset_changes(dir: &Path) -> Result<()> {
|
|
||||||
let out = new_binary_command(dir)
|
|
||||||
.await?
|
|
||||||
.args(["reset", "--hard", "HEAD"])
|
|
||||||
.output()
|
|
||||||
.await
|
|
||||||
.map_err(|e| GenericError(format!("failed to run git reset: {e}")))?;
|
|
||||||
|
|
||||||
if !out.status.success() {
|
|
||||||
let stderr = String::from_utf8_lossy(&out.stderr);
|
|
||||||
return Err(GenericError(format!("Failed to reset: {}", stderr.trim())));
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
@@ -18,8 +18,6 @@ pub struct GitStatusSummary {
|
|||||||
pub origins: Vec<String>,
|
pub origins: Vec<String>,
|
||||||
pub local_branches: Vec<String>,
|
pub local_branches: Vec<String>,
|
||||||
pub remote_branches: Vec<String>,
|
pub remote_branches: Vec<String>,
|
||||||
pub ahead: u32,
|
|
||||||
pub behind: u32,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, TS)]
|
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, TS)]
|
||||||
@@ -162,18 +160,6 @@ pub fn git_status(dir: &Path) -> crate::error::Result<GitStatusSummary> {
|
|||||||
let local_branches = local_branch_names(&repo)?;
|
let local_branches = local_branch_names(&repo)?;
|
||||||
let remote_branches = remote_branch_names(&repo)?;
|
let remote_branches = remote_branch_names(&repo)?;
|
||||||
|
|
||||||
// Compute ahead/behind relative to remote tracking branch
|
|
||||||
let (ahead, behind) = (|| -> Option<(usize, usize)> {
|
|
||||||
let head = repo.head().ok()?;
|
|
||||||
let local_oid = head.target()?;
|
|
||||||
let branch_name = head.shorthand()?;
|
|
||||||
let upstream_ref =
|
|
||||||
repo.find_branch(&format!("origin/{branch_name}"), git2::BranchType::Remote).ok()?;
|
|
||||||
let upstream_oid = upstream_ref.get().target()?;
|
|
||||||
repo.graph_ahead_behind(local_oid, upstream_oid).ok()
|
|
||||||
})()
|
|
||||||
.unwrap_or((0, 0));
|
|
||||||
|
|
||||||
Ok(GitStatusSummary {
|
Ok(GitStatusSummary {
|
||||||
entries,
|
entries,
|
||||||
origins,
|
origins,
|
||||||
@@ -182,7 +168,5 @@ pub fn git_status(dir: &Path) -> crate::error::Result<GitStatusSummary> {
|
|||||||
head_ref_shorthand,
|
head_ref_shorthand,
|
||||||
local_branches,
|
local_branches,
|
||||||
remote_branches,
|
remote_branches,
|
||||||
ahead: ahead as u32,
|
|
||||||
behind: behind as u32,
|
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ publish = false
|
|||||||
async-compression = { version = "0.4", features = ["tokio", "gzip", "deflate", "brotli", "zstd"] }
|
async-compression = { version = "0.4", features = ["tokio", "gzip", "deflate", "brotli", "zstd"] }
|
||||||
async-trait = "0.1"
|
async-trait = "0.1"
|
||||||
brotli = "7"
|
brotli = "7"
|
||||||
bytes = "1.11.1"
|
bytes = "1.5.0"
|
||||||
cookie = "0.18.1"
|
cookie = "0.18.1"
|
||||||
flate2 = "1"
|
flate2 = "1"
|
||||||
futures-util = "0.3"
|
futures-util = "0.3"
|
||||||
|
|||||||
@@ -168,7 +168,6 @@ impl<S: HttpSender> HttpTransaction<S> {
|
|||||||
response.drain().await?;
|
response.drain().await?;
|
||||||
|
|
||||||
// Update the request URL
|
// Update the request URL
|
||||||
let previous_url = current_url.clone();
|
|
||||||
current_url = if location.starts_with("http://") || location.starts_with("https://") {
|
current_url = if location.starts_with("http://") || location.starts_with("https://") {
|
||||||
// Absolute URL
|
// Absolute URL
|
||||||
location
|
location
|
||||||
@@ -182,8 +181,6 @@ impl<S: HttpSender> HttpTransaction<S> {
|
|||||||
format!("{}/{}", base_path, location)
|
format!("{}/{}", base_path, location)
|
||||||
};
|
};
|
||||||
|
|
||||||
Self::remove_sensitive_headers(&mut current_headers, &previous_url, ¤t_url);
|
|
||||||
|
|
||||||
// Determine redirect behavior based on status code and method
|
// Determine redirect behavior based on status code and method
|
||||||
let behavior = if status == 303 {
|
let behavior = if status == 303 {
|
||||||
// 303 See Other always changes to GET
|
// 303 See Other always changes to GET
|
||||||
@@ -223,33 +220,6 @@ impl<S: HttpSender> HttpTransaction<S> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Remove sensitive headers when redirecting to a different host.
|
|
||||||
/// This matches reqwest's `remove_sensitive_headers()` behavior and prevents
|
|
||||||
/// credentials from being forwarded to third-party servers (e.g., an
|
|
||||||
/// Authorization header sent from an API redirect to an S3 bucket).
|
|
||||||
fn remove_sensitive_headers(
|
|
||||||
headers: &mut Vec<(String, String)>,
|
|
||||||
previous_url: &str,
|
|
||||||
next_url: &str,
|
|
||||||
) {
|
|
||||||
let previous_host = Url::parse(previous_url).ok().and_then(|u| {
|
|
||||||
u.host_str().map(|h| format!("{}:{}", h, u.port_or_known_default().unwrap_or(0)))
|
|
||||||
});
|
|
||||||
let next_host = Url::parse(next_url).ok().and_then(|u| {
|
|
||||||
u.host_str().map(|h| format!("{}:{}", h, u.port_or_known_default().unwrap_or(0)))
|
|
||||||
});
|
|
||||||
if previous_host != next_host {
|
|
||||||
headers.retain(|h| {
|
|
||||||
let name_lower = h.0.to_lowercase();
|
|
||||||
name_lower != "authorization"
|
|
||||||
&& name_lower != "cookie"
|
|
||||||
&& name_lower != "cookie2"
|
|
||||||
&& name_lower != "proxy-authorization"
|
|
||||||
&& name_lower != "www-authenticate"
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Check if a status code indicates a redirect
|
/// Check if a status code indicates a redirect
|
||||||
fn is_redirect(status: u16) -> bool {
|
fn is_redirect(status: u16) -> bool {
|
||||||
matches!(status, 301 | 302 | 303 | 307 | 308)
|
matches!(status, 301 | 302 | 303 | 307 | 308)
|
||||||
@@ -299,20 +269,9 @@ mod tests {
|
|||||||
use tokio::io::AsyncRead;
|
use tokio::io::AsyncRead;
|
||||||
use tokio::sync::Mutex;
|
use tokio::sync::Mutex;
|
||||||
|
|
||||||
/// Captured request metadata for test assertions
|
|
||||||
#[derive(Debug, Clone)]
|
|
||||||
#[allow(dead_code)]
|
|
||||||
struct CapturedRequest {
|
|
||||||
url: String,
|
|
||||||
method: String,
|
|
||||||
headers: Vec<(String, String)>,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Mock sender for testing
|
/// Mock sender for testing
|
||||||
struct MockSender {
|
struct MockSender {
|
||||||
responses: Arc<Mutex<Vec<MockResponse>>>,
|
responses: Arc<Mutex<Vec<MockResponse>>>,
|
||||||
/// Captured requests for assertions
|
|
||||||
captured_requests: Arc<Mutex<Vec<CapturedRequest>>>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
struct MockResponse {
|
struct MockResponse {
|
||||||
@@ -323,10 +282,7 @@ mod tests {
|
|||||||
|
|
||||||
impl MockSender {
|
impl MockSender {
|
||||||
fn new(responses: Vec<MockResponse>) -> Self {
|
fn new(responses: Vec<MockResponse>) -> Self {
|
||||||
Self {
|
Self { responses: Arc::new(Mutex::new(responses)) }
|
||||||
responses: Arc::new(Mutex::new(responses)),
|
|
||||||
captured_requests: Arc::new(Mutex::new(Vec::new())),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -334,16 +290,9 @@ mod tests {
|
|||||||
impl HttpSender for MockSender {
|
impl HttpSender for MockSender {
|
||||||
async fn send(
|
async fn send(
|
||||||
&self,
|
&self,
|
||||||
request: SendableHttpRequest,
|
_request: SendableHttpRequest,
|
||||||
_event_tx: mpsc::Sender<HttpResponseEvent>,
|
_event_tx: mpsc::Sender<HttpResponseEvent>,
|
||||||
) -> Result<HttpResponse> {
|
) -> Result<HttpResponse> {
|
||||||
// Capture the request metadata for later assertions
|
|
||||||
self.captured_requests.lock().await.push(CapturedRequest {
|
|
||||||
url: request.url.clone(),
|
|
||||||
method: request.method.clone(),
|
|
||||||
headers: request.headers.clone(),
|
|
||||||
});
|
|
||||||
|
|
||||||
let mut responses = self.responses.lock().await;
|
let mut responses = self.responses.lock().await;
|
||||||
if responses.is_empty() {
|
if responses.is_empty() {
|
||||||
Err(crate::error::Error::RequestError("No more mock responses".to_string()))
|
Err(crate::error::Error::RequestError("No more mock responses".to_string()))
|
||||||
@@ -777,116 +726,4 @@ mod tests {
|
|||||||
assert!(result.is_ok());
|
assert!(result.is_ok());
|
||||||
assert_eq!(request_count.load(Ordering::SeqCst), 2);
|
assert_eq!(request_count.load(Ordering::SeqCst), 2);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn test_cross_origin_redirect_strips_auth_headers() {
|
|
||||||
// Redirect from api.example.com -> s3.amazonaws.com should strip Authorization
|
|
||||||
let responses = vec![
|
|
||||||
MockResponse {
|
|
||||||
status: 302,
|
|
||||||
headers: vec![(
|
|
||||||
"Location".to_string(),
|
|
||||||
"https://s3.amazonaws.com/bucket/file.pdf".to_string(),
|
|
||||||
)],
|
|
||||||
body: vec![],
|
|
||||||
},
|
|
||||||
MockResponse { status: 200, headers: Vec::new(), body: b"PDF content".to_vec() },
|
|
||||||
];
|
|
||||||
|
|
||||||
let sender = MockSender::new(responses);
|
|
||||||
let captured = sender.captured_requests.clone();
|
|
||||||
let transaction = HttpTransaction::new(sender);
|
|
||||||
|
|
||||||
let request = SendableHttpRequest {
|
|
||||||
url: "https://api.example.com/download".to_string(),
|
|
||||||
method: "GET".to_string(),
|
|
||||||
headers: vec![
|
|
||||||
("Authorization".to_string(), "Basic dXNlcjpwYXNz".to_string()),
|
|
||||||
("Accept".to_string(), "application/pdf".to_string()),
|
|
||||||
],
|
|
||||||
options: crate::types::SendableHttpRequestOptions {
|
|
||||||
follow_redirects: true,
|
|
||||||
..Default::default()
|
|
||||||
},
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
let (_tx, rx) = tokio::sync::watch::channel(false);
|
|
||||||
let (event_tx, _event_rx) = mpsc::channel(100);
|
|
||||||
let result = transaction.execute_with_cancellation(request, rx, event_tx).await.unwrap();
|
|
||||||
assert_eq!(result.status, 200);
|
|
||||||
|
|
||||||
let requests = captured.lock().await;
|
|
||||||
assert_eq!(requests.len(), 2);
|
|
||||||
|
|
||||||
// First request should have the Authorization header
|
|
||||||
assert!(
|
|
||||||
requests[0].headers.iter().any(|(k, _)| k.eq_ignore_ascii_case("authorization")),
|
|
||||||
"First request should have Authorization header"
|
|
||||||
);
|
|
||||||
|
|
||||||
// Second request (to different host) should NOT have the Authorization header
|
|
||||||
assert!(
|
|
||||||
!requests[1].headers.iter().any(|(k, _)| k.eq_ignore_ascii_case("authorization")),
|
|
||||||
"Redirected request to different host should NOT have Authorization header"
|
|
||||||
);
|
|
||||||
|
|
||||||
// Non-sensitive headers should still be present
|
|
||||||
assert!(
|
|
||||||
requests[1].headers.iter().any(|(k, _)| k.eq_ignore_ascii_case("accept")),
|
|
||||||
"Non-sensitive headers should be preserved across cross-origin redirects"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn test_same_origin_redirect_preserves_auth_headers() {
|
|
||||||
// Redirect within the same host should keep Authorization
|
|
||||||
let responses = vec![
|
|
||||||
MockResponse {
|
|
||||||
status: 302,
|
|
||||||
headers: vec![(
|
|
||||||
"Location".to_string(),
|
|
||||||
"https://api.example.com/v2/download".to_string(),
|
|
||||||
)],
|
|
||||||
body: vec![],
|
|
||||||
},
|
|
||||||
MockResponse { status: 200, headers: Vec::new(), body: b"OK".to_vec() },
|
|
||||||
];
|
|
||||||
|
|
||||||
let sender = MockSender::new(responses);
|
|
||||||
let captured = sender.captured_requests.clone();
|
|
||||||
let transaction = HttpTransaction::new(sender);
|
|
||||||
|
|
||||||
let request = SendableHttpRequest {
|
|
||||||
url: "https://api.example.com/v1/download".to_string(),
|
|
||||||
method: "GET".to_string(),
|
|
||||||
headers: vec![
|
|
||||||
("Authorization".to_string(), "Bearer token123".to_string()),
|
|
||||||
("Accept".to_string(), "application/json".to_string()),
|
|
||||||
],
|
|
||||||
options: crate::types::SendableHttpRequestOptions {
|
|
||||||
follow_redirects: true,
|
|
||||||
..Default::default()
|
|
||||||
},
|
|
||||||
..Default::default()
|
|
||||||
};
|
|
||||||
|
|
||||||
let (_tx, rx) = tokio::sync::watch::channel(false);
|
|
||||||
let (event_tx, _event_rx) = mpsc::channel(100);
|
|
||||||
let result = transaction.execute_with_cancellation(request, rx, event_tx).await.unwrap();
|
|
||||||
assert_eq!(result.status, 200);
|
|
||||||
|
|
||||||
let requests = captured.lock().await;
|
|
||||||
assert_eq!(requests.len(), 2);
|
|
||||||
|
|
||||||
// Both requests should have the Authorization header (same host)
|
|
||||||
assert!(
|
|
||||||
requests[0].headers.iter().any(|(k, _)| k.eq_ignore_ascii_case("authorization")),
|
|
||||||
"First request should have Authorization header"
|
|
||||||
);
|
|
||||||
assert!(
|
|
||||||
requests[1].headers.iter().any(|(k, _)| k.eq_ignore_ascii_case("authorization")),
|
|
||||||
"Redirected request to same host should preserve Authorization header"
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
+4
-10
@@ -66,9 +66,7 @@ export type DeleteModelRequest = { model: string, id: string, };
|
|||||||
|
|
||||||
export type DeleteModelResponse = { model: AnyModel, };
|
export type DeleteModelResponse = { model: AnyModel, };
|
||||||
|
|
||||||
export type DialogSize = "sm" | "md" | "lg" | "full" | "dynamic";
|
export type EditorLanguage = "text" | "javascript" | "json" | "html" | "xml" | "graphql" | "markdown";
|
||||||
|
|
||||||
export type EditorLanguage = "text" | "javascript" | "json" | "html" | "xml" | "graphql" | "markdown" | "c" | "clojure" | "csharp" | "go" | "http" | "java" | "kotlin" | "objective_c" | "ocaml" | "php" | "powershell" | "python" | "r" | "ruby" | "shell" | "swift";
|
|
||||||
|
|
||||||
export type EmptyPayload = {};
|
export type EmptyPayload = {};
|
||||||
|
|
||||||
@@ -174,11 +172,7 @@ hideGutter?: boolean,
|
|||||||
/**
|
/**
|
||||||
* Language for syntax highlighting
|
* Language for syntax highlighting
|
||||||
*/
|
*/
|
||||||
language?: EditorLanguage, readOnly?: boolean,
|
language?: EditorLanguage, readOnly?: boolean, completionOptions?: Array<GenericCompletionOption>,
|
||||||
/**
|
|
||||||
* Fixed number of visible rows
|
|
||||||
*/
|
|
||||||
rows?: number, completionOptions?: Array<GenericCompletionOption>,
|
|
||||||
/**
|
/**
|
||||||
* The name of the input. The value will be stored at this object attribute in the resulting data
|
* The name of the input. The value will be stored at this object attribute in the resulting data
|
||||||
*/
|
*/
|
||||||
@@ -482,9 +476,9 @@ label: string, title?: string, size?: WindowSize, dataDirKey?: string, };
|
|||||||
|
|
||||||
export type PluginContext = { id: string, label: string | null, workspaceId: string | null, };
|
export type PluginContext = { id: string, label: string | null, workspaceId: string | null, };
|
||||||
|
|
||||||
export type PromptFormRequest = { id: string, title: string, description?: string, inputs: Array<FormInput>, confirmText?: string, cancelText?: string, size?: DialogSize, };
|
export type PromptFormRequest = { id: string, title: string, description?: string, inputs: Array<FormInput>, confirmText?: string, cancelText?: string, };
|
||||||
|
|
||||||
export type PromptFormResponse = { values: { [key in string]?: JsonPrimitive } | null, done?: boolean, };
|
export type PromptFormResponse = { values: { [key in string]?: JsonPrimitive } | null, };
|
||||||
|
|
||||||
export type PromptTextRequest = { id: string, title: string, label: string, description?: string, defaultValue?: string, placeholder?: string,
|
export type PromptTextRequest = { id: string, title: string, label: string, description?: string, defaultValue?: string, placeholder?: string,
|
||||||
/**
|
/**
|
||||||
|
|||||||
+1
-1
@@ -49,7 +49,7 @@ export type HttpResponseEvent = { model: "http_response_event", id: string, crea
|
|||||||
* This mirrors `yaak_http::sender::HttpResponseEvent` but with serde support.
|
* This mirrors `yaak_http::sender::HttpResponseEvent` but with serde support.
|
||||||
* The `From` impl is in yaak-http to avoid circular dependencies.
|
* The `From` impl is in yaak-http to avoid circular dependencies.
|
||||||
*/
|
*/
|
||||||
export type HttpResponseEventData = { "type": "setting", name: string, value: string, } | { "type": "info", message: string, } | { "type": "redirect", url: string, status: number, behavior: string, } | { "type": "send_url", method: string, scheme: string, username: string, password: string, host: string, port: number, path: string, query: string, fragment: string, } | { "type": "receive_url", version: string, status: string, } | { "type": "header_up", name: string, value: string, } | { "type": "header_down", name: string, value: string, } | { "type": "chunk_sent", bytes: number, } | { "type": "chunk_received", bytes: number, } | { "type": "dns_resolved", hostname: string, addresses: Array<string>, duration: bigint, overridden: boolean, };
|
export type HttpResponseEventData = { "type": "setting", name: string, value: string, } | { "type": "info", message: string, } | { "type": "redirect", url: string, status: number, behavior: string, } | { "type": "send_url", method: string, path: string, } | { "type": "receive_url", version: string, status: string, } | { "type": "header_up", name: string, value: string, } | { "type": "header_down", name: string, value: string, } | { "type": "chunk_sent", bytes: number, } | { "type": "chunk_received", bytes: number, } | { "type": "dns_resolved", hostname: string, addresses: Array<string>, duration: bigint, overridden: boolean, };
|
||||||
|
|
||||||
export type HttpResponseHeader = { name: string, value: string, };
|
export type HttpResponseHeader = { name: string, value: string, };
|
||||||
|
|
||||||
|
|||||||
@@ -587,19 +587,6 @@ pub struct PromptFormRequest {
|
|||||||
pub confirm_text: Option<String>,
|
pub confirm_text: Option<String>,
|
||||||
#[ts(optional)]
|
#[ts(optional)]
|
||||||
pub cancel_text: Option<String>,
|
pub cancel_text: Option<String>,
|
||||||
#[ts(optional)]
|
|
||||||
pub size: Option<DialogSize>,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Clone, Serialize, Deserialize, TS)]
|
|
||||||
#[serde(rename_all = "snake_case")]
|
|
||||||
#[ts(export, export_to = "gen_events.ts")]
|
|
||||||
pub enum DialogSize {
|
|
||||||
Sm,
|
|
||||||
Md,
|
|
||||||
Lg,
|
|
||||||
Full,
|
|
||||||
Dynamic,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
|
#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
|
||||||
@@ -607,8 +594,6 @@ pub enum DialogSize {
|
|||||||
#[ts(export, export_to = "gen_events.ts")]
|
#[ts(export, export_to = "gen_events.ts")]
|
||||||
pub struct PromptFormResponse {
|
pub struct PromptFormResponse {
|
||||||
pub values: Option<HashMap<String, JsonPrimitive>>,
|
pub values: Option<HashMap<String, JsonPrimitive>>,
|
||||||
#[ts(optional)]
|
|
||||||
pub done: Option<bool>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
|
#[derive(Debug, Clone, Default, Serialize, Deserialize, TS)]
|
||||||
@@ -951,22 +936,6 @@ pub enum EditorLanguage {
|
|||||||
Xml,
|
Xml,
|
||||||
Graphql,
|
Graphql,
|
||||||
Markdown,
|
Markdown,
|
||||||
C,
|
|
||||||
Clojure,
|
|
||||||
Csharp,
|
|
||||||
Go,
|
|
||||||
Http,
|
|
||||||
Java,
|
|
||||||
Kotlin,
|
|
||||||
ObjectiveC,
|
|
||||||
Ocaml,
|
|
||||||
Php,
|
|
||||||
Powershell,
|
|
||||||
Python,
|
|
||||||
R,
|
|
||||||
Ruby,
|
|
||||||
Shell,
|
|
||||||
Swift,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Default for EditorLanguage {
|
impl Default for EditorLanguage {
|
||||||
@@ -997,10 +966,6 @@ pub struct FormInputEditor {
|
|||||||
#[ts(optional)]
|
#[ts(optional)]
|
||||||
pub read_only: Option<bool>,
|
pub read_only: Option<bool>,
|
||||||
|
|
||||||
/// Fixed number of visible rows
|
|
||||||
#[ts(optional)]
|
|
||||||
pub rows: Option<i32>,
|
|
||||||
|
|
||||||
#[ts(optional)]
|
#[ts(optional)]
|
||||||
pub completion_options: Option<Vec<GenericCompletionOption>>,
|
pub completion_options: Option<Vec<GenericCompletionOption>>,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -31,7 +31,7 @@ use std::time::Duration;
|
|||||||
use tokio::fs::read_dir;
|
use tokio::fs::read_dir;
|
||||||
use tokio::net::TcpListener;
|
use tokio::net::TcpListener;
|
||||||
use tokio::sync::mpsc::error::TrySendError;
|
use tokio::sync::mpsc::error::TrySendError;
|
||||||
use tokio::sync::{Mutex, mpsc, oneshot};
|
use tokio::sync::{Mutex, mpsc};
|
||||||
use tokio::time::{Instant, timeout};
|
use tokio::time::{Instant, timeout};
|
||||||
use yaak_models::models::Plugin;
|
use yaak_models::models::Plugin;
|
||||||
use yaak_models::util::generate_id;
|
use yaak_models::util::generate_id;
|
||||||
@@ -43,7 +43,6 @@ pub struct PluginManager {
|
|||||||
subscribers: Arc<Mutex<HashMap<String, mpsc::Sender<InternalEvent>>>>,
|
subscribers: Arc<Mutex<HashMap<String, mpsc::Sender<InternalEvent>>>>,
|
||||||
plugin_handles: Arc<Mutex<Vec<PluginHandle>>>,
|
plugin_handles: Arc<Mutex<Vec<PluginHandle>>>,
|
||||||
kill_tx: tokio::sync::watch::Sender<bool>,
|
kill_tx: tokio::sync::watch::Sender<bool>,
|
||||||
killed_rx: Arc<Mutex<Option<oneshot::Receiver<()>>>>,
|
|
||||||
ws_service: Arc<PluginRuntimeServerWebsocket>,
|
ws_service: Arc<PluginRuntimeServerWebsocket>,
|
||||||
vendored_plugin_dir: PathBuf,
|
vendored_plugin_dir: PathBuf,
|
||||||
pub(crate) installed_plugin_dir: PathBuf,
|
pub(crate) installed_plugin_dir: PathBuf,
|
||||||
@@ -71,7 +70,6 @@ impl PluginManager {
|
|||||||
) -> PluginManager {
|
) -> PluginManager {
|
||||||
let (events_tx, mut events_rx) = mpsc::channel(2048);
|
let (events_tx, mut events_rx) = mpsc::channel(2048);
|
||||||
let (kill_server_tx, kill_server_rx) = tokio::sync::watch::channel(false);
|
let (kill_server_tx, kill_server_rx) = tokio::sync::watch::channel(false);
|
||||||
let (killed_tx, killed_rx) = oneshot::channel();
|
|
||||||
|
|
||||||
let (client_disconnect_tx, mut client_disconnect_rx) = mpsc::channel(128);
|
let (client_disconnect_tx, mut client_disconnect_rx) = mpsc::channel(128);
|
||||||
let (client_connect_tx, mut client_connect_rx) = tokio::sync::watch::channel(false);
|
let (client_connect_tx, mut client_connect_rx) = tokio::sync::watch::channel(false);
|
||||||
@@ -83,7 +81,6 @@ impl PluginManager {
|
|||||||
subscribers: Default::default(),
|
subscribers: Default::default(),
|
||||||
ws_service: Arc::new(ws_service.clone()),
|
ws_service: Arc::new(ws_service.clone()),
|
||||||
kill_tx: kill_server_tx,
|
kill_tx: kill_server_tx,
|
||||||
killed_rx: Arc::new(Mutex::new(Some(killed_rx))),
|
|
||||||
vendored_plugin_dir,
|
vendored_plugin_dir,
|
||||||
installed_plugin_dir,
|
installed_plugin_dir,
|
||||||
dev_mode,
|
dev_mode,
|
||||||
@@ -144,15 +141,9 @@ impl PluginManager {
|
|||||||
});
|
});
|
||||||
|
|
||||||
// 2. Start Node.js runtime
|
// 2. Start Node.js runtime
|
||||||
start_nodejs_plugin_runtime(
|
start_nodejs_plugin_runtime(&node_bin_path, &plugin_runtime_main, addr, &kill_server_rx)
|
||||||
&node_bin_path,
|
.await
|
||||||
&plugin_runtime_main,
|
.unwrap();
|
||||||
addr,
|
|
||||||
&kill_server_rx,
|
|
||||||
killed_tx,
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.unwrap();
|
|
||||||
info!("Waiting for plugins to initialize");
|
info!("Waiting for plugins to initialize");
|
||||||
init_plugins_task.await.unwrap();
|
init_plugins_task.await.unwrap();
|
||||||
|
|
||||||
@@ -305,15 +296,8 @@ impl PluginManager {
|
|||||||
pub async fn terminate(&self) {
|
pub async fn terminate(&self) {
|
||||||
self.kill_tx.send_replace(true);
|
self.kill_tx.send_replace(true);
|
||||||
|
|
||||||
// Wait for the plugin runtime process to actually exit
|
// Give it a bit of time to kill
|
||||||
let killed_rx = self.killed_rx.lock().await.take();
|
tokio::time::sleep(Duration::from_millis(500)).await;
|
||||||
if let Some(rx) = killed_rx {
|
|
||||||
if timeout(Duration::from_secs(5), rx).await.is_err() {
|
|
||||||
warn!("Timed out waiting for plugin runtime to exit");
|
|
||||||
} else {
|
|
||||||
info!("Plugin runtime exited")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn reply(
|
pub async fn reply(
|
||||||
|
|||||||
@@ -4,7 +4,6 @@ use std::net::SocketAddr;
|
|||||||
use std::path::Path;
|
use std::path::Path;
|
||||||
use std::process::Stdio;
|
use std::process::Stdio;
|
||||||
use tokio::io::{AsyncBufReadExt, BufReader};
|
use tokio::io::{AsyncBufReadExt, BufReader};
|
||||||
use tokio::sync::oneshot;
|
|
||||||
use tokio::sync::watch::Receiver;
|
use tokio::sync::watch::Receiver;
|
||||||
use yaak_common::command::new_xplatform_command;
|
use yaak_common::command::new_xplatform_command;
|
||||||
|
|
||||||
@@ -20,7 +19,6 @@ pub async fn start_nodejs_plugin_runtime(
|
|||||||
plugin_runtime_main: &Path,
|
plugin_runtime_main: &Path,
|
||||||
addr: SocketAddr,
|
addr: SocketAddr,
|
||||||
kill_rx: &Receiver<bool>,
|
kill_rx: &Receiver<bool>,
|
||||||
killed_tx: oneshot::Sender<()>,
|
|
||||||
) -> Result<()> {
|
) -> Result<()> {
|
||||||
// HACK: Remove UNC prefix for Windows paths to pass to sidecar
|
// HACK: Remove UNC prefix for Windows paths to pass to sidecar
|
||||||
let plugin_runtime_main_str =
|
let plugin_runtime_main_str =
|
||||||
@@ -74,7 +72,6 @@ pub async fn start_nodejs_plugin_runtime(
|
|||||||
warn!("Failed to kill plugin runtime: {e}");
|
warn!("Failed to kill plugin runtime: {e}");
|
||||||
}
|
}
|
||||||
info!("Killed plugin runtime");
|
info!("Killed plugin runtime");
|
||||||
let _ = killed_tx.send(());
|
|
||||||
});
|
});
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
+41
-1
@@ -1,5 +1,31 @@
|
|||||||
import { invoke } from '@tauri-apps/api/core';
|
import { invoke } from '@tauri-apps/api/core';
|
||||||
import { WebsocketConnection } from '@yaakapp-internal/models';
|
import { WebsocketConnection, WebsocketEvent, WebsocketRequest } from '@yaakapp-internal/models';
|
||||||
|
|
||||||
|
export function upsertWebsocketRequest(
|
||||||
|
request: WebsocketRequest | Partial<Omit<WebsocketRequest, 'id'>>,
|
||||||
|
) {
|
||||||
|
return invoke('cmd_ws_upsert_request', {
|
||||||
|
request,
|
||||||
|
}) as Promise<WebsocketRequest>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function duplicateWebsocketRequest(requestId: string) {
|
||||||
|
return invoke('cmd_ws_duplicate_request', {
|
||||||
|
requestId,
|
||||||
|
}) as Promise<WebsocketRequest>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function deleteWebsocketRequest(requestId: string) {
|
||||||
|
return invoke('cmd_ws_delete_request', {
|
||||||
|
requestId,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
export function deleteWebsocketConnection(connectionId: string) {
|
||||||
|
return invoke('cmd_ws_delete_connection', {
|
||||||
|
connectionId,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
export function deleteWebsocketConnections(requestId: string) {
|
export function deleteWebsocketConnections(requestId: string) {
|
||||||
return invoke('cmd_ws_delete_connections', {
|
return invoke('cmd_ws_delete_connections', {
|
||||||
@@ -7,6 +33,20 @@ export function deleteWebsocketConnections(requestId: string) {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export function listWebsocketRequests({ workspaceId }: { workspaceId: string }) {
|
||||||
|
return invoke('cmd_ws_list_requests', { workspaceId }) as Promise<WebsocketRequest[]>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function listWebsocketEvents({ connectionId }: { connectionId: string }) {
|
||||||
|
return invoke('cmd_ws_list_events', { connectionId }) as Promise<WebsocketEvent[]>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export function listWebsocketConnections({ workspaceId }: { workspaceId: string }) {
|
||||||
|
return invoke('cmd_ws_list_connections', { workspaceId }) as Promise<
|
||||||
|
WebsocketConnection[]
|
||||||
|
>;
|
||||||
|
}
|
||||||
|
|
||||||
export function connectWebsocket({
|
export function connectWebsocket({
|
||||||
requestId,
|
requestId,
|
||||||
environmentId,
|
environmentId,
|
||||||
|
|||||||
Generated
+15
-221
@@ -13,7 +13,6 @@
|
|||||||
"packages/plugin-runtime-types",
|
"packages/plugin-runtime-types",
|
||||||
"plugins-external/mcp-server",
|
"plugins-external/mcp-server",
|
||||||
"plugins-external/template-function-faker",
|
"plugins-external/template-function-faker",
|
||||||
"plugins-external/httpsnippet",
|
|
||||||
"plugins/action-copy-curl",
|
"plugins/action-copy-curl",
|
||||||
"plugins/action-copy-grpcurl",
|
"plugins/action-copy-grpcurl",
|
||||||
"plugins/action-send-folder",
|
"plugins/action-send-folder",
|
||||||
@@ -63,13 +62,6 @@
|
|||||||
"crates/yaak-ws",
|
"crates/yaak-ws",
|
||||||
"src-web"
|
"src-web"
|
||||||
],
|
],
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/lang-go": "^6.0.1",
|
|
||||||
"@codemirror/lang-java": "^6.0.2",
|
|
||||||
"@codemirror/lang-php": "^6.0.2",
|
|
||||||
"@codemirror/lang-python": "^6.2.1",
|
|
||||||
"@codemirror/legacy-modes": "^6.5.2"
|
|
||||||
},
|
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@biomejs/biome": "^2.3.13",
|
"@biomejs/biome": "^2.3.13",
|
||||||
"@tauri-apps/cli": "^2.9.6",
|
"@tauri-apps/cli": "^2.9.6",
|
||||||
@@ -744,19 +736,6 @@
|
|||||||
"@lezer/css": "^1.1.7"
|
"@lezer/css": "^1.1.7"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@codemirror/lang-go": {
|
|
||||||
"version": "6.0.1",
|
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-go/-/lang-go-6.0.1.tgz",
|
|
||||||
"integrity": "sha512-7fNvbyNylvqCphW9HD6WFnRpcDjr+KXX/FgqXy5H5ZS0eC5edDljukm/yNgYkwTsgp2busdod50AOTIy6Jikfg==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/autocomplete": "^6.0.0",
|
|
||||||
"@codemirror/language": "^6.6.0",
|
|
||||||
"@codemirror/state": "^6.0.0",
|
|
||||||
"@lezer/common": "^1.0.0",
|
|
||||||
"@lezer/go": "^1.0.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@codemirror/lang-html": {
|
"node_modules/@codemirror/lang-html": {
|
||||||
"version": "6.4.11",
|
"version": "6.4.11",
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-html/-/lang-html-6.4.11.tgz",
|
"resolved": "https://registry.npmjs.org/@codemirror/lang-html/-/lang-html-6.4.11.tgz",
|
||||||
@@ -774,16 +753,6 @@
|
|||||||
"@lezer/html": "^1.3.12"
|
"@lezer/html": "^1.3.12"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@codemirror/lang-java": {
|
|
||||||
"version": "6.0.2",
|
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-java/-/lang-java-6.0.2.tgz",
|
|
||||||
"integrity": "sha512-m5Nt1mQ/cznJY7tMfQTJchmrjdjQ71IDs+55d1GAa8DGaB8JXWsVCkVT284C3RTASaY43YknrK2X3hPO/J3MOQ==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/language": "^6.0.0",
|
|
||||||
"@lezer/java": "^1.0.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@codemirror/lang-javascript": {
|
"node_modules/@codemirror/lang-javascript": {
|
||||||
"version": "6.2.4",
|
"version": "6.2.4",
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-javascript/-/lang-javascript-6.2.4.tgz",
|
"resolved": "https://registry.npmjs.org/@codemirror/lang-javascript/-/lang-javascript-6.2.4.tgz",
|
||||||
@@ -824,32 +793,6 @@
|
|||||||
"@lezer/markdown": "^1.0.0"
|
"@lezer/markdown": "^1.0.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@codemirror/lang-php": {
|
|
||||||
"version": "6.0.2",
|
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-php/-/lang-php-6.0.2.tgz",
|
|
||||||
"integrity": "sha512-ZKy2v1n8Fc8oEXj0Th0PUMXzQJ0AIR6TaZU+PbDHExFwdu+guzOA4jmCHS1Nz4vbFezwD7LyBdDnddSJeScMCA==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/lang-html": "^6.0.0",
|
|
||||||
"@codemirror/language": "^6.0.0",
|
|
||||||
"@codemirror/state": "^6.0.0",
|
|
||||||
"@lezer/common": "^1.0.0",
|
|
||||||
"@lezer/php": "^1.0.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@codemirror/lang-python": {
|
|
||||||
"version": "6.2.1",
|
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-python/-/lang-python-6.2.1.tgz",
|
|
||||||
"integrity": "sha512-IRjC8RUBhn9mGR9ywecNhB51yePWCGgvHfY1lWN/Mrp3cKuHr0isDKia+9HnvhiWNnMpbGhWrkhuWOc09exRyw==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/autocomplete": "^6.3.2",
|
|
||||||
"@codemirror/language": "^6.8.0",
|
|
||||||
"@codemirror/state": "^6.0.0",
|
|
||||||
"@lezer/common": "^1.2.1",
|
|
||||||
"@lezer/python": "^1.1.4"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@codemirror/lang-xml": {
|
"node_modules/@codemirror/lang-xml": {
|
||||||
"version": "6.1.0",
|
"version": "6.1.0",
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lang-xml/-/lang-xml-6.1.0.tgz",
|
"resolved": "https://registry.npmjs.org/@codemirror/lang-xml/-/lang-xml-6.1.0.tgz",
|
||||||
@@ -893,15 +836,6 @@
|
|||||||
"style-mod": "^4.0.0"
|
"style-mod": "^4.0.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@codemirror/legacy-modes": {
|
|
||||||
"version": "6.5.2",
|
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/legacy-modes/-/legacy-modes-6.5.2.tgz",
|
|
||||||
"integrity": "sha512-/jJbwSTazlQEDOQw2FJ8LEEKVS72pU0lx6oM54kGpL8t/NJ2Jda3CZ4pcltiKTdqYSRk3ug1B3pil1gsjA6+8Q==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/language": "^6.0.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@codemirror/lint": {
|
"node_modules/@codemirror/lint": {
|
||||||
"version": "6.9.2",
|
"version": "6.9.2",
|
||||||
"resolved": "https://registry.npmjs.org/@codemirror/lint/-/lint-6.9.2.tgz",
|
"resolved": "https://registry.npmjs.org/@codemirror/lint/-/lint-6.9.2.tgz",
|
||||||
@@ -1480,9 +1414,9 @@
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@hono/node-server": {
|
"node_modules/@hono/node-server": {
|
||||||
"version": "1.19.9",
|
"version": "1.19.8",
|
||||||
"resolved": "https://registry.npmjs.org/@hono/node-server/-/node-server-1.19.9.tgz",
|
"resolved": "https://registry.npmjs.org/@hono/node-server/-/node-server-1.19.8.tgz",
|
||||||
"integrity": "sha512-vHL6w3ecZsky+8P5MD+eFfaGTyCeOHUIFYMGpQGbrBTSmNNoxv0if69rEZ5giu36weC5saFuznL411gRX7bJDw==",
|
"integrity": "sha512-0/g2lIOPzX8f3vzW1ggQgvG5mjtFBDBHFAzI5SFAi2DzSqS9luJwqg9T6O/gKYLi+inS7eNxBeIFkkghIPvrMA==",
|
||||||
"license": "MIT",
|
"license": "MIT",
|
||||||
"engines": {
|
"engines": {
|
||||||
"node": ">=18.14.1"
|
"node": ">=18.14.1"
|
||||||
@@ -1636,17 +1570,6 @@
|
|||||||
"lezer-generator": "src/lezer-generator.cjs"
|
"lezer-generator": "src/lezer-generator.cjs"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@lezer/go": {
|
|
||||||
"version": "1.0.1",
|
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/go/-/go-1.0.1.tgz",
|
|
||||||
"integrity": "sha512-xToRsYxwsgJNHTgNdStpcvmbVuKxTapV0dM0wey1geMMRc9aggoVyKgzYp41D2/vVOx+Ii4hmE206kvxIXBVXQ==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@lezer/common": "^1.2.0",
|
|
||||||
"@lezer/highlight": "^1.0.0",
|
|
||||||
"@lezer/lr": "^1.3.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@lezer/highlight": {
|
"node_modules/@lezer/highlight": {
|
||||||
"version": "1.2.3",
|
"version": "1.2.3",
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/highlight/-/highlight-1.2.3.tgz",
|
"resolved": "https://registry.npmjs.org/@lezer/highlight/-/highlight-1.2.3.tgz",
|
||||||
@@ -1667,17 +1590,6 @@
|
|||||||
"@lezer/lr": "^1.0.0"
|
"@lezer/lr": "^1.0.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@lezer/java": {
|
|
||||||
"version": "1.1.3",
|
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/java/-/java-1.1.3.tgz",
|
|
||||||
"integrity": "sha512-yHquUfujwg6Yu4Fd1GNHCvidIvJwi/1Xu2DaKl/pfWIA2c1oXkVvawH3NyXhCaFx4OdlYBVX5wvz2f7Aoa/4Xw==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@lezer/common": "^1.2.0",
|
|
||||||
"@lezer/highlight": "^1.0.0",
|
|
||||||
"@lezer/lr": "^1.0.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@lezer/javascript": {
|
"node_modules/@lezer/javascript": {
|
||||||
"version": "1.5.4",
|
"version": "1.5.4",
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/javascript/-/javascript-1.5.4.tgz",
|
"resolved": "https://registry.npmjs.org/@lezer/javascript/-/javascript-1.5.4.tgz",
|
||||||
@@ -1719,28 +1631,6 @@
|
|||||||
"@lezer/highlight": "^1.0.0"
|
"@lezer/highlight": "^1.0.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@lezer/php": {
|
|
||||||
"version": "1.0.5",
|
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/php/-/php-1.0.5.tgz",
|
|
||||||
"integrity": "sha512-W7asp9DhM6q0W6DYNwIkLSKOvxlXRrif+UXBMxzsJUuqmhE7oVU+gS3THO4S/Puh7Xzgm858UNaFi6dxTP8dJA==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@lezer/common": "^1.2.0",
|
|
||||||
"@lezer/highlight": "^1.0.0",
|
|
||||||
"@lezer/lr": "^1.1.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@lezer/python": {
|
|
||||||
"version": "1.1.18",
|
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/python/-/python-1.1.18.tgz",
|
|
||||||
"integrity": "sha512-31FiUrU7z9+d/ElGQLJFXl+dKOdx0jALlP3KEOsGTex8mvj+SoE1FgItcHWK/axkxCHGUSpqIHt6JAWfWu9Rhg==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"@lezer/common": "^1.2.0",
|
|
||||||
"@lezer/highlight": "^1.0.0",
|
|
||||||
"@lezer/lr": "^1.0.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@lezer/xml": {
|
"node_modules/@lezer/xml": {
|
||||||
"version": "1.0.6",
|
"version": "1.0.6",
|
||||||
"resolved": "https://registry.npmjs.org/@lezer/xml/-/xml-1.0.6.tgz",
|
"resolved": "https://registry.npmjs.org/@lezer/xml/-/xml-1.0.6.tgz",
|
||||||
@@ -1785,12 +1675,12 @@
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@modelcontextprotocol/sdk": {
|
"node_modules/@modelcontextprotocol/sdk": {
|
||||||
"version": "1.26.0",
|
"version": "1.25.2",
|
||||||
"resolved": "https://registry.npmjs.org/@modelcontextprotocol/sdk/-/sdk-1.26.0.tgz",
|
"resolved": "https://registry.npmjs.org/@modelcontextprotocol/sdk/-/sdk-1.25.2.tgz",
|
||||||
"integrity": "sha512-Y5RmPncpiDtTXDbLKswIJzTqu2hyBKxTNsgKqKclDbhIgg1wgtf1fRuvxgTnRfcnxtvvgbIEcqUOzZrJ6iSReg==",
|
"integrity": "sha512-LZFeo4F9M5qOhC/Uc1aQSrBHxMrvxett+9KLHt7OhcExtoiRN9DKgbZffMP/nxjutWDQpfMDfP3nkHI4X9ijww==",
|
||||||
"license": "MIT",
|
"license": "MIT",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@hono/node-server": "^1.19.9",
|
"@hono/node-server": "^1.19.7",
|
||||||
"ajv": "^8.17.1",
|
"ajv": "^8.17.1",
|
||||||
"ajv-formats": "^3.0.1",
|
"ajv-formats": "^3.0.1",
|
||||||
"content-type": "^1.0.5",
|
"content-type": "^1.0.5",
|
||||||
@@ -1798,15 +1688,14 @@
|
|||||||
"cross-spawn": "^7.0.5",
|
"cross-spawn": "^7.0.5",
|
||||||
"eventsource": "^3.0.2",
|
"eventsource": "^3.0.2",
|
||||||
"eventsource-parser": "^3.0.0",
|
"eventsource-parser": "^3.0.0",
|
||||||
"express": "^5.2.1",
|
"express": "^5.0.1",
|
||||||
"express-rate-limit": "^8.2.1",
|
"express-rate-limit": "^7.5.0",
|
||||||
"hono": "^4.11.4",
|
"jose": "^6.1.1",
|
||||||
"jose": "^6.1.3",
|
|
||||||
"json-schema-typed": "^8.0.2",
|
"json-schema-typed": "^8.0.2",
|
||||||
"pkce-challenge": "^5.0.0",
|
"pkce-challenge": "^5.0.0",
|
||||||
"raw-body": "^3.0.0",
|
"raw-body": "^3.0.0",
|
||||||
"zod": "^3.25 || ^4.0",
|
"zod": "^3.25 || ^4.0",
|
||||||
"zod-to-json-schema": "^3.25.1"
|
"zod-to-json-schema": "^3.25.0"
|
||||||
},
|
},
|
||||||
"engines": {
|
"engines": {
|
||||||
"node": ">=18"
|
"node": ">=18"
|
||||||
@@ -2132,19 +2021,6 @@
|
|||||||
"node": ">=16.9"
|
"node": ">=16.9"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/@readme/httpsnippet": {
|
|
||||||
"version": "11.0.0",
|
|
||||||
"resolved": "https://registry.npmjs.org/@readme/httpsnippet/-/httpsnippet-11.0.0.tgz",
|
|
||||||
"integrity": "sha512-XSyaAsJkZfmMO9R4WDlVJARZgd4wlImftSkMkKclidniXA1h6DTya9iTqJenQo9mHQLh3u6kAC3CDRaIV+LbLw==",
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"qs": "^6.11.2",
|
|
||||||
"stringify-object": "^3.3.0"
|
|
||||||
},
|
|
||||||
"engines": {
|
|
||||||
"node": ">=18"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/@replit/codemirror-emacs": {
|
"node_modules/@replit/codemirror-emacs": {
|
||||||
"version": "6.1.0",
|
"version": "6.1.0",
|
||||||
"resolved": "https://registry.npmjs.org/@replit/codemirror-emacs/-/codemirror-emacs-6.1.0.tgz",
|
"resolved": "https://registry.npmjs.org/@replit/codemirror-emacs/-/codemirror-emacs-6.1.0.tgz",
|
||||||
@@ -4168,10 +4044,6 @@
|
|||||||
"resolved": "plugins/filter-xpath",
|
"resolved": "plugins/filter-xpath",
|
||||||
"link": true
|
"link": true
|
||||||
},
|
},
|
||||||
"node_modules/@yaak/httpsnippet": {
|
|
||||||
"resolved": "plugins-external/httpsnippet",
|
|
||||||
"link": true
|
|
||||||
},
|
|
||||||
"node_modules/@yaak/importer-curl": {
|
"node_modules/@yaak/importer-curl": {
|
||||||
"resolved": "plugins/importer-curl",
|
"resolved": "plugins/importer-curl",
|
||||||
"link": true
|
"link": true
|
||||||
@@ -6993,13 +6865,10 @@
|
|||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/express-rate-limit": {
|
"node_modules/express-rate-limit": {
|
||||||
"version": "8.2.1",
|
"version": "7.5.1",
|
||||||
"resolved": "https://registry.npmjs.org/express-rate-limit/-/express-rate-limit-8.2.1.tgz",
|
"resolved": "https://registry.npmjs.org/express-rate-limit/-/express-rate-limit-7.5.1.tgz",
|
||||||
"integrity": "sha512-PCZEIEIxqwhzw4KF0n7QF4QqruVTcF73O5kFKUnGOyjbCCgizBBiFaYpd/fnBLUMPw/BWw9OsiN7GgrNYr7j6g==",
|
"integrity": "sha512-7iN8iPMDzOMHPUYllBEsQdWVB6fPDMPqwjBaFrgr4Jgr/+okjvzAy+UHlYYL/Vs0OsOrMkwS6PJDkFlJwoxUnw==",
|
||||||
"license": "MIT",
|
"license": "MIT",
|
||||||
"dependencies": {
|
|
||||||
"ip-address": "10.0.1"
|
|
||||||
},
|
|
||||||
"engines": {
|
"engines": {
|
||||||
"node": ">= 16"
|
"node": ">= 16"
|
||||||
},
|
},
|
||||||
@@ -7538,12 +7407,6 @@
|
|||||||
"url": "https://github.com/sponsors/ljharb"
|
"url": "https://github.com/sponsors/ljharb"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/get-own-enumerable-property-symbols": {
|
|
||||||
"version": "3.0.2",
|
|
||||||
"resolved": "https://registry.npmjs.org/get-own-enumerable-property-symbols/-/get-own-enumerable-property-symbols-3.0.2.tgz",
|
|
||||||
"integrity": "sha512-I0UBV/XOz1XkIJHEUDMZAbzCThU/H8DxmSfmdGcKPnVhu2VfFqr34jr9777IyaTYvxjedWhqVIilEDsCdP5G6g==",
|
|
||||||
"license": "ISC"
|
|
||||||
},
|
|
||||||
"node_modules/get-proto": {
|
"node_modules/get-proto": {
|
||||||
"version": "1.0.1",
|
"version": "1.0.1",
|
||||||
"resolved": "https://registry.npmjs.org/get-proto/-/get-proto-1.0.1.tgz",
|
"resolved": "https://registry.npmjs.org/get-proto/-/get-proto-1.0.1.tgz",
|
||||||
@@ -8272,15 +8135,6 @@
|
|||||||
"node": ">= 0.4"
|
"node": ">= 0.4"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/ip-address": {
|
|
||||||
"version": "10.0.1",
|
|
||||||
"resolved": "https://registry.npmjs.org/ip-address/-/ip-address-10.0.1.tgz",
|
|
||||||
"integrity": "sha512-NWv9YLW4PoW2B7xtzaS3NCot75m6nK7Icdv0o3lfMceJVRfSoQwqD4wEH5rLwoKJwUiZ/rfpiVBhnaF0FK4HoA==",
|
|
||||||
"license": "MIT",
|
|
||||||
"engines": {
|
|
||||||
"node": ">= 12"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/ip-bigint": {
|
"node_modules/ip-bigint": {
|
||||||
"version": "7.3.0",
|
"version": "7.3.0",
|
||||||
"resolved": "https://registry.npmjs.org/ip-bigint/-/ip-bigint-7.3.0.tgz",
|
"resolved": "https://registry.npmjs.org/ip-bigint/-/ip-bigint-7.3.0.tgz",
|
||||||
@@ -8662,15 +8516,6 @@
|
|||||||
"url": "https://github.com/sponsors/ljharb"
|
"url": "https://github.com/sponsors/ljharb"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/is-obj": {
|
|
||||||
"version": "1.0.1",
|
|
||||||
"resolved": "https://registry.npmjs.org/is-obj/-/is-obj-1.0.1.tgz",
|
|
||||||
"integrity": "sha512-l4RyHgRqGN4Y3+9JHVrNqO+tN0rV5My76uW5/nuO4K1b6vw5G8d/cmFjP9tRfEsdhZNt0IFdZuK/c2Vr4Nb+Qg==",
|
|
||||||
"license": "MIT",
|
|
||||||
"engines": {
|
|
||||||
"node": ">=0.10.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/is-plain-obj": {
|
"node_modules/is-plain-obj": {
|
||||||
"version": "4.1.0",
|
"version": "4.1.0",
|
||||||
"resolved": "https://registry.npmjs.org/is-plain-obj/-/is-plain-obj-4.1.0.tgz",
|
"resolved": "https://registry.npmjs.org/is-plain-obj/-/is-plain-obj-4.1.0.tgz",
|
||||||
@@ -13931,29 +13776,6 @@
|
|||||||
"url": "https://github.com/sponsors/wooorm"
|
"url": "https://github.com/sponsors/wooorm"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"node_modules/stringify-object": {
|
|
||||||
"version": "3.3.0",
|
|
||||||
"resolved": "https://registry.npmjs.org/stringify-object/-/stringify-object-3.3.0.tgz",
|
|
||||||
"integrity": "sha512-rHqiFh1elqCQ9WPLIC8I0Q/g/wj5J1eMkyoiD6eoQApWHP0FtlK7rqnhmabL5VUY9JQCcqwwvlOaSuutekgyrw==",
|
|
||||||
"license": "BSD-2-Clause",
|
|
||||||
"dependencies": {
|
|
||||||
"get-own-enumerable-property-symbols": "^3.0.0",
|
|
||||||
"is-obj": "^1.0.1",
|
|
||||||
"is-regexp": "^1.0.0"
|
|
||||||
},
|
|
||||||
"engines": {
|
|
||||||
"node": ">=4"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/stringify-object/node_modules/is-regexp": {
|
|
||||||
"version": "1.0.0",
|
|
||||||
"resolved": "https://registry.npmjs.org/is-regexp/-/is-regexp-1.0.0.tgz",
|
|
||||||
"integrity": "sha512-7zjFAPO4/gwyQAAgRRmqeEeyIICSdmCqa3tsVHMdBzaXXRiqopZL4Cyghg/XulGWrtABTpbnYYzzIRffLkP4oA==",
|
|
||||||
"license": "MIT",
|
|
||||||
"engines": {
|
|
||||||
"node": ">=0.10.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"node_modules/strip-ansi": {
|
"node_modules/strip-ansi": {
|
||||||
"version": "7.1.2",
|
"version": "7.1.2",
|
||||||
"resolved": "https://registry.npmjs.org/strip-ansi/-/strip-ansi-7.1.2.tgz",
|
"resolved": "https://registry.npmjs.org/strip-ansi/-/strip-ansi-7.1.2.tgz",
|
||||||
@@ -15953,41 +15775,13 @@
|
|||||||
"undici-types": "~7.16.0"
|
"undici-types": "~7.16.0"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"plugins-external/httpsnippet": {
|
|
||||||
"name": "@yaak/httpsnippet",
|
|
||||||
"version": "1.0.0",
|
|
||||||
"dependencies": {
|
|
||||||
"@readme/httpsnippet": "^11.0.0"
|
|
||||||
},
|
|
||||||
"devDependencies": {
|
|
||||||
"@types/node": "^22.0.0",
|
|
||||||
"typescript": "^5.9.3"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"plugins-external/httpsnippet/node_modules/@types/node": {
|
|
||||||
"version": "22.19.9",
|
|
||||||
"resolved": "https://registry.npmjs.org/@types/node/-/node-22.19.9.tgz",
|
|
||||||
"integrity": "sha512-PD03/U8g1F9T9MI+1OBisaIARhSzeidsUjQaf51fOxrfjeiKN9bLVO06lHuHYjxdnqLWJijJHfqXPSJri2EM2A==",
|
|
||||||
"dev": true,
|
|
||||||
"license": "MIT",
|
|
||||||
"dependencies": {
|
|
||||||
"undici-types": "~6.21.0"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"plugins-external/httpsnippet/node_modules/undici-types": {
|
|
||||||
"version": "6.21.0",
|
|
||||||
"resolved": "https://registry.npmjs.org/undici-types/-/undici-types-6.21.0.tgz",
|
|
||||||
"integrity": "sha512-iwDZqg0QAGrg9Rav5H4n0M64c3mkR59cJ6wQp+7C4nI0gsmExaedaYLNO44eT4AtBBwjbTiGPMlt2Md0T9H9JQ==",
|
|
||||||
"dev": true,
|
|
||||||
"license": "MIT"
|
|
||||||
},
|
|
||||||
"plugins-external/mcp-server": {
|
"plugins-external/mcp-server": {
|
||||||
"name": "@yaak/mcp-server",
|
"name": "@yaak/mcp-server",
|
||||||
"version": "0.1.7",
|
"version": "0.1.7",
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@hono/mcp": "^0.2.3",
|
"@hono/mcp": "^0.2.3",
|
||||||
"@hono/node-server": "^1.19.7",
|
"@hono/node-server": "^1.19.7",
|
||||||
"@modelcontextprotocol/sdk": "^1.26.0",
|
"@modelcontextprotocol/sdk": "^1.25.2",
|
||||||
"hono": "^4.11.7",
|
"hono": "^4.11.7",
|
||||||
"zod": "^3.25.76"
|
"zod": "^3.25.76"
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -12,7 +12,6 @@
|
|||||||
"packages/plugin-runtime-types",
|
"packages/plugin-runtime-types",
|
||||||
"plugins-external/mcp-server",
|
"plugins-external/mcp-server",
|
||||||
"plugins-external/template-function-faker",
|
"plugins-external/template-function-faker",
|
||||||
"plugins-external/httpsnippet",
|
|
||||||
"plugins/action-copy-curl",
|
"plugins/action-copy-curl",
|
||||||
"plugins/action-copy-grpcurl",
|
"plugins/action-copy-grpcurl",
|
||||||
"plugins/action-send-folder",
|
"plugins/action-send-folder",
|
||||||
@@ -105,12 +104,5 @@
|
|||||||
"npm-run-all": "^4.1.5",
|
"npm-run-all": "^4.1.5",
|
||||||
"typescript": "^5.8.3",
|
"typescript": "^5.8.3",
|
||||||
"vitest": "^3.2.4"
|
"vitest": "^3.2.4"
|
||||||
},
|
|
||||||
"dependencies": {
|
|
||||||
"@codemirror/lang-go": "^6.0.1",
|
|
||||||
"@codemirror/lang-java": "^6.0.2",
|
|
||||||
"@codemirror/lang-php": "^6.0.2",
|
|
||||||
"@codemirror/lang-python": "^6.2.1",
|
|
||||||
"@codemirror/legacy-modes": "^6.5.2"
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
+4
-10
@@ -66,9 +66,7 @@ export type DeleteModelRequest = { model: string, id: string, };
|
|||||||
|
|
||||||
export type DeleteModelResponse = { model: AnyModel, };
|
export type DeleteModelResponse = { model: AnyModel, };
|
||||||
|
|
||||||
export type DialogSize = "sm" | "md" | "lg" | "full" | "dynamic";
|
export type EditorLanguage = "text" | "javascript" | "json" | "html" | "xml" | "graphql" | "markdown";
|
||||||
|
|
||||||
export type EditorLanguage = "text" | "javascript" | "json" | "html" | "xml" | "graphql" | "markdown" | "c" | "clojure" | "csharp" | "go" | "http" | "java" | "kotlin" | "objective_c" | "ocaml" | "php" | "powershell" | "python" | "r" | "ruby" | "shell" | "swift";
|
|
||||||
|
|
||||||
export type EmptyPayload = {};
|
export type EmptyPayload = {};
|
||||||
|
|
||||||
@@ -174,11 +172,7 @@ hideGutter?: boolean,
|
|||||||
/**
|
/**
|
||||||
* Language for syntax highlighting
|
* Language for syntax highlighting
|
||||||
*/
|
*/
|
||||||
language?: EditorLanguage, readOnly?: boolean,
|
language?: EditorLanguage, readOnly?: boolean, completionOptions?: Array<GenericCompletionOption>,
|
||||||
/**
|
|
||||||
* Fixed number of visible rows
|
|
||||||
*/
|
|
||||||
rows?: number, completionOptions?: Array<GenericCompletionOption>,
|
|
||||||
/**
|
/**
|
||||||
* The name of the input. The value will be stored at this object attribute in the resulting data
|
* The name of the input. The value will be stored at this object attribute in the resulting data
|
||||||
*/
|
*/
|
||||||
@@ -482,9 +476,9 @@ label: string, title?: string, size?: WindowSize, dataDirKey?: string, };
|
|||||||
|
|
||||||
export type PluginContext = { id: string, label: string | null, workspaceId: string | null, };
|
export type PluginContext = { id: string, label: string | null, workspaceId: string | null, };
|
||||||
|
|
||||||
export type PromptFormRequest = { id: string, title: string, description?: string, inputs: Array<FormInput>, confirmText?: string, cancelText?: string, size?: DialogSize, };
|
export type PromptFormRequest = { id: string, title: string, description?: string, inputs: Array<FormInput>, confirmText?: string, cancelText?: string, };
|
||||||
|
|
||||||
export type PromptFormResponse = { values: { [key in string]?: JsonPrimitive } | null, done?: boolean, };
|
export type PromptFormResponse = { values: { [key in string]?: JsonPrimitive } | null, };
|
||||||
|
|
||||||
export type PromptTextRequest = { id: string, title: string, label: string, description?: string, defaultValue?: string, placeholder?: string,
|
export type PromptTextRequest = { id: string, title: string, label: string, description?: string, defaultValue?: string, placeholder?: string,
|
||||||
/**
|
/**
|
||||||
|
|||||||
+1
-1
@@ -49,7 +49,7 @@ export type HttpResponseEvent = { model: "http_response_event", id: string, crea
|
|||||||
* This mirrors `yaak_http::sender::HttpResponseEvent` but with serde support.
|
* This mirrors `yaak_http::sender::HttpResponseEvent` but with serde support.
|
||||||
* The `From` impl is in yaak-http to avoid circular dependencies.
|
* The `From` impl is in yaak-http to avoid circular dependencies.
|
||||||
*/
|
*/
|
||||||
export type HttpResponseEventData = { "type": "setting", name: string, value: string, } | { "type": "info", message: string, } | { "type": "redirect", url: string, status: number, behavior: string, } | { "type": "send_url", method: string, scheme: string, username: string, password: string, host: string, port: number, path: string, query: string, fragment: string, } | { "type": "receive_url", version: string, status: string, } | { "type": "header_up", name: string, value: string, } | { "type": "header_down", name: string, value: string, } | { "type": "chunk_sent", bytes: number, } | { "type": "chunk_received", bytes: number, } | { "type": "dns_resolved", hostname: string, addresses: Array<string>, duration: bigint, overridden: boolean, };
|
export type HttpResponseEventData = { "type": "setting", name: string, value: string, } | { "type": "info", message: string, } | { "type": "redirect", url: string, status: number, behavior: string, } | { "type": "send_url", method: string, path: string, } | { "type": "receive_url", version: string, status: string, } | { "type": "header_up", name: string, value: string, } | { "type": "header_down", name: string, value: string, } | { "type": "chunk_sent", bytes: number, } | { "type": "chunk_received", bytes: number, } | { "type": "dns_resolved", hostname: string, addresses: Array<string>, duration: bigint, overridden: boolean, };
|
||||||
|
|
||||||
export type HttpResponseHeader = { name: string, value: string, };
|
export type HttpResponseHeader = { name: string, value: string, };
|
||||||
|
|
||||||
|
|||||||
@@ -1,12 +1,10 @@
|
|||||||
import type {
|
import type {
|
||||||
FindHttpResponsesRequest,
|
FindHttpResponsesRequest,
|
||||||
FindHttpResponsesResponse,
|
FindHttpResponsesResponse,
|
||||||
FormInput,
|
|
||||||
GetCookieValueRequest,
|
GetCookieValueRequest,
|
||||||
GetCookieValueResponse,
|
GetCookieValueResponse,
|
||||||
GetHttpRequestByIdRequest,
|
GetHttpRequestByIdRequest,
|
||||||
GetHttpRequestByIdResponse,
|
GetHttpRequestByIdResponse,
|
||||||
JsonPrimitive,
|
|
||||||
ListCookieNamesResponse,
|
ListCookieNamesResponse,
|
||||||
ListFoldersRequest,
|
ListFoldersRequest,
|
||||||
ListFoldersResponse,
|
ListFoldersResponse,
|
||||||
@@ -29,39 +27,6 @@ import type {
|
|||||||
} from '../bindings/gen_events.ts';
|
} from '../bindings/gen_events.ts';
|
||||||
import type { Folder, HttpRequest } from '../bindings/gen_models.ts';
|
import type { Folder, HttpRequest } from '../bindings/gen_models.ts';
|
||||||
import type { JsonValue } from '../bindings/serde_json/JsonValue';
|
import type { JsonValue } from '../bindings/serde_json/JsonValue';
|
||||||
import type { MaybePromise } from '../helpers';
|
|
||||||
|
|
||||||
export type CallPromptFormDynamicArgs = {
|
|
||||||
values: { [key in string]?: JsonPrimitive };
|
|
||||||
};
|
|
||||||
|
|
||||||
type AddDynamicMethod<T> = {
|
|
||||||
dynamic?: (
|
|
||||||
ctx: Context,
|
|
||||||
args: CallPromptFormDynamicArgs,
|
|
||||||
) => MaybePromise<Partial<T> | null | undefined>;
|
|
||||||
};
|
|
||||||
|
|
||||||
// biome-ignore lint/suspicious/noExplicitAny: distributive conditional type pattern
|
|
||||||
type AddDynamic<T> = T extends any
|
|
||||||
? T extends { inputs?: FormInput[] }
|
|
||||||
? Omit<T, 'inputs'> & {
|
|
||||||
inputs: Array<AddDynamic<FormInput>>;
|
|
||||||
dynamic?: (
|
|
||||||
ctx: Context,
|
|
||||||
args: CallPromptFormDynamicArgs,
|
|
||||||
) => MaybePromise<
|
|
||||||
Partial<Omit<T, 'inputs'> & { inputs: Array<AddDynamic<FormInput>> }> | null | undefined
|
|
||||||
>;
|
|
||||||
}
|
|
||||||
: T & AddDynamicMethod<T>
|
|
||||||
: never;
|
|
||||||
|
|
||||||
export type DynamicPromptFormArg = AddDynamic<FormInput>;
|
|
||||||
|
|
||||||
type DynamicPromptFormRequest = Omit<PromptFormRequest, 'inputs'> & {
|
|
||||||
inputs: DynamicPromptFormArg[];
|
|
||||||
};
|
|
||||||
|
|
||||||
export type WorkspaceHandle = Pick<WorkspaceInfo, 'id' | 'name'>;
|
export type WorkspaceHandle = Pick<WorkspaceInfo, 'id' | 'name'>;
|
||||||
|
|
||||||
@@ -74,7 +39,7 @@ export interface Context {
|
|||||||
};
|
};
|
||||||
prompt: {
|
prompt: {
|
||||||
text(args: PromptTextRequest): Promise<PromptTextResponse['value']>;
|
text(args: PromptTextRequest): Promise<PromptTextResponse['value']>;
|
||||||
form(args: DynamicPromptFormRequest): Promise<PromptFormResponse['values']>;
|
form(args: PromptFormRequest): Promise<PromptFormResponse['values']>;
|
||||||
};
|
};
|
||||||
store: {
|
store: {
|
||||||
set<T>(key: string, value: T): Promise<void>;
|
set<T>(key: string, value: T): Promise<void>;
|
||||||
|
|||||||
@@ -2,22 +2,21 @@ import type { AuthenticationPlugin } from './AuthenticationPlugin';
|
|||||||
|
|
||||||
import type { Context } from './Context';
|
import type { Context } from './Context';
|
||||||
import type { FilterPlugin } from './FilterPlugin';
|
import type { FilterPlugin } from './FilterPlugin';
|
||||||
import type { FolderActionPlugin } from './FolderActionPlugin';
|
|
||||||
import type { GrpcRequestActionPlugin } from './GrpcRequestActionPlugin';
|
import type { GrpcRequestActionPlugin } from './GrpcRequestActionPlugin';
|
||||||
import type { HttpRequestActionPlugin } from './HttpRequestActionPlugin';
|
import type { HttpRequestActionPlugin } from './HttpRequestActionPlugin';
|
||||||
|
import type { WebsocketRequestActionPlugin } from './WebsocketRequestActionPlugin';
|
||||||
|
import type { WorkspaceActionPlugin } from './WorkspaceActionPlugin';
|
||||||
|
import type { FolderActionPlugin } from './FolderActionPlugin';
|
||||||
import type { ImporterPlugin } from './ImporterPlugin';
|
import type { ImporterPlugin } from './ImporterPlugin';
|
||||||
import type { TemplateFunctionPlugin } from './TemplateFunctionPlugin';
|
import type { TemplateFunctionPlugin } from './TemplateFunctionPlugin';
|
||||||
import type { ThemePlugin } from './ThemePlugin';
|
import type { ThemePlugin } from './ThemePlugin';
|
||||||
import type { WebsocketRequestActionPlugin } from './WebsocketRequestActionPlugin';
|
|
||||||
import type { WorkspaceActionPlugin } from './WorkspaceActionPlugin';
|
|
||||||
|
|
||||||
export type { Context };
|
export type { Context };
|
||||||
export type { DynamicAuthenticationArg } from './AuthenticationPlugin';
|
|
||||||
export type { CallPromptFormDynamicArgs, DynamicPromptFormArg } from './Context';
|
|
||||||
export type { DynamicTemplateFunctionArg } from './TemplateFunctionPlugin';
|
export type { DynamicTemplateFunctionArg } from './TemplateFunctionPlugin';
|
||||||
|
export type { DynamicAuthenticationArg } from './AuthenticationPlugin';
|
||||||
export type { TemplateFunctionPlugin };
|
export type { TemplateFunctionPlugin };
|
||||||
export type { FolderActionPlugin } from './FolderActionPlugin';
|
|
||||||
export type { WorkspaceActionPlugin } from './WorkspaceActionPlugin';
|
export type { WorkspaceActionPlugin } from './WorkspaceActionPlugin';
|
||||||
|
export type { FolderActionPlugin } from './FolderActionPlugin';
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The global structure of a Yaak plugin
|
* The global structure of a Yaak plugin
|
||||||
|
|||||||
@@ -1,12 +1,7 @@
|
|||||||
import console from 'node:console';
|
import console from 'node:console';
|
||||||
import { type Stats, statSync, watch } from 'node:fs';
|
import { type Stats, statSync, watch } from 'node:fs';
|
||||||
import path from 'node:path';
|
import path from 'node:path';
|
||||||
import type {
|
import type { Context, PluginDefinition } from '@yaakapp/api';
|
||||||
CallPromptFormDynamicArgs,
|
|
||||||
Context,
|
|
||||||
DynamicPromptFormArg,
|
|
||||||
PluginDefinition,
|
|
||||||
} from '@yaakapp/api';
|
|
||||||
import {
|
import {
|
||||||
applyFormInputDefaults,
|
applyFormInputDefaults,
|
||||||
validateTemplateFunctionArgs,
|
validateTemplateFunctionArgs,
|
||||||
@@ -17,7 +12,6 @@ import type {
|
|||||||
DeleteModelResponse,
|
DeleteModelResponse,
|
||||||
FindHttpResponsesResponse,
|
FindHttpResponsesResponse,
|
||||||
Folder,
|
Folder,
|
||||||
FormInput,
|
|
||||||
GetCookieValueRequest,
|
GetCookieValueRequest,
|
||||||
GetCookieValueResponse,
|
GetCookieValueResponse,
|
||||||
GetHttpRequestByIdResponse,
|
GetHttpRequestByIdResponse,
|
||||||
@@ -61,7 +55,6 @@ export class PluginInstance {
|
|||||||
#mod: PluginDefinition;
|
#mod: PluginDefinition;
|
||||||
#pluginToAppEvents: EventChannel;
|
#pluginToAppEvents: EventChannel;
|
||||||
#appToPluginEvents: EventChannel;
|
#appToPluginEvents: EventChannel;
|
||||||
#pendingDynamicForms = new Map<string, DynamicPromptFormArg[]>();
|
|
||||||
|
|
||||||
constructor(workerData: PluginWorkerData, pluginEvents: EventChannel) {
|
constructor(workerData: PluginWorkerData, pluginEvents: EventChannel) {
|
||||||
this.#workerData = workerData;
|
this.#workerData = workerData;
|
||||||
@@ -113,7 +106,6 @@ export class PluginInstance {
|
|||||||
|
|
||||||
async terminate() {
|
async terminate() {
|
||||||
await this.#mod?.dispose?.();
|
await this.#mod?.dispose?.();
|
||||||
this.#pendingDynamicForms.clear();
|
|
||||||
this.#unimportModule();
|
this.#unimportModule();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -307,7 +299,7 @@ export class PluginInstance {
|
|||||||
const replyPayload: InternalEventPayload = {
|
const replyPayload: InternalEventPayload = {
|
||||||
type: 'get_template_function_config_response',
|
type: 'get_template_function_config_response',
|
||||||
pluginRefId: this.#workerData.pluginRefId,
|
pluginRefId: this.#workerData.pluginRefId,
|
||||||
function: { ...fn, args: stripDynamicCallbacks(resolvedArgs) },
|
function: { ...fn, args: resolvedArgs },
|
||||||
};
|
};
|
||||||
this.#sendPayload(context, replyPayload, replyId);
|
this.#sendPayload(context, replyPayload, replyId);
|
||||||
return;
|
return;
|
||||||
@@ -334,7 +326,7 @@ export class PluginInstance {
|
|||||||
|
|
||||||
const replyPayload: InternalEventPayload = {
|
const replyPayload: InternalEventPayload = {
|
||||||
type: 'get_http_authentication_config_response',
|
type: 'get_http_authentication_config_response',
|
||||||
args: stripDynamicCallbacks(resolvedArgs),
|
args: resolvedArgs,
|
||||||
actions: resolvedActions,
|
actions: resolvedActions,
|
||||||
pluginRefId: this.#workerData.pluginRefId,
|
pluginRefId: this.#workerData.pluginRefId,
|
||||||
};
|
};
|
||||||
@@ -672,66 +664,10 @@ export class PluginInstance {
|
|||||||
return reply.value;
|
return reply.value;
|
||||||
},
|
},
|
||||||
form: async (args) => {
|
form: async (args) => {
|
||||||
// Resolve dynamic callbacks on initial inputs using default values
|
const reply: PromptFormResponse = await this.#sendForReply(context, {
|
||||||
const defaults = applyFormInputDefaults(args.inputs, {});
|
type: 'prompt_form_request',
|
||||||
const callArgs: CallPromptFormDynamicArgs = { values: defaults };
|
...args,
|
||||||
const resolvedInputs = await applyDynamicFormInput(
|
|
||||||
this.#newCtx(context),
|
|
||||||
args.inputs,
|
|
||||||
callArgs,
|
|
||||||
);
|
|
||||||
const strippedInputs = stripDynamicCallbacks(resolvedInputs);
|
|
||||||
|
|
||||||
// Build the event manually so we can get the event ID for keying
|
|
||||||
const eventToSend = this.#buildEventToSend(
|
|
||||||
context,
|
|
||||||
{ type: 'prompt_form_request', ...args, inputs: strippedInputs },
|
|
||||||
null,
|
|
||||||
);
|
|
||||||
|
|
||||||
// Store original inputs (with dynamic callbacks) for later resolution
|
|
||||||
this.#pendingDynamicForms.set(eventToSend.id, args.inputs);
|
|
||||||
|
|
||||||
const reply = await new Promise<PromptFormResponse>((resolve) => {
|
|
||||||
const cb = (event: InternalEvent) => {
|
|
||||||
if (event.replyId !== eventToSend.id) return;
|
|
||||||
|
|
||||||
if (event.payload.type === 'prompt_form_response') {
|
|
||||||
const { done, values } = event.payload as PromptFormResponse;
|
|
||||||
if (done) {
|
|
||||||
// Final response — resolve the promise and clean up
|
|
||||||
this.#appToPluginEvents.unlisten(cb);
|
|
||||||
this.#pendingDynamicForms.delete(eventToSend.id);
|
|
||||||
resolve({ values } as PromptFormResponse);
|
|
||||||
} else {
|
|
||||||
// Intermediate value change — resolve dynamic inputs and send back
|
|
||||||
// Skip empty values (fired on initial mount before user interaction)
|
|
||||||
const storedInputs = this.#pendingDynamicForms.get(eventToSend.id);
|
|
||||||
if (storedInputs && values && Object.keys(values).length > 0) {
|
|
||||||
const ctx = this.#newCtx(context);
|
|
||||||
const callArgs: CallPromptFormDynamicArgs = { values };
|
|
||||||
applyDynamicFormInput(ctx, storedInputs, callArgs)
|
|
||||||
.then((resolvedInputs) => {
|
|
||||||
const stripped = stripDynamicCallbacks(resolvedInputs);
|
|
||||||
this.#sendPayload(
|
|
||||||
context,
|
|
||||||
{ type: 'prompt_form_request', ...args, inputs: stripped },
|
|
||||||
eventToSend.id,
|
|
||||||
);
|
|
||||||
})
|
|
||||||
.catch((err) => {
|
|
||||||
console.error('Failed to resolve dynamic form inputs', err);
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
};
|
|
||||||
this.#appToPluginEvents.listen(cb);
|
|
||||||
|
|
||||||
// Send the initial event after we start listening (to prevent race)
|
|
||||||
this.#sendEvent(eventToSend);
|
|
||||||
});
|
});
|
||||||
|
|
||||||
return reply.values;
|
return reply.values;
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -852,12 +788,12 @@ export class PluginInstance {
|
|||||||
const { folders } = await this.#sendForReply<ListFoldersResponse>(context, payload);
|
const { folders } = await this.#sendForReply<ListFoldersResponse>(context, payload);
|
||||||
return folders.find((f) => f.id === args.id) ?? null;
|
return folders.find((f) => f.id === args.id) ?? null;
|
||||||
},
|
},
|
||||||
create: async ({ name, ...args }) => {
|
create: async (args) => {
|
||||||
const payload = {
|
const payload = {
|
||||||
type: 'upsert_model_request',
|
type: 'upsert_model_request',
|
||||||
model: {
|
model: {
|
||||||
|
name: '',
|
||||||
...args,
|
...args,
|
||||||
name: name ?? '',
|
|
||||||
id: '',
|
id: '',
|
||||||
model: 'folder',
|
model: 'folder',
|
||||||
},
|
},
|
||||||
@@ -970,17 +906,6 @@ export class PluginInstance {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
function stripDynamicCallbacks(inputs: { dynamic?: unknown }[]): FormInput[] {
|
|
||||||
return inputs.map((input) => {
|
|
||||||
// biome-ignore lint/suspicious/noExplicitAny: stripping dynamic from union type
|
|
||||||
const { dynamic, ...rest } = input as any;
|
|
||||||
if ('inputs' in rest && Array.isArray(rest.inputs)) {
|
|
||||||
rest.inputs = stripDynamicCallbacks(rest.inputs);
|
|
||||||
}
|
|
||||||
return rest as FormInput;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
function genId(len = 5): string {
|
function genId(len = 5): string {
|
||||||
const alphabet = '01234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
|
const alphabet = '01234567890abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
|
||||||
let id = '';
|
let id = '';
|
||||||
|
|||||||
@@ -1,21 +1,9 @@
|
|||||||
import type {
|
import type { Context, DynamicAuthenticationArg, DynamicTemplateFunctionArg } from '@yaakapp/api';
|
||||||
CallPromptFormDynamicArgs,
|
|
||||||
Context,
|
|
||||||
DynamicAuthenticationArg,
|
|
||||||
DynamicPromptFormArg,
|
|
||||||
DynamicTemplateFunctionArg,
|
|
||||||
} from '@yaakapp/api';
|
|
||||||
import type {
|
import type {
|
||||||
CallHttpAuthenticationActionArgs,
|
CallHttpAuthenticationActionArgs,
|
||||||
CallTemplateFunctionArgs,
|
CallTemplateFunctionArgs,
|
||||||
} from '@yaakapp-internal/plugins';
|
} from '@yaakapp-internal/plugins';
|
||||||
|
|
||||||
type AnyDynamicArg = DynamicTemplateFunctionArg | DynamicAuthenticationArg | DynamicPromptFormArg;
|
|
||||||
type AnyCallArgs =
|
|
||||||
| CallTemplateFunctionArgs
|
|
||||||
| CallHttpAuthenticationActionArgs
|
|
||||||
| CallPromptFormDynamicArgs;
|
|
||||||
|
|
||||||
export async function applyDynamicFormInput(
|
export async function applyDynamicFormInput(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
args: DynamicTemplateFunctionArg[],
|
args: DynamicTemplateFunctionArg[],
|
||||||
@@ -30,40 +18,30 @@ export async function applyDynamicFormInput(
|
|||||||
|
|
||||||
export async function applyDynamicFormInput(
|
export async function applyDynamicFormInput(
|
||||||
ctx: Context,
|
ctx: Context,
|
||||||
args: DynamicPromptFormArg[],
|
args: (DynamicTemplateFunctionArg | DynamicAuthenticationArg)[],
|
||||||
callArgs: CallPromptFormDynamicArgs,
|
callArgs: CallTemplateFunctionArgs | CallHttpAuthenticationActionArgs,
|
||||||
): Promise<DynamicPromptFormArg[]>;
|
): Promise<(DynamicTemplateFunctionArg | DynamicAuthenticationArg)[]> {
|
||||||
|
const resolvedArgs: (DynamicTemplateFunctionArg | DynamicAuthenticationArg)[] = [];
|
||||||
export async function applyDynamicFormInput(
|
|
||||||
ctx: Context,
|
|
||||||
args: AnyDynamicArg[],
|
|
||||||
callArgs: AnyCallArgs,
|
|
||||||
): Promise<AnyDynamicArg[]> {
|
|
||||||
const resolvedArgs: AnyDynamicArg[] = [];
|
|
||||||
for (const { dynamic, ...arg } of args) {
|
for (const { dynamic, ...arg } of args) {
|
||||||
const dynamicResult =
|
const dynamicResult =
|
||||||
typeof dynamic === 'function'
|
typeof dynamic === 'function'
|
||||||
? await dynamic(
|
? await dynamic(
|
||||||
ctx,
|
ctx,
|
||||||
callArgs as CallTemplateFunctionArgs &
|
callArgs as CallTemplateFunctionArgs & CallHttpAuthenticationActionArgs,
|
||||||
CallHttpAuthenticationActionArgs &
|
|
||||||
CallPromptFormDynamicArgs,
|
|
||||||
)
|
)
|
||||||
: undefined;
|
: undefined;
|
||||||
|
|
||||||
const newArg = {
|
const newArg = {
|
||||||
...arg,
|
...arg,
|
||||||
...dynamicResult,
|
...dynamicResult,
|
||||||
} as AnyDynamicArg;
|
} as DynamicTemplateFunctionArg | DynamicAuthenticationArg;
|
||||||
|
|
||||||
if ('inputs' in newArg && Array.isArray(newArg.inputs)) {
|
if ('inputs' in newArg && Array.isArray(newArg.inputs)) {
|
||||||
try {
|
try {
|
||||||
newArg.inputs = await applyDynamicFormInput(
|
newArg.inputs = await applyDynamicFormInput(
|
||||||
ctx,
|
ctx,
|
||||||
newArg.inputs as DynamicTemplateFunctionArg[],
|
newArg.inputs as DynamicTemplateFunctionArg[],
|
||||||
callArgs as CallTemplateFunctionArgs &
|
callArgs as CallTemplateFunctionArgs & CallHttpAuthenticationActionArgs,
|
||||||
CallHttpAuthenticationActionArgs &
|
|
||||||
CallPromptFormDynamicArgs,
|
|
||||||
);
|
);
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.error('Failed to apply dynamic form input', e);
|
console.error('Failed to apply dynamic form input', e);
|
||||||
|
|||||||
@@ -1,9 +0,0 @@
|
|||||||
# Yaak HTTP Snippet Plugin
|
|
||||||
|
|
||||||
Generate code snippets for HTTP requests in various languages and frameworks,
|
|
||||||
powered by [httpsnippet](https://github.com/Kong/httpsnippet).
|
|
||||||
|
|
||||||
## Supported Languages
|
|
||||||
|
|
||||||
C, Clojure, C#, Go, HTTP, Java, JavaScript, Kotlin, Node.js, Objective-C,
|
|
||||||
OCaml, PHP, PowerShell, Python, R, Ruby, Shell, and Swift.
|
|
||||||
@@ -1,24 +0,0 @@
|
|||||||
{
|
|
||||||
"name": "@yaak/httpsnippet",
|
|
||||||
"private": true,
|
|
||||||
"version": "1.0.1",
|
|
||||||
"displayName": "HTTP Snippet",
|
|
||||||
"description": "Generate code snippets for HTTP requests in various languages and frameworks",
|
|
||||||
"minYaakVersion": "2026.2.0-beta.10",
|
|
||||||
"repository": {
|
|
||||||
"type": "git",
|
|
||||||
"url": "https://github.com/mountain-loop/yaak.git",
|
|
||||||
"directory": "plugins-external/httpsnippet"
|
|
||||||
},
|
|
||||||
"scripts": {
|
|
||||||
"build": "yaakcli build",
|
|
||||||
"dev": "yaakcli dev"
|
|
||||||
},
|
|
||||||
"dependencies": {
|
|
||||||
"@readme/httpsnippet": "^11.0.0"
|
|
||||||
},
|
|
||||||
"devDependencies": {
|
|
||||||
"@types/node": "^22.0.0",
|
|
||||||
"typescript": "^5.9.3"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,314 +0,0 @@
|
|||||||
import { availableTargets, type HarRequest, HTTPSnippet } from '@readme/httpsnippet';
|
|
||||||
import type { EditorLanguage, HttpRequest, PluginDefinition } from '@yaakapp/api';
|
|
||||||
|
|
||||||
// Get all available targets and build select options
|
|
||||||
const targets = availableTargets();
|
|
||||||
|
|
||||||
// Targets to exclude from the language list
|
|
||||||
const excludedTargets = new Set(['json']);
|
|
||||||
|
|
||||||
// Build language (target) options
|
|
||||||
const languageOptions = targets
|
|
||||||
.filter((target) => !excludedTargets.has(target.key))
|
|
||||||
.map((target) => ({
|
|
||||||
label: target.title,
|
|
||||||
value: target.key,
|
|
||||||
}));
|
|
||||||
|
|
||||||
// Preferred clients per target (shown first in the list)
|
|
||||||
const preferredClients: Record<string, string> = {
|
|
||||||
javascript: 'fetch',
|
|
||||||
node: 'fetch',
|
|
||||||
};
|
|
||||||
|
|
||||||
// Get client options for a given target key
|
|
||||||
function getClientOptions(targetKey: string) {
|
|
||||||
const target = targets.find((t) => t.key === targetKey);
|
|
||||||
if (!target) return [];
|
|
||||||
const preferred = preferredClients[targetKey];
|
|
||||||
return target.clients
|
|
||||||
.map((client) => ({
|
|
||||||
label: client.title,
|
|
||||||
value: client.key,
|
|
||||||
}))
|
|
||||||
.sort((a, b) => {
|
|
||||||
if (a.value === preferred) return -1;
|
|
||||||
if (b.value === preferred) return 1;
|
|
||||||
return 0;
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// Get default client for a target
|
|
||||||
function getDefaultClient(targetKey: string): string {
|
|
||||||
const options = getClientOptions(targetKey);
|
|
||||||
return options[0]?.value ?? '';
|
|
||||||
}
|
|
||||||
|
|
||||||
// Defaults
|
|
||||||
const defaultTarget = 'javascript';
|
|
||||||
|
|
||||||
// Map httpsnippet target key to editor language for syntax highlighting
|
|
||||||
const editorLanguageMap: Record<string, EditorLanguage> = {
|
|
||||||
c: 'c',
|
|
||||||
clojure: 'clojure',
|
|
||||||
csharp: 'csharp',
|
|
||||||
go: 'go',
|
|
||||||
http: 'http',
|
|
||||||
java: 'java',
|
|
||||||
javascript: 'javascript',
|
|
||||||
kotlin: 'kotlin',
|
|
||||||
node: 'javascript',
|
|
||||||
objc: 'objective_c',
|
|
||||||
ocaml: 'ocaml',
|
|
||||||
php: 'php',
|
|
||||||
powershell: 'powershell',
|
|
||||||
python: 'python',
|
|
||||||
r: 'r',
|
|
||||||
ruby: 'ruby',
|
|
||||||
shell: 'shell',
|
|
||||||
swift: 'swift',
|
|
||||||
};
|
|
||||||
|
|
||||||
function getEditorLanguage(targetKey: string): EditorLanguage {
|
|
||||||
return editorLanguageMap[targetKey] ?? 'text';
|
|
||||||
}
|
|
||||||
|
|
||||||
// Convert Yaak HttpRequest to HAR format
|
|
||||||
function toHarRequest(request: Partial<HttpRequest>) {
|
|
||||||
// Build URL with query parameters
|
|
||||||
let finalUrl = request.url || '';
|
|
||||||
const urlParams = (request.urlParameters ?? []).filter((p) => p.enabled !== false && !!p.name);
|
|
||||||
if (urlParams.length > 0) {
|
|
||||||
const [base, hash] = finalUrl.split('#');
|
|
||||||
const separator = base?.includes('?') ? '&' : '?';
|
|
||||||
const queryString = urlParams
|
|
||||||
.map((p) => `${encodeURIComponent(p.name)}=${encodeURIComponent(p.value)}`)
|
|
||||||
.join('&');
|
|
||||||
finalUrl = base + separator + queryString + (hash ? `#${hash}` : '');
|
|
||||||
}
|
|
||||||
|
|
||||||
// Build headers array
|
|
||||||
const headers: Array<{ name: string; value: string }> = (request.headers ?? [])
|
|
||||||
.filter((h) => h.enabled !== false && !!h.name)
|
|
||||||
.map((h) => ({ name: h.name, value: h.value }));
|
|
||||||
|
|
||||||
// Handle authentication
|
|
||||||
if (request.authentication?.disabled !== true) {
|
|
||||||
if (request.authenticationType === 'basic') {
|
|
||||||
const credentials = btoa(
|
|
||||||
`${request.authentication?.username ?? ''}:${request.authentication?.password ?? ''}`,
|
|
||||||
);
|
|
||||||
headers.push({ name: 'Authorization', value: `Basic ${credentials}` });
|
|
||||||
} else if (request.authenticationType === 'bearer') {
|
|
||||||
const prefix = request.authentication?.prefix ?? 'Bearer';
|
|
||||||
const token = request.authentication?.token ?? '';
|
|
||||||
headers.push({ name: 'Authorization', value: `${prefix} ${token}`.trim() });
|
|
||||||
} else if (request.authenticationType === 'apikey') {
|
|
||||||
if (request.authentication?.location === 'header') {
|
|
||||||
headers.push({
|
|
||||||
name: request.authentication?.key ?? 'X-Api-Key',
|
|
||||||
value: request.authentication?.value ?? '',
|
|
||||||
});
|
|
||||||
} else if (request.authentication?.location === 'query') {
|
|
||||||
const sep = finalUrl.includes('?') ? '&' : '?';
|
|
||||||
finalUrl = [
|
|
||||||
finalUrl,
|
|
||||||
sep,
|
|
||||||
encodeURIComponent(request.authentication?.key ?? 'token'),
|
|
||||||
'=',
|
|
||||||
encodeURIComponent(request.authentication?.value ?? ''),
|
|
||||||
].join('');
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Build HAR request object
|
|
||||||
const har: Record<string, unknown> = {
|
|
||||||
method: request.method || 'GET',
|
|
||||||
url: finalUrl,
|
|
||||||
headers,
|
|
||||||
};
|
|
||||||
|
|
||||||
// Handle request body
|
|
||||||
const bodyType = request.bodyType ?? 'none';
|
|
||||||
if (bodyType !== 'none' && request.body) {
|
|
||||||
if (bodyType === 'application/x-www-form-urlencoded' && Array.isArray(request.body.form)) {
|
|
||||||
const params = request.body.form
|
|
||||||
.filter((p: { enabled?: boolean; name?: string }) => p.enabled !== false && !!p.name)
|
|
||||||
.map((p: { name: string; value: string }) => ({ name: p.name, value: p.value }));
|
|
||||||
har.postData = {
|
|
||||||
mimeType: 'application/x-www-form-urlencoded',
|
|
||||||
params,
|
|
||||||
};
|
|
||||||
} else if (bodyType === 'multipart/form-data' && Array.isArray(request.body.form)) {
|
|
||||||
const params = request.body.form
|
|
||||||
.filter((p: { enabled?: boolean; name?: string }) => p.enabled !== false && !!p.name)
|
|
||||||
.map((p: { name: string; value: string; file?: string; contentType?: string }) => {
|
|
||||||
const param: Record<string, string> = { name: p.name, value: p.value || '' };
|
|
||||||
if (p.file) param.fileName = p.file;
|
|
||||||
if (p.contentType) param.contentType = p.contentType;
|
|
||||||
return param;
|
|
||||||
});
|
|
||||||
har.postData = {
|
|
||||||
mimeType: 'multipart/form-data',
|
|
||||||
params,
|
|
||||||
};
|
|
||||||
} else if (bodyType === 'graphql' && typeof request.body.query === 'string') {
|
|
||||||
const body = {
|
|
||||||
query: request.body.query || '',
|
|
||||||
variables: maybeParseJSON(request.body.variables, undefined),
|
|
||||||
};
|
|
||||||
har.postData = {
|
|
||||||
mimeType: 'application/json',
|
|
||||||
text: JSON.stringify(body),
|
|
||||||
};
|
|
||||||
} else if (typeof request.body.text === 'string') {
|
|
||||||
har.postData = {
|
|
||||||
mimeType: bodyType,
|
|
||||||
text: request.body.text,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return har;
|
|
||||||
}
|
|
||||||
|
|
||||||
function maybeParseJSON<T>(v: unknown, fallback: T): T | unknown {
|
|
||||||
if (typeof v !== 'string') return fallback;
|
|
||||||
try {
|
|
||||||
return JSON.parse(v);
|
|
||||||
} catch {
|
|
||||||
return fallback;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
export const plugin: PluginDefinition = {
|
|
||||||
httpRequestActions: [
|
|
||||||
{
|
|
||||||
label: 'Generate Code Snippet',
|
|
||||||
icon: 'copy',
|
|
||||||
async onSelect(ctx, args) {
|
|
||||||
// Render the request with variables resolved
|
|
||||||
const renderedRequest = await ctx.httpRequest.render({
|
|
||||||
httpRequest: args.httpRequest,
|
|
||||||
purpose: 'send',
|
|
||||||
});
|
|
||||||
|
|
||||||
// Convert to HAR format
|
|
||||||
const harRequest = toHarRequest(renderedRequest) as HarRequest;
|
|
||||||
|
|
||||||
// Get previously selected language or use defaults
|
|
||||||
const storedTarget = await ctx.store.get<string>('selectedTarget');
|
|
||||||
const initialTarget = storedTarget || defaultTarget;
|
|
||||||
const storedClient = await ctx.store.get<string>(`selectedClient:${initialTarget}`);
|
|
||||||
const initialClient = storedClient || getDefaultClient(initialTarget);
|
|
||||||
|
|
||||||
// Create snippet generator
|
|
||||||
const snippet = new HTTPSnippet(harRequest);
|
|
||||||
const generateSnippet = (target: string, client: string): string => {
|
|
||||||
const result = snippet.convert(target as any, client);
|
|
||||||
return (Array.isArray(result) ? result.join('\n') : result || '').replace(/\r\n/g, '\n');
|
|
||||||
};
|
|
||||||
|
|
||||||
// Generate initial code preview
|
|
||||||
let initialCode = '';
|
|
||||||
try {
|
|
||||||
initialCode = generateSnippet(initialTarget, initialClient);
|
|
||||||
} catch {
|
|
||||||
initialCode = '// Error generating snippet';
|
|
||||||
}
|
|
||||||
|
|
||||||
// Show dialog with language/library selectors and code preview
|
|
||||||
const result = await ctx.prompt.form({
|
|
||||||
id: 'httpsnippet',
|
|
||||||
title: 'Generate Code Snippet',
|
|
||||||
confirmText: 'Copy to Clipboard',
|
|
||||||
cancelText: 'Cancel',
|
|
||||||
size: 'md',
|
|
||||||
inputs: [
|
|
||||||
{
|
|
||||||
type: 'h_stack',
|
|
||||||
inputs: [
|
|
||||||
{
|
|
||||||
type: 'select',
|
|
||||||
name: 'target',
|
|
||||||
label: 'Language',
|
|
||||||
defaultValue: initialTarget,
|
|
||||||
options: languageOptions,
|
|
||||||
},
|
|
||||||
{
|
|
||||||
type: 'select',
|
|
||||||
name: `client-${initialTarget}`,
|
|
||||||
label: 'Library',
|
|
||||||
defaultValue: initialClient,
|
|
||||||
options: getClientOptions(initialTarget),
|
|
||||||
dynamic(_ctx, { values }) {
|
|
||||||
const targetKey = String(values.target || defaultTarget);
|
|
||||||
const options = getClientOptions(targetKey);
|
|
||||||
return {
|
|
||||||
name: `client-${targetKey}`,
|
|
||||||
options,
|
|
||||||
defaultValue: options[0]?.value ?? '',
|
|
||||||
};
|
|
||||||
},
|
|
||||||
},
|
|
||||||
],
|
|
||||||
},
|
|
||||||
{
|
|
||||||
type: 'editor',
|
|
||||||
name: 'code',
|
|
||||||
label: 'Preview',
|
|
||||||
language: getEditorLanguage(initialTarget),
|
|
||||||
defaultValue: initialCode,
|
|
||||||
readOnly: true,
|
|
||||||
rows: 15,
|
|
||||||
dynamic(_ctx, { values }) {
|
|
||||||
const targetKey = String(values.target || defaultTarget);
|
|
||||||
const clientKey = String(
|
|
||||||
values[`client-${targetKey}`] || getDefaultClient(targetKey),
|
|
||||||
);
|
|
||||||
let code: string;
|
|
||||||
try {
|
|
||||||
code = generateSnippet(targetKey, clientKey);
|
|
||||||
} catch {
|
|
||||||
code = '// Error generating snippet';
|
|
||||||
}
|
|
||||||
return {
|
|
||||||
defaultValue: code,
|
|
||||||
language: getEditorLanguage(targetKey),
|
|
||||||
};
|
|
||||||
},
|
|
||||||
},
|
|
||||||
],
|
|
||||||
});
|
|
||||||
|
|
||||||
if (result) {
|
|
||||||
// Store the selected language and library for next time
|
|
||||||
const selectedTarget = String(result.target || initialTarget);
|
|
||||||
const selectedClient = String(
|
|
||||||
result[`client-${selectedTarget}`] || getDefaultClient(selectedTarget),
|
|
||||||
);
|
|
||||||
await ctx.store.set('selectedTarget', selectedTarget);
|
|
||||||
await ctx.store.set(`selectedClient:${selectedTarget}`, selectedClient);
|
|
||||||
|
|
||||||
// Generate snippet for the selected language
|
|
||||||
try {
|
|
||||||
const codeText = generateSnippet(selectedTarget, selectedClient);
|
|
||||||
await ctx.clipboard.copyText(codeText);
|
|
||||||
await ctx.toast.show({
|
|
||||||
message: 'Code snippet copied to clipboard',
|
|
||||||
icon: 'copy',
|
|
||||||
color: 'success',
|
|
||||||
});
|
|
||||||
} catch (err) {
|
|
||||||
await ctx.toast.show({
|
|
||||||
message: `Failed to generate snippet: ${err}`,
|
|
||||||
icon: 'alert_triangle',
|
|
||||||
color: 'danger',
|
|
||||||
});
|
|
||||||
}
|
|
||||||
}
|
|
||||||
},
|
|
||||||
},
|
|
||||||
],
|
|
||||||
};
|
|
||||||
@@ -1,10 +1,10 @@
|
|||||||
{
|
{
|
||||||
"name": "@yaak/mcp-server",
|
"name": "@yaak/mcp-server",
|
||||||
"private": true,
|
"private": true,
|
||||||
"version": "0.2.1",
|
"version": "0.1.7",
|
||||||
"displayName": "MCP Server",
|
"displayName": "MCP Server",
|
||||||
"description": "Expose Yaak functionality via Model Context Protocol",
|
"description": "Expose Yaak functionality via Model Context Protocol",
|
||||||
"minYaakVersion": "2026.1.0",
|
"minYaakVersion": "2025.10.0-beta.6",
|
||||||
"repository": {
|
"repository": {
|
||||||
"type": "git",
|
"type": "git",
|
||||||
"url": "https://github.com/mountain-loop/yaak.git",
|
"url": "https://github.com/mountain-loop/yaak.git",
|
||||||
@@ -17,7 +17,7 @@
|
|||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@hono/mcp": "^0.2.3",
|
"@hono/mcp": "^0.2.3",
|
||||||
"@hono/node-server": "^1.19.7",
|
"@hono/node-server": "^1.19.7",
|
||||||
"@modelcontextprotocol/sdk": "^1.26.0",
|
"@modelcontextprotocol/sdk": "^1.25.2",
|
||||||
"hono": "^4.11.7",
|
"hono": "^4.11.7",
|
||||||
"zod": "^3.25.76"
|
"zod": "^3.25.76"
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -10,7 +10,7 @@ export const plugin: PluginDefinition = {
|
|||||||
async onSelect(ctx, args) {
|
async onSelect(ctx, args) {
|
||||||
const rendered_request = await ctx.httpRequest.render({
|
const rendered_request = await ctx.httpRequest.render({
|
||||||
httpRequest: args.httpRequest,
|
httpRequest: args.httpRequest,
|
||||||
purpose: 'send',
|
purpose: 'preview',
|
||||||
});
|
});
|
||||||
const data = await convertToCurl(rendered_request);
|
const data = await convertToCurl(rendered_request);
|
||||||
await ctx.clipboard.copyText(data);
|
await ctx.clipboard.copyText(data);
|
||||||
|
|||||||
@@ -184,18 +184,6 @@ export function buildHostedCallbackRedirectUri(localPort: number, localPath: str
|
|||||||
return `${HOSTED_CALLBACK_URL}?redirect_to=${encodeURIComponent(localRedirectUri)}`;
|
return `${HOSTED_CALLBACK_URL}?redirect_to=${encodeURIComponent(localRedirectUri)}`;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Stop the active callback server if one is running.
|
|
||||||
* Called during plugin dispose to ensure the server is cleaned up before the process exits.
|
|
||||||
*/
|
|
||||||
export function stopActiveServer(): void {
|
|
||||||
if (activeServer) {
|
|
||||||
console.log('[oauth2] Stopping active callback server during dispose');
|
|
||||||
activeServer.stop();
|
|
||||||
activeServer = null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Open an authorization URL in the system browser, start a local callback server,
|
* Open an authorization URL in the system browser, start a local callback server,
|
||||||
* and wait for the OAuth provider to redirect back.
|
* and wait for the OAuth provider to redirect back.
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import type {
|
|||||||
JsonPrimitive,
|
JsonPrimitive,
|
||||||
PluginDefinition,
|
PluginDefinition,
|
||||||
} from '@yaakapp/api';
|
} from '@yaakapp/api';
|
||||||
import { DEFAULT_LOCALHOST_PORT, HOSTED_CALLBACK_URL, stopActiveServer } from './callbackServer';
|
import { DEFAULT_LOCALHOST_PORT, HOSTED_CALLBACK_URL } from './callbackServer';
|
||||||
import {
|
import {
|
||||||
type CallbackType,
|
type CallbackType,
|
||||||
DEFAULT_PKCE_METHOD,
|
DEFAULT_PKCE_METHOD,
|
||||||
@@ -78,9 +78,6 @@ const accessTokenUrls = [
|
|||||||
];
|
];
|
||||||
|
|
||||||
export const plugin: PluginDefinition = {
|
export const plugin: PluginDefinition = {
|
||||||
dispose() {
|
|
||||||
stopActiveServer();
|
|
||||||
},
|
|
||||||
authentication: {
|
authentication: {
|
||||||
name: 'oauth2',
|
name: 'oauth2',
|
||||||
label: 'OAuth 2.0',
|
label: 'OAuth 2.0',
|
||||||
|
|||||||
@@ -55,7 +55,6 @@ const requestArg: FormInput = {
|
|||||||
type: 'http_request',
|
type: 'http_request',
|
||||||
name: 'request',
|
name: 'request',
|
||||||
label: 'Request',
|
label: 'Request',
|
||||||
defaultValue: '', // Make it not select the active one by default
|
|
||||||
};
|
};
|
||||||
|
|
||||||
export const plugin: PluginDefinition = {
|
export const plugin: PluginDefinition = {
|
||||||
|
|||||||
@@ -1,14 +1,21 @@
|
|||||||
import { getModel } from '@yaakapp-internal/models';
|
import { getModel } from '@yaakapp-internal/models';
|
||||||
|
import { Icon } from '../components/core/Icon';
|
||||||
|
import { HStack } from '../components/core/Stacks';
|
||||||
import type { FolderSettingsTab } from '../components/FolderSettingsDialog';
|
import type { FolderSettingsTab } from '../components/FolderSettingsDialog';
|
||||||
import { FolderSettingsDialog } from '../components/FolderSettingsDialog';
|
import { FolderSettingsDialog } from '../components/FolderSettingsDialog';
|
||||||
import { showDialog } from '../lib/dialog';
|
import { showDialog } from '../lib/dialog';
|
||||||
|
import { resolvedModelName } from '../lib/resolvedModelName';
|
||||||
|
|
||||||
export function openFolderSettings(folderId: string, tab?: FolderSettingsTab) {
|
export function openFolderSettings(folderId: string, tab?: FolderSettingsTab) {
|
||||||
const folder = getModel('folder', folderId);
|
const folder = getModel('folder', folderId);
|
||||||
if (folder == null) return;
|
|
||||||
showDialog({
|
showDialog({
|
||||||
id: 'folder-settings',
|
id: 'folder-settings',
|
||||||
title: null,
|
title: (
|
||||||
|
<HStack space={2} alignItems="center">
|
||||||
|
<Icon icon="folder_cog" size="xl" color="secondary" />
|
||||||
|
{resolvedModelName(folder)}
|
||||||
|
</HStack>
|
||||||
|
),
|
||||||
size: 'lg',
|
size: 'lg',
|
||||||
className: 'h-[50rem]',
|
className: 'h-[50rem]',
|
||||||
noPadding: true,
|
noPadding: true,
|
||||||
|
|||||||
@@ -317,7 +317,7 @@ function TextArg({
|
|||||||
autocompleteFunctions,
|
autocompleteFunctions,
|
||||||
autocompleteVariables,
|
autocompleteVariables,
|
||||||
};
|
};
|
||||||
if (autocompleteVariables || autocompleteFunctions || arg.completionOptions) {
|
if (autocompleteVariables || autocompleteFunctions) {
|
||||||
return <Input {...props} />;
|
return <Input {...props} />;
|
||||||
}
|
}
|
||||||
return <PlainInput {...props} />;
|
return <PlainInput {...props} />;
|
||||||
@@ -360,9 +360,8 @@ function EditorArg({
|
|||||||
className={classNames(
|
className={classNames(
|
||||||
'border border-border rounded-md overflow-hidden px-2 py-1',
|
'border border-border rounded-md overflow-hidden px-2 py-1',
|
||||||
'focus-within:border-border-focus',
|
'focus-within:border-border-focus',
|
||||||
!arg.rows && 'max-h-[10rem]', // So it doesn't take up too much space
|
'max-h-[10rem]', // So it doesn't take up too much space
|
||||||
)}
|
)}
|
||||||
style={arg.rows ? { height: `${arg.rows * 1.4 + 0.75}rem` } : undefined}
|
|
||||||
>
|
>
|
||||||
<Editor
|
<Editor
|
||||||
id={id}
|
id={id}
|
||||||
|
|||||||
@@ -1,25 +1,15 @@
|
|||||||
import {
|
import { createWorkspaceModel, foldersAtom, patchModel } from '@yaakapp-internal/models';
|
||||||
createWorkspaceModel,
|
|
||||||
foldersAtom,
|
|
||||||
patchModel,
|
|
||||||
} from '@yaakapp-internal/models';
|
|
||||||
import { useAtomValue } from 'jotai';
|
import { useAtomValue } from 'jotai';
|
||||||
import { Fragment, useMemo } from 'react';
|
import { useMemo } from 'react';
|
||||||
import { useAuthTab } from '../hooks/useAuthTab';
|
import { useAuthTab } from '../hooks/useAuthTab';
|
||||||
import { useEnvironmentsBreakdown } from '../hooks/useEnvironmentsBreakdown';
|
import { useEnvironmentsBreakdown } from '../hooks/useEnvironmentsBreakdown';
|
||||||
import { useHeadersTab } from '../hooks/useHeadersTab';
|
import { useHeadersTab } from '../hooks/useHeadersTab';
|
||||||
import { useInheritedHeaders } from '../hooks/useInheritedHeaders';
|
import { useInheritedHeaders } from '../hooks/useInheritedHeaders';
|
||||||
import { useModelAncestors } from '../hooks/useModelAncestors';
|
|
||||||
import { deleteModelWithConfirm } from '../lib/deleteModelWithConfirm';
|
|
||||||
import { hideDialog } from '../lib/dialog';
|
|
||||||
import { CopyIconButton } from './CopyIconButton';
|
|
||||||
import { Button } from './core/Button';
|
import { Button } from './core/Button';
|
||||||
import { CountBadge } from './core/CountBadge';
|
import { CountBadge } from './core/CountBadge';
|
||||||
import { Icon } from './core/Icon';
|
|
||||||
import { InlineCode } from './core/InlineCode';
|
|
||||||
import { Input } from './core/Input';
|
import { Input } from './core/Input';
|
||||||
import { Link } from './core/Link';
|
import { Link } from './core/Link';
|
||||||
import { HStack, VStack } from './core/Stacks';
|
import { VStack } from './core/Stacks';
|
||||||
import type { TabItem } from './core/Tabs/Tabs';
|
import type { TabItem } from './core/Tabs/Tabs';
|
||||||
import { TabContent, Tabs } from './core/Tabs/Tabs';
|
import { TabContent, Tabs } from './core/Tabs/Tabs';
|
||||||
import { EmptyStateText } from './EmptyStateText';
|
import { EmptyStateText } from './EmptyStateText';
|
||||||
@@ -47,8 +37,6 @@ export type FolderSettingsTab =
|
|||||||
export function FolderSettingsDialog({ folderId, tab }: Props) {
|
export function FolderSettingsDialog({ folderId, tab }: Props) {
|
||||||
const folders = useAtomValue(foldersAtom);
|
const folders = useAtomValue(foldersAtom);
|
||||||
const folder = folders.find((f) => f.id === folderId) ?? null;
|
const folder = folders.find((f) => f.id === folderId) ?? null;
|
||||||
const ancestors = useModelAncestors(folder);
|
|
||||||
const breadcrumbs = useMemo(() => ancestors.toReversed(), [ancestors]);
|
|
||||||
const authTab = useAuthTab(TAB_AUTH, folder);
|
const authTab = useAuthTab(TAB_AUTH, folder);
|
||||||
const headersTab = useHeadersTab(TAB_HEADERS, folder);
|
const headersTab = useHeadersTab(TAB_HEADERS, folder);
|
||||||
const inheritedHeaders = useInheritedHeaders(folder);
|
const inheritedHeaders = useInheritedHeaders(folder);
|
||||||
@@ -79,132 +67,76 @@ export function FolderSettingsDialog({ folderId, tab }: Props) {
|
|||||||
if (folder == null) return null;
|
if (folder == null) return null;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="h-full flex flex-col">
|
<Tabs
|
||||||
<div className="flex items-center gap-3 px-6 pr-10 mt-4 mb-2 min-w-0 text-xl">
|
defaultValue={tab ?? TAB_GENERAL}
|
||||||
<Icon icon="folder_cog" size="lg" color="secondary" className="flex-shrink-0" />
|
label="Folder Settings"
|
||||||
<div className="flex items-center gap-1.5 font-semibold text-text min-w-0 overflow-hidden flex-1">
|
className="pt-2 pb-2 pl-3 pr-1"
|
||||||
{breadcrumbs.map((item, index) => (
|
layout="horizontal"
|
||||||
<Fragment key={item.id}>
|
addBorders
|
||||||
{index > 0 && (
|
tabs={tabs}
|
||||||
<Icon
|
>
|
||||||
icon="chevron_right"
|
<TabContent value={TAB_AUTH} className="overflow-y-auto h-full px-4">
|
||||||
size="lg"
|
<HttpAuthenticationEditor model={folder} />
|
||||||
className="opacity-50 flex-shrink-0"
|
</TabContent>
|
||||||
/>
|
<TabContent value={TAB_GENERAL} className="overflow-y-auto h-full px-4">
|
||||||
)}
|
<VStack space={3} className="pb-3 h-full">
|
||||||
<span className="text-text-subtle truncate min-w-0" title={item.name}>
|
<Input
|
||||||
{item.name}
|
label="Folder Name"
|
||||||
</span>
|
defaultValue={folder.name}
|
||||||
</Fragment>
|
onChange={(name) => patchModel(folder, { name })}
|
||||||
))}
|
stateKey={`name.${folder.id}`}
|
||||||
{breadcrumbs.length > 0 && (
|
|
||||||
<Icon icon="chevron_right" size="lg" className="opacity-50 flex-shrink-0" />
|
|
||||||
)}
|
|
||||||
<span
|
|
||||||
className="whitespace-nowrap"
|
|
||||||
title={folder.name}
|
|
||||||
>
|
|
||||||
{folder.name}
|
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<Tabs
|
|
||||||
defaultValue={tab ?? TAB_GENERAL}
|
|
||||||
label="Folder Settings"
|
|
||||||
className="pt-2 pb-2 pl-3 pr-1 flex-1"
|
|
||||||
layout="horizontal"
|
|
||||||
addBorders
|
|
||||||
tabs={tabs}
|
|
||||||
>
|
|
||||||
<TabContent value={TAB_AUTH} className="overflow-y-auto h-full px-4">
|
|
||||||
<HttpAuthenticationEditor model={folder} />
|
|
||||||
</TabContent>
|
|
||||||
<TabContent value={TAB_GENERAL} className="overflow-y-auto h-full px-4">
|
|
||||||
<div className="grid grid-rows-[auto_minmax(0,1fr)_auto] gap-3 pb-3 h-full">
|
|
||||||
<Input
|
|
||||||
label="Folder Name"
|
|
||||||
defaultValue={folder.name}
|
|
||||||
onChange={(name) => patchModel(folder, { name })}
|
|
||||||
stateKey={`name.${folder.id}`}
|
|
||||||
/>
|
|
||||||
<MarkdownEditor
|
|
||||||
name="folder-description"
|
|
||||||
placeholder="Folder description"
|
|
||||||
className="border border-border px-2"
|
|
||||||
defaultValue={folder.description}
|
|
||||||
stateKey={`description.${folder.id}`}
|
|
||||||
onChange={(description) => patchModel(folder, { description })}
|
|
||||||
/>
|
|
||||||
<HStack alignItems="center" justifyContent="between" className="w-full">
|
|
||||||
<Button
|
|
||||||
onClick={async () => {
|
|
||||||
const didDelete = await deleteModelWithConfirm(folder);
|
|
||||||
if (didDelete) {
|
|
||||||
hideDialog('folder-settings');
|
|
||||||
}
|
|
||||||
}}
|
|
||||||
color="danger"
|
|
||||||
variant="border"
|
|
||||||
size="xs"
|
|
||||||
>
|
|
||||||
Delete Folder
|
|
||||||
</Button>
|
|
||||||
<InlineCode className="flex gap-1 items-center text-primary pl-2.5">
|
|
||||||
{folder.id}
|
|
||||||
<CopyIconButton
|
|
||||||
className="opacity-70 !text-primary"
|
|
||||||
size="2xs"
|
|
||||||
iconSize="sm"
|
|
||||||
title="Copy folder ID"
|
|
||||||
text={folder.id}
|
|
||||||
/>
|
|
||||||
</InlineCode>
|
|
||||||
</HStack>
|
|
||||||
</div>
|
|
||||||
</TabContent>
|
|
||||||
<TabContent value={TAB_HEADERS} className="overflow-y-auto h-full px-4">
|
|
||||||
<HeadersEditor
|
|
||||||
inheritedHeaders={inheritedHeaders}
|
|
||||||
forceUpdateKey={folder.id}
|
|
||||||
headers={folder.headers}
|
|
||||||
onChange={(headers) => patchModel(folder, { headers })}
|
|
||||||
stateKey={`headers.${folder.id}`}
|
|
||||||
/>
|
/>
|
||||||
</TabContent>
|
<MarkdownEditor
|
||||||
<TabContent value={TAB_VARIABLES} className="overflow-y-auto h-full px-4">
|
name="folder-description"
|
||||||
{folderEnvironment == null ? (
|
placeholder="Folder description"
|
||||||
<EmptyStateText>
|
className="border border-border px-2"
|
||||||
<VStack alignItems="center" space={1.5}>
|
defaultValue={folder.description}
|
||||||
<p>
|
stateKey={`description.${folder.id}`}
|
||||||
Override{' '}
|
onChange={(description) => patchModel(folder, { description })}
|
||||||
<Link href="https://yaak.app/docs/using-yaak/environments-and-variables">
|
/>
|
||||||
Variables
|
</VStack>
|
||||||
</Link>{' '}
|
</TabContent>
|
||||||
for requests within this folder.
|
<TabContent value={TAB_HEADERS} className="overflow-y-auto h-full px-4">
|
||||||
</p>
|
<HeadersEditor
|
||||||
<Button
|
inheritedHeaders={inheritedHeaders}
|
||||||
variant="border"
|
forceUpdateKey={folder.id}
|
||||||
size="sm"
|
headers={folder.headers}
|
||||||
onClick={async () => {
|
onChange={(headers) => patchModel(folder, { headers })}
|
||||||
await createWorkspaceModel({
|
stateKey={`headers.${folder.id}`}
|
||||||
workspaceId: folder.workspaceId,
|
/>
|
||||||
parentModel: 'folder',
|
</TabContent>
|
||||||
parentId: folder.id,
|
<TabContent value={TAB_VARIABLES} className="overflow-y-auto h-full px-4">
|
||||||
model: 'environment',
|
{folderEnvironment == null ? (
|
||||||
name: 'Folder Environment',
|
<EmptyStateText>
|
||||||
});
|
<VStack alignItems="center" space={1.5}>
|
||||||
}}
|
<p>
|
||||||
>
|
Override{' '}
|
||||||
Create Folder Environment
|
<Link href="https://yaak.app/docs/using-yaak/environments-and-variables">
|
||||||
</Button>
|
Variables
|
||||||
</VStack>
|
</Link>{' '}
|
||||||
</EmptyStateText>
|
for requests within this folder.
|
||||||
) : (
|
</p>
|
||||||
<EnvironmentEditor hideName environment={folderEnvironment} />
|
<Button
|
||||||
)}
|
variant="border"
|
||||||
</TabContent>
|
size="sm"
|
||||||
</Tabs>
|
onClick={async () => {
|
||||||
</div>
|
await createWorkspaceModel({
|
||||||
|
workspaceId: folder.workspaceId,
|
||||||
|
parentModel: 'folder',
|
||||||
|
parentId: folder.id,
|
||||||
|
model: 'environment',
|
||||||
|
name: 'Folder Environment',
|
||||||
|
});
|
||||||
|
}}
|
||||||
|
>
|
||||||
|
Create Folder Environment
|
||||||
|
</Button>
|
||||||
|
</VStack>
|
||||||
|
</EmptyStateText>
|
||||||
|
) : (
|
||||||
|
<EnvironmentEditor hideName environment={folderEnvironment} />
|
||||||
|
)}
|
||||||
|
</TabContent>
|
||||||
|
</Tabs>
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -239,7 +239,7 @@ export function HttpResponsePane({ style, className, activeRequestId }: Props) {
|
|||||||
<HttpMultipartViewer response={activeResponse} />
|
<HttpMultipartViewer response={activeResponse} />
|
||||||
) : mimeType?.match(/pdf/i) ? (
|
) : mimeType?.match(/pdf/i) ? (
|
||||||
<EnsureCompleteResponse response={activeResponse} Component={PdfViewer} />
|
<EnsureCompleteResponse response={activeResponse} Component={PdfViewer} />
|
||||||
) : mimeType?.match(/csv|tab-separated/i) && viewMode === 'pretty' ? (
|
) : mimeType?.match(/csv|tab-separated/i) ? (
|
||||||
<HttpCsvViewer className="pb-2" response={activeResponse} />
|
<HttpCsvViewer className="pb-2" response={activeResponse} />
|
||||||
) : (
|
) : (
|
||||||
<HTMLOrTextViewer
|
<HTMLOrTextViewer
|
||||||
|
|||||||
@@ -135,7 +135,6 @@ export function Workspace() {
|
|||||||
open={!floatingSidebarHidden}
|
open={!floatingSidebarHidden}
|
||||||
portalName="sidebar"
|
portalName="sidebar"
|
||||||
onClose={() => setFloatingSidebarHidden(true)}
|
onClose={() => setFloatingSidebarHidden(true)}
|
||||||
zIndex={20}
|
|
||||||
>
|
>
|
||||||
<m.div
|
<m.div
|
||||||
initial={{ opacity: 0, x: -20 }}
|
initial={{ opacity: 0, x: -20 }}
|
||||||
|
|||||||
@@ -5,7 +5,6 @@ import { useMemo } from 'react';
|
|||||||
import { Overlay } from '../Overlay';
|
import { Overlay } from '../Overlay';
|
||||||
import { Heading } from './Heading';
|
import { Heading } from './Heading';
|
||||||
import { IconButton } from './IconButton';
|
import { IconButton } from './IconButton';
|
||||||
import { DialogSize } from '@yaakapp-internal/plugins';
|
|
||||||
|
|
||||||
export interface DialogProps {
|
export interface DialogProps {
|
||||||
children: ReactNode;
|
children: ReactNode;
|
||||||
@@ -15,7 +14,7 @@ export interface DialogProps {
|
|||||||
title?: ReactNode;
|
title?: ReactNode;
|
||||||
description?: ReactNode;
|
description?: ReactNode;
|
||||||
className?: string;
|
className?: string;
|
||||||
size?: DialogSize;
|
size?: 'sm' | 'md' | 'lg' | 'full' | 'dynamic';
|
||||||
hideX?: boolean;
|
hideX?: boolean;
|
||||||
noPadding?: boolean;
|
noPadding?: boolean;
|
||||||
noScroll?: boolean;
|
noScroll?: boolean;
|
||||||
|
|||||||
@@ -5,14 +5,11 @@ import {
|
|||||||
completionKeymap,
|
completionKeymap,
|
||||||
} from '@codemirror/autocomplete';
|
} from '@codemirror/autocomplete';
|
||||||
import { history, historyKeymap } from '@codemirror/commands';
|
import { history, historyKeymap } from '@codemirror/commands';
|
||||||
import { go } from '@codemirror/lang-go';
|
|
||||||
import { java } from '@codemirror/lang-java';
|
|
||||||
import { javascript } from '@codemirror/lang-javascript';
|
import { javascript } from '@codemirror/lang-javascript';
|
||||||
import { json } from '@codemirror/lang-json';
|
import { json } from '@codemirror/lang-json';
|
||||||
import { markdown } from '@codemirror/lang-markdown';
|
import { markdown } from '@codemirror/lang-markdown';
|
||||||
import { php } from '@codemirror/lang-php';
|
|
||||||
import { python } from '@codemirror/lang-python';
|
|
||||||
import { xml } from '@codemirror/lang-xml';
|
import { xml } from '@codemirror/lang-xml';
|
||||||
|
import type { LanguageSupport } from '@codemirror/language';
|
||||||
import {
|
import {
|
||||||
bracketMatching,
|
bracketMatching,
|
||||||
codeFolding,
|
codeFolding,
|
||||||
@@ -20,19 +17,8 @@ import {
|
|||||||
foldKeymap,
|
foldKeymap,
|
||||||
HighlightStyle,
|
HighlightStyle,
|
||||||
indentOnInput,
|
indentOnInput,
|
||||||
LanguageSupport,
|
|
||||||
StreamLanguage,
|
|
||||||
syntaxHighlighting,
|
syntaxHighlighting,
|
||||||
} from '@codemirror/language';
|
} from '@codemirror/language';
|
||||||
import { c, csharp, kotlin, objectiveC } from '@codemirror/legacy-modes/mode/clike';
|
|
||||||
import { clojure } from '@codemirror/legacy-modes/mode/clojure';
|
|
||||||
import { http } from '@codemirror/legacy-modes/mode/http';
|
|
||||||
import { oCaml } from '@codemirror/legacy-modes/mode/mllike';
|
|
||||||
import { powerShell } from '@codemirror/legacy-modes/mode/powershell';
|
|
||||||
import { r } from '@codemirror/legacy-modes/mode/r';
|
|
||||||
import { ruby } from '@codemirror/legacy-modes/mode/ruby';
|
|
||||||
import { shell } from '@codemirror/legacy-modes/mode/shell';
|
|
||||||
import { swift } from '@codemirror/legacy-modes/mode/swift';
|
|
||||||
import { linter, lintGutter, lintKeymap } from '@codemirror/lint';
|
import { linter, lintGutter, lintKeymap } from '@codemirror/lint';
|
||||||
|
|
||||||
import { search, searchKeymap } from '@codemirror/search';
|
import { search, searchKeymap } from '@codemirror/search';
|
||||||
@@ -97,10 +83,6 @@ const syntaxTheme = EditorView.theme({}, { dark: true });
|
|||||||
|
|
||||||
const closeBracketsExtensions: Extension = [closeBrackets(), keymap.of([...closeBracketsKeymap])];
|
const closeBracketsExtensions: Extension = [closeBrackets(), keymap.of([...closeBracketsKeymap])];
|
||||||
|
|
||||||
const legacyLang = (mode: Parameters<typeof StreamLanguage.define>[0]) => {
|
|
||||||
return () => new LanguageSupport(StreamLanguage.define(mode));
|
|
||||||
};
|
|
||||||
|
|
||||||
const syntaxExtensions: Record<
|
const syntaxExtensions: Record<
|
||||||
NonNullable<EditorProps['language']>,
|
NonNullable<EditorProps['language']>,
|
||||||
null | (() => LanguageSupport)
|
null | (() => LanguageSupport)
|
||||||
@@ -116,22 +98,6 @@ const syntaxExtensions: Record<
|
|||||||
text: text,
|
text: text,
|
||||||
timeline: timeline,
|
timeline: timeline,
|
||||||
markdown: markdown,
|
markdown: markdown,
|
||||||
c: legacyLang(c),
|
|
||||||
clojure: legacyLang(clojure),
|
|
||||||
csharp: legacyLang(csharp),
|
|
||||||
go: go,
|
|
||||||
http: legacyLang(http),
|
|
||||||
java: java,
|
|
||||||
kotlin: legacyLang(kotlin),
|
|
||||||
objective_c: legacyLang(objectiveC),
|
|
||||||
ocaml: legacyLang(oCaml),
|
|
||||||
php: php,
|
|
||||||
powershell: legacyLang(powerShell),
|
|
||||||
python: python,
|
|
||||||
r: legacyLang(r),
|
|
||||||
ruby: legacyLang(ruby),
|
|
||||||
shell: legacyLang(shell),
|
|
||||||
swift: legacyLang(swift),
|
|
||||||
};
|
};
|
||||||
|
|
||||||
const closeBracketsFor: (keyof typeof syntaxExtensions)[] = ['json', 'javascript', 'graphql'];
|
const closeBracketsFor: (keyof typeof syntaxExtensions)[] = ['json', 'javascript', 'graphql'];
|
||||||
|
|||||||
@@ -111,7 +111,6 @@ import {
|
|||||||
RefreshCcwIcon,
|
RefreshCcwIcon,
|
||||||
RefreshCwIcon,
|
RefreshCwIcon,
|
||||||
RocketIcon,
|
RocketIcon,
|
||||||
RotateCcwIcon,
|
|
||||||
Rows2Icon,
|
Rows2Icon,
|
||||||
SaveIcon,
|
SaveIcon,
|
||||||
SearchIcon,
|
SearchIcon,
|
||||||
@@ -250,7 +249,6 @@ const icons = {
|
|||||||
puzzle: PuzzleIcon,
|
puzzle: PuzzleIcon,
|
||||||
refresh: RefreshCwIcon,
|
refresh: RefreshCwIcon,
|
||||||
rocket: RocketIcon,
|
rocket: RocketIcon,
|
||||||
rotate_ccw: RotateCcwIcon,
|
|
||||||
rows_2: Rows2Icon,
|
rows_2: Rows2Icon,
|
||||||
save: SaveIcon,
|
save: SaveIcon,
|
||||||
search: SearchIcon,
|
search: SearchIcon,
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
import type { FormInput, JsonPrimitive } from '@yaakapp-internal/plugins';
|
import type { FormInput, JsonPrimitive } from '@yaakapp-internal/plugins';
|
||||||
import type { FormEvent } from 'react';
|
import type { FormEvent } from 'react';
|
||||||
import { useCallback, useEffect, useRef, useState } from 'react';
|
import { useCallback, useRef, useState } from 'react';
|
||||||
import { generateId } from '../../lib/generateId';
|
import { generateId } from '../../lib/generateId';
|
||||||
import { DynamicForm } from '../DynamicForm';
|
import { DynamicForm } from '../DynamicForm';
|
||||||
import { Button } from './Button';
|
import { Button } from './Button';
|
||||||
@@ -12,21 +12,16 @@ export interface PromptProps {
|
|||||||
onResult: (value: Record<string, JsonPrimitive> | null) => void;
|
onResult: (value: Record<string, JsonPrimitive> | null) => void;
|
||||||
confirmText?: string;
|
confirmText?: string;
|
||||||
cancelText?: string;
|
cancelText?: string;
|
||||||
onValuesChange?: (values: Record<string, JsonPrimitive>) => void;
|
|
||||||
onInputsUpdated?: (cb: (inputs: FormInput[]) => void) => void;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
export function Prompt({
|
export function Prompt({
|
||||||
onCancel,
|
onCancel,
|
||||||
inputs: initialInputs,
|
inputs,
|
||||||
onResult,
|
onResult,
|
||||||
confirmText = 'Confirm',
|
confirmText = 'Confirm',
|
||||||
cancelText = 'Cancel',
|
cancelText = 'Cancel',
|
||||||
onValuesChange,
|
|
||||||
onInputsUpdated,
|
|
||||||
}: PromptProps) {
|
}: PromptProps) {
|
||||||
const [value, setValue] = useState<Record<string, JsonPrimitive>>({});
|
const [value, setValue] = useState<Record<string, JsonPrimitive>>({});
|
||||||
const [inputs, setInputs] = useState<FormInput[]>(initialInputs);
|
|
||||||
const handleSubmit = useCallback(
|
const handleSubmit = useCallback(
|
||||||
(e: FormEvent<HTMLFormElement>) => {
|
(e: FormEvent<HTMLFormElement>) => {
|
||||||
e.preventDefault();
|
e.preventDefault();
|
||||||
@@ -35,16 +30,6 @@ export function Prompt({
|
|||||||
[onResult, value],
|
[onResult, value],
|
||||||
);
|
);
|
||||||
|
|
||||||
// Register callback for external input updates (from plugin dynamic resolution)
|
|
||||||
useEffect(() => {
|
|
||||||
onInputsUpdated?.(setInputs);
|
|
||||||
}, [onInputsUpdated]);
|
|
||||||
|
|
||||||
// Notify of value changes for dynamic resolution
|
|
||||||
useEffect(() => {
|
|
||||||
onValuesChange?.(value);
|
|
||||||
}, [value, onValuesChange]);
|
|
||||||
|
|
||||||
const id = `prompt.form.${useRef(generateId()).current}`;
|
const id = `prompt.form.${useRef(generateId()).current}`;
|
||||||
|
|
||||||
return (
|
return (
|
||||||
|
|||||||
@@ -1,66 +0,0 @@
|
|||||||
import classNames from 'classnames';
|
|
||||||
import type { ReactNode } from 'react';
|
|
||||||
|
|
||||||
export interface RadioCardOption<T extends string> {
|
|
||||||
value: T;
|
|
||||||
label: ReactNode;
|
|
||||||
description?: ReactNode;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface RadioCardsProps<T extends string> {
|
|
||||||
value: T | null;
|
|
||||||
onChange: (value: T) => void;
|
|
||||||
options: RadioCardOption<T>[];
|
|
||||||
name: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export function RadioCards<T extends string>({
|
|
||||||
value,
|
|
||||||
onChange,
|
|
||||||
options,
|
|
||||||
name,
|
|
||||||
}: RadioCardsProps<T>) {
|
|
||||||
return (
|
|
||||||
<div className="flex flex-col gap-2">
|
|
||||||
{options.map((option) => {
|
|
||||||
const selected = value === option.value;
|
|
||||||
return (
|
|
||||||
<label
|
|
||||||
key={option.value}
|
|
||||||
className={classNames(
|
|
||||||
'flex items-start gap-3 p-3 rounded-lg border cursor-pointer',
|
|
||||||
'transition-colors',
|
|
||||||
selected
|
|
||||||
? 'border-border-focus'
|
|
||||||
: 'border-border-subtle hocus:border-text-subtlest',
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
<input
|
|
||||||
type="radio"
|
|
||||||
name={name}
|
|
||||||
value={option.value}
|
|
||||||
checked={selected}
|
|
||||||
onChange={() => onChange(option.value)}
|
|
||||||
className="sr-only"
|
|
||||||
/>
|
|
||||||
<div
|
|
||||||
className={classNames(
|
|
||||||
'mt-1 w-4 h-4 flex-shrink-0 rounded-full border',
|
|
||||||
'flex items-center justify-center',
|
|
||||||
selected ? 'border-focus' : 'border-border',
|
|
||||||
)}
|
|
||||||
>
|
|
||||||
{selected && <div className="w-2 h-2 rounded-full bg-text" />}
|
|
||||||
</div>
|
|
||||||
<div className="flex flex-col gap-0.5">
|
|
||||||
<span className="font-semibold text-text">{option.label}</span>
|
|
||||||
{option.description && (
|
|
||||||
<span className="text-sm text-text-subtle">{option.description}</span>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
</label>
|
|
||||||
);
|
|
||||||
})}
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
@@ -20,7 +20,7 @@ import { InlineCode } from '../core/InlineCode';
|
|||||||
import { gitCallbacks } from './callbacks';
|
import { gitCallbacks } from './callbacks';
|
||||||
import { GitCommitDialog } from './GitCommitDialog';
|
import { GitCommitDialog } from './GitCommitDialog';
|
||||||
import { GitRemotesDialog } from './GitRemotesDialog';
|
import { GitRemotesDialog } from './GitRemotesDialog';
|
||||||
import { handlePullResult, handlePushResult } from './git-util';
|
import { handlePullResult } from './git-util';
|
||||||
import { HistoryDialog } from './HistoryDialog';
|
import { HistoryDialog } from './HistoryDialog';
|
||||||
|
|
||||||
export function GitDropdown() {
|
export function GitDropdown() {
|
||||||
@@ -48,7 +48,6 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
push,
|
push,
|
||||||
pull,
|
pull,
|
||||||
checkout,
|
checkout,
|
||||||
resetChanges,
|
|
||||||
init,
|
init,
|
||||||
},
|
},
|
||||||
] = useGit(syncDir, gitCallbacks(syncDir));
|
] = useGit(syncDir, gitCallbacks(syncDir));
|
||||||
@@ -73,9 +72,6 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
const currentBranch = status.data.headRefShorthand;
|
const currentBranch = status.data.headRefShorthand;
|
||||||
const hasChanges = status.data.entries.some((e) => e.status !== 'current');
|
|
||||||
const hasRemotes = (status.data.origins ?? []).length > 0;
|
|
||||||
const { ahead, behind } = status.data;
|
|
||||||
|
|
||||||
const tryCheckout = (branch: string, force: boolean) => {
|
const tryCheckout = (branch: string, force: boolean) => {
|
||||||
checkout.mutate(
|
checkout.mutate(
|
||||||
@@ -172,13 +168,12 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
{ type: 'separator' },
|
{ type: 'separator' },
|
||||||
{
|
{
|
||||||
label: 'Push',
|
label: 'Push',
|
||||||
disabled: !hasRemotes || ahead === 0,
|
|
||||||
leftSlot: <Icon icon="arrow_up_from_line" />,
|
leftSlot: <Icon icon="arrow_up_from_line" />,
|
||||||
waitForOnSelect: true,
|
waitForOnSelect: true,
|
||||||
async onSelect() {
|
async onSelect() {
|
||||||
await push.mutateAsync(undefined, {
|
await push.mutateAsync(undefined, {
|
||||||
disableToastError: true,
|
disableToastError: true,
|
||||||
onSuccess: handlePushResult,
|
onSuccess: handlePullResult,
|
||||||
onError(err) {
|
onError(err) {
|
||||||
showErrorToast({
|
showErrorToast({
|
||||||
id: 'git-push-error',
|
id: 'git-push-error',
|
||||||
@@ -191,7 +186,7 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
label: 'Pull',
|
label: 'Pull',
|
||||||
disabled: !hasRemotes || behind === 0,
|
hidden: (status.data?.origins ?? []).length === 0,
|
||||||
leftSlot: <Icon icon="arrow_down_to_line" />,
|
leftSlot: <Icon icon="arrow_down_to_line" />,
|
||||||
waitForOnSelect: true,
|
waitForOnSelect: true,
|
||||||
async onSelect() {
|
async onSelect() {
|
||||||
@@ -210,7 +205,6 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
},
|
},
|
||||||
{
|
{
|
||||||
label: 'Commit...',
|
label: 'Commit...',
|
||||||
disabled: !hasChanges,
|
|
||||||
leftSlot: <Icon icon="git_commit_vertical" />,
|
leftSlot: <Icon icon="git_commit_vertical" />,
|
||||||
onSelect() {
|
onSelect() {
|
||||||
showDialog({
|
showDialog({
|
||||||
@@ -224,41 +218,6 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
});
|
});
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
{
|
|
||||||
label: 'Reset Changes',
|
|
||||||
hidden: !hasChanges,
|
|
||||||
leftSlot: <Icon icon="rotate_ccw" />,
|
|
||||||
color: 'danger',
|
|
||||||
async onSelect() {
|
|
||||||
const confirmed = await showConfirm({
|
|
||||||
id: 'git-reset-changes',
|
|
||||||
title: 'Reset Changes',
|
|
||||||
description: 'This will discard all uncommitted changes. This cannot be undone.',
|
|
||||||
confirmText: 'Reset',
|
|
||||||
color: 'danger',
|
|
||||||
});
|
|
||||||
if (!confirmed) return;
|
|
||||||
|
|
||||||
await resetChanges.mutateAsync(undefined, {
|
|
||||||
disableToastError: true,
|
|
||||||
onSuccess() {
|
|
||||||
showToast({
|
|
||||||
id: 'git-reset-success',
|
|
||||||
message: 'Changes have been reset',
|
|
||||||
color: 'success',
|
|
||||||
});
|
|
||||||
sync({ force: true });
|
|
||||||
},
|
|
||||||
onError(err) {
|
|
||||||
showErrorToast({
|
|
||||||
id: 'git-reset-error',
|
|
||||||
title: 'Error resetting changes',
|
|
||||||
message: String(err),
|
|
||||||
});
|
|
||||||
},
|
|
||||||
});
|
|
||||||
},
|
|
||||||
},
|
|
||||||
{ type: 'separator', label: 'Branches', hidden: localBranches.length < 1 },
|
{ type: 'separator', label: 'Branches', hidden: localBranches.length < 1 },
|
||||||
...localBranches.map((branch) => {
|
...localBranches.map((branch) => {
|
||||||
const isCurrent = currentBranch === branch;
|
const isCurrent = currentBranch === branch;
|
||||||
@@ -504,14 +463,8 @@ function SyncDropdownWithSyncDir({ syncDir }: { syncDir: string }) {
|
|||||||
return (
|
return (
|
||||||
<Dropdown fullWidth items={items} onOpen={fetchAll.mutate}>
|
<Dropdown fullWidth items={items} onOpen={fetchAll.mutate}>
|
||||||
<GitMenuButton>
|
<GitMenuButton>
|
||||||
<InlineCode className="flex items-center gap-1">
|
<InlineCode>{currentBranch}</InlineCode>
|
||||||
<Icon icon="git_branch" size="xs" className="opacity-50" />
|
<Icon icon="git_branch" size="sm" />
|
||||||
{currentBranch}
|
|
||||||
</InlineCode>
|
|
||||||
<div className="flex items-center gap-1.5">
|
|
||||||
{ahead > 0 && <span className="text-xs flex items-center gap-0.5"><span className="text-primary">↗</span>{ahead}</span>}
|
|
||||||
{behind > 0 && <span className="text-xs flex items-center gap-0.5"><span className="text-info">↙</span>{behind}</span>}
|
|
||||||
</div>
|
|
||||||
</GitMenuButton>
|
</GitMenuButton>
|
||||||
</Dropdown>
|
</Dropdown>
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -1,8 +1,5 @@
|
|||||||
import type { GitCallbacks } from '@yaakapp-internal/git';
|
import type { GitCallbacks } from '@yaakapp-internal/git';
|
||||||
import { sync } from '../../init/sync';
|
|
||||||
import { promptCredentials } from './credentials';
|
import { promptCredentials } from './credentials';
|
||||||
import { promptDivergedStrategy } from './diverged';
|
|
||||||
import { promptUncommittedChangesStrategy } from './uncommitted';
|
|
||||||
import { addGitRemote } from './showAddRemoteDialog';
|
import { addGitRemote } from './showAddRemoteDialog';
|
||||||
|
|
||||||
export function gitCallbacks(dir: string): GitCallbacks {
|
export function gitCallbacks(dir: string): GitCallbacks {
|
||||||
@@ -15,12 +12,5 @@ export function gitCallbacks(dir: string): GitCallbacks {
|
|||||||
if (creds == null) throw new Error('Cancelled credentials prompt');
|
if (creds == null) throw new Error('Cancelled credentials prompt');
|
||||||
return creds;
|
return creds;
|
||||||
},
|
},
|
||||||
promptDiverged: async ({ remote, branch }) => {
|
|
||||||
return promptDivergedStrategy({ remote, branch });
|
|
||||||
},
|
|
||||||
promptUncommittedChanges: async () => {
|
|
||||||
return promptUncommittedChangesStrategy();
|
|
||||||
},
|
|
||||||
forceSync: () => sync({ force: true }),
|
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,102 +0,0 @@
|
|||||||
import type { DivergedStrategy } from '@yaakapp-internal/git';
|
|
||||||
import { useState } from 'react';
|
|
||||||
import { showDialog } from '../../lib/dialog';
|
|
||||||
import { Button } from '../core/Button';
|
|
||||||
import { InlineCode } from '../core/InlineCode';
|
|
||||||
import { RadioCards } from '../core/RadioCards';
|
|
||||||
import { HStack } from '../core/Stacks';
|
|
||||||
|
|
||||||
type Resolution = 'force_reset' | 'merge';
|
|
||||||
|
|
||||||
const resolutionLabel: Record<Resolution, string> = {
|
|
||||||
force_reset: 'Force Pull',
|
|
||||||
merge: 'Merge',
|
|
||||||
};
|
|
||||||
|
|
||||||
interface DivergedDialogProps {
|
|
||||||
remote: string;
|
|
||||||
branch: string;
|
|
||||||
onResult: (strategy: DivergedStrategy) => void;
|
|
||||||
onHide: () => void;
|
|
||||||
}
|
|
||||||
|
|
||||||
function DivergedDialog({ remote, branch, onResult, onHide }: DivergedDialogProps) {
|
|
||||||
const [selected, setSelected] = useState<Resolution | null>(null);
|
|
||||||
|
|
||||||
const handleSubmit = () => {
|
|
||||||
if (selected == null) return;
|
|
||||||
onResult(selected);
|
|
||||||
onHide();
|
|
||||||
};
|
|
||||||
|
|
||||||
const handleCancel = () => {
|
|
||||||
onResult('cancel');
|
|
||||||
onHide();
|
|
||||||
};
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="flex flex-col gap-4 mb-4">
|
|
||||||
<p className="text-text-subtle">
|
|
||||||
Your local branch has diverged from{' '}
|
|
||||||
<InlineCode>
|
|
||||||
{remote}/{branch}
|
|
||||||
</InlineCode>. How would you like to resolve this?
|
|
||||||
</p>
|
|
||||||
<RadioCards
|
|
||||||
name="diverged-strategy"
|
|
||||||
value={selected}
|
|
||||||
onChange={setSelected}
|
|
||||||
options={[
|
|
||||||
{
|
|
||||||
value: 'merge',
|
|
||||||
label: 'Merge Commit',
|
|
||||||
description: 'Combining local and remote changes into a single merge commit',
|
|
||||||
},
|
|
||||||
{
|
|
||||||
value: 'force_reset',
|
|
||||||
label: 'Force Pull',
|
|
||||||
description: 'Discard local commits and reset to match the remote branch',
|
|
||||||
},
|
|
||||||
]}
|
|
||||||
/>
|
|
||||||
<HStack space={2} justifyContent="start" className="flex-row-reverse">
|
|
||||||
<Button
|
|
||||||
color={selected === 'force_reset' ? 'danger' : 'primary'}
|
|
||||||
disabled={selected == null}
|
|
||||||
onClick={handleSubmit}
|
|
||||||
>
|
|
||||||
{selected != null ? resolutionLabel[selected] : 'Select an option'}
|
|
||||||
</Button>
|
|
||||||
<Button variant="border" onClick={handleCancel}>
|
|
||||||
Cancel
|
|
||||||
</Button>
|
|
||||||
</HStack>
|
|
||||||
</div>
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
export async function promptDivergedStrategy({
|
|
||||||
remote,
|
|
||||||
branch,
|
|
||||||
}: {
|
|
||||||
remote: string;
|
|
||||||
branch: string;
|
|
||||||
}): Promise<DivergedStrategy> {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
showDialog({
|
|
||||||
id: 'git-diverged',
|
|
||||||
title: 'Branches Diverged',
|
|
||||||
hideX: true,
|
|
||||||
size: 'sm',
|
|
||||||
disableBackdropClose: true,
|
|
||||||
onClose: () => resolve('cancel'),
|
|
||||||
render: ({ hide }) =>
|
|
||||||
DivergedDialog({
|
|
||||||
remote,
|
|
||||||
branch,
|
|
||||||
onHide: hide,
|
|
||||||
onResult: resolve,
|
|
||||||
}),
|
|
||||||
});
|
|
||||||
});
|
|
||||||
}
|
|
||||||
@@ -26,11 +26,5 @@ export function handlePullResult(r: PullResult) {
|
|||||||
case 'up_to_date':
|
case 'up_to_date':
|
||||||
showToast({ id: 'pull-nothing', message: 'Already up-to-date', color: 'info' });
|
showToast({ id: 'pull-nothing', message: 'Already up-to-date', color: 'info' });
|
||||||
break;
|
break;
|
||||||
case 'diverged':
|
|
||||||
// Handled by mutation callback before reaching here
|
|
||||||
break;
|
|
||||||
case 'uncommitted_changes':
|
|
||||||
// Handled by mutation callback before reaching here
|
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,13 +0,0 @@
|
|||||||
import type { UncommittedChangesStrategy } from '@yaakapp-internal/git';
|
|
||||||
import { showConfirm } from '../../lib/confirm';
|
|
||||||
|
|
||||||
export async function promptUncommittedChangesStrategy(): Promise<UncommittedChangesStrategy> {
|
|
||||||
const confirmed = await showConfirm({
|
|
||||||
id: 'git-uncommitted-changes',
|
|
||||||
title: 'Uncommitted Changes',
|
|
||||||
description: 'You have uncommitted changes. Commit or reset your changes before pulling.',
|
|
||||||
confirmText: 'Reset and Pull',
|
|
||||||
color: 'danger',
|
|
||||||
});
|
|
||||||
return confirmed ? 'reset' : 'cancel';
|
|
||||||
}
|
|
||||||
@@ -1,12 +1,6 @@
|
|||||||
import { emit } from '@tauri-apps/api/event';
|
import { emit } from '@tauri-apps/api/event';
|
||||||
import { openUrl } from '@tauri-apps/plugin-opener';
|
import { openUrl } from '@tauri-apps/plugin-opener';
|
||||||
import { debounce } from '@yaakapp-internal/lib';
|
import type { InternalEvent, ShowToastRequest } from '@yaakapp-internal/plugins';
|
||||||
import type {
|
|
||||||
FormInput,
|
|
||||||
InternalEvent,
|
|
||||||
JsonPrimitive,
|
|
||||||
ShowToastRequest,
|
|
||||||
} from '@yaakapp-internal/plugins';
|
|
||||||
import { updateAllPlugins } from '@yaakapp-internal/plugins';
|
import { updateAllPlugins } from '@yaakapp-internal/plugins';
|
||||||
import type {
|
import type {
|
||||||
PluginUpdateNotification,
|
PluginUpdateNotification,
|
||||||
@@ -38,9 +32,6 @@ export function initGlobalListeners() {
|
|||||||
|
|
||||||
listenToTauriEvent('settings', () => openSettings.mutate(null));
|
listenToTauriEvent('settings', () => openSettings.mutate(null));
|
||||||
|
|
||||||
// Track active dynamic form dialogs so follow-up input updates can reach them
|
|
||||||
const activeForms = new Map<string, (inputs: FormInput[]) => void>();
|
|
||||||
|
|
||||||
// Listen for plugin events
|
// Listen for plugin events
|
||||||
listenToTauriEvent<InternalEvent>('plugin_event', async ({ payload: event }) => {
|
listenToTauriEvent<InternalEvent>('plugin_event', async ({ payload: event }) => {
|
||||||
if (event.payload.type === 'prompt_text_request') {
|
if (event.payload.type === 'prompt_text_request') {
|
||||||
@@ -58,47 +49,26 @@ export function initGlobalListeners() {
|
|||||||
};
|
};
|
||||||
await emit(event.id, result);
|
await emit(event.id, result);
|
||||||
} else if (event.payload.type === 'prompt_form_request') {
|
} else if (event.payload.type === 'prompt_form_request') {
|
||||||
if (event.replyId != null) {
|
|
||||||
// Follow-up update from plugin runtime — update the active dialog's inputs
|
|
||||||
const updateInputs = activeForms.get(event.replyId);
|
|
||||||
if (updateInputs) {
|
|
||||||
updateInputs(event.payload.inputs);
|
|
||||||
}
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Initial request — show the dialog with bidirectional support
|
|
||||||
const emitFormResponse = (values: Record<string, JsonPrimitive> | null, done: boolean) => {
|
|
||||||
const result: InternalEvent = {
|
|
||||||
id: generateId(),
|
|
||||||
replyId: event.id,
|
|
||||||
pluginName: event.pluginName,
|
|
||||||
pluginRefId: event.pluginRefId,
|
|
||||||
context: event.context,
|
|
||||||
payload: {
|
|
||||||
type: 'prompt_form_response',
|
|
||||||
values,
|
|
||||||
done,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
emit(event.id, result);
|
|
||||||
};
|
|
||||||
|
|
||||||
const values = await showPromptForm({
|
const values = await showPromptForm({
|
||||||
id: event.payload.id,
|
id: event.payload.id,
|
||||||
title: event.payload.title,
|
title: event.payload.title,
|
||||||
description: event.payload.description,
|
description: event.payload.description,
|
||||||
size: event.payload.size,
|
|
||||||
inputs: event.payload.inputs,
|
inputs: event.payload.inputs,
|
||||||
confirmText: event.payload.confirmText,
|
confirmText: event.payload.confirmText,
|
||||||
cancelText: event.payload.cancelText,
|
cancelText: event.payload.cancelText,
|
||||||
onValuesChange: debounce((values) => emitFormResponse(values, false), 150),
|
|
||||||
onInputsUpdated: (cb) => activeForms.set(event.id, cb),
|
|
||||||
});
|
});
|
||||||
|
const result: InternalEvent = {
|
||||||
// Clean up and send final response
|
id: generateId(),
|
||||||
activeForms.delete(event.id);
|
replyId: event.id,
|
||||||
emitFormResponse(values, true);
|
pluginName: event.pluginName,
|
||||||
|
pluginRefId: event.pluginRefId,
|
||||||
|
context: event.context,
|
||||||
|
payload: {
|
||||||
|
type: 'prompt_form_response',
|
||||||
|
values,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
await emit(event.id, result);
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|||||||
@@ -1,32 +1,21 @@
|
|||||||
import type { FormInput, JsonPrimitive } from '@yaakapp-internal/plugins';
|
|
||||||
import type { DialogProps } from '../components/core/Dialog';
|
import type { DialogProps } from '../components/core/Dialog';
|
||||||
import type { PromptProps } from '../components/core/Prompt';
|
import type { PromptProps } from '../components/core/Prompt';
|
||||||
import { Prompt } from '../components/core/Prompt';
|
import { Prompt } from '../components/core/Prompt';
|
||||||
import { showDialog } from './dialog';
|
import { showDialog } from './dialog';
|
||||||
|
|
||||||
type FormArgs = Pick<DialogProps, 'title' | 'description' | 'size'> &
|
type FormArgs = Pick<DialogProps, 'title' | 'description'> &
|
||||||
Omit<PromptProps, 'onClose' | 'onCancel' | 'onResult'> & {
|
Omit<PromptProps, 'onClose' | 'onCancel' | 'onResult'> & {
|
||||||
id: string;
|
id: string;
|
||||||
onValuesChange?: (values: Record<string, JsonPrimitive>) => void;
|
|
||||||
onInputsUpdated?: (cb: (inputs: FormInput[]) => void) => void;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
export async function showPromptForm({
|
export async function showPromptForm({ id, title, description, ...props }: FormArgs) {
|
||||||
id,
|
|
||||||
title,
|
|
||||||
description,
|
|
||||||
size,
|
|
||||||
onValuesChange,
|
|
||||||
onInputsUpdated,
|
|
||||||
...props
|
|
||||||
}: FormArgs) {
|
|
||||||
return new Promise((resolve: PromptProps['onResult']) => {
|
return new Promise((resolve: PromptProps['onResult']) => {
|
||||||
showDialog({
|
showDialog({
|
||||||
id,
|
id,
|
||||||
title,
|
title,
|
||||||
description,
|
description,
|
||||||
hideX: true,
|
hideX: true,
|
||||||
size: size ?? 'sm',
|
size: 'sm',
|
||||||
disableBackdropClose: true, // Prevent accidental dismisses
|
disableBackdropClose: true, // Prevent accidental dismisses
|
||||||
onClose: () => {
|
onClose: () => {
|
||||||
// Click backdrop, close, or escape
|
// Click backdrop, close, or escape
|
||||||
@@ -43,8 +32,6 @@ export async function showPromptForm({
|
|||||||
resolve(v);
|
resolve(v);
|
||||||
hide();
|
hide();
|
||||||
},
|
},
|
||||||
onValuesChange,
|
|
||||||
onInputsUpdated,
|
|
||||||
...props,
|
...props,
|
||||||
}),
|
}),
|
||||||
});
|
});
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ type TauriCmd =
|
|||||||
| 'cmd_call_workspace_action'
|
| 'cmd_call_workspace_action'
|
||||||
| 'cmd_call_folder_action'
|
| 'cmd_call_folder_action'
|
||||||
| 'cmd_check_for_updates'
|
| 'cmd_check_for_updates'
|
||||||
|
| 'cmd_create_grpc_request'
|
||||||
| 'cmd_curl_to_request'
|
| 'cmd_curl_to_request'
|
||||||
| 'cmd_decrypt_template'
|
| 'cmd_decrypt_template'
|
||||||
| 'cmd_default_headers'
|
| 'cmd_default_headers'
|
||||||
@@ -47,7 +48,9 @@ type TauriCmd =
|
|||||||
| 'cmd_save_response'
|
| 'cmd_save_response'
|
||||||
| 'cmd_secure_template'
|
| 'cmd_secure_template'
|
||||||
| 'cmd_send_ephemeral_request'
|
| 'cmd_send_ephemeral_request'
|
||||||
|
| 'cmd_send_folder'
|
||||||
| 'cmd_send_http_request'
|
| 'cmd_send_http_request'
|
||||||
|
| 'cmd_show_workspace_key'
|
||||||
| 'cmd_template_function_summaries'
|
| 'cmd_template_function_summaries'
|
||||||
| 'cmd_template_function_config'
|
| 'cmd_template_function_config'
|
||||||
| 'cmd_template_tokens_to_string';
|
| 'cmd_template_tokens_to_string';
|
||||||
|
|||||||
Reference in New Issue
Block a user