Merge from vscode 5e80bf449c995aa32a59254c0ff845d37da11b70 (#9317)

extensions/github-authentication/src/githubServer.ts

@@ -0,0 +1,114 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import * as https from 'https';
+import * as vscode from 'vscode';
+import * as uuid from 'uuid';
+import { PromiseAdapter, promiseFromEvent } from './common/utils';
+import Logger from './common/logger';
+import ClientRegistrar, { ClientDetails } from './common/clientRegistrar';
+
+class UriEventHandler extends vscode.EventEmitter<vscode.Uri> implements vscode.UriHandler {
+	public handleUri(uri: vscode.Uri) {
+		this.fire(uri);
+	}
+}
+
+export const uriHandler = new UriEventHandler;
+
+const exchangeCodeForToken: (state: string, clientDetails: ClientDetails) => PromiseAdapter<vscode.Uri, string> =
+	(state, clientDetails) => async (uri, resolve, reject) => {
+		Logger.info('Exchanging code for token...');
+		const query = parseQuery(uri);
+		const code = query.code;
+
+		if (query.state !== state) {
+			reject('Received mismatched state');
+			return;
+		}
+
+		const post = https.request({
+			host: 'github.com',
+			path: `/login/oauth/access_token?client_id=${clientDetails.id}&client_secret=${clientDetails.secret}&state=${query.state}&code=${code}`,
+			method: 'POST',
+			headers: {
+				Accept: 'application/json'
+			}
+		}, result => {
+			const buffer: Buffer[] = [];
+			result.on('data', (chunk: Buffer) => {
+				buffer.push(chunk);
+			});
+			result.on('end', () => {
+				if (result.statusCode === 200) {
+					const json = JSON.parse(Buffer.concat(buffer).toString());
+					Logger.info('Token exchange success!');
+					resolve(json.access_token);
+				} else {
+					reject(new Error(result.statusMessage));
+				}
+			});
+		});
+
+		post.end();
+		post.on('error', err => {
+			reject(err);
+		});
+	};
+
+function parseQuery(uri: vscode.Uri) {
+	return uri.query.split('&').reduce((prev: any, current) => {
+		const queryString = current.split('=');
+		prev[queryString[0]] = queryString[1];
+		return prev;
+	}, {});
+}
+
+export class GitHubServer {
+	public async login(scopes: string): Promise<string> {
+		Logger.info('Logging in...');
+		const state = uuid();
+		const callbackUri = await vscode.env.asExternalUri(vscode.Uri.parse(`${vscode.env.uriScheme}://vscode.github-authentication/did-authenticate`));
+		const clientDetails = ClientRegistrar.getClientDetails(callbackUri);
+		const uri = vscode.Uri.parse(`https://github.com/login/oauth/authorize?redirect_uri=${encodeURIComponent(callbackUri.toString())}&scope=${scopes}&state=${state}&client_id=${clientDetails.id}`);
+
+		vscode.env.openExternal(uri);
+		return promiseFromEvent(uriHandler.event, exchangeCodeForToken(state, clientDetails));
+	}
+
+	public async getUserInfo(token: string): Promise<{ id: string, accountName: string }> {
+		return new Promise((resolve, reject) => {
+			Logger.info('Getting account info...');
+			const post = https.request({
+				host: 'api.github.com',
+				path: `/user`,
+				method: 'GET',
+				headers: {
+					Authorization: `token ${token}`,
+					'User-Agent': 'Visual-Studio-Code'
+				}
+			}, result => {
+				const buffer: Buffer[] = [];
+				result.on('data', (chunk: Buffer) => {
+					buffer.push(chunk);
+				});
+				result.on('end', () => {
+					if (result.statusCode === 200) {
+						const json = JSON.parse(Buffer.concat(buffer).toString());
+						Logger.info('Got account info!');
+						resolve({ id: json.id, accountName: json.login });
+					} else {
+						reject(new Error(result.statusMessage));
+					}
+				});
+			});
+
+			post.end();
+			post.on('error', err => {
+				reject(err);
+			});
+		});
+	}
+}