99 lines
2.9 KiB
TypeScript
99 lines
2.9 KiB
TypeScript
|
/*---------------------------------------------------------------------------------------------
|
||
|
|
* Copyright (c) Microsoft Corporation. All rights reserved.
|
||
|
|
* Licensed under the MIT License. See License.txt in the project root for license information.
|
||
|
|
*--------------------------------------------------------------------------------------------*/
|
||
|
|
|
||
|
|
import { window, InputBoxOptions, Uri, OutputChannel, Disposable, workspace } from 'vscode';
|
||
|
|
import { IDisposable, EmptyDisposable, toDisposable } from './util';
|
||
|
|
import * as path from 'path';
|
||
|
|
import { IIPCHandler, IIPCServer, createIPCServer } from './ipc/ipcServer';
|
||
|
|
import { CredentialsProvider, Credentials } from './api/git';
|
||
|
|
|
||
|
|
export class Askpass implements IIPCHandler {
|
||
|
|
|
||
|
|
private disposable: IDisposable = EmptyDisposable;
|
||
|
|
private cache = new Map<string, Credentials>();
|
||
|
|
private credentialsProviders = new Set<CredentialsProvider>();
|
||
|
|
|
||
|
|
static async create(outputChannel: OutputChannel, context?: string): Promise<Askpass> {
|
||
|
|
try {
|
||
|
|
return new Askpass(await createIPCServer(context));
|
||
|
|
} catch (err) {
|
||
|
|
outputChannel.appendLine(`[error] Failed to create git askpass IPC: ${err}`);
|
||
|
|
return new Askpass();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private constructor(private ipc?: IIPCServer) {
|
||
|
|
if (ipc) {
|
||
|
|
this.disposable = ipc.registerHandler('askpass', this);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
async handle({ request, host }: { request: string, host: string }): Promise<string> {
|
||
|
|
const config = workspace.getConfiguration('git', null);
|
||
|
|
const enabled = config.get<boolean>('enabled');
|
||
|
|
|
||
|
|
if (!enabled) {
|
||
|
|
return '';
|
||
|
|
}
|
||
|
|
|
||
|
|
const uri = Uri.parse(host);
|
||
|
|
const authority = uri.authority.replace(/^.*@/, '');
|
||
|
|
const password = /password/i.test(request);
|
||
|
|
const cached = this.cache.get(authority);
|
||
|
|
|
||
|
|
if (cached && password) {
|
||
|
|
this.cache.delete(authority);
|
||
|
|
return cached.password;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!password) {
|
||
|
|
for (const credentialsProvider of this.credentialsProviders) {
|
||
|
|
try {
|
||
|
|
const credentials = await credentialsProvider.getCredentials(uri);
|
||
|
|
|
||
|
|
if (credentials) {
|
||
|
|
this.cache.set(authority, credentials);
|
||
|
|
setTimeout(() => this.cache.delete(authority), 60_000);
|
||
|
|
return credentials.username;
|
||
|
|
}
|
||
|
|
} catch { }
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const options: InputBoxOptions = {
|
||
|
|
password,
|
||
|
|
placeHolder: request,
|
||
|
|
prompt: `Git: ${host}`,
|
||
|
|
ignoreFocusOut: true
|
||
|
|
};
|
||
|
|
|
||
|
|
return await window.showInputBox(options) || '';
|
||
|
|
}
|
||
|
|
|
||
|
|
getEnv(): { [key: string]: string; } {
|
||
|
|
if (!this.ipc) {
|
||
|
|
return {
|
||
|
|
GIT_ASKPASS: path.join(__dirname, 'askpass-empty.sh')
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
return {
|
||
|
|
...this.ipc.getEnv(),
|
||
|
|
GIT_ASKPASS: path.join(__dirname, 'askpass.sh'),
|
||
|
|
VSCODE_GIT_ASKPASS_NODE: process.execPath,
|
||
|
|
VSCODE_GIT_ASKPASS_MAIN: path.join(__dirname, 'askpass-main.js')
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
registerCredentialsProvider(provider: CredentialsProvider): Disposable {
|
||
|
|
this.credentialsProviders.add(provider);
|
||
|
|
return toDisposable(() => this.credentialsProviders.delete(provider));
|
||
|
|
}
|
||
|
|
|
||
|
|
dispose(): void {
|
||
|
|
this.disposable.dispose();
|
||
|
|
}
|
||
|
|
}
|