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

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,15 +5,24 @@
  * 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) {
+export function getPatternAnalysisAvailable(
+  licensing: LicensingPluginStart,
+  application: ApplicationStart
+) {
   const lic = firstValueFrom(licensing.license$);
   return async (dataView: DataView) => {
     const isPlatinum = (await lic).hasAtLeast('platinum');
+    const aiopsEnabled = application.capabilities.aiops.enabled;
+    if (!aiopsEnabled) {
+      return false;
+    }
+
     return (
       isPlatinum &&
       dataView.isTimeBased() &&

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

@@ -33,8 +33,9 @@ export class AiopsPlugin
     Promise.all([firstValueFrom(licensing.license$), core.getStartServices()]).then(
       ([license, [coreStart, pluginStart]]) => {
         const { canUseAiops } = coreStart.application.capabilities.ml;
+        const aiopsEnabled = coreStart.application.capabilities.aiops.enabled;
 
-        if (license.hasAtLeast('platinum') && canUseAiops) {
+        if (license.hasAtLeast('platinum') && canUseAiops && aiopsEnabled) {
           if (embeddable) {
             registerEmbeddables(embeddable, core);
           }
@@ -58,7 +59,7 @@ export class AiopsPlugin
         const { getPatternAnalysisAvailable } = await import(
           './components/log_categorization/log_categorization_enabled'
         );
-        return getPatternAnalysisAvailable(plugins.licensing);
+        return getPatternAnalysisAvailable(plugins.licensing, core.application);
       },
       PatternAnalysisComponent,
     };

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,31 @@
+/*
+ * 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 type { AiopsPluginStartDeps } from '../types';
+
+export const setupCapabilities = (
+  core: Pick<CoreSetup<AiopsPluginStartDeps>, 'capabilities' | 'getStartServices'>,
+  enabled: boolean
+) => {
+  core.capabilities.registerProvider(() => {
+    return {
+      aiops: {
+        enabled,
+      },
+    };
+  });
+
+  core.capabilities.registerSwitcher(
+    async (request, capabilities, useDefaultCapabilities) => {
+      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/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) => {