support search for table

package.json

@@ -74,6 +74,11 @@
                 "command": "mysql.selectTop1000",
                 "title": "Select Top 1000",
                 "category": "MySQL"
+            },
+            {
+                "command": "mysql.keywordFilter",
+                "title": "Filter",
+                "category": "MySQL"
             }
         ],
         "keybindings": [
@@ -109,6 +114,11 @@
                     "when": "view == mysql && viewItem == connection",
                     "group": "mysql@2"
                 },
+                {
+                    "command": "mysql.keywordFilter",
+                    "when": "view == mysql && viewItem == connection",
+                    "group": "mysql@2"
+                },
                 {
                     "command": "mysql.newQuery",
                     "when": "view == mysql && viewItem == database",
@@ -119,6 +129,11 @@
                     "when": "view == mysql && viewItem == database",
                     "group": "mysql@1"
                 },
+                {
+                    "command": "mysql.keywordFilter",
+                    "when": "view == mysql && viewItem == database",
+                    "group": "mysql@2"
+                },
                 {
                     "command": "mysql.selectTop1000",
                     "when": "view == mysql && viewItem == table",

src/extension.ts

@@ -23,6 +23,10 @@ export function activate(context: vscode.ExtensionContext) {
         mysqlTreeDataProvider.addConnection();
     }));
 
+    context.subscriptions.push(vscode.commands.registerCommand("mysql.keywordFilter", (databaseOrConnectionNode: DatabaseNode | ConnectionNode) => {
+        mysqlTreeDataProvider.keywordFilter(databaseOrConnectionNode);
+    }));
+
     context.subscriptions.push(vscode.commands.registerCommand("mysql.deleteConnection", (connectionNode: ConnectionNode) => {
         connectionNode.deleteConnection(context, mysqlTreeDataProvider);
     }));

src/model/connectionNode.ts

@@ -13,6 +13,8 @@ import { InfoNode } from "./infoNode";
 import { INode } from "./INode";
 
 export class ConnectionNode implements INode {
+    public keyword;
+
     constructor(private readonly id: string, private readonly host: string, private readonly user: string,
                 private readonly password: string, private readonly port: string,
                 private readonly certPath: string) {
@@ -23,7 +25,7 @@ export class ConnectionNode implements INode {
             label: this.host,
             collapsibleState: vscode.TreeItemCollapsibleState.Collapsed,
             contextValue: "connection",
-            iconPath: path.join(__filename, "..", "..", "..", "resources", "server.png"),
+            iconPath: path.join(__filename, "..", "..", "..", "resources", this.keyword ? "b_search.png" : "server.png") ,
         };
     }
 
@@ -36,7 +38,7 @@ export class ConnectionNode implements INode {
             certPath: this.certPath,
         });
 
-        return Utility.queryPromise<any[]>(connection, "SHOW DATABASES")
+        return Utility.queryPromise<any[]>(connection, "SHOW DATABASES" + (this.keyword ? ` LIKE '%${this.keyword}%'` : ""))
             .then((databases) => {
                 return databases.map<DatabaseNode>((database) => {
                     return new DatabaseNode(this.host, this.user, this.password, this.port, database.Database, this.certPath);

src/model/databaseNode.ts

@@ -5,11 +5,14 @@ import * as vscode from "vscode";
 import { AppInsightsClient } from "../common/appInsightsClient";
 import { Global } from "../common/global";
 import { Utility } from "../common/utility";
+import { MySQLTreeDataProvider } from "../mysqlTreeDataProvider";
 import { InfoNode } from "./infoNode";
 import { INode } from "./INode";
 import { TableNode } from "./tableNode";
 
 export class DatabaseNode implements INode {
+    public keyword;
+
     constructor(private readonly host: string, private readonly user: string,
                 private readonly password: string, private readonly port: string, private readonly database: string,
                 private readonly certPath: string) {
@@ -20,7 +23,7 @@ export class DatabaseNode implements INode {
             label: this.database,
             collapsibleState: vscode.TreeItemCollapsibleState.Collapsed,
             contextValue: "database",
-            iconPath: path.join(__filename, "..", "..", "..", "resources", "database.svg"),
+            iconPath: path.join(__filename, "..", "..", "..", "resources", this.keyword ? "b_search.png" : "database.svg"),
         };
     }
 
@@ -33,8 +36,8 @@ export class DatabaseNode implements INode {
             database: this.database,
             certPath: this.certPath,
         });
-
-        return Utility.queryPromise<any[]>(connection, `SELECT TABLE_NAME FROM information_schema.TABLES  WHERE TABLE_SCHEMA = '${this.database}' LIMIT ${Utility.maxTableCount}`)
+        const filter = this.keyword ? `TABLE_NAME LIKE '%${this.keyword}%'` : "";
+        return Utility.queryPromise<any[]>(connection, `SELECT TABLE_NAME FROM information_schema.TABLES WHERE TABLE_SCHEMA = '${this.database}' ${filter} LIMIT ${Utility.maxTableCount}`)
             .then((tables) => {
                 return tables.map<TableNode>((table) => {
                     return new TableNode(this.host, this.user, this.password, this.port, this.database, table.TABLE_NAME, this.certPath);

src/mysqlTreeDataProvider.ts

@@ -6,6 +6,7 @@ import { Constants } from "./common/constants";
 import { Global } from "./common/global";
 import { IConnection } from "./model/connection";
 import { ConnectionNode } from "./model/connectionNode";
+import { DatabaseNode } from "./model/databaseNode";
 import { INode } from "./model/INode";
 
 export class MySQLTreeDataProvider implements vscode.TreeDataProvider<INode> {
@@ -76,6 +77,19 @@ export class MySQLTreeDataProvider implements vscode.TreeDataProvider<INode> {
         AppInsightsClient.sendEvent("addConnection.end");
     }
 
+    public async keywordFilter(node: DatabaseNode | ConnectionNode) {
+        vscode.window.showInputBox({
+           prompt: "" , placeHolder: "keyword", value: node.keyword,
+           validateInput: (keyword: string) => {
+                if (node.keyword !== keyword) {
+                    node.keyword = keyword;
+                    this.refresh(node);
+                }
+                return "";
+           },
+       });
+   }
+
     public refresh(element?: INode): void {
         this._onDidChangeTreeData.fire(element);
     }