From 5eed81cf9f17c7b84f923cecf7428d925283a90f Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 20 Jan 2022 15:15:49 +0100
Subject: [PATCH] feat(diagram): filter columns

---
 packages/web/src/designer/Designer.svelte     | 26 ++++++++++++++++++
 .../web/src/designer/DesignerTable.svelte     | 27 ++++++++++++++++++-
 2 files changed, 52 insertions(+), 1 deletion(-)

diff --git a/packages/web/src/designer/Designer.svelte b/packages/web/src/designer/Designer.svelte
index 535a47f9c..131360d72 100644
--- a/packages/web/src/designer/Designer.svelte
+++ b/packages/web/src/designer/Designer.svelte
@@ -732,6 +732,32 @@
             },
           ],
         },
+        {
+          text: `Columns - ${_.startCase(value?.style?.filterColumns || 'all')}`,
+          submenu: [
+            {
+              text: 'All',
+              onClick: changeStyleFunc('filterColumns', null),
+            },
+            {
+              text: 'Primary Key',
+              onClick: changeStyleFunc('filterColumns', 'primaryKey'),
+            },
+            {
+              text: 'All Keys',
+              onClick: changeStyleFunc('filterColumns', 'allKeys'),
+            },
+            {
+              text: 'Not Null',
+              onClick: changeStyleFunc('filterColumns', 'notNull'),
+            },
+            {
+              text: 'Keys And Not Null',
+              onClick: changeStyleFunc('filterColumns', 'keysAndNotNull'),
+            },
+            
+          ],
+        },
       ],
     ];
   }
diff --git a/packages/web/src/designer/DesignerTable.svelte b/packages/web/src/designer/DesignerTable.svelte
index 70faa6ea8..484d04073 100644
--- a/packages/web/src/designer/DesignerTable.svelte
+++ b/packages/web/src/designer/DesignerTable.svelte
@@ -1,5 +1,6 @@
 <script lang="ts">
   import { presetDarkPalettes, presetPalettes } from '@ant-design/colors';
+  import { computeDbDiffRows } from 'dbgate-tools';
 
   import { tick } from 'svelte';
   import { createDatabaseObjectMenu } from '../appobj/DatabaseObjectAppObject.svelte';
@@ -54,7 +55,7 @@
 
   $: pureName = table?.pureName;
   $: alias = table?.alias;
-  $: columns = table?.columns as any[];
+  $: columns = (table?.columns as any[]).filter(x => shouldShowColumn(table, x, designer?.style));
   $: designerId = table?.designerId;
   $: objectTypeField = table?.objectTypeField;
   $: left = table?.left;
@@ -83,6 +84,30 @@
     return res;
   }
 
+  function shouldShowColumn(table, column, style) {
+    if (!settings?.chooseColumnProperties) {
+      return true;
+    }
+    switch (style?.filterColumns || 'all') {
+      case 'primaryKey':
+        return table?.primaryKey?.columns?.find(x => x.columnName == column?.columnName);
+      case 'allKeys':
+        return (
+          table?.primaryKey?.columns?.find(x => x.columnName == column?.columnName) ||
+          table?.foreignKeys?.find(fk => fk.columns.find(x => x.columnName == column?.columnName))
+        );
+      case 'keysAndNotNull':
+        return (
+          column?.notNull ||
+          table?.primaryKey?.columns?.find(x => x.columnName == column?.columnName) ||
+          table?.foreignKeys?.find(fk => fk.columns.find(x => x.columnName == column?.columnName))
+        );
+      case 'notNull':
+        return column?.notNull;
+    }
+    return true;
+  }
+
   function handleMoveStart() {
     if (settings?.canSelectTables) {
       onMoveStart();