Adds "limit" to load all for branches/tags

src/git/git.ts

@@ -1020,7 +1020,22 @@ export namespace Git {
 		return git<string>({ cwd: repoPath }, 'reset', '-q', '--', fileName);
 	}
 
-	export async function rev_list(
+	export async function rev_list__count(repoPath: string, ref: string): Promise<number | undefined> {
+		let data = await git<string>(
+			{ cwd: repoPath, errors: GitErrorHandling.Ignore },
+			'rev-list',
+			'--count',
+			ref,
+			'--',
+		);
+		data = data.trim();
+		if (data.length === 0) return undefined;
+
+		const result = parseInt(data, 10);
+		return isNaN(result) ? undefined : result;
+	}
+
+	export async function rev_list__left_right(
 		repoPath: string,
 		refs: string[],
 	): Promise<{ ahead: number; behind: number } | undefined> {

src/git/gitService.ts

@@ -1318,7 +1318,12 @@ export class GitService implements Disposable {
 		repoPath: string,
 		refs: string[],
 	): Promise<{ ahead: number; behind: number } | undefined> {
-		return Git.rev_list(repoPath, refs);
+		return Git.rev_list__left_right(repoPath, refs);
+	}
+
+	@log()
+	getCommitCount(repoPath: string, ref: string): Promise<number | undefined> {
+		return Git.rev_list__count(repoPath, ref);
 	}
 
 	@log()

src/views/nodes/branchNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { BranchesView } from '../branchesView';
 import { BranchTrackingStatusNode } from './branchTrackingStatusNode';
 import { CommitNode } from './commitNode';
@@ -198,7 +198,11 @@ export class BranchNode
 			);
 
 			if (log.hasMore) {
-				children.push(new LoadMoreNode(this.view, this, children[children.length - 1]));
+				children.push(
+					new LoadMoreNode(this.view, this, children[children.length - 1], undefined, () =>
+						Container.git.getCommitCount(this.branch.repoPath, this.branch.name),
+					),
+				);
 			}
 
 			this._children = children;
@@ -350,8 +354,14 @@ export class BranchNode
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log == null || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);

src/views/nodes/branchTrackingStatusNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { ThemeIcon, TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { ThemeIcon, TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { BranchNode } from './branchNode';
 import { BranchTrackingStatusFilesNode } from './branchTrackingStatusFilesNode';
 import { CommitNode } from './commitNode';
@@ -232,8 +232,14 @@ export class BranchTrackingStatusNode extends ViewNode implement
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log == null || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);

src/views/nodes/common.ts

@@ -143,12 +143,18 @@ export abstract class PagerNode extends ViewNode {
 		protected readonly message: string,
 		protected readonly previousNode?: ViewNode,
 		protected readonly pageSize: number = Container.config.views.pageItemLimit,
+		protected readonly countFn?: () => Promise<number | undefined>,
 	) {
 		super(unknownGitUri, view, parent);
 	}
 
-	loadAll() {
-		return this.view.loadMoreNodeChildren(this.parent! as ViewNode & PageableViewNode, 0, this.previousNode);
+	async loadAll() {
+		const count = (await this.countFn?.()) ?? 0;
+		return this.view.loadMoreNodeChildren(
+			this.parent! as ViewNode & PageableViewNode,
+			count > 5000 ? 5000 : 0,
+			this.previousNode,
+		);
 	}
 
 	loadMore() {
@@ -180,7 +186,13 @@ export abstract class PagerNode extends ViewNode {
 }
 
 export class LoadMoreNode extends PagerNode {
-	constructor(view: View, parent: ViewNode & PageableViewNode, previousNode: ViewNode, pageSize?: number) {
+	constructor(
+		view: View,
+		parent: ViewNode & PageableViewNode,
+		previousNode: ViewNode,
+		pageSize?: number,
+		countFn?: () => Promise<number | undefined>,
+	) {
 		super(
 			view,
 			parent,
@@ -189,6 +201,7 @@ export class LoadMoreNode extends PagerNode {
 				: 'Load more',
 			previousNode,
 			pageSize,
+			countFn,
 		);
 	}
 }

src/views/nodes/contributorNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { CommitNode } from './commitNode';
 import { LoadMoreNode, MessageNode } from './common';
 import { GlyphChars } from '../../constants';
@@ -113,8 +113,14 @@ export class ContributorNode extends ViewNode
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log == null || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);

src/views/nodes/fileHistoryNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { Disposable, TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { Disposable, TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { CommitFileNode } from './commitFileNode';
 import { LoadMoreNode, MessageNode } from './common';
 import { Container } from '../../container';
@@ -185,8 +185,14 @@ export class FileHistoryNode extends SubscribeableViewNode implements PageableVi
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log == null || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);

src/views/nodes/lineHistoryNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { Disposable, Selection, TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { Disposable, Selection, TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { CommitFileNode } from './commitFileNode';
 import { LoadMoreNode, MessageNode } from './common';
 import { Container } from '../../container';
@@ -317,8 +317,14 @@ export class LineHistoryNode extends SubscribeableViewNode implements PageableVi
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log == null || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);

src/views/nodes/reflogRecordNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { CommitNode } from './commitNode';
 import { LoadMoreNode, MessageNode } from './common';
 import { GlyphChars } from '../../constants';
@@ -104,8 +104,14 @@ export class ReflogRecordNode extends ViewNode implements Pageab
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log === undefined || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);

src/views/nodes/tagNode.ts

@@ -1,5 +1,5 @@
 'use strict';
-import { TreeItem, TreeItemCollapsibleState } from 'vscode';
+import { TreeItem, TreeItemCollapsibleState, window } from 'vscode';
 import { CommitNode } from './commitNode';
 import { LoadMoreNode, MessageNode } from './common';
 import { ViewBranchesLayout } from '../../configuration';
@@ -57,7 +57,11 @@ export class TagNode extends ViewRefNode
 		];
 
 		if (log.hasMore) {
-			children.push(new LoadMoreNode(this.view, this, children[children.length - 1]));
+			children.push(
+				new LoadMoreNode(this.view, this, children[children.length - 1], undefined, () =>
+					Container.git.getCommitCount(this.tag.repoPath, this.tag.name),
+				),
+			);
 		}
 		return children;
 	}
@@ -105,8 +109,14 @@ export class TagNode extends ViewRefNode
 	}
 
 	limit: number | undefined = this.view.getNodeLastKnownLimit(this);
+	@gate()
 	async loadMore(limit?: number | { until?: any }) {
-		let log = await this.getLog();
+		let log = await window.withProgress(
+			{
+				location: { viewId: this.view.id },
+			},
+			() => this.getLog(),
+		);
 		if (log == null || !log.hasMore) return;
 
 		log = await log.more?.(limit ?? this.view.config.pageItemLimit);