[Security Assistant] Move security AI assistant button into global na…

…v bar (elastic#203060)

## Summary

More changes are needed within the observability and search solution to
close the issue fully.

Summarise your PR. If it involves visual changes include a screenshot or
gif.

Move the security AI assistant button from the solution header bar into
the global nav bar. This is part of the AI assistant unification
initiative.

### How to Test
- Start kibana
- Go to one of the security solution pages (e.g. attack discovery)
- AI assistant button should be in the global nav bar. Clicking it opens
the assistant.

- The button can also be tested for security serverless deployment. It
should look like the screenshot bellow.

### Checklist

Check the PR satisfies following conditions. 

Reviewers should verify this PR satisfies this list as well.

- [x] Any text added follows [EUI's writing
guidelines](https://elastic.github.io/eui/#/guidelines/writing), uses
sentence case text and includes [i18n
support](https://github.com/elastic/kibana/blob/main/packages/kbn-i18n/README.md)
- [x] [Unit or functional
tests](https://www.elastic.co/guide/en/kibana/master/development-tests.html)
were updated or added to match the most common scenarios
- [x] This was checked for breaking HTTP API changes, and any breaking
changes have been approved by the breaking-change committee. The
`release_note:breaking` label should be applied in these situations.
- [x] [Flaky Test
Runner](https://ci-stats.kibana.dev/trigger_flaky_test_runner/1) was
used on any tests changed
- [x] The PR description includes the appropriate Release Notes section,
and the correct `release_note:*` label is applied per the
[guidelines](https://www.elastic.co/guide/en/kibana/master/contributing.html#kibana-release-notes-process)

### Identify risks

Does this PR introduce any risks? For example, consider risks like hard
to test bugs, performance regression, potential of data loss.

Describe the risk, its severity, and mitigation for each identified
risk. Invite stakeholders and evaluate how to proceed before merging.

- [ ] [See some risk
examples](https://github.com/elastic/kibana/blob/main/RISK_MATRIX.mdx)
- [ ] ...

Classic:

![image](https://github.com/user-attachments/assets/b2a9c982-bc54-42f4-ab59-6f0c99d4d899)

![image](https://github.com/user-attachments/assets/1ae36af0-5d1a-4519-844a-563074646ddf)

Serverless:

![image](https://github.com/user-attachments/assets/345280df-0e70-4203-b0d8-48ad11753f74)

![image](https://github.com/user-attachments/assets/7425c886-4528-4987-a00a-48bdc71728c7)

Old:
<img width="1728" alt="image"
src="https://github.com/user-attachments/assets/5ef568c6-2d31-47da-8f5f-87dfdf10cb5c">

---------

Co-authored-by: kibanamachine <[email protected]>
Co-authored-by: Elastic Machine <[email protected]>

x-pack/platform/packages/shared/kbn-elastic-assistant/impl/assistant/prompt_editor/system_prompt/index.test.tsx

@@ -15,6 +15,8 @@ import { DEFAULT_CONVERSATION_TITLE } from '../../use_conversation/translations'
 import { TestProviders } from '../../../mock/test_providers/test_providers';
 import { WELCOME_CONVERSATION } from '../../use_conversation/sample_conversations';
 import { PromptResponse } from '@kbn/elastic-assistant-common';
+import { chromeServiceMock } from '@kbn/core-chrome-browser-mocks';
+import { of } from 'rxjs';
 
 const BASE_CONVERSATION: Conversation = {
   ...WELCOME_CONVERSATION,
@@ -36,6 +38,13 @@ const mockUseAssistantContext = {
   setConversations: jest.fn(),
   setAllSystemPrompts: jest.fn(),
   allSystemPrompts: mockSystemPrompts,
+  chrome: {
+    getChromeStyle$: jest.fn(() => of('classic')),
+    navControls: chromeServiceMock.createStartContract().navControls,
+  },
+  assistantAvailability: {
+    hasAssistantPrivilege: true,
+  },
 };
 
 jest.mock('../../../assistant_context', () => {

x-pack/platform/packages/shared/kbn-elastic-assistant/impl/assistant/quick_prompts/quick_prompt_settings/quick_prompt_settings.test.tsx

@@ -11,6 +11,8 @@ import { QuickPromptSettings } from './quick_prompt_settings';
 import { TestProviders } from '../../../mock/test_providers/test_providers';
 import { MOCK_QUICK_PROMPTS } from '../../../mock/quick_prompt';
 import { mockPromptContexts } from '../../../mock/prompt_context';
+import { chromeServiceMock } from '@kbn/core-chrome-browser-mocks';
+import { of } from 'rxjs';
 
 const onSelectedQuickPromptChange = jest.fn();
 const setPromptsBulkActions = jest.fn();
@@ -28,6 +30,13 @@ const testProps = {
 };
 const mockContext = {
   basePromptContexts: MOCK_QUICK_PROMPTS,
+  chrome: {
+    getChromeStyle$: jest.fn(() => of('classic')),
+    navControls: chromeServiceMock.createStartContract().navControls,
+  },
+  assistantAvailability: {
+    hasAssistantPrivilege: true,
+  },
 };
 
 jest.mock('../../../assistant_context', () => ({

x-pack/platform/packages/shared/kbn-elastic-assistant/impl/assistant/quick_prompts/quick_prompts.test.tsx

@@ -11,6 +11,8 @@ import { TestProviders } from '../../mock/test_providers/test_providers';
 import { MOCK_QUICK_PROMPTS } from '../../mock/quick_prompt';
 import { QUICK_PROMPTS_TAB } from '../settings/const';
 import { QuickPrompts } from './quick_prompts';
+import { of } from 'rxjs';
+import { chromeServiceMock } from '@kbn/core-chrome-browser-mocks';
 
 const setInput = jest.fn();
 const setIsSettingsModalVisible = jest.fn();
@@ -26,6 +28,13 @@ const mockUseAssistantContext = {
   setSelectedSettingsTab,
   promptContexts: {},
   allQuickPrompts: MOCK_QUICK_PROMPTS,
+  chrome: {
+    getChromeStyle$: jest.fn(() => of('classic')),
+    navControls: chromeServiceMock.createStartContract().navControls,
+  },
+  assistantAvailability: {
+    hasAssistantPrivilege: true,
+  },
 };
 
 const testTitle = 'SPL_QUERY_CONVERSION_TITLE';

x-pack/platform/packages/shared/kbn-elastic-assistant/impl/assistant_context/assistant_nav_link.test.tsx

@@ -0,0 +1,161 @@
+/*
+ * 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 React from 'react';
+import { render, renderHook } from '@testing-library/react';
+import { AssistantNavLink } from './assistant_nav_link';
+import { chromeServiceMock } from '@kbn/core-chrome-browser-mocks';
+import { ChromeNavControl } from '@kbn/core/public';
+import { createHtmlPortalNode, OutPortal } from 'react-reverse-portal';
+import { of } from 'rxjs';
+import { useAssistantContext } from '.';
+
+const MockNavigationBar = OutPortal;
+
+const mockShowAssistantOverlay = jest.fn();
+const mockNavControls = chromeServiceMock.createStartContract().navControls;
+const mockGetChromeStyle = jest.fn();
+
+const mockAssistantContext = {
+  chrome: {
+    getChromeStyle$: mockGetChromeStyle,
+    navControls: mockNavControls,
+  },
+  showAssistantOverlay: mockShowAssistantOverlay,
+  assistantAvailability: {
+    hasAssistantPrivilege: true,
+  },
+};
+
+jest.mock('.', () => {
+  return {
+    ...jest.requireActual('.'),
+    useAssistantContext: jest.fn(),
+  };
+});
+
+describe('AssistantNavLink', () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+    mockGetChromeStyle.mockReturnValue(of('classic'));
+    (useAssistantContext as jest.Mock).mockReturnValue({
+      ...mockAssistantContext,
+    });
+  });
+
+  it('should register link in nav bar', () => {
+    render(<AssistantNavLink />);
+    expect(mockNavControls.registerRight).toHaveBeenCalledTimes(1);
+  });
+
+  it('button has transparent background in project navigation', () => {
+    const { result: portalNode } = renderHook(() =>
+      React.useMemo(() => createHtmlPortalNode(), [])
+    );
+
+    mockGetChromeStyle.mockReturnValue(of('project'));
+
+    mockNavControls.registerRight.mockImplementation((chromeNavControl: ChromeNavControl) => {
+      chromeNavControl.mount(portalNode.current.element);
+    });
+
+    const { queryByTestId } = render(
+      <>
+        <MockNavigationBar node={portalNode.current} />
+        <AssistantNavLink />
+      </>
+    );
+    expect(queryByTestId('assistantNavLink')).not.toHaveStyle(
+      'background-color: rgb(204, 228, 245)'
+    );
+  });
+
+  it('button has opaque background in classic navigation', () => {
+    const { result: portalNode } = renderHook(() =>
+      React.useMemo(() => createHtmlPortalNode(), [])
+    );
+
+    mockGetChromeStyle.mockReturnValue(of('classic'));
+
+    mockNavControls.registerRight.mockImplementation((chromeNavControl: ChromeNavControl) => {
+      chromeNavControl.mount(portalNode.current.element);
+    });
+
+    const { queryByTestId } = render(
+      <>
+        <MockNavigationBar node={portalNode.current} />
+        <AssistantNavLink />
+      </>
+    );
+    expect(queryByTestId('assistantNavLink')).toHaveStyle('background-color: rgb(204, 228, 245)');
+  });
+
+  it('should render the header link text', () => {
+    const { result: portalNode } = renderHook(() =>
+      React.useMemo(() => createHtmlPortalNode(), [])
+    );
+
+    mockNavControls.registerRight.mockImplementation((chromeNavControl: ChromeNavControl) => {
+      chromeNavControl.mount(portalNode.current.element);
+    });
+
+    const { queryByText, queryByTestId } = render(
+      <>
+        <MockNavigationBar node={portalNode.current} />
+        <AssistantNavLink />
+      </>
+    );
+    expect(queryByTestId('assistantNavLink')).toBeInTheDocument();
+    expect(queryByText('AI Assistant')).toBeInTheDocument();
+  });
+
+  it('should not render the header link if not authorized', () => {
+    const { result: portalNode } = renderHook(() =>
+      React.useMemo(() => createHtmlPortalNode(), [])
+    );
+
+    mockNavControls.registerRight.mockImplementation((chromeNavControl: ChromeNavControl) => {
+      chromeNavControl.mount(portalNode.current.element);
+    });
+
+    (useAssistantContext as jest.Mock).mockReturnValue({
+      ...mockAssistantContext,
+      assistantAvailability: {
+        hasAssistantPrivilege: false,
+      },
+    });
+
+    const { queryByText, queryByTestId } = render(
+      <>
+        <MockNavigationBar node={portalNode.current} />
+        <AssistantNavLink />
+      </>
+    );
+    expect(queryByTestId('assistantNavLink')).not.toBeInTheDocument();
+    expect(queryByText('AI Assistant')).not.toBeInTheDocument();
+  });
+
+  it('should call the assistant overlay to show on click', () => {
+    const { result: portalNode } = renderHook(() =>
+      React.useMemo(() => createHtmlPortalNode(), [])
+    );
+
+    mockNavControls.registerRight.mockImplementation((chromeNavControl: ChromeNavControl) => {
+      chromeNavControl.mount(portalNode.current.element);
+    });
+
+    const { queryByTestId } = render(
+      <>
+        <MockNavigationBar node={portalNode.current} />
+        <AssistantNavLink />
+      </>
+    );
+    queryByTestId('assistantNavLink')?.click();
+    expect(mockShowAssistantOverlay).toHaveBeenCalledTimes(1);
+    expect(mockShowAssistantOverlay).toHaveBeenCalledWith({ showOverlay: true });
+  });
+});

x-pack/platform/packages/shared/kbn-elastic-assistant/impl/assistant_context/assistant_nav_link.tsx

@@ -0,0 +1,94 @@
+/*
+ * 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 { FC } from 'react';
+import React, { useCallback, useEffect, useState } from 'react';
+import ReactDOM from 'react-dom';
+import { createHtmlPortalNode, OutPortal, InPortal } from 'react-reverse-portal';
+import { EuiToolTip, EuiButton, EuiFlexGroup, EuiFlexItem, EuiButtonEmpty } from '@elastic/eui';
+import { i18n } from '@kbn/i18n';
+import { ChromeStyle } from '@kbn/core-chrome-browser';
+import { AssistantIcon } from '@kbn/ai-assistant-icon';
+import { useAssistantContext } from '.';
+
+const isMac = navigator.platform.toLowerCase().indexOf('mac') >= 0;
+
+const TOOLTIP_CONTENT = i18n.translate(
+  'xpack.elasticAssistant.assistantContext.assistantNavLinkShortcutTooltip',
+  {
+    values: { keyboardShortcut: isMac ? '⌘ ;' : 'Ctrl ;' },
+    defaultMessage: 'Keyboard shortcut {keyboardShortcut}',
+  }
+);
+const LINK_LABEL = i18n.translate('xpack.elasticAssistant.assistantContext.assistantNavLink', {
+  defaultMessage: 'AI Assistant',
+});
+
+export const AssistantNavLink: FC = () => {
+  const { chrome, showAssistantOverlay, assistantAvailability, currentAppId } =
+    useAssistantContext();
+  const portalNode = React.useMemo(() => createHtmlPortalNode(), []);
+  const [chromeStyle, setChromeStyle] = useState<ChromeStyle | undefined>(undefined);
+
+  // useObserverable would change the order of re-renders that are tested against closely.
+  useEffect(() => {
+    const s = chrome.getChromeStyle$().subscribe(setChromeStyle);
+    return () => s.unsubscribe();
+  }, [chrome]);
+
+  useEffect(() => {
+    const registerPortalNode = () => {
+      chrome.navControls.registerRight({
+        mount: (element: HTMLElement) => {
+          ReactDOM.render(<OutPortal node={portalNode} />, element);
+          return () => ReactDOM.unmountComponentAtNode(element);
+        },
+        // right before the user profile
+        order: 1001,
+      });
+    };
+
+    if (
+      assistantAvailability.hasAssistantPrivilege &&
+      chromeStyle &&
+      currentAppId !== 'management'
+    ) {
+      registerPortalNode();
+    }
+  }, [chrome, portalNode, assistantAvailability.hasAssistantPrivilege, chromeStyle, currentAppId]);
+
+  const showOverlay = useCallback(
+    () => showAssistantOverlay({ showOverlay: true }),
+    [showAssistantOverlay]
+  );
+
+  if (!assistantAvailability.hasAssistantPrivilege || !chromeStyle) {
+    return null;
+  }
+
+  const EuiButtonBasicOrEmpty = chromeStyle === 'project' ? EuiButtonEmpty : EuiButton;
+
+  return (
+    <InPortal node={portalNode}>
+      <EuiToolTip content={TOOLTIP_CONTENT}>
+        <EuiButtonBasicOrEmpty
+          onClick={showOverlay}
+          color="primary"
+          size="s"
+          data-test-subj="assistantNavLink"
+        >
+          <EuiFlexGroup gutterSize="s" alignItems="center">
+            <EuiFlexItem grow={false}>
+              <AssistantIcon size="m" />
+            </EuiFlexItem>
+            <EuiFlexItem grow={false}>{LINK_LABEL}</EuiFlexItem>
+          </EuiFlexGroup>
+        </EuiButtonBasicOrEmpty>
+      </EuiToolTip>
+    </InPortal>
+  );
+};

x-pack/platform/packages/shared/kbn-elastic-assistant/impl/assistant_context/index.tsx

@@ -14,7 +14,7 @@ import useLocalStorage from 'react-use/lib/useLocalStorage';
 import useSessionStorage from 'react-use/lib/useSessionStorage';
 import type { DocLinksStart } from '@kbn/core-doc-links-browser';
 import { AssistantFeatures, defaultAssistantFeatures } from '@kbn/elastic-assistant-common';
-import { NavigateToAppOptions, UserProfileService } from '@kbn/core/public';
+import { ChromeStart, NavigateToAppOptions, UserProfileService } from '@kbn/core/public';
 import { useQuery } from '@tanstack/react-query';
 import { updatePromptContexts } from './helpers';
 import type {
@@ -43,6 +43,7 @@ import {
 import { useCapabilities } from '../assistant/api/capabilities/use_capabilities';
 import { WELCOME_CONVERSATION_TITLE } from '../assistant/use_conversation/translations';
 import { SettingsTabs } from '../assistant/settings/types';
+import { AssistantNavLink } from './assistant_nav_link';
 
 export interface ShowAssistantOverlayProps {
   showOverlay: boolean;
@@ -77,6 +78,7 @@ export interface AssistantProviderProps {
   toasts?: IToasts;
   currentAppId: string;
   userProfileService: UserProfileService;
+  chrome: ChromeStart;
 }
 
 export interface UserAvatar {
@@ -128,6 +130,7 @@ export interface UseAssistantContext {
   currentAppId: string;
   codeBlockRef: React.MutableRefObject<(codeBlock: string) => void>;
   userProfileService: UserProfileService;
+  chrome: ChromeStart;
 }
 
 const AssistantContext = React.createContext<UseAssistantContext | undefined>(undefined);
@@ -151,6 +154,7 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
   toasts,
   currentAppId,
   userProfileService,
+  chrome,
 }) => {
   /**
    * Session storage for traceOptions, including APM URL and LangSmith Project/API Key
@@ -303,6 +307,7 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
       currentAppId,
       codeBlockRef,
       userProfileService,
+      chrome,
     }),
     [
       actionTypeRegistry,
@@ -338,10 +343,16 @@ export const AssistantProvider: React.FC<AssistantProviderProps> = ({
       currentAppId,
       codeBlockRef,
       userProfileService,
+      chrome,
     ]
   );
 
-  return <AssistantContext.Provider value={value}>{children}</AssistantContext.Provider>;
+  return (
+    <AssistantContext.Provider value={value}>
+      <AssistantNavLink />
+      {children}
+    </AssistantContext.Provider>
+  );
 };
 
 export const useAssistantContext = () => {