From 24182b3f1b46e13253fcb9d56ef9783b0d6a20dd Mon Sep 17 00:00:00 2001
From: Andrew Kingston <andrew@kingston.dev>
Date: Thu, 27 Jul 2023 14:22:13 +0100
Subject: [PATCH] Fix refreshing table definition

---
 .../modals/grid/GridCreateColumnModal.svelte        |  4 +++-
 .../src/components/grid/stores/rows.js              | 13 +++++++++----
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/packages/builder/src/components/backend/DataTable/modals/grid/GridCreateColumnModal.svelte b/packages/builder/src/components/backend/DataTable/modals/grid/GridCreateColumnModal.svelte
index 9e386ae140..a58f4d76db 100644
--- a/packages/builder/src/components/backend/DataTable/modals/grid/GridCreateColumnModal.svelte
+++ b/packages/builder/src/components/backend/DataTable/modals/grid/GridCreateColumnModal.svelte
@@ -11,5 +11,7 @@
 </script>
 
 <Modal bind:this={modal}>
-  <CreateEditColumn on:updatecolumns={rows.actions.refreshTableDefinition} />
+  <CreateEditColumn
+    on:updatecolumns={rows.actions.refreshDatasourceDefinition}
+  />
 </Modal>
diff --git a/packages/frontend-core/src/components/grid/stores/rows.js b/packages/frontend-core/src/components/grid/stores/rows.js
index 80e4e6039d..23eae75428 100644
--- a/packages/frontend-core/src/components/grid/stores/rows.js
+++ b/packages/frontend-core/src/components/grid/stores/rows.js
@@ -482,9 +482,14 @@ export const createActions = context => {
   }
 
   // Refreshes the schema of the data fetch subscription
-  const refreshTableDefinition = async () => {
-    const definition = await API.fetchTableDefinition(get(tableId))
-    table.set(definition)
+  const refreshDatasourceDefinition = async () => {
+    const $datasource = get(datasource)
+    if ($datasource.type === "table") {
+      table.set(await API.fetchTableDefinition($datasource.tableId))
+    } else if ($datasource.type === "viewV2") {
+      // const definition = await API.viewsV2.(get(tableId))
+      // table.set(definition)
+    }
   }
 
   // Checks if we have a row with a certain ID
@@ -520,7 +525,7 @@ export const createActions = context => {
         refreshRow,
         replaceRow,
         refreshData,
-        refreshTableDefinition,
+        refreshDatasourceDefinition,
       },
     },
   }