repository.ts 38.7 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';

J
Joao Moreno 已提交
8
import { commands, Uri, Command, EventEmitter, Event, scm, SourceControl, SourceControlInputBox, SourceControlResourceGroup, SourceControlResourceState, SourceControlResourceDecorations, SourceControlInputBoxValidation, Disposable, ProgressLocation, window, workspace, WorkspaceEdit, ThemeColor, DecorationData, Memento, SourceControlInputBoxValidationType } from 'vscode';
J
Joao Moreno 已提交
9
import { Repository as BaseRepository, Ref, Branch, Remote, Commit, GitErrorCodes, Stash, RefType, GitError, Submodule, DiffOptions } from './git';
J
Joao Moreno 已提交
10
import { anyEvent, filterEvent, eventToPromise, dispose, find, isDescendant, IDisposable, onceEvent, EmptyDisposable, debounceEvent } from './util';
J
Joao Moreno 已提交
11
import { memoize, throttle, debounce } from './decorators';
J
Joao Moreno 已提交
12
import { toGitUri } from './uri';
J
Joao Moreno 已提交
13
import { AutoFetcher } from './autofetch';
J
Joao Moreno 已提交
14
import * as path from 'path';
J
Joao Moreno 已提交
15
import * as nls from 'vscode-nls';
16
import * as fs from 'fs';
M
Matt Bierner 已提交
17
import { StatusBarCommands } from './statusbar';
J
Joao Moreno 已提交
18

J
Joao Moreno 已提交
19 20
const timeout = (millis: number) => new Promise(c => setTimeout(c, millis));

J
Joao Moreno 已提交
21
const localize = nls.loadMessageBundle();
J
Joao Moreno 已提交
22 23 24 25 26 27
const iconsRootPath = path.join(path.dirname(__dirname), 'resources', 'icons');

function getIconUri(iconName: string, theme: string): Uri {
	return Uri.file(path.join(iconsRootPath, theme, `${iconName}.svg`));
}

J
Joao 已提交
28
export enum RepositoryState {
J
Joao Moreno 已提交
29
	Idle,
J
Joao Moreno 已提交
30
	Disposed
J
Joao Moreno 已提交
31 32
}

J
Joao Moreno 已提交
33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
export enum Status {
	INDEX_MODIFIED,
	INDEX_ADDED,
	INDEX_DELETED,
	INDEX_RENAMED,
	INDEX_COPIED,

	MODIFIED,
	DELETED,
	UNTRACKED,
	IGNORED,

	ADDED_BY_US,
	ADDED_BY_THEM,
	DELETED_BY_US,
	DELETED_BY_THEM,
	BOTH_ADDED,
	BOTH_DELETED,
	BOTH_MODIFIED
}

J
Joao Moreno 已提交
54 55 56 57 58 59
export enum ResourceGroupType {
	Merge,
	Index,
	WorkingTree
}

J
Joao Moreno 已提交
60
export class Resource implements SourceControlResourceState {
J
Joao Moreno 已提交
61

62
	@memoize
J
Joao Moreno 已提交
63
	get resourceUri(): Uri {
J
Joao Moreno 已提交
64
		if (this.renameResourceUri && (this._type === Status.MODIFIED || this._type === Status.DELETED || this._type === Status.INDEX_RENAMED || this._type === Status.INDEX_COPIED)) {
J
Joao Moreno 已提交
65 66 67 68
			return this.renameResourceUri;
		}

		return this._resourceUri;
69 70 71
	}

	@memoize
J
Joao Moreno 已提交
72 73 74 75 76 77
	get command(): Command {
		return {
			command: 'git.openResource',
			title: localize('open', "Open"),
			arguments: [this]
		};
J
Joao Moreno 已提交
78 79
	}

J
Joao Moreno 已提交
80
	get resourceGroupType(): ResourceGroupType { return this._resourceGroupType; }
J
Joao Moreno 已提交
81
	get type(): Status { return this._type; }
J
Joao Moreno 已提交
82 83
	get original(): Uri { return this._resourceUri; }
	get renameResourceUri(): Uri | undefined { return this._renameResourceUri; }
J
Joao Moreno 已提交
84

85
	private static Icons: any = {
J
Joao Moreno 已提交
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
		light: {
			Modified: getIconUri('status-modified', 'light'),
			Added: getIconUri('status-added', 'light'),
			Deleted: getIconUri('status-deleted', 'light'),
			Renamed: getIconUri('status-renamed', 'light'),
			Copied: getIconUri('status-copied', 'light'),
			Untracked: getIconUri('status-untracked', 'light'),
			Ignored: getIconUri('status-ignored', 'light'),
			Conflict: getIconUri('status-conflict', 'light'),
		},
		dark: {
			Modified: getIconUri('status-modified', 'dark'),
			Added: getIconUri('status-added', 'dark'),
			Deleted: getIconUri('status-deleted', 'dark'),
			Renamed: getIconUri('status-renamed', 'dark'),
			Copied: getIconUri('status-copied', 'dark'),
			Untracked: getIconUri('status-untracked', 'dark'),
			Ignored: getIconUri('status-ignored', 'dark'),
			Conflict: getIconUri('status-conflict', 'dark')
		}
	};

J
Joao Moreno 已提交
108
	private getIconPath(theme: string): Uri {
J
Joao Moreno 已提交
109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128
		switch (this.type) {
			case Status.INDEX_MODIFIED: return Resource.Icons[theme].Modified;
			case Status.MODIFIED: return Resource.Icons[theme].Modified;
			case Status.INDEX_ADDED: return Resource.Icons[theme].Added;
			case Status.INDEX_DELETED: return Resource.Icons[theme].Deleted;
			case Status.DELETED: return Resource.Icons[theme].Deleted;
			case Status.INDEX_RENAMED: return Resource.Icons[theme].Renamed;
			case Status.INDEX_COPIED: return Resource.Icons[theme].Copied;
			case Status.UNTRACKED: return Resource.Icons[theme].Untracked;
			case Status.IGNORED: return Resource.Icons[theme].Ignored;
			case Status.BOTH_DELETED: return Resource.Icons[theme].Conflict;
			case Status.ADDED_BY_US: return Resource.Icons[theme].Conflict;
			case Status.DELETED_BY_THEM: return Resource.Icons[theme].Conflict;
			case Status.ADDED_BY_THEM: return Resource.Icons[theme].Conflict;
			case Status.DELETED_BY_US: return Resource.Icons[theme].Conflict;
			case Status.BOTH_ADDED: return Resource.Icons[theme].Conflict;
			case Status.BOTH_MODIFIED: return Resource.Icons[theme].Conflict;
		}
	}

129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
	private get tooltip(): string {
		switch (this.type) {
			case Status.INDEX_MODIFIED: return localize('index modified', "Index Modified");
			case Status.MODIFIED: return localize('modified', "Modified");
			case Status.INDEX_ADDED: return localize('index added', "Index Added");
			case Status.INDEX_DELETED: return localize('index deleted', "Index Deleted");
			case Status.DELETED: return localize('deleted', "Deleted");
			case Status.INDEX_RENAMED: return localize('index renamed', "Index Renamed");
			case Status.INDEX_COPIED: return localize('index copied', "Index Copied");
			case Status.UNTRACKED: return localize('untracked', "Untracked");
			case Status.IGNORED: return localize('ignored', "Ignored");
			case Status.BOTH_DELETED: return localize('both deleted', "Both Deleted");
			case Status.ADDED_BY_US: return localize('added by us', "Added By Us");
			case Status.DELETED_BY_THEM: return localize('deleted by them', "Deleted By Them");
			case Status.ADDED_BY_THEM: return localize('added by them', "Added By Them");
			case Status.DELETED_BY_US: return localize('deleted by us', "Deleted By Us");
			case Status.BOTH_ADDED: return localize('both added', "Both Added");
			case Status.BOTH_MODIFIED: return localize('both modified', "Both Modified");
			default: return '';
		}
	}

J
Joao Moreno 已提交
151 152 153 154 155 156
	private get strikeThrough(): boolean {
		switch (this.type) {
			case Status.DELETED:
			case Status.BOTH_DELETED:
			case Status.DELETED_BY_THEM:
			case Status.DELETED_BY_US:
J
Joao Moreno 已提交
157
			case Status.INDEX_DELETED:
J
Joao Moreno 已提交
158 159 160 161 162 163
				return true;
			default:
				return false;
		}
	}

164 165
	@memoize
	private get faded(): boolean {
166 167 168 169
		// TODO@joao
		return false;
		// const workspaceRootPath = this.workspaceRoot.fsPath;
		// return this.resourceUri.fsPath.substr(0, workspaceRootPath.length) !== workspaceRootPath;
170 171
	}

J
Joao Moreno 已提交
172
	get decorations(): SourceControlResourceDecorations {
173 174
		const light = this._useIcons ? { iconPath: this.getIconPath('light') } : undefined;
		const dark = this._useIcons ? { iconPath: this.getIconPath('dark') } : undefined;
175
		const tooltip = this.tooltip;
176 177
		const strikeThrough = this.strikeThrough;
		const faded = this.faded;
178 179
		const letter = this.letter;
		const color = this.color;
J
Joao Moreno 已提交
180

181
		return { strikeThrough, faded, tooltip, light, dark, letter, color, source: 'git.resource' /*todo@joh*/ };
182 183
	}

N
Nick Snyder 已提交
184
	get letter(): string {
185
		switch (this.type) {
186 187 188 189 190 191 192 193 194 195
			case Status.INDEX_MODIFIED:
			case Status.MODIFIED:
				return 'M';
			case Status.INDEX_ADDED:
				return 'A';
			case Status.INDEX_DELETED:
			case Status.DELETED:
				return 'D';
			case Status.INDEX_RENAMED:
				return 'R';
196
			case Status.UNTRACKED:
197 198 199 200 201 202 203 204 205 206 207 208 209 210 211
				return 'U';
			case Status.IGNORED:
				return 'I';
			case Status.INDEX_COPIED:
			case Status.BOTH_DELETED:
			case Status.ADDED_BY_US:
			case Status.DELETED_BY_THEM:
			case Status.ADDED_BY_THEM:
			case Status.DELETED_BY_US:
			case Status.BOTH_ADDED:
			case Status.BOTH_MODIFIED:
				return 'C';
		}
	}

N
Nick Snyder 已提交
212
	get color(): ThemeColor {
213
		switch (this.type) {
214 215
			case Status.INDEX_MODIFIED:
			case Status.MODIFIED:
J
Johannes Rieken 已提交
216
				return new ThemeColor('gitDecoration.modifiedResourceForeground');
217 218
			case Status.INDEX_DELETED:
			case Status.DELETED:
J
Johannes Rieken 已提交
219
				return new ThemeColor('gitDecoration.deletedResourceForeground');
220 221
			case Status.INDEX_ADDED:
				return new ThemeColor('gitDecoration.addedResourceForeground');
222 223
			case Status.INDEX_RENAMED: // todo@joh - special color?
			case Status.UNTRACKED:
J
Johannes Rieken 已提交
224
				return new ThemeColor('gitDecoration.untrackedResourceForeground');
225
			case Status.IGNORED:
J
Johannes Rieken 已提交
226
				return new ThemeColor('gitDecoration.ignoredResourceForeground');
227 228 229 230 231 232 233 234
			case Status.INDEX_COPIED:
			case Status.BOTH_DELETED:
			case Status.ADDED_BY_US:
			case Status.DELETED_BY_THEM:
			case Status.ADDED_BY_THEM:
			case Status.DELETED_BY_US:
			case Status.BOTH_ADDED:
			case Status.BOTH_MODIFIED:
J
Johannes Rieken 已提交
235
				return new ThemeColor('gitDecoration.conflictingResourceForeground');
236
		}
J
Joao Moreno 已提交
237 238
	}

J
Johannes Rieken 已提交
239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
	get priority(): number {
		switch (this.type) {
			case Status.INDEX_MODIFIED:
			case Status.MODIFIED:
				return 2;
			case Status.IGNORED:
				return 3;
			case Status.INDEX_COPIED:
			case Status.BOTH_DELETED:
			case Status.ADDED_BY_US:
			case Status.DELETED_BY_THEM:
			case Status.ADDED_BY_THEM:
			case Status.DELETED_BY_US:
			case Status.BOTH_ADDED:
			case Status.BOTH_MODIFIED:
				return 4;
			default:
				return 1;
		}
	}

N
Nick Snyder 已提交
260
	get resourceDecoration(): DecorationData {
261 262 263
		const title = this.tooltip;
		const abbreviation = this.letter;
		const color = this.color;
J
Johannes Rieken 已提交
264
		const priority = this.priority;
265
		return { bubble: true, source: 'git.resource', title, abbreviation, color, priority };
266 267
	}

268
	constructor(
J
Joao Moreno 已提交
269
		private _resourceGroupType: ResourceGroupType,
270 271
		private _resourceUri: Uri,
		private _type: Status,
272
		private _useIcons: boolean,
273 274
		private _renameResourceUri?: Uri
	) { }
J
Joao Moreno 已提交
275 276
}

277
export enum Operation {
J
Joao Moreno 已提交
278
	Status = 'Status',
279
	Diff = 'Diff',
J
Joao Moreno 已提交
280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
	Add = 'Add',
	RevertFiles = 'RevertFiles',
	Commit = 'Commit',
	Clean = 'Clean',
	Branch = 'Branch',
	Checkout = 'Checkout',
	Reset = 'Reset',
	Fetch = 'Fetch',
	Pull = 'Pull',
	Push = 'Push',
	Sync = 'Sync',
	Show = 'Show',
	Stage = 'Stage',
	GetCommitTemplate = 'GetCommitTemplate',
	DeleteBranch = 'DeleteBranch',
295
	RenameBranch = 'RenameBranch',
296
	DeleteRef = 'DeleteRef',
J
Joao Moreno 已提交
297 298 299 300
	Merge = 'Merge',
	Ignore = 'Ignore',
	Tag = 'Tag',
	Stash = 'Stash',
J
Joao Moreno 已提交
301
	CheckIgnore = 'CheckIgnore',
J
Joao Moreno 已提交
302
	GetObjectDetails = 'GetObjectDetails',
303 304
	SubmoduleUpdate = 'SubmoduleUpdate',
	RebaseContinue = 'RebaseContinue',
305 306
}

J
Joao Moreno 已提交
307 308 309 310
function isReadOnly(operation: Operation): boolean {
	switch (operation) {
		case Operation.Show:
		case Operation.GetCommitTemplate:
311
		case Operation.CheckIgnore:
J
Joao Moreno 已提交
312
		case Operation.GetObjectDetails:
J
Joao Moreno 已提交
313 314 315 316 317 318
			return true;
		default:
			return false;
	}
}

319 320 321
function shouldShowProgress(operation: Operation): boolean {
	switch (operation) {
		case Operation.Fetch:
322
		case Operation.CheckIgnore:
J
Joao Moreno 已提交
323
		case Operation.GetObjectDetails:
J
Joao Moreno 已提交
324
		case Operation.Show:
325 326 327 328 329 330
			return false;
		default:
			return true;
	}
}

331
export interface Operations {
332
	isIdle(): boolean;
J
Joao Moreno 已提交
333
	shouldShowProgress(): boolean;
334 335 336 337 338
	isRunning(operation: Operation): boolean;
}

class OperationsImpl implements Operations {

J
Joao Moreno 已提交
339
	private operations = new Map<Operation, number>();
340

J
Joao Moreno 已提交
341 342
	start(operation: Operation): void {
		this.operations.set(operation, (this.operations.get(operation) || 0) + 1);
343 344
	}

J
Joao Moreno 已提交
345 346 347 348 349 350 351 352
	end(operation: Operation): void {
		const count = (this.operations.get(operation) || 0) - 1;

		if (count <= 0) {
			this.operations.delete(operation);
		} else {
			this.operations.set(operation, count);
		}
353 354 355
	}

	isRunning(operation: Operation): boolean {
J
Joao Moreno 已提交
356
		return this.operations.has(operation);
357
	}
358 359

	isIdle(): boolean {
J
Joao Moreno 已提交
360 361 362 363 364 365 366 367 368
		const operations = this.operations.keys();

		for (const operation of operations) {
			if (!isReadOnly(operation)) {
				return false;
			}
		}

		return true;
369
	}
J
Joao Moreno 已提交
370 371 372 373 374 375 376 377 378 379 380 381

	shouldShowProgress(): boolean {
		const operations = this.operations.keys();

		for (const operation of operations) {
			if (shouldShowProgress(operation)) {
				return true;
			}
		}

		return false;
	}
382 383
}

J
Joao Moreno 已提交
384 385 386 387
export interface CommitOptions {
	all?: boolean;
	amend?: boolean;
	signoff?: boolean;
388
	signCommit?: boolean;
J
Joao Moreno 已提交
389 390
}

J
Joao Moreno 已提交
391 392 393 394
export interface GitResourceGroup extends SourceControlResourceGroup {
	resourceStates: Resource[];
}

J
Joao Moreno 已提交
395 396 397 398 399
export interface OperationResult {
	operation: Operation;
	error: any;
}

J
Joao Moreno 已提交
400 401 402 403
class ProgressManager {

	private disposable: IDisposable = EmptyDisposable;

J
Joao Moreno 已提交
404
	constructor(repository: Repository) {
J
Joao Moreno 已提交
405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422
		const start = onceEvent(filterEvent(repository.onDidChangeOperations, () => repository.operations.shouldShowProgress()));
		const end = onceEvent(filterEvent(debounceEvent(repository.onDidChangeOperations, 300), () => !repository.operations.shouldShowProgress()));

		const setup = () => {
			this.disposable = start(() => {
				const promise = eventToPromise(end).then(() => setup());
				window.withProgress({ location: ProgressLocation.SourceControl }, () => promise);
			});
		};

		setup();
	}

	dispose(): void {
		this.disposable.dispose();
	}
}

J
Joao Moreno 已提交
423
export class Repository implements Disposable {
J
Joao Moreno 已提交
424

425 426
	private static readonly InputValidationLength = 72;

J
Joao Moreno 已提交
427 428 429
	private _onDidChangeRepository = new EventEmitter<Uri>();
	readonly onDidChangeRepository: Event<Uri> = this._onDidChangeRepository.event;

J
Joao 已提交
430 431
	private _onDidChangeState = new EventEmitter<RepositoryState>();
	readonly onDidChangeState: Event<RepositoryState> = this._onDidChangeState.event;
J
Joao Moreno 已提交
432

J
Joao Moreno 已提交
433
	private _onDidChangeStatus = new EventEmitter<void>();
J
Joao Moreno 已提交
434
	readonly onDidRunGitStatus: Event<void> = this._onDidChangeStatus.event;
J
Joao Moreno 已提交
435

J
Joao Moreno 已提交
436 437 438
	private _onDidChangeOriginalResource = new EventEmitter<Uri>();
	readonly onDidChangeOriginalResource: Event<Uri> = this._onDidChangeOriginalResource.event;

439 440 441
	private _onRunOperation = new EventEmitter<Operation>();
	readonly onRunOperation: Event<Operation> = this._onRunOperation.event;

J
Joao Moreno 已提交
442 443
	private _onDidRunOperation = new EventEmitter<OperationResult>();
	readonly onDidRunOperation: Event<OperationResult> = this._onDidRunOperation.event;
444 445 446 447 448 449

	@memoize
	get onDidChangeOperations(): Event<void> {
		return anyEvent(this.onRunOperation as Event<any>, this.onDidRunOperation as Event<any>);
	}

J
Joao Moreno 已提交
450 451 452
	private _sourceControl: SourceControl;
	get sourceControl(): SourceControl { return this._sourceControl; }

J
Joao Moreno 已提交
453 454
	get inputBox(): SourceControlInputBox { return this._sourceControl.inputBox; }

J
Joao Moreno 已提交
455 456
	private _mergeGroup: SourceControlResourceGroup;
	get mergeGroup(): GitResourceGroup { return this._mergeGroup as GitResourceGroup; }
J
Joao Moreno 已提交
457

J
Joao Moreno 已提交
458 459
	private _indexGroup: SourceControlResourceGroup;
	get indexGroup(): GitResourceGroup { return this._indexGroup as GitResourceGroup; }
J
Joao Moreno 已提交
460

J
Joao Moreno 已提交
461 462
	private _workingTreeGroup: SourceControlResourceGroup;
	get workingTreeGroup(): GitResourceGroup { return this._workingTreeGroup as GitResourceGroup; }
J
Joao Moreno 已提交
463

J
Joao Moreno 已提交
464 465 466 467 468 469 470 471 472 473 474 475 476 477 478
	private _HEAD: Branch | undefined;
	get HEAD(): Branch | undefined {
		return this._HEAD;
	}

	private _refs: Ref[] = [];
	get refs(): Ref[] {
		return this._refs;
	}

	private _remotes: Remote[] = [];
	get remotes(): Remote[] {
		return this._remotes;
	}

479 480 481 482 483
	private _submodules: Submodule[] = [];
	get submodules(): Submodule[] {
		return this._submodules;
	}

484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499
	private _rebaseCommit: Commit | undefined = undefined;

	set rebaseCommit(rebaseCommit: Commit | undefined) {
		if (this._rebaseCommit && !rebaseCommit) {
			this.inputBox.value = '';
		} else if (rebaseCommit && (!this._rebaseCommit || this._rebaseCommit.hash !== rebaseCommit.hash)) {
			this.inputBox.value = rebaseCommit.message;
		}

		this._rebaseCommit = rebaseCommit;
	}

	get rebaseCommit(): Commit | undefined {
		return this._rebaseCommit;
	}

500 501 502
	private _operations = new OperationsImpl();
	get operations(): Operations { return this._operations; }

J
Joao 已提交
503 504 505
	private _state = RepositoryState.Idle;
	get state(): RepositoryState { return this._state; }
	set state(state: RepositoryState) {
J
Joao Moreno 已提交
506 507
		this._state = state;
		this._onDidChangeState.fire(state);
J
Joao Moreno 已提交
508 509 510 511

		this._HEAD = undefined;
		this._refs = [];
		this._remotes = [];
J
Joao Moreno 已提交
512 513 514 515
		this.mergeGroup.resourceStates = [];
		this.indexGroup.resourceStates = [];
		this.workingTreeGroup.resourceStates = [];
		this._sourceControl.count = 0;
J
Joao Moreno 已提交
516 517
	}

518 519 520 521
	get root(): string {
		return this.repository.root;
	}

522 523
	private isRepositoryHuge = false;
	private didWarnAboutLimit = false;
J
Joao Moreno 已提交
524
	private disposables: Disposable[] = [];
J
Joao Moreno 已提交
525

526
	constructor(
J
Joao Moreno 已提交
527 528
		private readonly repository: BaseRepository,
		globalState: Memento
529
	) {
J
Joao Moreno 已提交
530 531 532
		const fsWatcher = workspace.createFileSystemWatcher('**');
		this.disposables.push(fsWatcher);

533
		const onWorkspaceChange = anyEvent(fsWatcher.onDidChange, fsWatcher.onDidCreate, fsWatcher.onDidDelete);
J
Joao Moreno 已提交
534
		const onRepositoryChange = filterEvent(onWorkspaceChange, uri => isDescendant(repository.root, uri.fsPath));
J
Joao Moreno 已提交
535
		const onRelevantRepositoryChange = filterEvent(onRepositoryChange, uri => !/\/\.git(\/index\.lock)?$/.test(uri.path));
J
Joao Moreno 已提交
536
		onRelevantRepositoryChange(this.onFSChange, this, this.disposables);
537

J
Joao Moreno 已提交
538
		const onRelevantGitChange = filterEvent(onRelevantRepositoryChange, uri => /\/\.git\//.test(uri.path));
J
Joao Moreno 已提交
539 540
		onRelevantGitChange(this._onDidChangeRepository.fire, this._onDidChangeRepository, this.disposables);

J
Joao Moreno 已提交
541
		this._sourceControl = scm.createSourceControl('git', 'Git', Uri.file(repository.root));
542
		this._sourceControl.inputBox.placeholder = localize('commitMessage', "Message (press {0} to commit)");
J
Joao Moreno 已提交
543
		this._sourceControl.acceptInputCommand = { command: 'git.commitWithInput', title: localize('commit', "Commit"), arguments: [this._sourceControl] };
J
Joao Moreno 已提交
544
		this._sourceControl.quickDiffProvider = this;
J
Joao Moreno 已提交
545
		this._sourceControl.inputBox.validateInput = this.validateInput.bind(this);
J
Joao Moreno 已提交
546 547 548 549 550 551 552 553 554 555 556 557 558
		this.disposables.push(this._sourceControl);

		this._mergeGroup = this._sourceControl.createResourceGroup('merge', localize('merge changes', "Merge Changes"));
		this._indexGroup = this._sourceControl.createResourceGroup('index', localize('staged changes', "Staged Changes"));
		this._workingTreeGroup = this._sourceControl.createResourceGroup('workingTree', localize('changes', "Changes"));

		this.mergeGroup.hideWhenEmpty = true;
		this.indexGroup.hideWhenEmpty = true;

		this.disposables.push(this.mergeGroup);
		this.disposables.push(this.indexGroup);
		this.disposables.push(this.workingTreeGroup);

J
Joao Moreno 已提交
559
		this.disposables.push(new AutoFetcher(this, globalState));
J
Joao Moreno 已提交
560

561 562 563 564 565 566 567 568 569 570
		// https://github.com/Microsoft/vscode/issues/39039
		const onSuccessfulPush = filterEvent(this.onDidRunOperation, e => e.operation === Operation.Push && !e.error);
		onSuccessfulPush(() => {
			const gitConfig = workspace.getConfiguration('git');

			if (gitConfig.get<boolean>('showPushSuccessNotification')) {
				window.showInformationMessage(localize('push success', "Successfully pushed."));
			}
		}, null, this.disposables);

J
Joao Moreno 已提交
571 572 573 574 575
		const statusBar = new StatusBarCommands(this);
		this.disposables.push(statusBar);
		statusBar.onDidChange(() => this._sourceControl.statusBarCommands = statusBar.commands, null, this.disposables);
		this._sourceControl.statusBarCommands = statusBar.commands;

J
Joao Moreno 已提交
576 577 578
		const progressManager = new ProgressManager(this);
		this.disposables.push(progressManager);

J
Joao Moreno 已提交
579
		this.updateCommitTemplate();
J
Joao Moreno 已提交
580
		this.status();
J
Joao Moreno 已提交
581 582
	}

583
	validateInput(text: string, position: number): SourceControlInputBoxValidation | undefined {
584 585 586 587 588 589 590 591 592
		if (this.rebaseCommit) {
			if (this.rebaseCommit.message !== text) {
				return {
					message: localize('commit in rebase', "It's not possible to change the commit message in the middle of a rebase. Please complete the rebase operation and use interactive rebase instead."),
					type: SourceControlInputBoxValidationType.Warning
				};
			}
		}

593 594 595 596 597 598 599
		const config = workspace.getConfiguration('git');
		const setting = config.get<'always' | 'warn' | 'off'>('inputValidation');

		if (setting === 'off') {
			return;
		}

J
Joao Moreno 已提交
600
		if (/^\s+$/.test(text)) {
601
			return {
J
Joao Moreno 已提交
602
				message: localize('commitMessageWhitespacesOnlyWarning', "Current commit message only contains whitespace characters"),
603 604 605 606
				type: SourceControlInputBoxValidationType.Warning
			};
		}

607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635
		let start = 0, end;
		let match: RegExpExecArray | null;
		const regex = /\r?\n/g;

		while ((match = regex.exec(text)) && position > match.index) {
			start = match.index + match[0].length;
		}

		end = match ? match.index : text.length;

		const line = text.substring(start, end);

		if (line.length <= Repository.InputValidationLength) {
			if (setting !== 'always') {
				return;
			}

			return {
				message: localize('commitMessageCountdown', "{0} characters left in current line", Repository.InputValidationLength - line.length),
				type: SourceControlInputBoxValidationType.Information
			};
		} else {
			return {
				message: localize('commitMessageWarning', "{0} characters over {1} in current line", line.length - Repository.InputValidationLength, Repository.InputValidationLength),
				type: SourceControlInputBoxValidationType.Warning
			};
		}
	}

J
Joao Moreno 已提交
636 637 638 639 640
	provideOriginalResource(uri: Uri): Uri | undefined {
		if (uri.scheme !== 'file') {
			return;
		}

641
		return toGitUri(uri, '', { replaceFileExtension: true });
J
Joao Moreno 已提交
642 643 644 645 646 647 648 649 650 651
	}

	private async updateCommitTemplate(): Promise<void> {
		try {
			this._sourceControl.commitTemplate = await this.repository.getCommitTemplate();
		} catch (e) {
			// noop
		}
	}

J
Joao Moreno 已提交
652
	@throttle
J
Joao Moreno 已提交
653 654
	async status(): Promise<void> {
		await this.run(Operation.Status);
J
Joao Moreno 已提交
655
	}
J
Joao Moreno 已提交
656

657 658 659 660
	diff(path: string, options: DiffOptions = {}): Promise<string> {
		return this.run(Operation.Diff, () => this.repository.diff(path, options));
	}

J
Joao Moreno 已提交
661
	async add(resources: Uri[]): Promise<void> {
662
		await this.run(Operation.Add, () => this.repository.add(resources.map(r => r.fsPath)));
J
Joao Moreno 已提交
663 664
	}

J
Joao Moreno 已提交
665 666
	async stage(resource: Uri, contents: string): Promise<void> {
		const relativePath = path.relative(this.repository.root, resource.fsPath).replace(/\\/g, '/');
J
Joao Moreno 已提交
667
		await this.run(Operation.Stage, () => this.repository.stage(relativePath, contents));
J
Joao Moreno 已提交
668
		this._onDidChangeOriginalResource.fire(resource);
J
Joao Moreno 已提交
669 670
	}

J
Joao Moreno 已提交
671 672
	async revert(resources: Uri[]): Promise<void> {
		await this.run(Operation.RevertFiles, () => this.repository.revert('HEAD', resources.map(r => r.fsPath)));
J
Joao Moreno 已提交
673
	}
J
Joao Moreno 已提交
674

J
Joao Moreno 已提交
675
	async commit(message: string, opts: CommitOptions = Object.create(null)): Promise<void> {
676 677 678 679 680
		if (this.rebaseCommit) {
			await this.run(Operation.RebaseContinue, async () => {
				if (opts.all) {
					await this.repository.add([]);
				}
J
Joao Moreno 已提交
681

682 683 684 685 686 687 688 689 690 691 692
				await this.repository.rebaseContinue();
			});
		} else {
			await this.run(Operation.Commit, async () => {
				if (opts.all) {
					await this.repository.add([]);
				}

				await this.repository.commit(message, opts);
			});
		}
J
Joao Moreno 已提交
693
	}
J
Joao Moreno 已提交
694

J
Joao Moreno 已提交
695
	async clean(resources: Uri[]): Promise<void> {
696 697 698
		await this.run(Operation.Clean, async () => {
			const toClean: string[] = [];
			const toCheckout: string[] = [];
699
			const submodulesToUpdate: string[] = [];
700 701

			resources.forEach(r => {
702 703 704 705 706 707 708 709 710
				const fsPath = r.fsPath;

				for (const submodule of this.submodules) {
					if (path.join(this.root, submodule.path) === fsPath) {
						submodulesToUpdate.push(fsPath);
						return;
					}
				}

711
				const raw = r.toString();
J
Joao Moreno 已提交
712
				const scmResource = find(this.workingTreeGroup.resourceStates, sr => sr.resourceUri.toString() === raw);
713 714 715 716 717 718

				if (!scmResource) {
					return;
				}

				switch (scmResource.type) {
719 720
					case Status.UNTRACKED:
					case Status.IGNORED:
721
						toClean.push(fsPath);
722 723 724
						break;

					default:
725
						toCheckout.push(fsPath);
726 727 728
						break;
				}
			});
J
Joao Moreno 已提交
729

730
			const promises: Promise<void>[] = [];
J
Joao Moreno 已提交
731

732 733 734
			if (toClean.length > 0) {
				promises.push(this.repository.clean(toClean));
			}
J
Joao Moreno 已提交
735

736 737 738
			if (toCheckout.length > 0) {
				promises.push(this.repository.checkout('', toCheckout));
			}
J
Joao Moreno 已提交
739

740 741 742 743
			if (submodulesToUpdate.length > 0) {
				promises.push(this.repository.updateSubmodules(submodulesToUpdate));
			}

744 745
			await Promise.all(promises);
		});
J
Joao Moreno 已提交
746
	}
J
Joao Moreno 已提交
747

J
Joao Moreno 已提交
748
	async branch(name: string): Promise<void> {
J
Joao Moreno 已提交
749
		await this.run(Operation.Branch, () => this.repository.branch(name, true));
J
Joao Moreno 已提交
750 751
	}

752 753
	async deleteBranch(name: string, force?: boolean): Promise<void> {
		await this.run(Operation.DeleteBranch, () => this.repository.deleteBranch(name, force));
M
Maik Riechert 已提交
754 755
	}

756 757 758 759
	async renameBranch(name: string): Promise<void> {
		await this.run(Operation.RenameBranch, () => this.repository.renameBranch(name));
	}

J
Joao Moreno 已提交
760 761
	async merge(ref: string): Promise<void> {
		await this.run(Operation.Merge, () => this.repository.merge(ref));
762 763
	}

J
Joao Moreno 已提交
764 765
	async tag(name: string, message?: string): Promise<void> {
		await this.run(Operation.Tag, () => this.repository.tag(name, message));
766 767
	}

J
Joao Moreno 已提交
768
	async checkout(treeish: string): Promise<void> {
J
Joao Moreno 已提交
769
		await this.run(Operation.Checkout, () => this.repository.checkout(treeish, []));
J
Joao Moreno 已提交
770
	}
J
Joao Moreno 已提交
771

J
Joao Moreno 已提交
772 773 774 775 776 777 778 779
	async getCommit(ref: string): Promise<Commit> {
		return await this.repository.getCommit(ref);
	}

	async reset(treeish: string, hard?: boolean): Promise<void> {
		await this.run(Operation.Reset, () => this.repository.reset(treeish, hard));
	}

780 781 782 783
	async deleteRef(ref: string): Promise<void> {
		await this.run(Operation.DeleteRef, () => this.repository.deleteRef(ref));
	}

J
Joao Moreno 已提交
784
	@throttle
J
Joao Moreno 已提交
785
	async fetch(): Promise<void> {
K
Keegan Carruthers-Smith 已提交
786
		await this.run(Operation.Fetch, () => this.repository.fetch());
J
Joao Moreno 已提交
787 788
	}

J
Joao Moreno 已提交
789
	@throttle
J
Joao Moreno 已提交
790
	async pullWithRebase(head: Branch | undefined): Promise<void> {
J
Joao Moreno 已提交
791 792
		let remote: string | undefined;
		let branch: string | undefined;
J
Joao Moreno 已提交
793

J
Joao Moreno 已提交
794 795 796
		if (head && head.name && head.upstream) {
			remote = head.upstream.remote;
			branch = `${head.upstream.name}`;
J
Joao Moreno 已提交
797 798 799
		}

		await this.run(Operation.Pull, () => this.repository.pull(true, remote, branch));
J
Joao Moreno 已提交
800 801 802
	}

	@throttle
J
Joao Moreno 已提交
803
	async pull(head: Branch | undefined): Promise<void> {
J
Joao Moreno 已提交
804 805 806
		let remote: string | undefined;
		let branch: string | undefined;

J
Joao Moreno 已提交
807 808 809
		if (head && head.name && head.upstream) {
			remote = head.upstream.remote;
			branch = `${head.upstream.name}`;
J
Joao Moreno 已提交
810 811 812
		}

		await this.run(Operation.Pull, () => this.repository.pull(false, remote, branch));
J
Joao Moreno 已提交
813 814
	}

815
	async pullFrom(rebase?: boolean, remote?: string, branch?: string): Promise<void> {
M
Matt Shirley 已提交
816
		await this.run(Operation.Pull, () => this.repository.pull(rebase, remote, branch));
J
Joao Moreno 已提交
817 818
	}

J
Joao Moreno 已提交
819
	@throttle
J
Joao Moreno 已提交
820
	async push(head: Branch): Promise<void> {
J
Joao Moreno 已提交
821 822 823
		let remote: string | undefined;
		let branch: string | undefined;

J
Joao Moreno 已提交
824 825 826
		if (head && head.name && head.upstream) {
			remote = head.upstream.remote;
			branch = `${head.name}:${head.upstream.name}`;
J
Joao Moreno 已提交
827 828 829 830 831
		}

		await this.run(Operation.Push, () => this.repository.push(remote, branch));
	}

J
Joao Moreno 已提交
832 833
	async pushTo(remote?: string, name?: string, setUpstream: boolean = false): Promise<void> {
		await this.run(Operation.Push, () => this.repository.push(remote, name, setUpstream));
834 835
	}

836 837
	async pushTags(remote?: string): Promise<void> {
		await this.run(Operation.Push, () => this.repository.push(remote, undefined, false, true));
838 839
	}

J
Joao Moreno 已提交
840 841 842 843 844 845 846 847 848 849 850
	@throttle
	sync(head: Branch): Promise<void> {
		return this._sync(head, false);
	}

	@throttle
	async syncRebase(head: Branch): Promise<void> {
		return this._sync(head, true);
	}

	private async _sync(head: Branch, rebase: boolean): Promise<void> {
851
		let remoteName: string | undefined;
J
Joao Moreno 已提交
852 853 854 855
		let pullBranch: string | undefined;
		let pushBranch: string | undefined;

		if (head.name && head.upstream) {
856
			remoteName = head.upstream.remote;
J
Joao Moreno 已提交
857 858 859 860
			pullBranch = `${head.upstream.name}`;
			pushBranch = `${head.name}:${head.upstream.name}`;
		}

861
		await this.run(Operation.Sync, async () => {
862
			await this.repository.pull(rebase, remoteName, pullBranch);
863

864
			const remote = this.remotes.find(r => r.name === remoteName);
J
Joao Moreno 已提交
865 866 867 868 869 870

			if (remote && remote.isReadOnly) {
				return;
			}

			const shouldPush = this.HEAD && (typeof this.HEAD.ahead === 'number' ? this.HEAD.ahead > 0 : true);
871 872

			if (shouldPush) {
873
				await this.repository.push(remoteName, pushBranch);
874 875
			}
		});
J
Joao Moreno 已提交
876 877
	}

878
	async show(ref: string, filePath: string): Promise<string> {
879 880
		return await this.run(Operation.Show, async () => {
			const relativePath = path.relative(this.repository.root, filePath).replace(/\\/g, '/');
J
Joao Moreno 已提交
881
			const configFiles = workspace.getConfiguration('files', Uri.file(filePath));
J
Joao Moreno 已提交
882
			const defaultEncoding = configFiles.get<string>('encoding');
883 884
			const autoGuessEncoding = configFiles.get<boolean>('autoGuessEncoding');

885 886 887 888 889 890 891 892 893 894
			try {
				return await this.repository.bufferString(`${ref}:${relativePath}`, defaultEncoding, autoGuessEncoding);
			} catch (err) {
				if (err.gitErrorCode === GitErrorCodes.WrongCase) {
					const gitRelativePath = await this.repository.getGitRelativePath(ref, relativePath);
					return await this.repository.bufferString(`${ref}:${gitRelativePath}`, defaultEncoding, autoGuessEncoding);
				}

				throw err;
			}
J
Joao Moreno 已提交
895 896 897 898
		});
	}

	async buffer(ref: string, filePath: string): Promise<Buffer> {
J
Joao Moreno 已提交
899
		return this.run(Operation.Show, () => {
J
Joao Moreno 已提交
900
			const relativePath = path.relative(this.repository.root, filePath).replace(/\\/g, '/');
J
Joao Moreno 已提交
901
			return this.repository.buffer(`${ref}:${relativePath}`);
J
Joao Moreno 已提交
902 903 904
		});
	}

J
Joao Moreno 已提交
905 906
	getObjectDetails(ref: string, filePath: string): Promise<{ mode: string, object: string, size: number }> {
		return this.run(Operation.GetObjectDetails, () => this.repository.getObjectDetails(ref, filePath));
J
Joao Moreno 已提交
907 908 909 910 911 912
	}

	detectObjectType(object: string): Promise<{ mimetype: string, encoding?: string }> {
		return this.run(Operation.Show, () => this.repository.detectObjectType(object));
	}

J
Joao Moreno 已提交
913 914 915 916
	async getStashes(): Promise<Stash[]> {
		return await this.repository.getStashes();
	}

917 918
	async createStash(message?: string, includeUntracked?: boolean): Promise<void> {
		return await this.run(Operation.Stash, () => this.repository.createStash(message, includeUntracked));
J
Joao Moreno 已提交
919 920 921 922
	}

	async popStash(index?: number): Promise<void> {
		return await this.run(Operation.Stash, () => this.repository.popStash(index));
923 924
	}

J
Joao Moreno 已提交
925 926 927 928
	async getCommitTemplate(): Promise<string> {
		return await this.run(Operation.GetCommitTemplate, async () => this.repository.getCommitTemplate());
	}

J
Joao Moreno 已提交
929
	async ignore(files: Uri[]): Promise<void> {
N
NKumar2 已提交
930
		return await this.run(Operation.Ignore, async () => {
J
Joao Moreno 已提交
931 932
			const ignoreFile = `${this.repository.root}${path.sep}.gitignore`;
			const textToAppend = files
J
Joao Moreno 已提交
933
				.map(uri => path.relative(this.repository.root, uri.fsPath).replace(/\\/g, '/'))
J
Joao Moreno 已提交
934
				.join('\n');
N
NKumar2 已提交
935

J
Joao Moreno 已提交
936 937 938
			const document = await new Promise(c => fs.exists(ignoreFile, c))
				? await workspace.openTextDocument(ignoreFile)
				: await workspace.openTextDocument(Uri.file(ignoreFile).with({ scheme: 'untitled' }));
939

J
Joao Moreno 已提交
940
			await window.showTextDocument(document);
J
Joao Moreno 已提交
941

J
Joao Moreno 已提交
942
			const edit = new WorkspaceEdit();
J
Joao Moreno 已提交
943 944
			const lastLine = document.lineAt(document.lineCount - 1);
			const text = lastLine.isEmptyOrWhitespace ? `${textToAppend}\n` : `\n${textToAppend}\n`;
945

J
Joao Moreno 已提交
946
			edit.insert(document.uri, lastLine.range.end, text);
J
Joao Moreno 已提交
947
			workspace.applyEdit(edit);
N
NKumar2 已提交
948 949 950
		});
	}

J
Johannes Rieken 已提交
951
	checkIgnore(filePaths: string[]): Promise<Set<string>> {
952
		return this.run(Operation.CheckIgnore, () => {
J
Johannes Rieken 已提交
953 954
			return new Promise<Set<string>>((resolve, reject) => {

J
Joao Moreno 已提交
955 956
				filePaths = filePaths
					.filter(filePath => isDescendant(this.root, filePath));
957

958 959
				if (filePaths.length === 0) {
					// nothing left
C
cleidigh 已提交
960
					return resolve(new Set<string>());
961 962
				}

963 964 965
				// https://git-scm.com/docs/git-check-ignore#git-check-ignore--z
				const child = this.repository.stream(['check-ignore', '-z', '--stdin'], { stdio: [null, null, null] });
				child.stdin.end(filePaths.join('\0'), 'utf8');
J
Johannes Rieken 已提交
966

967
				const onExit = (exitCode: number) => {
J
Johannes Rieken 已提交
968 969 970 971
					if (exitCode === 1) {
						// nothing ignored
						resolve(new Set<string>());
					} else if (exitCode === 0) {
972 973
						// paths are separated by the null-character
						resolve(new Set<string>(data.split('\0')));
J
Johannes Rieken 已提交
974
					} else {
J
Joao Moreno 已提交
975 976 977 978 979
						if (/ is in submodule /.test(stderr)) {
							reject(new GitError({ stdout: data, stderr, exitCode, gitErrorCode: GitErrorCodes.IsInSubmodule }));
						} else {
							reject(new GitError({ stdout: data, stderr, exitCode }));
						}
J
Johannes Rieken 已提交
980 981 982 983 984 985 986 987 988 989 990
					}
				};

				let data = '';
				const onStdoutData = (raw: string) => {
					data += raw;
				};

				child.stdout.setEncoding('utf8');
				child.stdout.on('data', onStdoutData);

991 992 993
				let stderr: string = '';
				child.stderr.setEncoding('utf8');
				child.stderr.on('data', raw => stderr += raw);
J
Johannes Rieken 已提交
994 995 996 997 998 999 1000

				child.on('error', reject);
				child.on('exit', onExit);
			});
		});
	}

J
Joao Moreno 已提交
1001
	private async run<T>(operation: Operation, runOperation: () => Promise<T> = () => Promise.resolve<any>(null)): Promise<T> {
J
Joao 已提交
1002
		if (this.state !== RepositoryState.Idle) {
1003 1004 1005
			throw new Error('Repository not initialized');
		}

J
Joao Moreno 已提交
1006
		let error: any = null;
J
Joao Moreno 已提交
1007

J
Joao Moreno 已提交
1008 1009
		this._operations.start(operation);
		this._onRunOperation.fire(operation);
J
Joao Moreno 已提交
1010

J
Joao Moreno 已提交
1011 1012
		try {
			const result = await this.retryRun(runOperation);
J
Joao Moreno 已提交
1013

J
Joao Moreno 已提交
1014 1015 1016
			if (!isReadOnly(operation)) {
				await this.updateModelState();
			}
J
Joao Moreno 已提交
1017

J
Joao Moreno 已提交
1018 1019 1020
			return result;
		} catch (err) {
			error = err;
J
Joao Moreno 已提交
1021

J
Joao Moreno 已提交
1022 1023
			if (err.gitErrorCode === GitErrorCodes.NotAGitRepository) {
				this.state = RepositoryState.Disposed;
J
Joao Moreno 已提交
1024
			}
1025

J
Joao Moreno 已提交
1026 1027 1028 1029 1030
			throw err;
		} finally {
			this._operations.end(operation);
			this._onDidRunOperation.fire({ operation, error });
		}
J
Joao Moreno 已提交
1031
	}
1032

1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050
	private async retryRun<T>(runOperation: () => Promise<T> = () => Promise.resolve<any>(null)): Promise<T> {
		let attempt = 0;

		while (true) {
			try {
				attempt++;
				return await runOperation();
			} catch (err) {
				if (err.gitErrorCode === GitErrorCodes.RepositoryIsLocked && attempt <= 10) {
					// quatratic backoff
					await timeout(Math.pow(attempt, 2) * 50);
				} else {
					throw err;
				}
			}
		}
	}

J
Joao Moreno 已提交
1051
	@throttle
1052
	private async updateModelState(): Promise<void> {
1053 1054 1055
		const { status, didHitLimit } = await this.repository.getStatus();
		const config = workspace.getConfiguration('git');
		const shouldIgnore = config.get<boolean>('ignoreLimitWarning') === true;
J
Johannes Rieken 已提交
1056
		const useIcons = !config.get<boolean>('decorations.enabled', true);
1057 1058 1059 1060

		this.isRepositoryHuge = didHitLimit;

		if (didHitLimit && !shouldIgnore && !this.didWarnAboutLimit) {
B
Benjamin Pasero 已提交
1061
			const neverAgain = { title: localize('neveragain', "Don't Show Again") };
1062

1063
			window.showWarningMessage(localize('huge', "The git repository at '{0}' has too many active changes, only a subset of Git features will be enabled.", this.repository.root), neverAgain).then(result => {
1064 1065 1066 1067 1068 1069 1070 1071
				if (result === neverAgain) {
					config.update('ignoreLimitWarning', true, false);
				}
			});

			this.didWarnAboutLimit = true;
		}

J
Joao Moreno 已提交
1072
		let HEAD: Branch | undefined;
J
Joao Moreno 已提交
1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087

		try {
			HEAD = await this.repository.getHEAD();

			if (HEAD.name) {
				try {
					HEAD = await this.repository.getBranch(HEAD.name);
				} catch (err) {
					// noop
				}
			}
		} catch (err) {
			// noop
		}

1088
		const [refs, remotes, submodules, rebaseCommit] = await Promise.all([this.repository.getRefs(), this.repository.getRemotes(), this.repository.getSubmodules(), this.getRebaseCommit()]);
J
Joao Moreno 已提交
1089 1090 1091 1092

		this._HEAD = HEAD;
		this._refs = refs;
		this._remotes = remotes;
1093
		this._submodules = submodules;
1094
		this.rebaseCommit = rebaseCommit;
J
Joao Moreno 已提交
1095 1096 1097 1098 1099 1100

		const index: Resource[] = [];
		const workingTree: Resource[] = [];
		const merge: Resource[] = [];

		status.forEach(raw => {
J
Joao Moreno 已提交
1101 1102
			const uri = Uri.file(path.join(this.repository.root, raw.path));
			const renameUri = raw.rename ? Uri.file(path.join(this.repository.root, raw.rename)) : undefined;
J
Joao Moreno 已提交
1103 1104

			switch (raw.x + raw.y) {
1105 1106 1107 1108 1109 1110 1111 1112 1113
				case '??': return workingTree.push(new Resource(ResourceGroupType.WorkingTree, uri, Status.UNTRACKED, useIcons));
				case '!!': return workingTree.push(new Resource(ResourceGroupType.WorkingTree, uri, Status.IGNORED, useIcons));
				case 'DD': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.BOTH_DELETED, useIcons));
				case 'AU': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.ADDED_BY_US, useIcons));
				case 'UD': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.DELETED_BY_THEM, useIcons));
				case 'UA': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.ADDED_BY_THEM, useIcons));
				case 'DU': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.DELETED_BY_US, useIcons));
				case 'AA': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.BOTH_ADDED, useIcons));
				case 'UU': return merge.push(new Resource(ResourceGroupType.Merge, uri, Status.BOTH_MODIFIED, useIcons));
J
Joao Moreno 已提交
1114 1115 1116
			}

			switch (raw.x) {
J
Joao Moreno 已提交
1117
				case 'M': index.push(new Resource(ResourceGroupType.Index, uri, Status.INDEX_MODIFIED, useIcons)); break;
1118 1119 1120 1121
				case 'A': index.push(new Resource(ResourceGroupType.Index, uri, Status.INDEX_ADDED, useIcons)); break;
				case 'D': index.push(new Resource(ResourceGroupType.Index, uri, Status.INDEX_DELETED, useIcons)); break;
				case 'R': index.push(new Resource(ResourceGroupType.Index, uri, Status.INDEX_RENAMED, useIcons, renameUri)); break;
				case 'C': index.push(new Resource(ResourceGroupType.Index, uri, Status.INDEX_COPIED, useIcons, renameUri)); break;
J
Joao Moreno 已提交
1122 1123 1124
			}

			switch (raw.y) {
1125 1126
				case 'M': workingTree.push(new Resource(ResourceGroupType.WorkingTree, uri, Status.MODIFIED, useIcons, renameUri)); break;
				case 'D': workingTree.push(new Resource(ResourceGroupType.WorkingTree, uri, Status.DELETED, useIcons, renameUri)); break;
J
Joao Moreno 已提交
1127 1128 1129
			}
		});

J
Joao Moreno 已提交
1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145
		// set resource groups
		this.mergeGroup.resourceStates = merge;
		this.indexGroup.resourceStates = index;
		this.workingTreeGroup.resourceStates = workingTree;

		// set count badge
		const countBadge = workspace.getConfiguration('git').get<string>('countBadge');
		let count = merge.length + index.length + workingTree.length;

		switch (countBadge) {
			case 'off': count = 0; break;
			case 'tracked': count = count - workingTree.filter(r => r.type === Status.UNTRACKED || r.type === Status.IGNORED).length; break;
		}

		this._sourceControl.count = count;

J
Joao Moreno 已提交
1146 1147 1148
		// Disable `Discard All Changes` for "fresh" repositories
		// https://github.com/Microsoft/vscode/issues/43066
		commands.executeCommand('setContext', 'gitFreshRepository', !this._HEAD || !this._HEAD.commit);
J
Joao Moreno 已提交
1149

J
Joao Moreno 已提交
1150
		this._onDidChangeStatus.fire();
J
Joao Moreno 已提交
1151 1152
	}

1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163
	private async getRebaseCommit(): Promise<Commit | undefined> {
		const rebaseHeadPath = path.join(this.repository.root, '.git', 'REBASE_HEAD');

		try {
			const rebaseHead = await new Promise<string>((c, e) => fs.readFile(rebaseHeadPath, 'utf8', (err, result) => err ? e(err) : c(result)));
			return await this.getCommit(rebaseHead.trim());
		} catch (err) {
			return undefined;
		}
	}

J
Joao Moreno 已提交
1164
	private onFSChange(uri: Uri): void {
J
Joao Moreno 已提交
1165 1166 1167 1168 1169 1170 1171
		const config = workspace.getConfiguration('git');
		const autorefresh = config.get<boolean>('autorefresh');

		if (!autorefresh) {
			return;
		}

1172 1173 1174 1175
		if (this.isRepositoryHuge) {
			return;
		}

J
Joao Moreno 已提交
1176 1177 1178 1179 1180 1181 1182
		if (!this.operations.isIdle()) {
			return;
		}

		this.eventuallyUpdateWhenIdleAndWait();
	}

1183
	@debounce(1000)
J
Joao Moreno 已提交
1184
	private eventuallyUpdateWhenIdleAndWait(): void {
1185 1186 1187
		this.updateWhenIdleAndWait();
	}

J
Joao Moreno 已提交
1188
	@throttle
1189
	private async updateWhenIdleAndWait(): Promise<void> {
J
Joao 已提交
1190
		await this.whenIdleAndFocused();
J
Joao Moreno 已提交
1191
		await this.status();
J
Joao Moreno 已提交
1192
		await timeout(5000);
1193 1194
	}

J
Joao Moreno 已提交
1195
	async whenIdleAndFocused(): Promise<void> {
J
Joao 已提交
1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208
		while (true) {
			if (!this.operations.isIdle()) {
				await eventToPromise(this.onDidRunOperation);
				continue;
			}

			if (!window.state.focused) {
				const onDidFocusWindow = filterEvent(window.onDidChangeWindowState, e => e.focused);
				await eventToPromise(onDidFocusWindow);
				continue;
			}

			return;
1209 1210 1211
		}
	}

J
Joao 已提交
1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238
	get headLabel(): string {
		const HEAD = this.HEAD;

		if (!HEAD) {
			return '';
		}

		const tag = this.refs.filter(iref => iref.type === RefType.Tag && iref.commit === HEAD.commit)[0];
		const tagName = tag && tag.name;
		const head = HEAD.name || tagName || (HEAD.commit || '').substr(0, 8);

		return head
			+ (this.workingTreeGroup.resourceStates.length > 0 ? '*' : '')
			+ (this.indexGroup.resourceStates.length > 0 ? '+' : '')
			+ (this.mergeGroup.resourceStates.length > 0 ? '!' : '');
	}

	get syncLabel(): string {
		if (!this.HEAD
			|| !this.HEAD.name
			|| !this.HEAD.commit
			|| !this.HEAD.upstream
			|| !(this.HEAD.ahead || this.HEAD.behind)
		) {
			return '';
		}

1239 1240 1241
		const remoteName = this.HEAD && this.HEAD.remote || this.HEAD.upstream.remote;
		const remote = this.remotes.find(r => r.name === remoteName);

J
Joao Moreno 已提交
1242
		if (remote && remote.isReadOnly) {
1243 1244 1245
			return `${this.HEAD.behind}↓`;
		}

J
Joao 已提交
1246 1247 1248
		return `${this.HEAD.behind}${this.HEAD.ahead}↑`;
	}

J
Joao Moreno 已提交
1249
	dispose(): void {
J
Joao Moreno 已提交
1250
		this.disposables = dispose(this.disposables);
J
Joao Moreno 已提交
1251
	}
1252
}