Adds repo visibility to FeatureAccess

Author: d13

src/features.ts

@@ -1,3 +1,4 @@
+import type { RepositoryVisibility } from './git/gitProvider';
 import type { RequiredSubscriptionPlans, Subscription } from './subscription';
 
 export const enum Features {
@@ -7,10 +8,19 @@ export const enum Features {
 }
 
 export type FeatureAccess =
-	| { allowed: true; subscription: { current: Subscription; required?: undefined } }
-	| { allowed: false; subscription: { current: Subscription; required?: RequiredSubscriptionPlans } };
+	| {
+			allowed: true;
+			subscription: { current: Subscription; required?: undefined };
+			visibility?: RepositoryVisibility;
+	  }
+	| {
+			allowed: false;
+			subscription: { current: Subscription; required?: RequiredSubscriptionPlans };
+			visibility?: RepositoryVisibility;
+	  };
 
 export const enum PlusFeatures {
 	Timeline = 'timeline',
 	Worktrees = 'worktrees',
+	Graph = 'graph',
 }

src/git/gitProviderService.ts

@@ -16,7 +16,8 @@ import { ContextKeys, CoreGitConfiguration, GlyphChars, Schemes } from '../const
 import type { Container } from '../container';
 import { setContext } from '../context';
 import { AccessDeniedError, ProviderNotFoundError } from '../errors';
-import type { FeatureAccess, Features, PlusFeatures } from '../features';
+import type { FeatureAccess, Features } from '../features';
+import { PlusFeatures } from '../features';
 import { Logger } from '../logger';
 import type { SubscriptionChangeEvent } from '../plus/subscription/subscriptionService';
 import type { RepoComparisonKey } from '../repositories';
@@ -552,18 +553,32 @@ export class GitProviderService implements Disposable {
 					if (visibility !== RepositoryVisibility.Private) {
 						switch (plan) {
 							case SubscriptionPlanId.Free:
+								if (feature === PlusFeatures.Graph) {
+									return {
+										allowed: true,
+										subscription: { current: subscription },
+										visibility: visibility,
+									};
+								}
+
 								return {
 									allowed: false,
 									subscription: { current: subscription, required: SubscriptionPlanId.FreePlus },
+									visibility: visibility,
 								};
 							case SubscriptionPlanId.FreePlus:
-								return { allowed: true, subscription: { current: subscription } };
+								return {
+									allowed: true,
+									subscription: { current: subscription },
+									visibility: visibility,
+								};
 						}
 					}
 
 					return {
 						allowed: false,
 						subscription: { current: subscription, required: SubscriptionPlanId.Pro },
+						visibility: visibility,
 					};
 				});