Adds tracking of active repo

Author: d13

src/webviews/apps/plus/home/components/active-work.ts

@@ -37,9 +37,7 @@ export class GlActiveWork extends SignalWatcher(LitElement) {
 	override connectedCallback() {
 		super.connectedCallback();
 
-		if (this._overviewState.state.value == null) {
-			this._overviewState.run();
-		}
+		this._overviewState.run();
 	}
 
 	override render() {

src/webviews/apps/plus/home/components/overview.ts

@@ -33,9 +33,7 @@ export class GlOverview extends SignalWatcher(LitElement) {
 	override connectedCallback() {
 		super.connectedCallback();
 
-		if (this._overviewState.state.value == null) {
-			this._overviewState.run();
-		}
+		this._overviewState.run();
 	}
 
 	override render() {

src/webviews/apps/plus/home/components/overviewState.ts

@@ -1,12 +1,15 @@
 import { createContext } from '@lit/context';
 import type { GetOverviewResponse } from '../../../../home/protocol';
-import { GetOverview } from '../../../../home/protocol';
+import { DidChangeRepositoryWip, GetOverview } from '../../../../home/protocol';
 import { AsyncComputedState } from '../../../shared/components/signal-utils';
+import type { Disposable } from '../../../shared/events';
 import type { HostIpc } from '../../../shared/ipc';
 
 export type Overview = GetOverviewResponse;
 
 export class OverviewState extends AsyncComputedState<Overview> {
+	private _disposable: Disposable | undefined;
+
 	constructor(
 		private _ipc: HostIpc,
 		options?: {
@@ -19,6 +22,18 @@ export class OverviewState extends AsyncComputedState<Overview> {
 
 			return rsp;
 		}, options);
+
+		this._disposable = this._ipc.onReceiveMessage(msg => {
+			switch (true) {
+				case DidChangeRepositoryWip.is(msg):
+					this.run(true);
+					break;
+			}
+		});
+	}
+
+	dispose() {
+		this._disposable?.dispose();
 	}
 }
 

src/webviews/apps/shared/components/signal-utils.ts

@@ -1,3 +1,4 @@
+import { signal } from '@lit-labs/signals';
 import { AsyncComputed } from 'signal-utils/async-computed';
 
 export const renderAsyncComputed = <T, R = unknown>(
@@ -27,10 +28,17 @@ export const renderAsyncComputed = <T, R = unknown>(
 };
 
 export class AsyncComputedState<T, R = unknown> {
+	private _invalidate = signal(0);
 	private _initial?: T;
 	private _state?: AsyncComputed<T>;
 	get state() {
-		this._state ??= new AsyncComputed(this._fetch, this._initial ? { initialValue: this._initial } : undefined);
+		this._state ??= new AsyncComputed(
+			(abortSignal: AbortSignal) => {
+				this._invalidate.get();
+				return this._fetch(abortSignal);
+			},
+			this._initial ? { initialValue: this._initial } : undefined,
+		);
 		return this._state;
 	}
 
@@ -48,10 +56,17 @@ export class AsyncComputedState<T, R = unknown> {
 			}
 		}
 	}
-	run() {
+	run(force = false) {
+		if (force) {
+			this.invalidate();
+		}
 		this.state.run();
 	}
 
+	invalidate() {
+		this._invalidate.set(Date.now());
+	}
+
 	render(config: {
 		initial?: () => R;
 		pending?: () => R;

src/webviews/home/homeWebview.ts

@@ -7,6 +7,8 @@ import type { Container } from '../../container';
 import type { BranchContributorOverview } from '../../git/gitProvider';
 import { sortBranches } from '../../git/models/branch';
 import type { PullRequest } from '../../git/models/pullRequest';
+import type { Repository } from '../../git/models/repository';
+import { RepositoryChange, RepositoryChangeComparisonMode } from '../../git/models/repository';
 import type { GitStatus } from '../../git/models/status';
 import { getOpenedWorktreesByBranch, groupWorktreesByBranch } from '../../git/models/worktree';
 import type { Subscription } from '../../plus/gk/account/subscription';
@@ -34,6 +36,7 @@ import {
 	DidChangeOrgSettings,
 	DidChangePreviewEnabled,
 	DidChangeRepositories,
+	DidChangeRepositoryWip,
 	DidChangeSubscription,
 	DidChangeWalkthroughProgress,
 	DidFocusAccount,
@@ -49,6 +52,8 @@ const emptyDisposable = Object.freeze({
 	},
 });
 
+type RepositorySubscription = { repo: Repository; subscription: Disposable };
+
 export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWebviewShowingArgs> {
 	private readonly _disposable: Disposable;
 	private _pendingFocusAccount = false;
@@ -138,7 +143,7 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 				void this.host.respond(GetLaunchpadSummary, e, await getLaunchpadSummary(this.container));
 				break;
 			case GetOverview.is(e):
-				void this.host.respond(GetOverview, e, await getBranchOverview(this.container));
+				void this.host.respond(GetOverview, e, await this.getBranchOverview());
 				break;
 		}
 	}
@@ -260,6 +265,63 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 		};
 	}
 
+	private async getBranchOverview(): Promise<GetOverviewResponse | undefined> {
+		const repo = this.getSelectedRepository();
+		if (repo == null) return undefined;
+
+		return getBranchOverview(repo, this.container);
+	}
+
+	private _repositorySubscription: RepositorySubscription | undefined;
+	private selectRepository(repoPath?: string) {
+		let repo: Repository | undefined;
+		if (repoPath != null) {
+			repo = this.container.git.getRepository(repoPath)!;
+		} else {
+			repo = this.container.git.highlander;
+			if (repo == null) {
+				repo = this.container.git.getBestRepositoryOrFirst();
+			}
+		}
+
+		if (this._repositorySubscription != null) {
+			this._repositorySubscription.subscription.dispose();
+			this._repositorySubscription = undefined;
+		}
+		if (repo != null) {
+			this._repositorySubscription = {
+				repo: repo,
+				subscription: this.subscribeToRepository(repo),
+			};
+		}
+
+		return repo;
+	}
+
+	private subscribeToRepository(repo: Repository): Disposable {
+		return Disposable.from(
+			repo.watchFileSystem(1000),
+			repo.onDidChangeFileSystem(() => this.onWipChanged(repo)),
+			repo.onDidChange(e => {
+				if (e.changed(RepositoryChange.Index, RepositoryChangeComparisonMode.Any)) {
+					this.onWipChanged(repo);
+				}
+			}),
+		);
+	}
+
+	private onWipChanged(_repo: Repository) {
+		void this.host.notify(DidChangeRepositoryWip, undefined);
+	}
+
+	private getSelectedRepository() {
+		if (this._repositorySubscription == null) {
+			this.selectRepository();
+		}
+
+		return this._repositorySubscription?.repo;
+	}
+
 	private _hostedIntegrationConnected: boolean | undefined;
 	private isAnyIntegrationConnected(force = false) {
 		if (this._hostedIntegrationConnected == null || force === true) {
@@ -346,14 +408,10 @@ interface BranchOverviewOptions {
 }
 
 async function getBranchOverview(
+	repo: Repository,
 	container: Container,
 	options?: BranchOverviewOptions,
 ): Promise<GetOverviewResponse | undefined> {
-	const repo = container.git.highlander;
-	if (repo == null) {
-		return undefined;
-	}
-
 	const [branchesResult, worktreesResult] = await Promise.allSettled([
 		repo.git.getBranches({ filter: b => !b.remote }),
 		repo.git.getWorktrees(),

src/webviews/home/protocol.ts

@@ -123,6 +123,8 @@ export const DismissWalkthroughSection = new IpcCommand<void>(scope, 'walkthroug
 
 export const DidChangePreviewEnabled = new IpcNotification<boolean>(scope, 'previewEnabled/didChange');
 
+export const DidChangeRepositoryWip = new IpcNotification<undefined>(scope, 'repository/wip/didChange');
+
 export interface DidChangeRepositoriesParams {
 	count: number;
 	openCount: number;