use extism_pdk::*;
use serde::{Deserialize, Serialize};
#[derive(Debug, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
struct ConfigGetRequest {
key: String,
}
#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "camelCase")]
struct ConfigGetResponse {
#[serde(default)]
value: String,
#[serde(default)]
exists: bool,
}
#[derive(Debug, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
struct ConfigGetIntRequest {
key: String,
}
#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "camelCase")]
struct ConfigGetIntResponse {
#[serde(default)]
value: i64,
#[serde(default)]
exists: bool,
}
#[derive(Debug, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
struct ConfigKeysRequest {
prefix: String,
}
#[derive(Debug, Clone, Deserialize)]
#[serde(rename_all = "camelCase")]
struct ConfigKeysResponse {
#[serde(default)]
keys: Vec<String>,
}
#[host_fn]
extern "ExtismHost" {
fn config_get(input: Json<ConfigGetRequest>) -> Json<ConfigGetResponse>;
fn config_getint(input: Json<ConfigGetIntRequest>) -> Json<ConfigGetIntResponse>;
fn config_keys(input: Json<ConfigKeysRequest>) -> Json<ConfigKeysResponse>;
}
pub fn get(key: &str) -> Result<Option<String>, Error> {
let response = unsafe {
config_get(Json(ConfigGetRequest {
key: key.to_owned(),
}))?
};
if response.0.exists {
Ok(Some(response.0.value))
} else {
Ok(None)
}
}
pub fn get_int(key: &str) -> Result<Option<i64>, Error> {
let response = unsafe {
config_getint(Json(ConfigGetIntRequest {
key: key.to_owned(),
}))?
};
if response.0.exists {
Ok(Some(response.0.value))
} else {
Ok(None)
}
}
pub fn keys(prefix: &str) -> Result<Vec<String>, Error> {
let response = unsafe {
config_keys(Json(ConfigKeysRequest {
prefix: prefix.to_owned(),
}))?
};
Ok(response.0.keys)
}