Support for filter

intersystems-community · Apr 20, 2024 · 24f3c8d · 24f3c8d
1 parent 50c52be
commit 24f3c8d
Show file tree

Hide file tree

Showing 6 changed files with 28 additions and 11 deletions.
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -9,8 +9,9 @@
 				"${workspaceFolder}/test/project.code-workspace",
 				"--extensionDevelopmentPath=${workspaceFolder}"
 			],
+			"sourceMaps": true,
 			"outFiles": [
-				"${workspaceFolder}/out/**/*.js"
+				"${workspaceFolder}/dist/**/*.js"
 			],
 			"preLaunchTask": "npm: webpack"
 		}

diff --git a/connection.schema.json b/connection.schema.json
@@ -37,6 +37,15 @@
       "title": "Show system items?",
       "type": "boolean",
       "default": false
+    },
+    "filter": {
+      "title": "Filter",
+      "type": "string",
+      "examples": [
+        "'Ens*",
+        "'HS*"
+      ],
+      "description": "* 0 or more characters, _ any one character, ' NOT pattern"
     }
   },
   "dependencies": {

diff --git a/package.json b/package.json
@@ -48,7 +48,7 @@
     "mtxr.sqltools"
   ],
   "activationEvents": [
-    "*",
+    "onStartupFinished",
     "onLanguage:sql",
     "onCommand:sqltools.*"
   ],

diff --git a/src/ls/driver.ts b/src/ls/driver.ts
@@ -13,6 +13,7 @@ export default class IRISDriver extends AbstractDriver<IRISdb, DriverOptions> im
 
   queries: IQueries = queries;
   private showSystem = false;
+  private filter = "";
 
   public async open() {
     if (this.connection) {
@@ -22,6 +23,7 @@ export default class IRISDriver extends AbstractDriver<IRISdb, DriverOptions> im
     const { namespace } = this.credentials;
     let config: IRISDirect;
     this.showSystem = this.credentials.showSystem || false;
+    this.filter = this.credentials.filter || "";
 
     if (this.credentials.serverName) {
       throw new Error("not supported");
@@ -113,6 +115,7 @@ export default class IRISDriver extends AbstractDriver<IRISdb, DriverOptions> im
 
   private async getSchemas({ item }: Arg0<IConnectionDriver['getChildrenForItem']>) {
     item['showSystem'] = this.showSystem;
+    item['filter'] = this.filter;
 
     switch (item.childType) {
       case ContextValue.TABLE:
@@ -127,6 +130,7 @@ export default class IRISDriver extends AbstractDriver<IRISdb, DriverOptions> im
 
   private async getChildrenForSchema({ item }: Arg0<IConnectionDriver['getChildrenForItem']>) {
     item['showSystem'] = this.showSystem;
+    item['filter'] = this.filter;
 
     switch (item.childType) {
       case ContextValue.TABLE:
@@ -151,11 +155,11 @@ export default class IRISDriver extends AbstractDriver<IRISdb, DriverOptions> im
       case ContextValue.DATABASE:
         // Syntatically, a schema in IRIS SQL resembles a database in other databases.
         // That's the simplest way to adapt IRIS SQL to the generic Hue parser vscode-sqltools uses.
-        return this.queryResults(this.queries.searchEverything({ search, showSystem: this.showSystem }));
+        return this.queryResults(this.queries.searchEverything({ search, showSystem: this.showSystem, filter: this.filter }));
       case ContextValue.TABLE:
       case ContextValue.FUNCTION:
       case ContextValue.VIEW:
-        const searchParams = { search, showSystem: this.showSystem, itemType, ...extraParams };
+        const searchParams = { search, showSystem: this.showSystem, filter: this.filter, itemType, ...extraParams };
         if (extraParams['database']) {
           searchParams['schema'] = extraParams['database'];
         }

diff --git a/src/ls/queries.ts b/src/ls/queries.ts
@@ -14,6 +14,7 @@ ValueColumn[ContextValue.FUNCTION] = "PROCEDURE_NAME";
 
 interface ISchema extends NSDatabase.ISchema {
   showSystem: boolean;
+  filter: string;
 }
 
 const describeTable: IQueries['describeTable'] = queryFactory`
@@ -173,7 +174,7 @@ DISTINCT BY(SCHEMA_NAME)
   '${ContextValue.SCHEMA}' as "type",
   '${type}' as "childType",
   'folder' as iconId
-FROM ${Functions[type]} (${p => p.showSystem ? 1 : 0})
+FROM ${Functions[type]} (${p => p.showSystem ? 1 : 0}, '${p => (p.filter && p.filter != "") ? `${p.filter.replace("'", "''")}` : "*"}')
 `;
 
 const fetchTableSchemas = fetchTypedSchemas(ContextValue.TABLE);

diff --git a/test/project.code-workspace b/test/project.code-workspace
@@ -42,16 +42,18 @@
         "username": "_SYSTEM"
       },
       {
-        "askForPassword": false,
+        "namespace": "USER",
         "connectionMethod": "Server and Port",
+        "showSystem": false,
+        "filter": "'Ens*",
+        "previewLimit": 50,
+        "server": "localhost",
+        "port": 52773,
+        "https": false,
+        "askForPassword": false,
         "driver": "InterSystems IRIS",
         "name": "InterSystems IRIS",
-        "namespace": "USER",
         "password": "SYS",
-        "port": 52773,
-        "previewLimit": 50,
-        "server": "localhost",
-        "showSystem": true,
         "username": "_SYSTEM"
       }
     ]