commands.ts 25.1 KB
Newer Older
J
Joao Moreno 已提交
1 2 3 4 5 6 7
/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the MIT License. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

'use strict';

8
import { Uri, commands, scm, Disposable, window, workspace, QuickPickItem, OutputChannel, Range, WorkspaceEdit, Position, LineChange, SourceControlResourceState } from 'vscode';
9
import { Ref, RefType, Git, GitErrorCodes } from './git';
10
import { Model, Resource, Status, CommitOptions, WorkingTreeGroup, IndexGroup, MergeGroup } from './model';
J
Joao Moreno 已提交
11
import { toGitUri, fromGitUri } from './uri';
12
import { applyLineChanges, intersectDiffWithRange, toLineRanges, invertLineChange } from './staging';
J
Joao Moreno 已提交
13
import * as path from 'path';
J
Joao Moreno 已提交
14
import * as os from 'os';
J
Joao Moreno 已提交
15
import TelemetryReporter from 'vscode-extension-telemetry';
J
Joao Moreno 已提交
16 17 18
import * as nls from 'vscode-nls';

const localize = nls.loadMessageBundle();
J
Joao Moreno 已提交
19

J
Joao Moreno 已提交
20 21 22 23 24 25 26
class CheckoutItem implements QuickPickItem {

	protected get shortCommit(): string { return (this.ref.commit || '').substr(0, 8); }
	protected get treeish(): string | undefined { return this.ref.name; }
	get label(): string { return this.ref.name || this.shortCommit; }
	get description(): string { return this.shortCommit; }

J
Joao Moreno 已提交
27
	constructor(protected ref: Ref) { }
J
Joao Moreno 已提交
28 29 30 31 32 33 34 35 36 37 38 39 40 41

	async run(model: Model): Promise<void> {
		const ref = this.treeish;

		if (!ref) {
			return;
		}

		await model.checkout(ref);
	}
}

class CheckoutTagItem extends CheckoutItem {

J
Joao Moreno 已提交
42 43 44
	get description(): string {
		return localize('tag at', "Tag at {0}", this.shortCommit);
	}
J
Joao Moreno 已提交
45 46 47 48
}

class CheckoutRemoteHeadItem extends CheckoutItem {

J
Joao Moreno 已提交
49 50 51
	get description(): string {
		return localize('remote branch at', "Remote branch at {0}", this.shortCommit);
	}
J
Joao Moreno 已提交
52 53 54 55 56 57 58 59 60 61 62

	protected get treeish(): string | undefined {
		if (!this.ref.name) {
			return;
		}

		const match = /^[^/]+\/(.*)$/.exec(this.ref.name);
		return match ? match[1] : this.ref.name;
	}
}

63 64 65 66 67 68 69 70 71
interface Command {
	commandId: string;
	key: string;
	method: Function;
	skipModelCheck: boolean;
	requiresDiffInformation: boolean;
}

const Commands: Command[] = [];
J
Joao Moreno 已提交
72

73
function command(commandId: string, skipModelCheck = false, requiresDiffInformation = false): Function {
J
Joao Moreno 已提交
74
	return (target: any, key: string, descriptor: any) => {
J
Joao Moreno 已提交
75 76 77 78
		if (!(typeof descriptor.value === 'function')) {
			throw new Error('not supported');
		}

79
		Commands.push({ commandId, key, method: descriptor.value, skipModelCheck, requiresDiffInformation });
J
Joao Moreno 已提交
80 81
	};
}
J
Joao Moreno 已提交
82

J
Joao Moreno 已提交
83
export class CommandCenter {
J
Joao Moreno 已提交
84

J
Joao Moreno 已提交
85
	private model: Model;
J
Joao Moreno 已提交
86
	private disposables: Disposable[];
J
Joao Moreno 已提交
87

J
Joao Moreno 已提交
88
	constructor(
J
Joao Moreno 已提交
89
		private git: Git,
J
Joao Moreno 已提交
90
		model: Model | undefined,
J
Joao Moreno 已提交
91 92
		private outputChannel: OutputChannel,
		private telemetryReporter: TelemetryReporter
J
Joao Moreno 已提交
93
	) {
J
Joao Moreno 已提交
94 95 96 97
		if (model) {
			this.model = model;
		}

J
Joao Moreno 已提交
98
		this.disposables = Commands
99 100 101 102 103 104 105 106 107
			.map(({ commandId, key, method, skipModelCheck, requiresDiffInformation }) => {
				const command = this.createCommand(commandId, key, method, skipModelCheck);

				if (requiresDiffInformation) {
					return commands.registerDiffInformationCommand(commandId, command);
				} else {
					return commands.registerCommand(commandId, command);
				}
			});
J
Joao Moreno 已提交
108 109
	}

J
Joao Moreno 已提交
110
	@command('git.refresh')
J
Joao Moreno 已提交
111
	async refresh(): Promise<void> {
J
Joao Moreno 已提交
112
		await this.model.status();
J
Joao Moreno 已提交
113
	}
J
Joao Moreno 已提交
114

J
Joao Moreno 已提交
115 116 117 118 119 120
	@command('git.openResource')
	async openResource(resource: Resource): Promise<void> {
		await this._openResource(resource);
	}

	private async _openResource(resource: Resource): Promise<void> {
J
Joao Moreno 已提交
121 122 123 124
		const left = this.getLeftResource(resource);
		const right = this.getRightResource(resource);
		const title = this.getTitle(resource);

J
Joao Moreno 已提交
125 126 127 128 129
		if (!right) {
			// TODO
			console.error('oh no');
			return;
		}
J
Joao Moreno 已提交
130

J
Joao Moreno 已提交
131
		if (!left) {
J
Joao Moreno 已提交
132
			return await commands.executeCommand<void>('vscode.open', right);
J
Joao Moreno 已提交
133 134
		}

J
Joao Moreno 已提交
135
		return await commands.executeCommand<void>('vscode.diff', left, right, title);
J
Joao Moreno 已提交
136 137 138 139 140 141
	}

	private getLeftResource(resource: Resource): Uri | undefined {
		switch (resource.type) {
			case Status.INDEX_MODIFIED:
			case Status.INDEX_RENAMED:
J
Joao Moreno 已提交
142
				return toGitUri(resource.original, 'HEAD');
J
Joao Moreno 已提交
143 144

			case Status.MODIFIED:
J
Joao Moreno 已提交
145
				return toGitUri(resource.resourceUri, '~');
J
Joao Moreno 已提交
146
		}
J
Joao Moreno 已提交
147
	}
J
Joao Moreno 已提交
148

J
Joao Moreno 已提交
149 150 151 152 153 154
	private getRightResource(resource: Resource): Uri | undefined {
		switch (resource.type) {
			case Status.INDEX_MODIFIED:
			case Status.INDEX_ADDED:
			case Status.INDEX_COPIED:
			case Status.INDEX_RENAMED:
J
Joao Moreno 已提交
155
				return toGitUri(resource.resourceUri, '');
J
Joao Moreno 已提交
156 157 158

			case Status.INDEX_DELETED:
			case Status.DELETED:
J
Joao Moreno 已提交
159
				return toGitUri(resource.resourceUri, 'HEAD');
J
Joao Moreno 已提交
160 161 162 163

			case Status.MODIFIED:
			case Status.UNTRACKED:
			case Status.IGNORED:
J
Joao Moreno 已提交
164 165
				const uriString = resource.resourceUri.toString();
				const [indexStatus] = this.model.indexGroup.resources.filter(r => r.resourceUri.toString() === uriString);
J
Joao Moreno 已提交
166

J
Joao Moreno 已提交
167 168
				if (indexStatus && indexStatus.renameResourceUri) {
					return indexStatus.renameResourceUri;
J
Joao Moreno 已提交
169 170
				}

J
Joao Moreno 已提交
171
				return resource.resourceUri;
J
Joao Moreno 已提交
172

J
Joao Moreno 已提交
173
			case Status.BOTH_MODIFIED:
J
Joao Moreno 已提交
174
				return resource.resourceUri;
J
Joao Moreno 已提交
175 176 177 178
		}
	}

	private getTitle(resource: Resource): string {
J
Joao Moreno 已提交
179
		const basename = path.basename(resource.resourceUri.fsPath);
J
Joao Moreno 已提交
180 181 182 183 184 185 186 187 188 189 190 191 192

		switch (resource.type) {
			case Status.INDEX_MODIFIED:
			case Status.INDEX_RENAMED:
				return `${basename} (Index)`;

			case Status.MODIFIED:
				return `${basename} (Working Tree)`;
		}

		return '';
	}

193 194
	@command('git.clone', true)
	async clone(): Promise<void> {
J
Joao Moreno 已提交
195
		const url = await window.showInputBox({
J
Joao Moreno 已提交
196 197
			prompt: localize('repourl', "Repository URL"),
			ignoreFocusOut: true
J
Joao Moreno 已提交
198 199 200
		});

		if (!url) {
201 202
			this.telemetryReporter.sendTelemetryEvent('clone', { outcome: 'no_URL' });
			return;
J
Joao Moreno 已提交
203 204
		}

205
		const config = workspace.getConfiguration('git');
J
Joao Moreno 已提交
206
		const value = config.get<string>('defaultCloneDirectory') || os.homedir();
207

J
Joao Moreno 已提交
208 209
		const parentPath = await window.showInputBox({
			prompt: localize('parent', "Parent Directory"),
J
Joao Moreno 已提交
210
			value,
J
Joao Moreno 已提交
211
			ignoreFocusOut: true
J
Joao Moreno 已提交
212 213 214
		});

		if (!parentPath) {
215 216
			this.telemetryReporter.sendTelemetryEvent('clone', { outcome: 'no_directory' });
			return;
J
Joao Moreno 已提交
217 218
		}

J
Joao Moreno 已提交
219
		const clonePromise = this.git.clone(url, parentPath);
J
Joao Moreno 已提交
220 221
		window.setStatusBarMessage(localize('cloning', "Cloning git repository..."), clonePromise);

222
		try {
223 224 225 226 227 228 229 230 231 232
			const repositoryPath = await clonePromise;

			const open = localize('openrepo', "Open Repository");
			const result = await window.showInformationMessage(localize('proposeopen', "Would you like to open the cloned repository?"), open);

			const openFolder = result === open;
			this.telemetryReporter.sendTelemetryEvent('clone', { outcome: 'success' }, { openFolder: openFolder ? 1 : 0 });
			if (openFolder) {
				commands.executeCommand('vscode.openFolder', Uri.file(repositoryPath));
			}
233 234
		} catch (err) {
			if (/already exists and is not an empty directory/.test(err && err.stderr || '')) {
235 236 237
				this.telemetryReporter.sendTelemetryEvent('clone', { outcome: 'directory_not_empty' });
			} else {
				this.telemetryReporter.sendTelemetryEvent('clone', { outcome: 'error' });
238 239
			}
			throw err;
J
Joao Moreno 已提交
240 241 242
		}
	}

J
Joao Moreno 已提交
243 244 245 246 247
	@command('git.init')
	async init(): Promise<void> {
		await this.model.init();
	}

J
Joao Moreno 已提交
248
	@command('git.openFile')
249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268
	async openFile(arg?: Resource | Uri): Promise<void> {
		let uri: Uri | undefined;

		if (arg instanceof Uri) {
			if (arg.scheme === 'git') {
				uri = Uri.file(fromGitUri(arg).path);
			} else if (arg.scheme === 'file') {
				uri = arg;
			}
		} else {
			let resource = arg;

			if (!(resource instanceof Resource)) {
				// can happen when called from a keybinding
				resource = this.getSCMResource();
			}

			if (resource) {
				uri = resource.resourceUri;
			}
J
Joao Moreno 已提交
269 270
		}

271
		if (!uri) {
J
Joao Moreno 已提交
272
			return;
J
Joao Moreno 已提交
273 274
		}

275
		return await commands.executeCommand<void>('vscode.open', uri);
J
Joao Moreno 已提交
276 277 278
	}

	@command('git.openChange')
279 280 281 282 283 284 285 286
	async openChange(arg?: Resource | Uri): Promise<void> {
		let resource: Resource | undefined = undefined;

		if (arg instanceof Resource) {
			resource = arg;
		} else if (arg instanceof Uri) {
			resource = this.getSCMResource(arg);
		} else {
J
Joao Moreno 已提交
287 288 289
			resource = this.getSCMResource();
		}

J
Joao Moreno 已提交
290 291
		if (!resource) {
			return;
J
Joao Moreno 已提交
292 293
		}

J
Joao Moreno 已提交
294
		return await this._openResource(resource);
J
Joao Moreno 已提交
295 296
	}

297 298 299
	@command('git.openFileFromUri')
	async openFileFromUri(uri?: Uri): Promise<void> {
		const resource = this.getSCMResource(uri);
J
Joao Moreno 已提交
300 301 302 303 304 305 306 307 308 309
		let uriToOpen: Uri | undefined;

		if (resource) {
			uriToOpen = resource.resourceUri;
		} else if (uri && uri.scheme === 'git') {
			const { path } = fromGitUri(uri);
			uriToOpen = Uri.file(path);
		} else if (uri && uri.scheme === 'file') {
			uriToOpen = uri;
		}
310

J
Joao Moreno 已提交
311
		if (!uriToOpen) {
312 313 314
			return;
		}

J
Joao Moreno 已提交
315
		return await commands.executeCommand<void>('vscode.open', uriToOpen);
316 317
	}

J
Joao Moreno 已提交
318
	@command('git.stage')
319
	async stage(...resourceStates: SourceControlResourceState[]): Promise<void> {
J
Joao Moreno 已提交
320
		if (resourceStates.length === 0 || !(resourceStates[0].resourceUri instanceof Uri)) {
321
			const resource = this.getSCMResource();
322 323 324 325 326 327 328 329

			if (!resource) {
				return;
			}

			resourceStates = [resource];
		}

330 331 332
		const resources = resourceStates
			.filter(s => s instanceof Resource && (s.resourceGroup instanceof WorkingTreeGroup || s.resourceGroup instanceof MergeGroup)) as Resource[];

333
		if (!resources.length) {
J
Joao Moreno 已提交
334 335
			return;
		}
J
Joao Moreno 已提交
336

337
		return await this.model.add(...resources);
J
Joao Moreno 已提交
338 339
	}

J
Joao Moreno 已提交
340
	@command('git.stageAll')
J
Joao Moreno 已提交
341
	async stageAll(): Promise<void> {
J
Joao Moreno 已提交
342 343 344
		return await this.model.add();
	}

345 346
	@command('git.stageSelectedRanges', false, true)
	async stageSelectedRanges(diffs: LineChange[]): Promise<void> {
J
Joao Moreno 已提交
347 348 349 350 351 352 353 354 355
		const textEditor = window.activeTextEditor;

		if (!textEditor) {
			return;
		}

		const modifiedDocument = textEditor.document;
		const modifiedUri = modifiedDocument.uri;

J
Joao Moreno 已提交
356
		if (modifiedUri.scheme !== 'file') {
J
Joao Moreno 已提交
357 358 359
			return;
		}

J
Joao Moreno 已提交
360
		const originalUri = toGitUri(modifiedUri, '~');
J
Joao Moreno 已提交
361
		const originalDocument = await workspace.openTextDocument(originalUri);
362 363 364 365
		const selectedLines = toLineRanges(textEditor.selections, modifiedDocument);
		const selectedDiffs = diffs
			.map(diff => selectedLines.reduce<LineChange | null>((result, range) => result || intersectDiffWithRange(modifiedDocument, diff, range), null))
			.filter(d => !!d) as LineChange[];
J
Joao Moreno 已提交
366 367 368 369 370

		if (!selectedDiffs.length) {
			return;
		}

371 372
		const result = applyLineChanges(originalDocument, modifiedDocument, selectedDiffs);

J
Joao Moreno 已提交
373
		await this.model.stage(modifiedUri, result);
J
Joao Moreno 已提交
374
	}
J
Joao Moreno 已提交
375

376 377
	@command('git.revertSelectedRanges', false, true)
	async revertSelectedRanges(diffs: LineChange[]): Promise<void> {
J
Joao Moreno 已提交
378 379 380 381 382 383 384 385 386 387 388 389 390
		const textEditor = window.activeTextEditor;

		if (!textEditor) {
			return;
		}

		const modifiedDocument = textEditor.document;
		const modifiedUri = modifiedDocument.uri;

		if (modifiedUri.scheme !== 'file') {
			return;
		}

J
Joao Moreno 已提交
391
		const originalUri = toGitUri(modifiedUri, '~');
J
Joao Moreno 已提交
392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414
		const originalDocument = await workspace.openTextDocument(originalUri);
		const selections = textEditor.selections;
		const selectedDiffs = diffs.filter(diff => {
			const modifiedRange = diff.modifiedEndLineNumber === 0
				? new Range(modifiedDocument.lineAt(diff.modifiedStartLineNumber - 1).range.end, modifiedDocument.lineAt(diff.modifiedStartLineNumber).range.start)
				: new Range(modifiedDocument.lineAt(diff.modifiedStartLineNumber - 1).range.start, modifiedDocument.lineAt(diff.modifiedEndLineNumber - 1).range.end);

			return selections.every(selection => !selection.intersection(modifiedRange));
		});

		if (selectedDiffs.length === diffs.length) {
			return;
		}

		const basename = path.basename(modifiedUri.fsPath);
		const message = localize('confirm revert', "Are you sure you want to revert the selected changes in {0}?", basename);
		const yes = localize('revert', "Revert Changes");
		const pick = await window.showWarningMessage(message, { modal: true }, yes);

		if (pick !== yes) {
			return;
		}

415
		const result = applyLineChanges(originalDocument, modifiedDocument, selectedDiffs);
J
Joao Moreno 已提交
416 417 418 419 420
		const edit = new WorkspaceEdit();
		edit.replace(modifiedUri, new Range(new Position(0, 0), modifiedDocument.lineAt(modifiedDocument.lineCount - 1).range.end), result);
		workspace.applyEdit(edit);
	}

J
Joao Moreno 已提交
421
	@command('git.unstage')
422
	async unstage(...resourceStates: SourceControlResourceState[]): Promise<void> {
J
Joao Moreno 已提交
423
		if (resourceStates.length === 0 || !(resourceStates[0].resourceUri instanceof Uri)) {
424
			const resource = this.getSCMResource();
425 426 427 428 429 430 431 432

			if (!resource) {
				return;
			}

			resourceStates = [resource];
		}

433 434 435
		const resources = resourceStates
			.filter(s => s instanceof Resource && s.resourceGroup instanceof IndexGroup) as Resource[];

436
		if (!resources.length) {
J
Joao Moreno 已提交
437 438 439
			return;
		}

440
		return await this.model.revertFiles(...resources);
J
Joao Moreno 已提交
441 442
	}

J
Joao Moreno 已提交
443
	@command('git.unstageAll')
J
Joao Moreno 已提交
444
	async unstageAll(): Promise<void> {
J
Joao Moreno 已提交
445
		return await this.model.revertFiles();
J
Joao Moreno 已提交
446
	}
J
Joao Moreno 已提交
447

448 449
	@command('git.unstageSelectedRanges', false, true)
	async unstageSelectedRanges(diffs: LineChange[]): Promise<void> {
J
Joao Moreno 已提交
450 451 452 453 454 455 456 457 458
		const textEditor = window.activeTextEditor;

		if (!textEditor) {
			return;
		}

		const modifiedDocument = textEditor.document;
		const modifiedUri = modifiedDocument.uri;

459 460 461 462 463 464 465
		if (modifiedUri.scheme !== 'git') {
			return;
		}

		const { ref } = fromGitUri(modifiedUri);

		if (ref !== '') {
J
Joao Moreno 已提交
466 467 468
			return;
		}

J
Joao Moreno 已提交
469
		const originalUri = toGitUri(modifiedUri, 'HEAD');
J
Joao Moreno 已提交
470
		const originalDocument = await workspace.openTextDocument(originalUri);
471 472 473 474
		const selectedLines = toLineRanges(textEditor.selections, modifiedDocument);
		const selectedDiffs = diffs
			.map(diff => selectedLines.reduce<LineChange | null>((result, range) => result || intersectDiffWithRange(modifiedDocument, diff, range), null))
			.filter(d => !!d) as LineChange[];
J
Joao Moreno 已提交
475 476 477 478 479

		if (!selectedDiffs.length) {
			return;
		}

480 481
		const invertedDiffs = selectedDiffs.map(invertLineChange);
		const result = applyLineChanges(modifiedDocument, originalDocument, invertedDiffs);
J
Joao Moreno 已提交
482 483 484 485

		await this.model.stage(modifiedUri, result);
	}

J
Joao Moreno 已提交
486
	@command('git.clean')
487
	async clean(...resourceStates: SourceControlResourceState[]): Promise<void> {
J
Joao Moreno 已提交
488
		if (resourceStates.length === 0 || !(resourceStates[0].resourceUri instanceof Uri)) {
489
			const resource = this.getSCMResource();
490 491 492 493 494 495 496 497

			if (!resource) {
				return;
			}

			resourceStates = [resource];
		}

498 499 500
		const resources = resourceStates
			.filter(s => s instanceof Resource && s.resourceGroup instanceof WorkingTreeGroup) as Resource[];

501
		if (!resources.length) {
J
Joao Moreno 已提交
502 503
			return;
		}
J
Joao Moreno 已提交
504

505
		const message = resources.length === 1
J
Joao Moreno 已提交
506
			? localize('confirm discard', "Are you sure you want to discard changes in {0}?", path.basename(resources[0].resourceUri.fsPath))
507 508
			: localize('confirm discard multiple', "Are you sure you want to discard changes in {0} files?", resources.length);

509
		const yes = localize('discard', "Discard Changes");
J
Joao Moreno 已提交
510
		const pick = await window.showWarningMessage(message, { modal: true }, yes);
J
Joao Moreno 已提交
511

J
Joao Moreno 已提交
512 513 514 515
		if (pick !== yes) {
			return;
		}

516
		await this.model.clean(...resources);
J
Joao Moreno 已提交
517
	}
J
Joao Moreno 已提交
518

J
Joao Moreno 已提交
519
	@command('git.cleanAll')
J
Joao Moreno 已提交
520
	async cleanAll(): Promise<void> {
521 522
		const message = localize('confirm discard all', "Are you sure you want to discard ALL changes? This is IRREVERSIBLE!");
		const yes = localize('discardAll', "Discard ALL Changes");
J
Joao Moreno 已提交
523
		const pick = await window.showWarningMessage(message, { modal: true }, yes);
J
Joao Moreno 已提交
524 525 526 527 528

		if (pick !== yes) {
			return;
		}

J
Joao Moreno 已提交
529
		await this.model.clean(...this.model.workingTreeGroup.resources);
J
Joao Moreno 已提交
530 531
	}

J
Joao Moreno 已提交
532
	private async smartCommit(
533
		getCommitMessage: () => Promise<string | undefined>,
J
Joao Moreno 已提交
534 535
		opts?: CommitOptions
	): Promise<boolean> {
536 537 538
		const config = workspace.getConfiguration('git');
		const enableSmartCommit = config.get<boolean>('enableSmartCommit') === true;
		const noStagedChanges = this.model.indexGroup.resources.length === 0;
539
		const noUnstagedChanges = this.model.workingTreeGroup.resources.length === 0;
540 541

		// no changes, and the user has not configured to commit all in this case
542
		if (!noUnstagedChanges && noStagedChanges && !enableSmartCommit) {
543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559 560
			// prompt the user if we want to commit all or not

			const message = localize('no staged changes', "There are no staged changes to commit. Would you like to stage all changes and commit them?");
			const yes = localize('yes', "Yes");
			const always = localize('always', "Always");
			const pick = await window.showWarningMessage(message, { modal: true }, yes, always);

			if (pick === always) {
				// update preference to enable smart commit always
				config.update('enableSmartCommit', true, false);
			}
			else if (pick !== yes) {
				// do not commit on cancel
				return false;
			}
			// for yes or always, continue onto previous smart commit behavior
		}

J
Joao Moreno 已提交
561
		if (!opts) {
562
			opts = { all: noStagedChanges };
J
Joao Moreno 已提交
563 564 565 566
		}

		if (
			// no changes
567
			(noStagedChanges && noUnstagedChanges)
J
Joao Moreno 已提交
568
			// or no staged changes and not `all`
569
			|| (!opts.all && noStagedChanges)
J
Joao Moreno 已提交
570
		) {
J
Joao Moreno 已提交
571 572 573 574
			window.showInformationMessage(localize('no changes', "There are no changes to commit."));
			return false;
		}

J
Joao Moreno 已提交
575
		const message = await getCommitMessage();
J
Joao Moreno 已提交
576 577 578 579 580 581

		if (!message) {
			// TODO@joao: show modal dialog to confirm empty message commit
			return false;
		}

J
Joao Moreno 已提交
582
		await this.model.commit(message, opts);
J
Joao Moreno 已提交
583 584 585 586

		return true;
	}

J
Joao Moreno 已提交
587
	private async commitWithAnyInput(opts?: CommitOptions): Promise<void> {
588
		const message = scm.inputBox.value;
J
Joao Moreno 已提交
589
		const getCommitMessage = async () => {
J
Joao Moreno 已提交
590 591 592 593 594 595
			if (message) {
				return message;
			}

			return await window.showInputBox({
				placeHolder: localize('commit message', "Commit message"),
J
Joao Moreno 已提交
596 597
				prompt: localize('provide commit message', "Please provide a commit message"),
				ignoreFocusOut: true
J
Joao Moreno 已提交
598
			});
J
Joao Moreno 已提交
599 600 601
		};

		const didCommit = await this.smartCommit(getCommitMessage, opts);
J
Joao Moreno 已提交
602 603

		if (message && didCommit) {
J
Joao Moreno 已提交
604
			scm.inputBox.value = await this.model.getCommitTemplate();
J
Joao Moreno 已提交
605
		}
J
Joao Moreno 已提交
606 607
	}

J
Joao Moreno 已提交
608
	@command('git.commit')
J
Joao Moreno 已提交
609 610 611 612
	async commit(): Promise<void> {
		await this.commitWithAnyInput();
	}

J
Joao Moreno 已提交
613
	@command('git.commitWithInput')
J
Joao Moreno 已提交
614
	async commitWithInput(): Promise<void> {
J
Joao Moreno 已提交
615
		const didCommit = await this.smartCommit(async () => scm.inputBox.value);
J
Joao Moreno 已提交
616 617

		if (didCommit) {
J
Joao Moreno 已提交
618
			scm.inputBox.value = await this.model.getCommitTemplate();
J
Joao Moreno 已提交
619
		}
J
Joao Moreno 已提交
620 621
	}

J
Joao Moreno 已提交
622
	@command('git.commitStaged')
J
Joao Moreno 已提交
623
	async commitStaged(): Promise<void> {
J
Joao Moreno 已提交
624
		await this.commitWithAnyInput({ all: false });
J
Joao Moreno 已提交
625 626
	}

J
Joao Moreno 已提交
627
	@command('git.commitStagedSigned')
J
Joao Moreno 已提交
628
	async commitStagedSigned(): Promise<void> {
J
Joao Moreno 已提交
629
		await this.commitWithAnyInput({ all: false, signoff: true });
J
Joao Moreno 已提交
630 631
	}

J
Joao Moreno 已提交
632
	@command('git.commitAll')
J
Joao Moreno 已提交
633
	async commitAll(): Promise<void> {
J
Joao Moreno 已提交
634
		await this.commitWithAnyInput({ all: true });
J
Joao Moreno 已提交
635 636
	}

J
Joao Moreno 已提交
637
	@command('git.commitAllSigned')
J
Joao Moreno 已提交
638
	async commitAllSigned(): Promise<void> {
J
Joao Moreno 已提交
639
		await this.commitWithAnyInput({ all: true, signoff: true });
J
Joao Moreno 已提交
640 641
	}

J
Joao Moreno 已提交
642
	@command('git.undoCommit')
J
Joao Moreno 已提交
643
	async undoCommit(): Promise<void> {
J
Joao Moreno 已提交
644 645 646 647 648 649 650 651 652
		const HEAD = this.model.HEAD;

		if (!HEAD || !HEAD.commit) {
			return;
		}

		const commit = await this.model.getCommit('HEAD');
		await this.model.reset('HEAD~');
		scm.inputBox.value = commit.message;
J
Joao Moreno 已提交
653 654
	}

J
Joao Moreno 已提交
655
	@command('git.checkout')
J
Joao Moreno 已提交
656 657 658 659 660
	async checkout(treeish: string): Promise<void> {
		if (typeof treeish === 'string') {
			return await this.model.checkout(treeish);
		}

J
Joao Moreno 已提交
661
		const config = workspace.getConfiguration('git');
J
Joao Moreno 已提交
662
		const checkoutType = config.get<string>('checkoutType') || 'all';
J
Joao Moreno 已提交
663 664 665 666 667 668 669 670 671 672 673 674
		const includeTags = checkoutType === 'all' || checkoutType === 'tags';
		const includeRemotes = checkoutType === 'all' || checkoutType === 'remote';

		const heads = this.model.refs.filter(ref => ref.type === RefType.Head)
			.map(ref => new CheckoutItem(ref));

		const tags = (includeTags ? this.model.refs.filter(ref => ref.type === RefType.Tag) : [])
			.map(ref => new CheckoutTagItem(ref));

		const remoteHeads = (includeRemotes ? this.model.refs.filter(ref => ref.type === RefType.RemoteHead) : [])
			.map(ref => new CheckoutRemoteHeadItem(ref));

J
Joao Moreno 已提交
675 676 677
		const picks = [...heads, ...tags, ...remoteHeads];
		const placeHolder = 'Select a ref to checkout';
		const choice = await window.showQuickPick<CheckoutItem>(picks, { placeHolder });
J
Joao Moreno 已提交
678 679 680 681 682 683

		if (!choice) {
			return;
		}

		await choice.run(this.model);
J
Joao Moreno 已提交
684 685
	}

J
Joao Moreno 已提交
686
	@command('git.branch')
J
Joao Moreno 已提交
687 688
	async branch(): Promise<void> {
		const result = await window.showInputBox({
J
Joao Moreno 已提交
689
			placeHolder: localize('branch name', "Branch name"),
J
Joao Moreno 已提交
690 691
			prompt: localize('provide branch name', "Please provide a branch name"),
			ignoreFocusOut: true
J
Joao Moreno 已提交
692
		});
J
Joao Moreno 已提交
693

J
Joao Moreno 已提交
694 695 696
		if (!result) {
			return;
		}
J
Joao Moreno 已提交
697

J
Joao Moreno 已提交
698 699
		const name = result.replace(/^\.|\/\.|\.\.|~|\^|:|\/$|\.lock$|\.lock\/|\\|\*|\s|^\s*$|\.$/g, '-');
		await this.model.branch(name);
J
Joao Moreno 已提交
700 701
	}

J
Joao Moreno 已提交
702
	@command('git.pull')
J
Joao Moreno 已提交
703
	async pull(): Promise<void> {
J
Joao Moreno 已提交
704 705 706 707 708 709 710 711
		const remotes = this.model.remotes;

		if (remotes.length === 0) {
			window.showWarningMessage(localize('no remotes to pull', "Your repository has no remotes configured to pull from."));
			return;
		}

		await this.model.pull();
J
Joao Moreno 已提交
712 713
	}

J
Joao Moreno 已提交
714
	@command('git.pullRebase')
J
Joao Moreno 已提交
715
	async pullRebase(): Promise<void> {
J
Joao Moreno 已提交
716 717 718 719 720 721 722 723
		const remotes = this.model.remotes;

		if (remotes.length === 0) {
			window.showWarningMessage(localize('no remotes to pull', "Your repository has no remotes configured to pull from."));
			return;
		}

		await this.model.pull(true);
J
Joao Moreno 已提交
724 725
	}

J
Joao Moreno 已提交
726
	@command('git.push')
J
Joao Moreno 已提交
727
	async push(): Promise<void> {
J
Joao Moreno 已提交
728 729 730 731 732 733 734 735
		const remotes = this.model.remotes;

		if (remotes.length === 0) {
			window.showWarningMessage(localize('no remotes to push', "Your repository has no remotes configured to push to."));
			return;
		}

		await this.model.push();
J
Joao Moreno 已提交
736 737
	}

J
Joao Moreno 已提交
738
	@command('git.pushTo')
J
Joao Moreno 已提交
739
	async pushTo(): Promise<void> {
J
Joao Moreno 已提交
740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761
		const remotes = this.model.remotes;

		if (remotes.length === 0) {
			window.showWarningMessage(localize('no remotes to push', "Your repository has no remotes configured to push to."));
			return;
		}

		if (!this.model.HEAD || !this.model.HEAD.name) {
			window.showWarningMessage(localize('nobranch', "Please check out a branch to push to a remote."));
			return;
		}

		const branchName = this.model.HEAD.name;
		const picks = remotes.map(r => ({ label: r.name, description: r.url }));
		const placeHolder = localize('pick remote', "Pick a remote to publish the branch '{0}' to:", branchName);
		const pick = await window.showQuickPick(picks, { placeHolder });

		if (!pick) {
			return;
		}

		this.model.push(pick.label, branchName);
J
Joao Moreno 已提交
762 763
	}

J
Joao Moreno 已提交
764
	@command('git.sync')
J
Joao Moreno 已提交
765
	async sync(): Promise<void> {
J
Joao Moreno 已提交
766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787
		const HEAD = this.model.HEAD;

		if (!HEAD || !HEAD.upstream) {
			return;
		}

		const config = workspace.getConfiguration('git');
		const shouldPrompt = config.get<boolean>('confirmSync') === true;

		if (shouldPrompt) {
			const message = localize('sync is unpredictable', "This action will push and pull commits to and from '{0}'.", HEAD.upstream);
			const yes = localize('ok', "OK");
			const neverAgain = localize('never again', "OK, Never Show Again");
			const pick = await window.showWarningMessage(message, { modal: true }, yes, neverAgain);

			if (pick === neverAgain) {
				await config.update('confirmSync', false, true);
			} else if (pick !== yes) {
				return;
			}
		}

J
Joao Moreno 已提交
788 789 790
		await this.model.sync();
	}

J
Joao Moreno 已提交
791
	@command('git.publish')
J
Joao Moreno 已提交
792
	async publish(): Promise<void> {
J
Joao Moreno 已提交
793 794 795 796 797 798 799
		const remotes = this.model.remotes;

		if (remotes.length === 0) {
			window.showWarningMessage(localize('no remotes to publish', "Your repository has no remotes configured to publish to."));
			return;
		}

J
Joao Moreno 已提交
800 801
		const branchName = this.model.HEAD && this.model.HEAD.name || '';
		const picks = this.model.remotes.map(r => r.name);
J
Joao Moreno 已提交
802
		const placeHolder = localize('pick remote', "Pick a remote to publish the branch '{0}' to:", branchName);
J
Joao Moreno 已提交
803 804 805 806 807 808 809 810 811
		const choice = await window.showQuickPick(picks, { placeHolder });

		if (!choice) {
			return;
		}

		await this.model.push(choice, branchName, { setUpstream: true });
	}

J
Joao Moreno 已提交
812
	@command('git.showOutput')
J
Joao Moreno 已提交
813 814 815 816
	showOutput(): void {
		this.outputChannel.show();
	}

J
Joao Moreno 已提交
817
	private createCommand(id: string, key: string, method: Function, skipModelCheck: boolean): (...args: any[]) => any {
818
		const result = (...args) => {
J
Joao Moreno 已提交
819
			if (!skipModelCheck && !this.model) {
J
Joao Moreno 已提交
820 821 822 823
				window.showInformationMessage(localize('disabled', "Git is either disabled or not supported in this workspace"));
				return;
			}

J
Joao Moreno 已提交
824 825
			this.telemetryReporter.sendTelemetryEvent('git.command', { command: id });

J
Joao Moreno 已提交
826 827 828 829 830 831
			const result = Promise.resolve(method.apply(this, args));

			return result.catch(async err => {
				let message: string;

				switch (err.gitErrorCode) {
832
					case GitErrorCodes.DirtyWorkTree:
J
Joao Moreno 已提交
833 834
						message = localize('clean repo', "Please clean your repository working tree before checkout.");
						break;
835 836 837
					case GitErrorCodes.PushRejected:
						message = localize('cant push', "Can't push refs to remote. Run 'Pull' first to integrate your changes.");
						break;
J
Joao Moreno 已提交
838
					default:
839 840 841
						const hint = (err.stderr || err.message || String(err))
							.replace(/^error: /mi, '')
							.replace(/^> husky.*$/mi, '')
J
Joao Moreno 已提交
842
							.split(/[\r\n]/)
843 844 845 846 847 848
							.filter(line => !!line)
						[0];

						message = hint
							? localize('git error details', "Git: {0}", hint)
							: localize('git error', "Git error");
J
Joao Moreno 已提交
849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866

						break;
				}

				if (!message) {
					console.error(err);
					return;
				}

				const outputChannel = this.outputChannel as OutputChannel;
				const openOutputChannelChoice = localize('open git log', "Open Git Log");
				const choice = await window.showErrorMessage(message, openOutputChannelChoice);

				if (choice === openOutputChannelChoice) {
					outputChannel.show();
				}
			});
		};
867 868 869 870 871

		// patch this object, so people can call methods directly
		this[key] = result;

		return result;
J
Joao Moreno 已提交
872 873
	}

874 875
	private getSCMResource(uri?: Uri): Resource | undefined {
		uri = uri ? uri : window.activeTextEditor && window.activeTextEditor.document.uri;
J
Joao Moreno 已提交
876 877

		if (!uri) {
878
			return undefined;
J
Joao Moreno 已提交
879 880 881
		}

		if (uri.scheme === 'git') {
J
Joao Moreno 已提交
882 883
			const { path } = fromGitUri(uri);
			uri = Uri.file(path);
J
Joao Moreno 已提交
884 885 886 887 888
		}

		if (uri.scheme === 'file') {
			const uriString = uri.toString();

J
Joao Moreno 已提交
889 890
			return this.model.workingTreeGroup.resources.filter(r => r.resourceUri.toString() === uriString)[0]
				|| this.model.indexGroup.resources.filter(r => r.resourceUri.toString() === uriString)[0];
J
Joao Moreno 已提交
891 892 893
		}
	}

J
Joao Moreno 已提交
894 895 896
	dispose(): void {
		this.disposables.forEach(d => d.dispose());
	}
J
Joao Moreno 已提交
897
}