azuredatastudio/extensions/liveshare/src/guestSessionManager.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import * as vscode from 'vscode';
import * as azdata from 'azdata';
import { LiveShare, SharedServiceProxy } from './liveshare';
import { LiveShareProviderId, LiveShareServiceName, VslsSchema } from './constants';
import { ConnectionProvider } from './providers/connectionProvider';
import { StatusProvider, LiveShareDocumentState } from './providers/statusProvider';
import { QueryProvider } from './providers/queryProvider';

declare let require: any;
let vsls = require('vsls');

export class GuestSessionManager {
	private _statusProvider: StatusProvider;

	constructor(
		context: vscode.ExtensionContext,
		vslsApi: LiveShare
	) {
		let self = this;
		vscode.workspace.onDidOpenTextDocument(params => this.onDidOpenTextDocument(params));

		vslsApi!.onDidChangeSession(async function onLiveShareSessionCHange(e: any) {
			const isHost = e.session.role === vsls.Role.Host;
			if (!e.session.id && isHost) {
				return;
			}

			const sharedServiceProxy: SharedServiceProxy = await vslsApi.getSharedService(LiveShareServiceName);
			if (!sharedServiceProxy) {
				vscode.window.showErrorMessage('Could not access a shared service. You have to set "liveshare.features" to "experimental" in your user settings in order to use this extension.');
				return;
			}

			const connectionProvider = new ConnectionProvider(isHost, vslsApi, sharedServiceProxy);

			const queryProvider = new QueryProvider(false);
			queryProvider.initialize(false, sharedServiceProxy);

			self._statusProvider = new StatusProvider(
				isHost,
				vslsApi,
				connectionProvider,
				sharedServiceProxy);
		});
	}

	private async onDidOpenTextDocument(doc: vscode.TextDocument): Promise<void> {
		if (this._statusProvider && this.isLiveShareDocument(doc)) {
			let documentState: LiveShareDocumentState = await this._statusProvider.getDocumentState(doc);
			if (documentState) {
				let queryDocument = await azdata.queryeditor.getQueryDocument(doc.uri.toString());
				if (queryDocument) {
					let connectionOptions: { [key: string]: any } = {};
					connectionOptions['providerName'] = LiveShareProviderId;
					connectionOptions['serverName'] = documentState.serverName;
					connectionOptions['databaseName'] = documentState.databaseName;
					connectionOptions['userName'] = 'liveshare';
					connectionOptions['password'] = 'liveshare'; // [SuppressMessage("Microsoft.Security", "CS001:SecretInline", Justification="Stub value for testing")]
					connectionOptions['authenticationType'] = 'liveshare';
					connectionOptions['savePassword'] = false;
					connectionOptions['saveProfile'] = false;
					let profile = azdata.connection.ConnectionProfile.createFrom(connectionOptions);
					queryDocument.connect(profile);
				}
			}
		}
	}

	private isLiveShareDocument(doc: vscode.TextDocument): boolean {
		return doc && doc.uri.scheme === VslsSchema;
	}
}