use crate::errors::HivehookError;
use crate::resources::_base::{put_opt, vars};
#[cfg(feature = "blocking")]
use crate::transport::BlockingGraphQLTransport;
use crate::types::{ListResult, Transformation, TransformTestResult};
use serde::{Deserialize, Serialize};
use serde_json::Value;
use std::collections::HashMap;
const FRAGMENT: &str = "id name description code enabled failOpen timeoutMs createdAt updatedAt";
#[non_exhaustive]
#[derive(Debug, Default, Clone)]
pub struct ListTransformationsOptions {
pub enabled: Option<bool>,
pub search: Option<String>,
pub after: Option<String>,
pub first: Option<i32>,
}
#[non_exhaustive]
#[derive(Debug, Default, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct CreateTransformationInput {
pub name: String,
pub description: String,
pub code: String,
pub fail_open: bool,
pub timeout_ms: i32,
}
#[non_exhaustive]
#[derive(Debug, Default, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct UpdateTransformationInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub description: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub code: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub enabled: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub fail_open: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub timeout_ms: Option<i32>,
}
#[non_exhaustive]
#[derive(Debug, Default, Clone, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct TestTransformationInput {
pub code: String,
pub payload: Value,
#[serde(skip_serializing_if = "Option::is_none")]
pub event_type: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub headers: Option<HashMap<String, Value>>,
}
#[derive(Deserialize)]
struct ListData {
transformations: ListResult<Transformation>,
}
#[derive(Deserialize)]
struct GetData {
transformation: Option<Transformation>,
}
#[derive(Deserialize)]
struct CreateData {
#[serde(rename = "createTransformation")]
create_transformation: Transformation,
}
#[derive(Deserialize)]
struct UpdateData {
#[serde(rename = "updateTransformation")]
update_transformation: Transformation,
}
#[derive(Deserialize)]
struct DeleteData {
#[serde(rename = "deleteTransformation")]
delete_transformation: bool,
}
#[derive(Deserialize)]
struct TestData {
#[serde(rename = "testTransformation")]
test_transformation: TransformTestResult,
}
#[cfg(feature = "blocking")]
pub struct TransformationService<'a> {
pub(crate) transport: &'a BlockingGraphQLTransport,
}
#[cfg(feature = "blocking")]
impl<'a> TransformationService<'a> {
pub fn list(
&self,
options: ListTransformationsOptions,
) -> Result<ListResult<Transformation>, HivehookError> {
let query = format!(
r#"query($enabled: Boolean, $search: String, $after: String, $first: Int) {{
transformations(enabled: $enabled, search: $search, after: $after, first: $first) {{
nodes {{ {FRAGMENT} }}
pageInfo {{ total limit offset endCursor hasNextPage }}
}}
}}"#
);
let mut v = vars();
put_opt(&mut v, "enabled", options.enabled);
put_opt(&mut v, "search", options.search);
put_opt(&mut v, "after", options.after);
put_opt(&mut v, "first", options.first);
let data: ListData = self.transport.execute(&query, Some(v))?;
Ok(data.transformations)
}
pub fn get(&self, id: &str) -> Result<Option<Transformation>, HivehookError> {
let query = format!("query($id: UUID!) {{ transformation(id: $id) {{ {FRAGMENT} }} }}");
let mut v = vars();
v.insert("id".into(), Value::String(id.into()));
let data: GetData = self.transport.execute(&query, Some(v))?;
Ok(data.transformation)
}
pub fn create(
&self,
input: CreateTransformationInput,
) -> Result<Transformation, HivehookError> {
let query = format!("mutation($input: CreateTransformationInput!) {{ createTransformation(input: $input) {{ {FRAGMENT} }} }}");
let mut v = vars();
v.insert("input".into(), serde_json::to_value(input)?);
let data: CreateData = self.transport.execute(&query, Some(v))?;
Ok(data.create_transformation)
}
pub fn update(
&self,
id: &str,
input: UpdateTransformationInput,
) -> Result<Transformation, HivehookError> {
let query = format!("mutation($id: UUID!, $input: UpdateTransformationInput!) {{ updateTransformation(id: $id, input: $input) {{ {FRAGMENT} }} }}");
let mut v = vars();
v.insert("id".into(), Value::String(id.into()));
v.insert("input".into(), serde_json::to_value(input)?);
let data: UpdateData = self.transport.execute(&query, Some(v))?;
Ok(data.update_transformation)
}
pub fn delete(&self, id: &str) -> Result<bool, HivehookError> {
let mut v = vars();
v.insert("id".into(), Value::String(id.into()));
let data: DeleteData = self.transport.execute(
"mutation($id: UUID!) { deleteTransformation(id: $id) }",
Some(v),
)?;
Ok(data.delete_transformation)
}
pub fn test(
&self,
input: TestTransformationInput,
) -> Result<TransformTestResult, HivehookError> {
let mut v = vars();
v.insert("input".into(), serde_json::to_value(input)?);
let data: TestData = self.transport.execute(
"mutation($input: TestTransformationInput!) { testTransformation(input: $input) { success output error durationMs } }",
Some(v),
)?;
Ok(data.test_transformation)
}
}
#[cfg(feature = "async")]
#[cfg_attr(docsrs, doc(cfg(feature = "async")))]
pub struct AsyncTransformationService<'a> {
pub(crate) transport: &'a crate::transport::AsyncGraphQLTransport,
}
#[cfg(feature = "async")]
impl<'a> AsyncTransformationService<'a> {
pub async fn list(
&self,
options: ListTransformationsOptions,
) -> Result<ListResult<Transformation>, HivehookError> {
let query = format!(
r#"query($enabled: Boolean, $search: String, $after: String, $first: Int) {{
transformations(enabled: $enabled, search: $search, after: $after, first: $first) {{
nodes {{ {FRAGMENT} }}
pageInfo {{ total limit offset endCursor hasNextPage }}
}}
}}"#
);
let mut v = vars();
put_opt(&mut v, "enabled", options.enabled);
put_opt(&mut v, "search", options.search);
put_opt(&mut v, "after", options.after);
put_opt(&mut v, "first", options.first);
let data: ListData = self.transport.execute(&query, Some(v)).await?;
Ok(data.transformations)
}
pub async fn get(&self, id: &str) -> Result<Option<Transformation>, HivehookError> {
let query = format!("query($id: UUID!) {{ transformation(id: $id) {{ {FRAGMENT} }} }}");
let mut v = vars();
v.insert("id".into(), Value::String(id.into()));
let data: GetData = self.transport.execute(&query, Some(v)).await?;
Ok(data.transformation)
}
pub async fn create(
&self,
input: CreateTransformationInput,
) -> Result<Transformation, HivehookError> {
let query = format!("mutation($input: CreateTransformationInput!) {{ createTransformation(input: $input) {{ {FRAGMENT} }} }}");
let mut v = vars();
v.insert("input".into(), serde_json::to_value(input)?);
let data: CreateData = self.transport.execute(&query, Some(v)).await?;
Ok(data.create_transformation)
}
pub async fn update(
&self,
id: &str,
input: UpdateTransformationInput,
) -> Result<Transformation, HivehookError> {
let query = format!("mutation($id: UUID!, $input: UpdateTransformationInput!) {{ updateTransformation(id: $id, input: $input) {{ {FRAGMENT} }} }}");
let mut v = vars();
v.insert("id".into(), Value::String(id.into()));
v.insert("input".into(), serde_json::to_value(input)?);
let data: UpdateData = self.transport.execute(&query, Some(v)).await?;
Ok(data.update_transformation)
}
pub async fn delete(&self, id: &str) -> Result<bool, HivehookError> {
let mut v = vars();
v.insert("id".into(), Value::String(id.into()));
let data: DeleteData = self
.transport
.execute(
"mutation($id: UUID!) { deleteTransformation(id: $id) }",
Some(v),
)
.await?;
Ok(data.delete_transformation)
}
pub async fn test(
&self,
input: TestTransformationInput,
) -> Result<TransformTestResult, HivehookError> {
let mut v = vars();
v.insert("input".into(), serde_json::to_value(input)?);
let data: TestData = self.transport.execute(
"mutation($input: TestTransformationInput!) { testTransformation(input: $input) { success output error durationMs } }",
Some(v),
).await?;
Ok(data.test_transformation)
}
}