From 33c36c1112a3f7d695a3a230d7190d22bf8d5e08 Mon Sep 17 00:00:00 2001
From: Mel O'Hagan <mel@budibase.com>
Date: Wed, 12 Oct 2022 15:31:53 +0100
Subject: [PATCH] Include stage templates

---
 .../src/components/integration/index.svelte   | 19 +++++++++++++++++--
 packages/server/src/integrations/mongodb.ts   | 19 +++++++++++++++++++
 packages/types/src/sdk/datasources.ts         |  6 ++++++
 3 files changed, 42 insertions(+), 2 deletions(-)

diff --git a/packages/builder/src/components/integration/index.svelte b/packages/builder/src/components/integration/index.svelte
index c98746ae6e..d07f395a01 100644
--- a/packages/builder/src/components/integration/index.svelte
+++ b/packages/builder/src/components/integration/index.svelte
@@ -4,6 +4,7 @@
   import {
     Label,
     Input,
+    Select,
     Divider,
     Layout,
     Icon,
@@ -42,6 +43,19 @@
     }
   }
 
+  function setEditorTemplate(fromKey, toKey, index) {
+    if (
+      schema.steps.filter(step => step.key === fromKey)[0]?.template ===
+      query.fields.steps[index].value.value
+    ) {
+      query.fields.steps[index].value.value = schema.steps.filter(
+        step => step.key === toKey
+      )[0]?.template
+      flowEditors[index].update(query.fields.steps[index].value.value)
+    }
+    query.fields.steps[index].key = toKey
+  }
+
   $: shouldDisplayJsonBox =
     schema.type === QueryTypes.JSON && query.fields.extra?.actionType !== "flow"
 </script>
@@ -114,10 +128,11 @@
                 <Layout noPadding gap="S">
                   <div class="fields">
                     <div class="block-field">
-                      <Input
+                      <Select
                         value={step.key}
+                        options={schema.steps.map(s => s.key)}
                         on:change={({ detail }) => {
-                          query.fields.steps[index].key = detail
+                          setEditorTemplate(step.key, detail, index)
                         }}
                       />
                       <Editor
diff --git a/packages/server/src/integrations/mongodb.ts b/packages/server/src/integrations/mongodb.ts
index 85aa341935..1794615541 100644
--- a/packages/server/src/integrations/mongodb.ts
+++ b/packages/server/src/integrations/mongodb.ts
@@ -58,6 +58,25 @@ const SCHEMA: Integration = {
     },
     aggregate: {
       type: QueryType.JSON,
+      steps: [
+        {
+          key: "$addFields",
+          template: "{\n\t\n}",
+        },
+        {
+          key: "$bucket",
+          template: `{
+  "groupBy": "",
+  "boundaries": [],
+  "default": "",
+  "output": {}
+}`,
+        },
+        {
+          key: "$match",
+          template: "{\n\t\n}",
+        },
+      ]
     },
   },
   extra: {
diff --git a/packages/types/src/sdk/datasources.ts b/packages/types/src/sdk/datasources.ts
index d01d636b86..c890e57881 100644
--- a/packages/types/src/sdk/datasources.ts
+++ b/packages/types/src/sdk/datasources.ts
@@ -70,6 +70,11 @@ export enum FilterType {
   ONE_OF = "oneOf",
 }
 
+export interface StepDefinition {
+  key: string
+  template: string
+}
+
 export interface QueryDefinition {
   type: QueryType
   displayName?: string
@@ -77,6 +82,7 @@ export interface QueryDefinition {
   customisable?: boolean
   fields?: object
   urlDisplay?: boolean
+  steps?: Array<StepDefinition>
 }
 
 export interface ExtraQueryConfig {