From 42a7c70dd4af3634a2f09b8f4b3e2479d367ec49 Mon Sep 17 00:00:00 2001
From: Diego Imbert <diego@windmill.dev>
Date: Fri, 3 Jul 2026 11:52:21 +0200
Subject: [PATCH] refresh db manager schema on migrations

---
 .../src/lib/components/DBManagerDrawer.svelte     | 11 +++++++----
 .../DataTableMigrationsButton.svelte              | 15 +++++++++++++--
 2 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/frontend/src/lib/components/DBManagerDrawer.svelte b/frontend/src/lib/components/DBManagerDrawer.svelte
index b03014d251..b303671fe8 100644
--- a/frontend/src/lib/components/DBManagerDrawer.svelte
+++ b/frontend/src/lib/components/DBManagerDrawer.svelte
@@ -106,6 +106,11 @@
 		return toSourceIdentifier(input.resourcePath)
 	}
 
+	function refreshManager() {
+		dbManagerContent?.refresh()
+		dbManagerContent?.dbManager()?.dbTable()?.refresh()
+	}
+
 	async function handleExportSchema() {
 		const source = currentSourceIdentifier()
 		if (!source || !$workspaceStore) return
@@ -206,6 +211,7 @@
 				<DataTableMigrationsButton
 					workspace={$workspaceStore}
 					datatable={uriState.selectedDatatable}
+					onSchemaChanged={refreshManager}
 				/>
 			{/if}
 			{#if enableImportExport}
@@ -216,10 +222,7 @@
 			{/if}
 			<Button
 				loading={dbManagerContent?.isLoading() ?? false}
-				on:click={() => {
-					dbManagerContent?.refresh()
-					dbManagerContent?.dbManager()?.dbTable()?.refresh()
-				}}
+				on:click={refreshManager}
 				startIcon={{ icon: RefreshCcw }}
 				iconOnly
 				title="Refresh"
diff --git a/frontend/src/lib/components/workspaceSettings/DataTableMigrationsButton.svelte b/frontend/src/lib/components/workspaceSettings/DataTableMigrationsButton.svelte
index c4030d2212..1850dc6ee7 100644
--- a/frontend/src/lib/components/workspaceSettings/DataTableMigrationsButton.svelte
+++ b/frontend/src/lib/components/workspaceSettings/DataTableMigrationsButton.svelte
@@ -30,8 +30,16 @@
 	let {
 		workspace,
 		datatable,
-		disabled = false
-	}: { workspace: string; datatable: string; disabled?: boolean } = $props()
+		disabled = false,
+		onSchemaChanged
+	}: {
+		workspace: string
+		datatable: string
+		disabled?: boolean
+		/** Called after a migration run/revert actually changes the data table
+		 * schema, so an open viewer (e.g. the database manager) can refresh. */
+		onSchemaChanged?: () => void
+	} = $props()
 
 	let listOpen = $state(false)
 	let migrations = $state<DatatableMigrationWithStatus[]>([])
@@ -102,6 +110,7 @@
 					: 'No pending migrations to run'
 			)
 			await loadMigrations()
+			if (res.applied.length > 0) onSchemaChanged?.()
 		} catch (e: any) {
 			sendUserToast(`Failed to run migrations: ${e?.body ?? e?.message ?? e}`, true)
 		} finally {
@@ -121,6 +130,7 @@
 				res.applied.length > 0 ? `Applied ${res.applied[0].name}` : 'Migration already applied'
 			)
 			await loadMigrations()
+			if (res.applied.length > 0) onSchemaChanged?.()
 		} catch (e: any) {
 			sendUserToast(`Failed to run migration: ${e?.body ?? e?.message ?? e}`, true)
 		} finally {
@@ -159,6 +169,7 @@
 					: 'Migration was not applied'
 			)
 			await loadMigrations()
+			if (res.rolled_back.length > 0) onSchemaChanged?.()
 		} catch (e: any) {
 			sendUserToast(`Failed to revert migration: ${e?.body ?? e?.message ?? e}`, true)
 		} finally {