Merge pull request #42 from caido/amr/eng-878-sdk-client-add-workflowstestpassive-and-testactive

Add WorkflowSDK.testPassive() and testActive() methods

Author: amrelsagaei

packages/sdk-client/src/sdks/workflow.ts

@@ -1,13 +1,17 @@
 import { encodeBlob } from "@/convert/blob.js";
 import {
   mapToTestWorkflowConvertResult,
+  mapToTestWorkflowHttpResult,
   mapToWorkflow,
 } from "@/convert/workflow.js";
 import {
   CreateWorkflowDocument,
   DeleteWorkflowDocument,
   type GraphQLClient,
+  TestWorkflowActiveDocument,
   TestWorkflowConvertDocument,
+  TestWorkflowPassiveDocument,
+  ToggleWorkflowDocument,
   UpdateWorkflowDocument,
   WorkflowDocument,
   WorkflowsDocument,
@@ -17,12 +21,30 @@ import type {
   ID,
   TestWorkflowConvertOptions,
   TestWorkflowConvertResult,
+  TestWorkflowHttpOptions,
+  TestWorkflowHttpResult,
   UpdateWorkflowOptions,
 } from "@/types/index.js";
 import type { Workflow as WorkflowType } from "@/types/workflow.js";
 import { handleGraphQLError } from "@/utils/errors.js";
 import { isAbsent, isPresent } from "@/utils/optional.js";
 
+const buildTestWorkflowHttpInput = (options: TestWorkflowHttpOptions) => ({
+  definition: options.definition,
+  request: {
+    connectionInfo: {
+      host: options.request.connection.host,
+      port: options.request.connection.port,
+      isTLS: options.request.connection.isTLS,
+      SNI: options.request.connection.SNI,
+    },
+    raw: encodeBlob(options.request.raw),
+  },
+  response: isPresent(options.response)
+    ? { raw: encodeBlob(options.response.raw) }
+    : undefined,
+});
+
 /**
  * Higher-level SDK for workflow-related operations.
  */
@@ -119,6 +141,28 @@ export class WorkflowSDK {
     }
   }
 
+  /**
+   * Toggle a workflow's enabled state.
+   */
+  async toggle(id: ID, enabled: boolean): Promise<WorkflowType> {
+    const result = await this.graphql.mutation(ToggleWorkflowDocument, {
+      id,
+      enabled,
+    });
+
+    const payload = result.toggleWorkflow;
+
+    if (isPresent(payload.error)) {
+      handleGraphQLError(payload.error);
+    }
+
+    if (isAbsent(payload.workflow)) {
+      throw new Error("Toggle workflow returned no workflow");
+    }
+
+    return mapToWorkflow(payload.workflow);
+  }
+
   /**
    * Test a workflow against input without saving it.
    */
@@ -147,4 +191,44 @@ export class WorkflowSDK {
       }
     }
   }
+
+  /**
+   * Test a passive workflow against a request and optional response without
+   * saving it.
+   */
+  async testPassive(
+    options: TestWorkflowHttpOptions,
+  ): Promise<TestWorkflowHttpResult> {
+    const result = await this.graphql.mutation(TestWorkflowPassiveDocument, {
+      input: buildTestWorkflowHttpInput(options),
+    });
+
+    const payload = result.testWorkflowPassive;
+
+    if (isPresent(payload.error)) {
+      handleGraphQLError(payload.error);
+    }
+
+    return mapToTestWorkflowHttpResult(payload);
+  }
+
+  /**
+   * Test an active workflow against a request and optional response without
+   * saving it.
+   */
+  async testActive(
+    options: TestWorkflowHttpOptions,
+  ): Promise<TestWorkflowHttpResult> {
+    const result = await this.graphql.mutation(TestWorkflowActiveDocument, {
+      input: buildTestWorkflowHttpInput(options),
+    });
+
+    const payload = result.testWorkflowActive;
+
+    if (isPresent(payload.error)) {
+      handleGraphQLError(payload.error);
+    }
+
+    return mapToTestWorkflowHttpResult(payload);
+  }
 }