Updates Start Work and Associate Issue flows

Author: axosoft-ramint

src/commands/quickCommand.steps.ts

@@ -2705,6 +2705,28 @@ export async function* ensureAccessStep<
 				createQuickPickSeparator(),
 			);
 			break;
+		case PlusFeatures.StartWork:
+			directives.splice(
+				0,
+				0,
+				createDirectiveQuickPickItem(Directive.Noop, undefined, {
+					label: 'Start work on an issue from your connected integrations',
+					iconPath: new ThemeIcon('issues'),
+				}),
+				createQuickPickSeparator(),
+			);
+			break;
+		case PlusFeatures.AssociateIssueWithBranch:
+			directives.splice(
+				0,
+				0,
+				createDirectiveQuickPickItem(Directive.Noop, undefined, {
+					label: 'Connect your branches to their associated issues in Home view',
+					iconPath: new ThemeIcon('issues'),
+				}),
+				createQuickPickSeparator(),
+			);
+			break;
 		case PlusFeatures.Worktrees:
 			directives.splice(
 				0,

src/features.ts

@@ -41,6 +41,8 @@ export const enum PlusFeatures {
 	Worktrees = 'worktrees',
 	Graph = 'graph',
 	Launchpad = 'launchpad',
+	StartWork = 'startWork',
+	AssociateIssueWithBranch = 'associateIssueWithBranch',
 }
 
 export type FeaturePreviews = 'graph';

src/plus/startWork/startWork.ts

@@ -26,12 +26,14 @@ import {
 	OpenOnGitLabQuickInputButton,
 	OpenOnJiraQuickInputButton,
 } from '../../commands/quickCommand.buttons';
+import { ensureAccessStep } from '../../commands/quickCommand.steps';
 import { getSteps } from '../../commands/quickWizard.utils';
 import { proBadge } from '../../constants';
 import type { IntegrationId } from '../../constants.integrations';
 import { HostingIntegrationId, IssueIntegrationId, SelfHostedIntegrationId } from '../../constants.integrations';
 import type { Source, Sources, StartWorkTelemetryContext, TelemetryEvents } from '../../constants.telemetry';
 import type { Container } from '../../container';
+import { PlusFeatures } from '../../features';
 import type { Issue, IssueShape } from '../../git/models/issue';
 import type { GitBranchReference } from '../../git/models/reference';
 import type { Repository } from '../../git/models/repository';
@@ -214,6 +216,18 @@ export abstract class StartWorkBaseCommand extends QuickCommand<State> {
 				}
 			}
 
+			let plusFeature: PlusFeatures | undefined;
+			if (this.key === 'startWork') {
+				plusFeature = PlusFeatures.StartWork;
+			} else if (this.key === 'associateIssueWithBranch') {
+				plusFeature = PlusFeatures.AssociateIssueWithBranch;
+			}
+
+			if (plusFeature != null) {
+				const result = yield* ensureAccessStep(this.container, state, context, plusFeature);
+				if (result === StepResultBreak) continue;
+			}
+
 			if (state.counter < 1 || state.item == null) {
 				if (this.container.telemetry.enabled) {
 					this.container.telemetry.sendEvent(