Shows rich provider info for pro accounts in home

Author: d13

src/webviews/home/homeWebview.ts

@@ -24,6 +24,7 @@ import type { GitStatus } from '../../git/models/status';
 import type { GitWorktree } from '../../git/models/worktree';
 import { getOpenedWorktreesByBranch, groupWorktreesByBranch } from '../../git/models/worktree';
 import type { Subscription } from '../../plus/gk/account/subscription';
+import { isSubscriptionStatePaidOrTrial } from '../../plus/gk/account/subscription';
 import type { SubscriptionChangeEvent } from '../../plus/gk/account/subscriptionService';
 import { getLaunchpadSummary } from '../../plus/launchpad/utils';
 import type { ShowInCommitGraphCommandArgs } from '../../plus/webviews/graph/protocol';
@@ -446,12 +447,16 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 		}
 	}
 
-	private onSubscriptionChanged(e: SubscriptionChangeEvent) {
-		void this.notifyDidChangeSubscription(e.current);
+	private async onSubscriptionChanged(e: SubscriptionChangeEvent) {
+		await this.notifyDidChangeSubscription(e.current);
+
+		if (isSubscriptionStatePaidOrTrial(e.current.state) !== isSubscriptionStatePaidOrTrial(e.previous.state)) {
+			this.onOverviewRepoChanged('repo');
+		}
 	}
 
 	private async getState(subscription?: Subscription): Promise<State> {
-		const subResult = await this.getSubscription(subscription);
+		const subResult = await this.getSubscriptionState(subscription);
 
 		return {
 			...this.host.baseWebviewState,
@@ -504,7 +509,10 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 			branchesAndWorktrees,
 			this.container,
 			this._overviewBranchFilter,
-			forceWip ? { forceActive: true } : undefined,
+			{
+				forceActive: forceWip ? true : undefined,
+				isPro: await this.isSubscriptionPro(),
+			},
 		);
 		this._invalidateOverview = undefined;
 		if (overviewBranches == null) return undefined;
@@ -683,8 +691,28 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 		return this._hostedIntegrationConnected;
 	}
 
+	private _subscription: Subscription | undefined;
 	private async getSubscription(subscription?: Subscription) {
-		subscription ??= await this.container.subscription.getSubscription(true);
+		if (subscription != null) {
+			this._subscription = subscription;
+		} else if (this._subscription != null) {
+			subscription = this._subscription;
+		} else {
+			this._subscription = subscription = await this.container.subscription.getSubscription(true);
+		}
+
+		return this._subscription;
+	}
+
+	private async isSubscriptionPro() {
+		const subscription = await this.getSubscription();
+		if (subscription == null) return false;
+
+		return isSubscriptionStatePaidOrTrial(subscription.state);
+	}
+
+	private async getSubscriptionState(subscription?: Subscription) {
+		subscription = await this.getSubscription(subscription);
 
 		let avatar;
 		if (subscription.account?.email) {
@@ -755,7 +783,7 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 	}
 
 	private async notifyDidChangeSubscription(subscription?: Subscription) {
-		const subResult = await this.getSubscription(subscription);
+		const subResult = await this.getSubscriptionState(subscription);
 
 		void this.host.notify(DidChangeSubscription, {
 			subscription: subResult.subscription,
@@ -902,7 +930,7 @@ async function getOverviewBranches(
 	branchesData: RepositoryBranchData,
 	container: Container,
 	filters: OverviewFilters,
-	options?: { forceActive?: boolean },
+	options?: { forceActive?: boolean; isPro?: boolean },
 ): Promise<GetOverviewBranches | undefined> {
 	const { branches, worktreesByBranch } = branchesData;
 	if (branches.length === 0) return undefined;
@@ -928,7 +956,14 @@ async function getOverviewBranches(
 		const timestamp = branch.date?.getTime();
 		if (branch.current || wt?.opened) {
 			const forceOptions = options?.forceActive ? { force: true } : undefined;
-			prPromises.set(branch.id, branch.getAssociatedPullRequest({ avatarSize: 16 }));
+			if (options?.isPro !== false) {
+				prPromises.set(branch.id, branch.getAssociatedPullRequest({ avatarSize: 16 }));
+				contributorPromises.set(
+					branch.id,
+					container.git.getBranchContributorOverview(branch.repoPath, branch.ref),
+				);
+			}
+
 			if (wt != null) {
 				statusPromises.set(branch.id, wt.getStatus(forceOptions));
 			} else {
@@ -937,7 +972,6 @@ async function getOverviewBranches(
 				}
 				statusPromises.set(branch.id, repoStatusPromise);
 			}
-			contributorPromises.set(branch.id, container.git.getBranchContributorOverview(branch.repoPath, branch.ref));
 
 			overviewBranches.active.push({
 				id: branch.id,
@@ -954,11 +988,17 @@ async function getOverviewBranches(
 		}
 
 		if (timestamp != null && timestamp > recentThreshold) {
-			prPromises.set(branch.id, branch.getAssociatedPullRequest());
+			if (options?.isPro !== false) {
+				prPromises.set(branch.id, branch.getAssociatedPullRequest());
+				contributorPromises.set(
+					branch.id,
+					container.git.getBranchContributorOverview(branch.repoPath, branch.ref),
+				);
+			}
+
 			if (wt != null) {
 				statusPromises.set(branch.id, wt.getStatus());
 			}
-			contributorPromises.set(branch.id, container.git.getBranchContributorOverview(branch.repoPath, branch.ref));
 
 			overviewBranches.recent.push({
 				id: branch.id,
@@ -998,16 +1038,20 @@ async function getOverviewBranches(
 					? { name: wt.name, uri: wt.uri.toString() }
 					: undefined;
 
-				if (!branch.upstream?.missing) {
-					prPromises.set(branch.id, branch.getAssociatedPullRequest());
+				if (options?.isPro !== false) {
+					if (!branch.upstream?.missing) {
+						prPromises.set(branch.id, branch.getAssociatedPullRequest());
+					}
+
+					contributorPromises.set(
+						branch.id,
+						container.git.getBranchContributorOverview(branch.repoPath, branch.ref),
+					);
 				}
+
 				if (wt != null) {
 					statusPromises.set(branch.id, wt.getStatus());
 				}
-				contributorPromises.set(
-					branch.id,
-					container.git.getBranchContributorOverview(branch.repoPath, branch.ref),
-				);
 
 				overviewBranches.stale.push({
 					id: branch.id,