From 5180e7ad2784ef683c577c67a72efd2225393cce Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 2 Dec 2021 14:52:25 +0100
Subject: [PATCH] toggle dynamic structure

---
 packages/web/src/tabs/FreeTableTab.svelte | 43 +++++++++++++++++++++--
 packages/web/src/utility/contextMenu.ts   |  6 +++-
 2 files changed, 46 insertions(+), 3 deletions(-)

diff --git a/packages/web/src/tabs/FreeTableTab.svelte b/packages/web/src/tabs/FreeTableTab.svelte
index 3490bed76..4c5664d03 100644
--- a/packages/web/src/tabs/FreeTableTab.svelte
+++ b/packages/web/src/tabs/FreeTableTab.svelte
@@ -13,10 +13,23 @@
     testEnabled: () => getCurrentEditor() != null,
     onClick: () => getCurrentEditor().save(),
   });
+
+  registerCommand({
+    id: 'freeTable.toggleDynamicStructure',
+    category: 'Table data',
+    name: 'Toggle dynamic structure',
+    testEnabled: () => getCurrentEditor() != null,
+    onClick: () => getCurrentEditor().toggleDynamicStructure(),
+  });
 </script>
 
 <script lang="ts">
-  import { createFreeTableModel, FreeTableGridDisplay, runMacro } from 'dbgate-datalib';
+  import {
+    analyseCollectionDisplayColumns,
+    createFreeTableModel,
+    FreeTableGridDisplay,
+    runMacro,
+  } from 'dbgate-datalib';
   import { setContext } from 'svelte';
   import { writable } from 'svelte/store';
   import registerCommand from '../commands/registerCommand';
@@ -86,7 +99,33 @@
   const collapsedLeftColumnStore = writable(false);
   setContext('collapsedLeftColumnStore', collapsedLeftColumnStore);
 
-  registerMenu({ command: 'freeTable.save', tag: 'save' });
+  export function toggleDynamicStructure() {
+    let structure = $modelState.value.structure;
+    structure = { ...structure, __isDynamicStructure: !structure.__isDynamicStructure };
+    if (!structure.__isDynamicStructure) {
+      const columns = analyseCollectionDisplayColumns($modelState.value.rows, display);
+      structure = {
+        ...structure,
+        columns: columns
+          .filter(col => col.uniquePath.length == 1)
+          .map(col => ({
+            columnName: col.uniqueName,
+          })),
+      };
+    }
+    dispatchModel({
+      type: 'set',
+      value: {
+        ...$modelState.value,
+        structure,
+      },
+    });
+  }
+
+  registerMenu(
+    { command: 'freeTable.save', tag: 'save' },
+    { command: 'freeTable.toggleDynamicStructure', tag: 'export' }
+  );
 
   // display is overridden in FreeTableGridCore, this is because of column manager
   $: display = new FreeTableGridDisplay($modelState.value, $config, config.update, null, null);
diff --git a/packages/web/src/utility/contextMenu.ts b/packages/web/src/utility/contextMenu.ts
index 7e9f53775..a89c2759c 100644
--- a/packages/web/src/utility/contextMenu.ts
+++ b/packages/web/src/utility/contextMenu.ts
@@ -1,5 +1,6 @@
 import _ from 'lodash';
 import { getContext, setContext } from 'svelte';
+import invalidateCommands from '../commands/invalidateCommands';
 import { currentDropDownMenu } from '../stores';
 import getAsArray from './getAsArray';
 
@@ -9,9 +10,12 @@ export function registerMenu(...items) {
 }
 
 export default function contextMenu(node, items = []) {
-  const handleContextMenu = e => {
+  const handleContextMenu = async e => {
     e.preventDefault();
     e.stopPropagation();
+
+    await invalidateCommands();
+    
     if (items) {
       const left = e.pageX;
       const top = e.pageY;