Refactor out extensions-modules and rewrite mssql extension (#909)

* commting .d.ts changes * added serverinfo to .d.ts * maybe its working? * works * updated contrib * remove unnecessary code * fix compile errors * init * conitnue * on the way to working maybe? * close * EVERYTHING WORKS * moved src out of client folder * formatting * reenable logging * working on build file * fixed install service gulp command * fix the command to properly return promises * clean up code * add telemetry * formatting * added logging/telemetry/statusview * formatting * address comments * resolute vscode-language versions
2026-02-16 10:58:30 -05:00 · 2018-03-28 10:59:08 -07:00
parent 22c54a9917
commit 2414f43757
62 changed files with 3396 additions and 11239 deletions
--- a/extensions/mssql/src/main.ts
+++ b/extensions/mssql/src/main.ts
@@ -0,0 +1,136 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+'use strict';
+
+import * as vscode from 'vscode';
+import * as path from 'path';
+import { SqlOpsDataClient, ClientOptions } from 'dataprotocol-client';
+import { IConfig, ServerProvider, Events } from 'service-downloader';
+import { ServerOptions, TransportKind } from 'vscode-languageclient';
+
+import * as Constants from './constants';
+import ContextProvider from './contextProvider';
+import { CredentialStore } from './credentialstore/credentialstore';
+import { AzureResourceProvider } from './resourceProvider/resourceProvider';
+import * as Utils from './utils';
+import { Telemetry, LanguageClientErrorHandler } from './telemetry';
+import { TelemetryFeature } from './features';
+
+const baseConfig = require('./config.json');
+const outputChannel = vscode.window.createOutputChannel(Constants.serviceName);
+const statusView = vscode.window.createStatusBarItem(vscode.StatusBarAlignment.Left);
+
+export async function activate(context: vscode.ExtensionContext) {
+	// lets make sure we support this platform first
+	let supported = await Utils.verifyPlatform();
+
+	if (!supported) {
+		vscode.window.showErrorMessage('Unsupported platform');
+		return;
+	}
+
+	let config: IConfig = JSON.parse(JSON.stringify(baseConfig));
+	config.installDirectory = path.join(__dirname, config.installDirectory);
+	config.proxy = vscode.workspace.getConfiguration('http').get('proxy');
+	config.strictSSL = vscode.workspace.getConfiguration('http').get('proxyStrictSSL') || true;
+
+	const credentialsStore = new CredentialStore(config);
+	const resourceProvider = new AzureResourceProvider(config);
+	let languageClient: SqlOpsDataClient;
+
+	const serverdownloader = new ServerProvider(config);
+
+	serverdownloader.eventEmitter.onAny(generateHandleServerProviderEvent());
+
+	let clientOptions: ClientOptions = {
+		providerId: Constants.providerId,
+		errorHandler: new LanguageClientErrorHandler(),
+		features: [
+			// we only want to add new features
+			...SqlOpsDataClient.defaultFeatures,
+			TelemetryFeature
+		]
+	};
+
+	const installationStart = Date.now();
+	serverdownloader.getOrDownloadServer().then(e => {
+		const installationComplete = Date.now();
+		let serverOptions = generateServerOptions(e);
+		languageClient = new SqlOpsDataClient(Constants.serviceName, serverOptions, clientOptions);
+		const processStart = Date.now();
+		languageClient.onReady().then(() => {
+			const processEnd = Date.now();
+			statusView.text = 'Service Started';
+			setTimeout(() => {
+				statusView.hide();
+			}, 1500);
+			Telemetry.sendTelemetryEvent('startup/LanguageClientStarted', {
+				installationTime: String(installationComplete - installationStart),
+				processStartupTime: String(processEnd - processStart),
+				totalTime: String(processEnd - installationStart),
+				beginningTimestamp: String(installationStart)
+			});
+		});
+		statusView.show();
+		statusView.text = 'Starting service';
+		languageClient.start();
+		credentialsStore.start();
+		resourceProvider.start();
+	}, e => {
+		Telemetry.sendTelemetryEvent('ServiceInitializingFailed');
+		vscode.window.showErrorMessage('Failed to start Sql tools service');
+	});
+
+	let contextProvider = new ContextProvider();
+	context.subscriptions.push(contextProvider);
+	context.subscriptions.push(credentialsStore);
+	context.subscriptions.push(resourceProvider);
+	context.subscriptions.push({ dispose: () => languageClient.stop() });
+}
+
+function generateServerOptions(executablePath: string): ServerOptions {
+	let launchArgs = [];
+	launchArgs.push('--log-dir');
+	let logFileLocation = path.join(Utils.getDefaultLogLocation(), 'mssql');
+	launchArgs.push(logFileLocation);
+
+	return { command: executablePath, args: launchArgs, transport: TransportKind.stdio };
+}
+
+function generateHandleServerProviderEvent() {
+	let dots = 0;
+	return (e: string, ...args: any[]) => {
+		outputChannel.show();
+		statusView.show();
+		switch (e) {
+			case Events.INSTALL_START:
+				outputChannel.appendLine(`Installing ${Constants.serviceName} service to ${args[0]}`);
+				statusView.text = 'Installing Service';
+				break;
+			case Events.INSTALL_END:
+				outputChannel.appendLine('Installed');
+				break;
+			case Events.DOWNLOAD_START:
+				outputChannel.appendLine(`Downloading ${args[0]}`);
+				outputChannel.append(`(${Math.ceil(args[1] / 1024)} KB)`);
+				statusView.text = 'Downloading Service';
+				break;
+			case Events.DOWNLOAD_PROGRESS:
+				let newDots = Math.ceil(args[0] / 5);
+				if (newDots > dots) {
+					outputChannel.append('.'.repeat(newDots - dots));
+					dots = newDots;
+				}
+				break;
+			case Events.DOWNLOAD_END:
+				outputChannel.appendLine('Done!');
+				break;
+		}
+	};
+}
+
+// this method is called when your extension is deactivated
+export function deactivate(): void {
+}