From 4f5343ded9d4d8634b8f018ad8ce4884261b88ef Mon Sep 17 00:00:00 2001
From: Peter Clement <peter@budibase.com>
Date: Fri, 24 Sep 2021 10:01:53 +0100
Subject: [PATCH] fetch tables for datasource plus

---
 .../modals/DatasourceDetailsModal.svelte      | 28 +++++++++++++++----
 .../app/[application]/data/index.svelte       |  5 +---
 2 files changed, 24 insertions(+), 9 deletions(-)

diff --git a/packages/builder/src/components/backend/DatasourceNavigator/modals/DatasourceDetailsModal.svelte b/packages/builder/src/components/backend/DatasourceNavigator/modals/DatasourceDetailsModal.svelte
index dcdd2da497..6d494378b3 100644
--- a/packages/builder/src/components/backend/DatasourceNavigator/modals/DatasourceDetailsModal.svelte
+++ b/packages/builder/src/components/backend/DatasourceNavigator/modals/DatasourceDetailsModal.svelte
@@ -1,11 +1,10 @@
 <script>
   import { ModalContent, notifications } from "@budibase/bbui"
   import IntegrationConfigForm from "components/backend/DatasourceNavigator/TableIntegrationMenu/IntegrationConfigForm.svelte"
-  import { datasources } from "stores/backend"
+  import { datasources, tables } from "stores/backend"
   import { IntegrationNames } from "constants"
 
   export let integration
-
   function prepareData() {
     let datasource = {}
     let existingTypeCount = $datasources.list.filter(
@@ -20,25 +19,44 @@
     datasource.source = integration.type
     datasource.config = integration.config
     datasource.name = name
+    datasource.plus = integration.plus
 
     return datasource
   }
   async function saveDatasource() {
     try {
       // Create datasource
-      await datasources.save(prepareData())
+      const resp = await datasources.save(prepareData())
+
+      if (integration.plus) {
+        updateDatasourceSchema(resp)
+      }
+      await datasources.fetch()
+      await datasources.select(resp["_id"])
       notifications.success(`Datasource updated successfully.`)
     } catch (err) {
       notifications.error(`Error saving datasource: ${err}`)
     }
   }
+
+  async function updateDatasourceSchema(datasourceJson) {
+    try {
+      await datasources.updateSchema(datasourceJson)
+      notifications.success(`Datasource ${name} tables updated successfully.`)
+      await tables.fetch()
+    } catch (err) {
+      notifications.error(`Error updating datasource schema: ${err}`)
+    }
+  }
 </script>
 
 <ModalContent
   title="Add Data"
   onConfirm={() => saveDatasource()}
-  confirmText="Continue"
-  cancelText="Start from scratch"
+  confirmText={integration.plus
+    ? "Fetch tables from database"
+    : "Save and continue to query"}
+  cancelText="Back"
   size="M"
 >
   <IntegrationConfigForm
diff --git a/packages/builder/src/pages/builder/app/[application]/data/index.svelte b/packages/builder/src/pages/builder/app/[application]/data/index.svelte
index 2d1d94d267..ca457cd1aa 100644
--- a/packages/builder/src/pages/builder/app/[application]/data/index.svelte
+++ b/packages/builder/src/pages/builder/app/[application]/data/index.svelte
@@ -8,10 +8,7 @@
   $: setupComplete =
     $datasources.list.find(x => (x._id = "bb_internal")).entities.length > 1 ||
     $datasources.list.length >= 1
-  $: console.log($datasources.list.length >= 1)
-  $: console.log(
-    $datasources.list.find(x => (x._id = "bb_internal")).entities.length > 1
-  )
+
   onMount(() => {
     if (!setupComplete) {
       modal.show()