Insights: don't show "Ask AI" if disabled by enterprise

Fixed: 404735118
Change-Id: Ia61407364cfe77571bd6d498d0ad9a918829bbbc
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6414391
Reviewed-by: Alina Varkki <[email protected]>
Commit-Queue: Alina Varkki <[email protected]>
Auto-Submit: Jack Franklin <[email protected]>

Author: jackfranklin

front_end/panels/timeline/components/insights/BaseInsightComponent.test.ts

@@ -3,6 +3,7 @@
 // found in the LICENSE file.
 
 import type * as Common from '../../../../core/common/common.js';
+import * as Root from '../../../../core/root/root.js';
 import * as Trace from '../../../../models/trace/trace.js';
 import {dispatchClickEvent, renderElementIntoDOM} from '../../../../testing/DOMHelpers.js';
 import {describeWithEnvironment, updateHostConfig} from '../../../../testing/EnvironmentHelpers.js';
@@ -137,6 +138,22 @@ describeWithEnvironment('BaseInsightComponent', () => {
       assert.isOk(button);
     });
 
+    it('does not render the "Ask AI" button if disabled by enterprise policy', async () => {
+      updateHostConfig({
+        devToolsAiAssistancePerformanceAgent: {
+          enabled: true,
+          insightsEnabled: true,
+        },
+        aidaAvailability: {
+          enterprisePolicyValue: Root.Runtime.GenAiEnterprisePolicyValue.DISABLE,
+        }
+      });
+      const component = await renderComponent({insightHasAISupport: true});
+      assert.isOk(component.shadowRoot);
+      const button = component.shadowRoot.querySelector('devtools-button[data-insights-ask-ai]');
+      assert.isNull(button);
+    });
+
     it('does not show the button if the feature is enabled but the Insight does not support it', async () => {
       updateHostConfig({
         devToolsAiAssistancePerformanceAgent: {

front_end/panels/timeline/components/insights/BaseInsightComponent.ts

@@ -323,7 +323,10 @@ export abstract class BaseInsightComponent<T extends InsightModel> extends HTMLE
   }
 
   #canShowAskAI(): boolean {
-    return this.#insightsAskAiEnabled && this.hasAskAISupport;
+    const aiDisabledByEnterprisePolicy = Root.Runtime.hostConfig.aidaAvailability?.enterprisePolicyValue ===
+        Root.Runtime.GenAiEnterprisePolicyValue.DISABLE;
+
+    return !aiDisabledByEnterprisePolicy && this.#insightsAskAiEnabled && this.hasAskAISupport;
   }
 
   #renderInsightContent(insightModel: T): Lit.LitTemplate {