Reflects org settings in AI section of Home

Author: axosoft-ramint

src/webviews/apps/plus/shared/components/integrations-chip.ts

@@ -178,6 +178,10 @@ export class GLIntegrationsChip extends LitElement {
 		return this._state.ai;
 	}
 
+	private get aiEnabled() {
+		return this._state.orgSettings?.ai ?? true;
+	}
+
 	private get integrations() {
 		return this._state.integrations;
 	}
@@ -322,52 +326,58 @@ export class GLIntegrationsChip extends LitElement {
 
 	private renderAIStatus() {
 		return html`<span
-			class="integration status--${this.ai?.model != null ? 'connected' : 'disconnected'}"
+			class="integration status--${this.aiEnabled && this.ai?.model != null ? 'connected' : 'disconnected'}"
 			slot="anchor"
 		>
-			<code-icon icon="${this.ai?.model != null ? 'sparkle-filled' : 'sparkle'}"></code-icon>
+			<code-icon icon="${this.aiEnabled && this.ai?.model != null ? 'sparkle-filled' : 'sparkle'}"></code-icon>
 		</span>`;
 	}
 
 	private renderAIRow() {
 		const { model } = this.ai;
 
-		const connected = model != null;
-		const showLock = false;
+		const connectedAndEnabled = this.aiEnabled && model != null;
+		const showLock = !this.aiEnabled;
 		const showProBadge = false;
-		const icon = connected ? 'sparkle-filled' : 'sparkle'; // TODO: Provider?
+		const icon = connectedAndEnabled ? 'sparkle-filled' : 'sparkle'; // TODO: Provider?
 
 		return html`<div
-			class="integration-row integration-row--ai status--${connected ? 'connected' : 'disconnected'}${showLock
-				? ' is-locked'
-				: ''}"
+			class="integration-row integration-row--ai status--${connectedAndEnabled
+				? 'connected'
+				: 'disconnected'}${showLock ? ' is-locked' : ''}"
 		>
 			<span class="integration__icon"><code-icon icon="${icon}"></code-icon></span>
-			<span class="integration__content">
-				<span class="integration__title">
-					<span>${model?.provider.name ?? 'AI'}</span>
-					${showProBadge
-						? html` <gl-feature-badge
-								placement="right"
-								.source=${{ source: 'home', detail: 'integrations' } as const}
-								cloud
-						  ></gl-feature-badge>`
-						: nothing}
-				</span>
-				${model?.name ? html`<span class="integration__details">${model.name}</span>` : nothing}
-			</span>
-			<span class="integration__actions">
-				<gl-button
-					appearance="toolbar"
-					href="${createCommandLink<Source>('gitlens.switchAIModel', {
-						source: 'home',
-						detail: 'integrations',
-					})}"
-					tooltip="Switch AI Model"
-					aria-label="Switch AI Model"
-					><code-icon icon="arrow-swap"></code-icon
-				></gl-button>
-			</span>
+			${this.aiEnabled
+				? html`<span class="integration__content">
+							<span class="integration__title">
+								<span>${model?.provider.name ?? 'AI'}</span>
+								${showProBadge
+									? html` <gl-feature-badge
+											placement="right"
+											.source=${{ source: 'home', detail: 'integrations' } as const}
+											cloud
+									  ></gl-feature-badge>`
+									: nothing}
+							</span>
+							${model?.name ? html`<span class="integration__details">${model.name}</span>` : nothing}
+						</span>
+						<span class="integration__actions">
+							<gl-button
+								appearance="toolbar"
+								href="${createCommandLink<Source>('gitlens.switchAIModel', {
+									source: 'home',
+									detail: 'integrations',
+								})}"
+								tooltip="Switch AI Model"
+								aria-label="Switch AI Model"
+								><code-icon icon="arrow-swap"></code-icon
+							></gl-button>
+						</span>`
+				: html`<span class="integration__content">
+						<span class="integration_details"
+							>GitLens AI features have been disabled by your GitKraken admin</span
+						>
+				  </span>`}
 		</div>`;
 	}
 }

src/webviews/home/homeWebview.ts

@@ -664,6 +664,7 @@ export class HomeWebviewProvider implements WebviewProvider<State, State, HomeWe
 	private getOrgSettings(): State['orgSettings'] {
 		return {
 			drafts: getContext('gitlens:gk:organization:drafts:enabled', false),
+			ai: getContext('gitlens:gk:organization:ai:enabled', true),
 		};
 	}
 

src/webviews/home/protocol.ts

@@ -23,6 +23,7 @@ export interface State extends WebviewState {
 	subscription: Subscription;
 	orgSettings: {
 		drafts: boolean;
+		ai: boolean;
 	};
 	previewCollapsed: boolean;
 	integrationBannerCollapsed: boolean;