sample code to implement providers (#15929)

* sample code to implement providers

* context menu

* comments

* more comments

* move comments to README
This commit is contained in:
Alan Ren
2021-08-20 14:19:50 -07:00
committed by GitHub
parent ccaf111696
commit 1d7accadbf
7 changed files with 404 additions and 3 deletions

View File

@@ -0,0 +1,82 @@
/*---------------------------------------------------------------------------------------------
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the Source EULA. See License.txt in the project root for license information.
*--------------------------------------------------------------------------------------------*/
import * as azdata from 'azdata';
import * as vscode from 'vscode';
export class ConnectionProvider implements azdata.ConnectionProvider {
private onConnectionCompleteEmitter: vscode.EventEmitter<azdata.ConnectionInfoSummary> = new vscode.EventEmitter();
onConnectionComplete: vscode.Event<azdata.ConnectionInfoSummary> = this.onConnectionCompleteEmitter.event;
private onIntelliSenseCacheCompleteEmitter: vscode.EventEmitter<string> = new vscode.EventEmitter();
onIntelliSenseCacheComplete: vscode.Event<string> = this.onIntelliSenseCacheCompleteEmitter.event;
private onConnectionChangedEmitter: vscode.EventEmitter<azdata.ChangedConnectionInfo> = new vscode.EventEmitter();
onConnectionChanged: vscode.Event<azdata.ChangedConnectionInfo> = this.onConnectionChangedEmitter.event;
connect(connectionUri: string, connectionInfo: azdata.ConnectionInfo): Promise<boolean> {
this.onConnectionCompleteEmitter.fire({
connectionId: '',
ownerUri: connectionUri,
messages: '',
errorMessage: '',
errorNumber: 0,
connectionSummary: {
serverName: '',
userName: ''
},
serverInfo: {
serverReleaseVersion: 1,
engineEditionId: 1,
serverVersion: '1.0',
serverLevel: '',
serverEdition: '',
isCloud: true,
azureVersion: 1,
osVersion: '',
options: {}
}
});
return Promise.resolve(true);
}
disconnect(connectionUri: string): Promise<boolean> {
return Promise.resolve(true);
}
cancelConnect(connectionUri: string): Promise<boolean> {
return Promise.resolve(true);
}
listDatabases(connectionUri: string): Promise<azdata.ListDatabasesResult> {
return Promise.resolve({
databaseNames: ['master', 'msdb']
});
}
changeDatabase(connectionUri: string, newDatabase: string): Promise<boolean> {
return Promise.resolve(true);
}
rebuildIntelliSenseCache(connectionUri: string): Promise<void> {
return Promise.resolve();
}
getConnectionString(connectionUri: string, includePassword: boolean): Promise<string> {
return Promise.resolve('conn_string');
}
buildConnectionInfo?(connectionString: string): Promise<azdata.ConnectionInfo> {
return Promise.resolve({
options: []
});
}
registerOnConnectionComplete(handler: (connSummary: azdata.ConnectionInfoSummary) => any): void {
this.onConnectionComplete((e) => {
handler(e);
});
}
registerOnIntelliSenseCacheComplete(handler: (connectionUri: string) => any): void {
console.log('IntellisenseCache complete');
}
registerOnConnectionChanged(handler: (changedConnInfo: azdata.ChangedConnectionInfo) => any): void {
console.log('Connection changed');
}
handle?: number;
providerId: string = 'testProvider';
}

View File

@@ -13,6 +13,9 @@ import * as fs from 'fs';
import * as path from 'path';
import { TreeNode, TreeDataProvider } from './treeDataProvider';
import * as dashboard from './modelViewDashboard';
import { ConnectionProvider } from '../providers/connectionProvider';
import { IconProvider } from '../providers/IconProvider';
import { ObjectExplorerProvider } from '../providers/objectExplorerProvider';
/**
* The main controller class that initializes the extension
@@ -37,6 +40,13 @@ export default class MainController implements vscode.Disposable {
}
public activate(): Promise<boolean> {
const connectionProvider = new ConnectionProvider();
const iconProvider = new IconProvider();
const objectExplorer = new ObjectExplorerProvider();
azdata.dataprotocol.registerConnectionProvider(connectionProvider);
azdata.dataprotocol.registerIconProvider(iconProvider);
azdata.dataprotocol.registerObjectExplorerProvider(objectExplorer);
const buttonHtml = fs.readFileSync(path.join(__dirname, 'button.html')).toString();
const counterHtml = fs.readFileSync(path.join(__dirname, 'counter.html')).toString();
this.registerSqlServicesModelView();