SQL Operations Studio Public Preview 1 (0.23) release source code

extensions/mssql/client/src/credentialstore/credentialstore.ts

@@ -0,0 +1,73 @@
+/*---------------------------------------------------------------------------------------------
+*  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 Contracts from '../models/contracts';
+import { ICredentialStore } from './icredentialstore';
+import { SqlToolsServiceClient, Utils } from 'extensions-modules';
+import { LanguageClient } from 'dataprotocol-client';
+
+/**
+ * Implements a credential storage for Windows, Mac (darwin), or Linux.
+ *
+ * Allows a single credential to be stored per service (that is, one username per service);
+ */
+export class CredentialStore implements ICredentialStore {
+
+    public languageClient: LanguageClient;
+
+    constructor(private _client?: SqlToolsServiceClient) {
+        if (!this._client) {
+            this._client = SqlToolsServiceClient.instance;
+        }
+    }
+
+    /**
+     * Gets a credential saved in the credential store
+     *
+     * @param {string} credentialId the ID uniquely identifying this credential
+     * @returns {Promise<Credential>} Promise that resolved to the credential, or undefined if not found
+     */
+    public readCredential(credentialId: string): Promise<Contracts.Credential> {
+        Utils.logDebug(this.languageClient, 'MainController._extensionConstants');
+        let self = this;
+        let cred: Contracts.Credential = new Contracts.Credential();
+        cred.credentialId = credentialId;
+        return new Promise<Contracts.Credential>( (resolve, reject) => {
+            self._client
+            .sendRequest(Contracts.ReadCredentialRequest.type, cred, this.languageClient)
+            .then(returnedCred => {
+                resolve(<Contracts.Credential>returnedCred);
+            }, err => reject(err));
+        });
+    }
+
+    public saveCredential(credentialId: string, password: any): Promise<boolean> {
+        let self = this;
+        let cred: Contracts.Credential = new Contracts.Credential();
+        cred.credentialId = credentialId;
+        cred.password = password;
+        return new Promise<boolean>( (resolve, reject) => {
+            self._client
+            .sendRequest(Contracts.SaveCredentialRequest.type, cred, this.languageClient)
+            .then(status => {
+                resolve(<boolean>status);
+            }, err => reject(err));
+        });
+    }
+
+    public deleteCredential(credentialId: string): Promise<boolean> {
+        let self = this;
+        let cred: Contracts.Credential = new Contracts.Credential();
+        cred.credentialId = credentialId;
+        return new Promise<boolean>( (resolve, reject) => {
+            self._client
+            .sendRequest(Contracts.DeleteCredentialRequest.type, cred, this.languageClient)
+            .then(status => {
+                resolve(<boolean>status);
+            }, err => reject(err));
+        });
+    }
+}

extensions/mssql/client/src/credentialstore/icredentialstore.ts

@@ -0,0 +1,23 @@
+/*---------------------------------------------------------------------------------------------
+*  Copyright (c) Microsoft Corporation. All rights reserved.
+*  Licensed under the Source EULA. See License.txt in the project root for license information.
+*--------------------------------------------------------------------------------------------*/
+'use strict';
+
+
+// This code is originally from https://github.com/microsoft/vsts-vscode
+// License: https://github.com/Microsoft/vsts-vscode/blob/master/LICENSE.txt
+
+import { Credential } from '../models/contracts';
+
+/**
+ * A credential store that securely stores sensitive information in a platform-specific manner
+ *
+ * @export
+ * @interface ICredentialStore
+ */
+export interface ICredentialStore {
+    readCredential(credentialId: string): Promise<Credential>;
+    saveCredential(credentialId: string, password: any): Promise<boolean>;
+    deleteCredential(credentialId: string): Promise<boolean>;
+}