microsoft_vscode/extensions/git/src/askpass.ts

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

'use strict';

import { Disposable, window, InputBoxOptions } from 'vscode';
import * as path from 'path';
import * as http from 'http';

export class Askpass implements Disposable {

	private server: http.Server;
	private portPromise: Promise<number>;

	constructor() {
		this.server = http.createServer((req, res) => this.onRequest(req, res));
		this.server.listen(0, 'localhost');

		this.portPromise = new Promise(c => {
			this.server.on('listening', () => c(this.server.address().port));
		});

		this.server.on('error', err => console.error(err));
	}

	private onRequest(req: http.ServerRequest, res: http.ServerResponse): void {
		const chunks: string[] = [];
		req.setEncoding('utf8');
		req.on('data', (d: string) => chunks.push(d));
		req.on('end', () => {
			const {request, host} = JSON.parse(chunks.join(''));

			this.prompt(host, request).then(result => {
				res.writeHead(200);
				res.end(JSON.stringify(result));
			}, () => {
				res.writeHead(500);
				res.end();
			});
		});
	}

	private async prompt(host: string, request: string): Promise<string> {
		const options: InputBoxOptions = {
			password: /password/i.test(request),
			placeHolder: request,
			prompt: `Git: ${host}`,
			ignoreFocusOut: true
		};

		return await window.showInputBox(options) || '';
	}

	getEnv(): Promise<any> {
		return this.portPromise.then(port => ({
			ELECTRON_RUN_AS_NODE: '1',
			GIT_ASKPASS: path.join(__dirname, 'askpass.sh'),
			VSCODE_GIT_ASKPASS_NODE: process.execPath,
			VSCODE_GIT_ASKPASS_MAIN: path.join(__dirname, 'askpass-main.js'),
			VSCODE_GIT_ASKPASS_PORT: String(port)
		}));
	}

	dispose(): void {
		this.server.close();
	}
}