Add filter button

Signed-off-by: Sanjula Ganepola <Sanjula.Ganepola@ibm.com>

Author: SanjulaGanepola

package.json

@@ -539,6 +539,12 @@
         "category": "Db2 for i",
         "icon": "$(list-selection)"
       },
+      {
+        "command": "vscode-db2i.filterDatabaseObjectTypes",
+        "title": "Filter Database Object Types",
+        "category": "Db2 for i",
+        "icon": "$(filter)"
+      },
       {
         "command": "vscode-db2i.removeSchemaFromSchemaBrowser",
         "title": "Remove schema from view",
@@ -976,6 +982,10 @@
           "command": "vscode-db2i.importDataContextMenu",
           "when": "false"
         },
+        {
+          "command": "vscode-db2i.filterDatabaseObjectTypes",
+          "when": "never"
+        },
         {
           "command": "vscode-db2i.setSchemaFilter",
           "when": "never"
@@ -1140,14 +1150,19 @@
       ],
       "view/title": [
         {
-          "command": "vscode-db2i.refreshSchemaBrowser",
-          "group": "navigation",
+          "command": "vscode-db2i.manageSchemaBrowserList",
+          "group": "navigation@0",
+          "when": "view == schemaBrowser && vscode-db2i:manageSchemaBrowserEnabled"
+        },
+        {
+          "command": "vscode-db2i.filterDatabaseObjectTypes",
+          "group": "navigation@1",
           "when": "view == schemaBrowser"
         },
         {
-          "command": "vscode-db2i.manageSchemaBrowserList",
-          "group": "navigation",
-          "when": "view == schemaBrowser && vscode-db2i:manageSchemaBrowserEnabled"
+          "command": "vscode-db2i.refreshSchemaBrowser",
+          "group": "navigation@2",
+          "when": "view == schemaBrowser"
         },
         {
           "command": "vscode-db2i.resultset.settings",

src/views/schemaBrowser/contributes.json

@@ -23,6 +23,12 @@
         "category": "Db2 for i",
         "icon": "$(list-selection)"
       },
+      {
+        "command": "vscode-db2i.filterDatabaseObjectTypes",
+        "title": "Filter Database Object Types",
+        "category": "Db2 for i",
+        "icon": "$(filter)"
+      },
       {
         "command": "vscode-db2i.removeSchemaFromSchemaBrowser",
         "title": "Remove schema from view",
@@ -114,6 +120,10 @@
     ],
     "menus": {
       "commandPalette": [
+        {
+          "command": "vscode-db2i.filterDatabaseObjectTypes",
+          "when": "never"
+        },
         {
           "command": "vscode-db2i.setSchemaFilter",
           "when": "never"
@@ -173,14 +183,19 @@
       ],
       "view/title": [
         {
-          "command": "vscode-db2i.refreshSchemaBrowser",
-          "group": "navigation",
+          "command": "vscode-db2i.manageSchemaBrowserList",
+          "group": "navigation@0",
+          "when": "view == schemaBrowser && vscode-db2i:manageSchemaBrowserEnabled"
+        },
+        {
+          "command": "vscode-db2i.filterDatabaseObjectTypes",
+          "group": "navigation@1",
           "when": "view == schemaBrowser"
         },
         {
-          "command": "vscode-db2i.manageSchemaBrowserList",
-          "group": "navigation",
-          "when": "view == schemaBrowser && vscode-db2i:manageSchemaBrowserEnabled"
+          "command": "vscode-db2i.refreshSchemaBrowser",
+          "group": "navigation@2",
+          "when": "view == schemaBrowser"
         }
       ],
       "view/item/context": [

src/views/schemaBrowser/index.ts

@@ -1,5 +1,5 @@
 
-import { ThemeIcon, TreeItem, workspace } from "vscode"
+import { ThemeIcon, TreeItem, workspace, window } from "vscode"
 import * as vscode from "vscode"
 import Schemas, { AllSQLTypes, InternalTypes, SQL_ESCAPE_CHAR, SQLType } from "../../database/schemas";
 import Table from "../../database/table";
@@ -106,6 +106,30 @@ export default class schemaBrowser {
         }
       }),
 
+      vscode.commands.registerCommand(`vscode-db2i.filterDatabaseObjectTypes`, async () => {
+        const currentTypes = Configuration.get(`schemaBrowser.databaseObjectTypes`);
+        if (currentTypes) {
+          const quickPickItems = Object.keys(currentTypes).map((key) => {
+            return {
+              label: key, picked: currentTypes[key]
+            }
+          });
+          const selectedTypes = await window.showQuickPick(quickPickItems, {
+            title: `Select which database object types should be shown`,
+            canPickMany: true
+          });
+          if (selectedTypes) {
+            const unselectedTypes: Record<string, boolean> = {};
+            for (const key of Object.keys(currentTypes)) {
+              if (!selectedTypes.find(type => type.label === key)) {
+                unselectedTypes[key] = false;
+              }
+            }
+            Configuration.set(`schemaBrowser.databaseObjectTypes`, unselectedTypes);
+          }
+        }
+      }),
+
       vscode.commands.registerCommand(`vscode-db2i.removeSchemaFromSchemaBrowser`, async (node: SchemaItem) => {
         if (node) {
           //Running from right click