[ML] AIOps: Adding ability to disable AIOps in kibana

src/platform/plugins/shared/discover/public/__mocks__/services.ts

@@ -172,7 +172,7 @@ export function createDiscoverServicesMock(): DiscoverServices {
       FieldStatisticsTable: jest.fn(() => createElement('div')),
     },
     aiops: {
-      getPatternAnalysisAvailable: jest.fn().mockResolvedValue(jest.fn().mockResolvedValue(true)),
+      getPatternAnalysisAvailable: jest.fn().mockResolvedValue(jest.fn(() => true)),
       PatternAnalysisComponent: jest.fn(() => createElement('div')),
     },
     docLinks: docLinksServiceMock.createStartContract(),

src/platform/plugins/shared/discover/public/components/view_mode_toggle/view_mode_toggle.test.tsx

@@ -39,7 +39,7 @@ describe('Document view mode toggle component', () => {
       aiops: {
         getPatternAnalysisAvailable: jest
           .fn()
-          .mockResolvedValue(jest.fn().mockResolvedValue(useDataViewWithTextFields)),
+          .mockResolvedValue(jest.fn(() => useDataViewWithTextFields)),
       },
     };
 

src/platform/plugins/shared/discover/public/components/view_mode_toggle/view_mode_toggle.tsx

@@ -73,9 +73,8 @@ export const DocumentViewModeToggle = ({
       aiopsService
         .getPatternAnalysisAvailable()
         .then((patternAnalysisAvailable) => {
-          patternAnalysisAvailable(dataView)
-            .then(setShowPatternAnalysisTabWrapper)
-            .catch(() => setShowPatternAnalysisTabWrapper(false));
+          const available = patternAnalysisAvailable(dataView);
+          setShowPatternAnalysisTabWrapper(available);
         })
         .catch(() => setShowPatternAnalysisTabWrapper(false));
     },

x-pack/platform/plugins/shared/aiops/kibana.jsonc

@@ -11,6 +11,7 @@
     "id": "aiops",
     "browser": true,
     "server": true,
+    "configPath": ["xpack", "aiops"],
     "requiredPlugins": [
       "charts",
       "data",
@@ -38,4 +39,4 @@
       "cases"
     ]
   }
-}
+}

x-pack/platform/plugins/shared/aiops/public/components/log_categorization/log_categorization_enabled.ts

@@ -5,17 +5,18 @@
  * 2.0.
  */
 
+import type { ApplicationStart } from '@kbn/core/public';
 import type { DataView } from '@kbn/data-views-plugin/public';
 import { ES_FIELD_TYPES } from '@kbn/field-types';
-import type { LicensingPluginStart } from '@kbn/licensing-plugin/public';
-import { firstValueFrom } from 'rxjs';
 
-export function getPatternAnalysisAvailable(licensing: LicensingPluginStart) {
-  const lic = firstValueFrom(licensing.license$);
-  return async (dataView: DataView) => {
-    const isPlatinum = (await lic).hasAtLeast('platinum');
+export function getPatternAnalysisAvailable(application: ApplicationStart) {
+  return (dataView: DataView) => {
+    const aiopsEnabled = application.capabilities.aiops.enabled;
+    if (!aiopsEnabled) {
+      return false;
+    }
+
     return (
-      isPlatinum &&
       dataView.isTimeBased() &&
       dataView.fields.some((f) => f.esTypes?.includes(ES_FIELD_TYPES.TEXT))
     );

x-pack/platform/plugins/shared/aiops/public/plugin.tsx

@@ -7,7 +7,6 @@
 
 import type { CoreStart, Plugin } from '@kbn/core/public';
 import { type CoreSetup } from '@kbn/core/public';
-import { firstValueFrom } from 'rxjs';
 
 import { getChangePointDetectionComponent } from './shared_components';
 import { LogCategorizationForDiscover as PatternAnalysisComponent } from './shared_lazy_components';
@@ -26,29 +25,25 @@ export type AiopsCoreSetup = CoreSetup<AiopsPluginStartDeps, AiopsPluginStart>;
 export class AiopsPlugin
   implements Plugin<AiopsPluginSetup, AiopsPluginStart, AiopsPluginSetupDeps, AiopsPluginStartDeps>
 {
-  public setup(
-    core: AiopsCoreSetup,
-    { embeddable, cases, licensing, uiActions }: AiopsPluginSetupDeps
-  ) {
-    Promise.all([firstValueFrom(licensing.license$), core.getStartServices()]).then(
-      ([license, [coreStart, pluginStart]]) => {
-        const { canUseAiops } = coreStart.application.capabilities.ml;
-
-        if (license.hasAtLeast('platinum') && canUseAiops) {
-          if (embeddable) {
-            registerEmbeddables(embeddable, core);
-          }
+  public setup(core: AiopsCoreSetup, { embeddable, cases, uiActions }: AiopsPluginSetupDeps) {
+    core.getStartServices().then(([coreStart, pluginStart]) => {
+      const { canUseAiops } = coreStart.application.capabilities.ml;
+      const aiopsEnabled = coreStart.application.capabilities.aiops.enabled;
+
+      if (canUseAiops && aiopsEnabled) {
+        if (embeddable) {
+          registerEmbeddables(embeddable, core);
+        }
 
-          if (uiActions) {
-            registerAiopsUiActions(uiActions, coreStart, pluginStart);
-          }
+        if (uiActions) {
+          registerAiopsUiActions(uiActions, coreStart, pluginStart);
+        }
 
-          if (cases) {
-            registerCases(cases, coreStart, pluginStart);
-          }
+        if (cases) {
+          registerCases(cases, coreStart, pluginStart);
         }
       }
-    );
+    });
   }
 
   public start(core: CoreStart, plugins: AiopsPluginStartDeps): AiopsPluginStart {
@@ -58,7 +53,7 @@ export class AiopsPlugin
         const { getPatternAnalysisAvailable } = await import(
           './components/log_categorization/log_categorization_enabled'
         );
-        return getPatternAnalysisAvailable(plugins.licensing);
+        return getPatternAnalysisAvailable(core.application);
       },
       PatternAnalysisComponent,
     };

x-pack/platform/plugins/shared/aiops/public/types/index.ts

@@ -50,7 +50,7 @@ export interface AiopsPluginStartDeps {
 
 export type AiopsPluginSetup = void;
 export interface AiopsPluginStart {
-  getPatternAnalysisAvailable: () => Promise<(dataView: DataView) => Promise<boolean>>;
+  getPatternAnalysisAvailable: () => Promise<(dataView: DataView) => boolean>;
   PatternAnalysisComponent: React.ComponentType<LogCategorizationEmbeddableWrapperProps>;
   ChangePointDetectionComponent: ChangePointDetectionSharedComponent;
 }

x-pack/platform/plugins/shared/aiops/server/config_schema.ts

@@ -0,0 +1,23 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+import type { TypeOf } from '@kbn/config-schema';
+import { schema } from '@kbn/config-schema';
+import type { PluginConfigDescriptor } from '@kbn/core/server';
+
+export const configSchema = schema.object({
+  ui: schema.maybe(
+    schema.object({
+      enabled: schema.maybe(schema.boolean()),
+    })
+  ),
+});
+
+export type ConfigSchema = TypeOf<typeof configSchema>;
+
+export const config: PluginConfigDescriptor<ConfigSchema> = {
+  schema: configSchema,
+};

x-pack/platform/plugins/shared/aiops/server/index.ts

@@ -6,10 +6,13 @@
  */
 
 import type { PluginInitializerContext } from '@kbn/core/server';
+import type { ConfigSchema } from './config_schema';
 
-export async function plugin(initializerContext: PluginInitializerContext) {
+export async function plugin(initializerContext: PluginInitializerContext<ConfigSchema>) {
   const { AiopsPlugin } = await import('./plugin');
   return new AiopsPlugin(initializerContext);
 }
 
 export type { AiopsPluginSetup, AiopsPluginStart } from './types';
+
+export { config, type ConfigSchema } from './config_schema';

x-pack/platform/plugins/shared/aiops/server/lib/capabilities.ts

@@ -0,0 +1,39 @@
+/*
+ * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+ * or more contributor license agreements. Licensed under the Elastic License
+ * 2.0; you may not use this file except in compliance with the Elastic License
+ * 2.0.
+ */
+
+import type { CoreSetup } from '@kbn/core/server';
+import { firstValueFrom } from 'rxjs';
+import type { AiopsPluginStartDeps } from '../types';
+
+export const setupCapabilities = (core: CoreSetup<AiopsPluginStartDeps>, enabled: boolean) => {
+  core.capabilities.registerProvider(() => {
+    return {
+      aiops: {
+        enabled,
+      },
+    };
+  });
+
+  core.capabilities.registerSwitcher(
+    async (request, capabilities, useDefaultCapabilities) => {
+      const [, { licensing }] = await core.getStartServices();
+      const lic = await firstValueFrom(licensing.license$);
+      if (lic.hasAtLeast('platinum') === false) {
+        return {
+          aiops: {
+            enabled: false,
+          },
+        };
+      }
+
+      return {};
+    },
+    {
+      capabilityPath: 'aiops.*',
+    }
+  );
+};

x-pack/platform/plugins/shared/aiops/server/plugin.ts

@@ -29,23 +29,28 @@ import { defineRoute as defineLogRateAnalysisFieldCandidatesRoute } from './rout
 import { defineRoute as defineLogRateAnalysisRoute } from './routes/log_rate_analysis/define_route';
 import { defineRoute as defineCategorizationFieldValidationRoute } from './routes/categorization_field_validation/define_route';
 import { registerCasesPersistableState } from './register_cases';
+import type { ConfigSchema } from './config_schema';
+import { setupCapabilities } from './lib/capabilities';
 
 export class AiopsPlugin
   implements Plugin<AiopsPluginSetup, AiopsPluginStart, AiopsPluginSetupDeps, AiopsPluginStartDeps>
 {
   private readonly logger: Logger;
   private licenseSubscription: Subscription | null = null;
   private usageCounter?: UsageCounter;
+  private aiopsEnabled: boolean = true;
 
-  constructor(initializerContext: PluginInitializerContext) {
+  constructor(initializerContext: PluginInitializerContext<ConfigSchema>) {
     this.logger = initializerContext.logger.get();
+    this.aiopsEnabled = initializerContext.config.get().ui?.enabled ?? true;
   }
 
   public setup(
     core: CoreSetup<AiopsPluginStartDeps, AiopsPluginSetupDeps>,
     plugins: AiopsPluginSetupDeps
   ) {
-    this.logger.debug('aiops: Setup');
+    setupCapabilities(core, this.aiopsEnabled);
+
     this.usageCounter = plugins.usageCollection?.createUsageCounter(AIOPS_PLUGIN_ID);
 
     // Subscribe to license changes and store the current license in `currentLicense`.

x-pack/platform/plugins/shared/aiops/server/types.ts

@@ -6,7 +6,7 @@
  */
 
 import type { PluginSetup, PluginStart } from '@kbn/data-plugin/server';
-import type { LicensingPluginSetup } from '@kbn/licensing-plugin/server';
+import type { LicensingPluginSetup, LicensingPluginStart } from '@kbn/licensing-plugin/server';
 import type { CasesServerSetup } from '@kbn/cases-plugin/server';
 import type { UsageCollectionSetup } from '@kbn/usage-collection-plugin/server';
 
@@ -19,6 +19,7 @@ export interface AiopsPluginSetupDeps {
 
 export interface AiopsPluginStartDeps {
   data: PluginStart;
+  licensing: LicensingPluginStart;
 }
 
 /**

x-pack/platform/plugins/shared/aiops/tsconfig.json

@@ -76,6 +76,7 @@
     "@kbn/observability-ai-assistant-plugin",
     "@kbn/apm-utils",
     "@kbn/ml-field-stats-flyout",
+    "@kbn/config-schema",
   ],
   "exclude": [
     "target/**/*",

x-pack/platform/plugins/shared/ml/public/application/components/anomalies_table/links_menu.tsx

@@ -765,7 +765,7 @@ export const LinksMenuUI = (props: LinksMenuProps) => {
   };
 
   const { anomaly, showViewSeriesLink } = props;
-  const canUpdateJob = usePermissionCheck('canUpdateJob');
+  const [canUpdateJob, canUseAiops] = usePermissionCheck(['canUpdateJob', 'canUseAiops']);
   const canConfigureRules = isRuleSupported(anomaly.source) && canUpdateJob;
 
   const contextMenuItems = useMemo(() => {
@@ -929,7 +929,7 @@ export const LinksMenuUI = (props: LinksMenuProps) => {
       );
     }
 
-    if (openInLogRateAnalysisUrl) {
+    if (openInLogRateAnalysisUrl && canUseAiops) {
       items.push(
         <EuiContextMenuItem
           key="log_rate_analysis"
@@ -945,7 +945,7 @@ export const LinksMenuUI = (props: LinksMenuProps) => {
       );
     }
 
-    if (messageField !== null) {
+    if (messageField !== null && canUseAiops) {
       items.push(
         <EuiContextMenuItem
           key="run_pattern_analysis"

x-pack/platform/plugins/shared/ml/public/application/components/ml_page/side_nav.tsx

@@ -17,6 +17,7 @@ import { isFullLicense } from '../../license';
 import type { MlRoute } from '../../routing';
 import { ML_PAGES } from '../../../../common/constants/locator';
 import { useEnabledFeatures } from '../../contexts/ml';
+import { usePermissionCheck } from '../../capabilities/check_capabilities';
 
 export interface Tab {
   id: string;
@@ -38,6 +39,7 @@ export function useSideNavItems(activeRoute: MlRoute | undefined) {
 
   const mlFeaturesDisabled = !isFullLicense();
   const { isADEnabled, isDFAEnabled } = useEnabledFeatures();
+  const [canUseAiops] = usePermissionCheck(['canUseAiops']);
 
   const [globalState] = useUrlState('_g');
 
@@ -158,6 +160,10 @@ export function useSideNavItems(activeRoute: MlRoute | undefined) {
         : []),
     ];
 
+    if (canUseAiops === false) {
+      return mlTabs;
+    }
+
     mlTabs.push({
       id: 'aiops_section',
       name: i18n.translate('xpack.ml.navMenu.aiopsTabLinkText', {
@@ -203,7 +209,7 @@ export function useSideNavItems(activeRoute: MlRoute | undefined) {
     });
 
     return mlTabs;
-  }, [mlFeaturesDisabled, isADEnabled, isDFAEnabled]);
+  }, [mlFeaturesDisabled, isADEnabled, isDFAEnabled, canUseAiops]);
 
   const getTabItem: (tab: Tab) => EuiSideNavItemType<unknown> = useCallback(
     (tab: Tab) => {