feat: initial work on detecting vars

i want to throw errors when vars are missing, so that its easier to find
and fill them when running pw
This commit is contained in:
əlemi 2024-10-22 21:22:25 +02:00
parent ac92c53799
commit 40f2c0a6d4
3 changed files with 79 additions and 50 deletions

View file

@ -42,6 +42,9 @@ pub enum PostWomanError {
#[error("regex failed matching in content: {0}")]
NoMatch(String),
#[error("missing environment variable: {0}")]
MissingVar(#[from] crate::ext::FillError),
}
#[derive(Debug, thiserror::Error)]

View file

@ -1,3 +1,7 @@
use std::sync::OnceLock;
use crate::PostWomanError;
#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
#[serde(untagged)]
pub enum StringOr<T> {
@ -34,13 +38,40 @@ pub fn stringify_json(v: &serde_json::Value) -> String {
}
}
pub fn var_matcher() -> &'static regex::Regex {
static MATCHER : OnceLock<regex::Regex> = OnceLock::new();
MATCHER.get_or_init(|| regex::Regex::new(r"\$\{(.*)\}").expect("wrong matcher regex"))
}
// keep it as separate fn so we can change it everywhere easily
pub fn full_name(namespace: &str, name: &str) -> String {
format!("{namespace}:{name}")
}
pub trait FillableFromEnvironment {
fn fill(self, env: &toml::Table) -> Self;
#[derive(Debug, thiserror::Error)]
#[error("could not fill {0}")]
pub struct FillError(pub String);
pub trait FillableFromEnvironment: Sized {
fn fill(self, env: &toml::Table) -> Result<Self, FillError>;
fn replace(mut from: String, env: &toml::Table) -> Result<String, FillError> {
let placeholders: Vec<(String, String)> = var_matcher()
.captures_iter(&from)
.map(|m| m.extract())
.map(|(txt, [var])| (txt.to_string(), var.to_string()))
.collect();
// TODO can we avoid cloning all matches??? can't mutate `from` as captures_iter holds an
// immutable reference to original string
for (txt, var) in placeholders {
let value = env.get(&var).ok_or(FillError(var.to_string()))?;
from = from.replace(&txt, &stringify_toml(value));
}
Ok(from)
}
fn default_vars(env: &toml::Table) -> std::collections::HashMap<String, String> {
let mut vars: std::collections::HashMap<String, String> = std::collections::HashMap::default();

View file

@ -8,7 +8,7 @@ use jaq_interpret::FilterT;
use crate::errors::InvalidHeaderError;
use crate::{PostWomanError, APP_USER_AGENT};
use crate::ext::{stringify_json, FillableFromEnvironment, StringOr};
use crate::ext::{stringify_json, var_matcher, FillError, FillableFromEnvironment, StringOr};
use super::{ExtractorConfig, ClientConfig};
@ -146,71 +146,66 @@ impl EndpointConfig {
}
impl FillableFromEnvironment for EndpointConfig {
fn fill(mut self, env: &toml::Table) -> Self {
fn fill(mut self, env: &toml::Table) -> Result<Self, FillError> {
let vars = Self::default_vars(env);
for (k, v) in vars {
let k_var = format!("${{{k}}}");
self.path = self.path.replace(&k_var, &v);
self.path = Self::replace(self.path, env)?;
if let Some(method) = self.method {
self.method = Some(method.replace(&k_var, &v));
self.method = Some(Self::replace(method, env)?);
}
if let Some(b) = self.body {
match b {
StringOr::Str(body) => {
self.body = Some(StringOr::Str(body.replace(&k_var, &v)));
self.body = Some(StringOr::Str(Self::replace(body, env)?));
},
StringOr::T(json) => {
let wrap = toml::Value::Table(json.clone());
let toml::Value::Table(out) = replace_recursive(wrap, &k_var, &v)
let toml::Value::Table(out) = replace_recursive(wrap, env)?
else { unreachable!("we put in a table, we get out a table") };
self.body = Some(StringOr::T(out));
},
}
}
if let Some(query) = self.query {
self.query = Some(
query.into_iter()
.map(|x| x.replace(&k_var, &v))
.collect()
);
for q in query {
q = Self::replace(q, env)?;
}
self.query = Some(query);
}
if let Some(headers) = self.headers {
self.headers = Some(
headers.into_iter()
.map(|x| x.replace(&k_var, &v))
.collect()
);
for h in headers {
h = Self::replace(h, env)?;
}
self.headers = Some(headers);
}
Ok(self)
}
}
self
}
}
fn replace_recursive(element: toml::Value, from: &str, to: &str) -> toml::Value {
match element {
fn replace_recursive(element: toml::Value, env: &toml::Table) -> Result<toml::Value, FillError> {
Ok(match element {
toml::Value::Float(x) => toml::Value::Float(x),
toml::Value::Integer(x) => toml::Value::Integer(x),
toml::Value::Boolean(x) => toml::Value::Boolean(x),
toml::Value::Datetime(x) => toml::Value::Datetime(x),
toml::Value::String(x) => toml::Value::String(x.replace(from, to)),
toml::Value::Array(x) => toml::Value::Array(
x.into_iter().map(|x| replace_recursive(x, from, to)).collect()
),
toml::Value::String(x) => toml::Value::String(EndpointConfig::replace(x, env)?),
toml::Value::Array(arr) => {
for v in arr.iter_mut() {
*v = replace_recursive(v, env)?;
}
toml::Value::Array(arr)
},
toml::Value::Table(map) => {
let mut out = toml::map::Map::new();
for (k, v) in map {
let new_v = replace_recursive(v.clone(), from, to);
if k.contains(from) {
out.insert(k.replace(from, to), new_v);
} else {
out.insert(k.to_string(), new_v);
}
let new_v = replace_recursive(v.clone(), env)?;
let new_k = EndpointConfig::replace(k, env)?;
out.insert(new_k, new_v);
}
toml::Value::Table(out)
},
}
})
}
async fn format_body(res: reqwest::Response) -> Result<String, PostWomanError> {