Add option to remove temporary integration and component after a comp…

…onent:dev:test (#22)

* Add option to remove temporary integration and component after a component:dev:test

* PR Feedback

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@prismatic-io/prism",
-  "version": "4.6.3",
+  "version": "4.6.4",
   "description": "Build, deploy, and support integrations in Prismatic from the comfort of your command line",
   "keywords": [
     "prismatic",
@@ -163,4 +163,4 @@
       }
     }
   }
-}
+}

src/commands/components/dev/test.ts

@@ -20,7 +20,10 @@ import {
   ComponentTestInfo,
 } from "../../../utils/integration/definition";
 import { importDefinition } from "../../../utils/integration/import";
-import { runIntegrationFlow } from "../../../utils/integration/invoke";
+import {
+  deleteIntegration,
+  runIntegrationFlow,
+} from "../../../utils/integration/invoke";
 import { pollForActiveConfigVarState } from "../../../utils/integration/query";
 import { Expression } from "../../../utils/integration/export";
 import { exists } from "../../../fs";
@@ -30,6 +33,7 @@ import {
   printFinalStepResults,
   writeFinalStepResults,
 } from "../../../utils/execution/stepResults";
+import { deleteComponentByKey } from "../../../utils/component/deleteByKey";
 
 const setTimeoutPromise = promisify(setTimeout);
 
@@ -168,6 +172,12 @@ export default class TestCommand extends Command {
       default: false,
       description: "Print the results of the action to stdout",
     }),
+    "clean-up": Flags.boolean({
+      required: false,
+      default: false,
+      description:
+        "Clean up the integration and temporary component after running the action",
+    }),
   };
 
   async run() {
@@ -177,6 +187,7 @@ export default class TestCommand extends Command {
         build,
         "output-file": outputFile,
         "print-results": printResults,
+        "clean-up": cleanUp,
       },
     } = await this.parse(TestCommand);
 
@@ -379,6 +390,16 @@ export default class TestCommand extends Command {
       await printFinalStepResults(executionId);
     }
 
+    if (cleanUp) {
+      CliUx.ux.action.start(`Cleaning up test Integration (${integrationId})`);
+      await deleteIntegration(integrationId);
+      CliUx.ux.action.stop();
+
+      CliUx.ux.action.start(`Cleaning up test component (${definition.key})`);
+      await deleteComponentByKey(definition.key);
+      CliUx.ux.action.stop();
+    }
+
     CliUx.ux.action.stop();
   }
 }

src/commands/instances/create.ts

@@ -13,7 +13,8 @@ export default class CreateCommand extends Command {
     integration: Flags.string({
       char: "i",
       required: true,
-      description: "ID of the integration or a specific integration version ID this is an instance of",
+      description:
+        "ID of the integration or a specific integration version ID this is an instance of",
     }),
     customer: Flags.string({
       char: "c",

src/commands/instances/update.ts

@@ -36,9 +36,19 @@ export default class UpdateCommand extends Command {
 
     const result = await gqlRequest({
       document: gql`
-        mutation updateInstance($id: ID!, $name: String, $description: String, $version: ID! ) {
+        mutation updateInstance(
+          $id: ID!
+          $name: String
+          $description: String
+          $version: ID!
+        ) {
           updateInstance(
-            input: { id: $id, name: $name, description: $description, integration: $version}
+            input: {
+              id: $id
+              name: $name
+              description: $description
+              integration: $version
+            }
           ) {
             instance {
               id

src/utils/component/deleteByKey.ts

@@ -0,0 +1,33 @@
+import { gql, gqlRequest } from "../../graphql";
+
+export const deleteComponentByKey = async (key: string) => {
+  // Fetch a component by key
+  const result = await gqlRequest({
+    document: gql`
+      query component($key: String!) {
+        components(key: $key, public: false) {
+          nodes {
+            id
+          }
+        }
+      }
+    `,
+    variables: {
+      key,
+    },
+  });
+  // Delete the component by ID
+  await gqlRequest({
+    document: gql`
+      mutation deleteComponent($id: ID!) {
+        deleteComponent(input: { id: $id }) {
+          errors {
+            field
+            messages
+          }
+        }
+      }
+    `,
+    variables: { id: result.components.nodes[0].id },
+  });
+};

src/utils/integration/invoke.ts

@@ -49,6 +49,23 @@ export const getIntegrationFlow = async (
   return flows[0].id;
 };
 
+/** Delete a specified integration by ID */
+export const deleteIntegration = async (integrationId: string) => {
+  await gqlRequest({
+    document: gql`
+      mutation deleteIntegration($id: ID!) {
+        deleteIntegration(input: { id: $id }) {
+          errors {
+            field
+            messages
+          }
+        }
+      }
+    `,
+    variables: { id: integrationId },
+  });
+};
+
 interface IntegrationFlowRunProps {
   integrationId: string;
   flowId?: string;