feat(discover): register page context for AI chatbot in classic Discover page

Signed-off-by: tq0905 <1352711780@qq.com>

Author: tq0905

src/plugins/discover/opensearch_dashboards.json

@@ -16,7 +16,7 @@
     "visualizations",
     "usageCollection"
   ],
-  "optionalPlugins": ["home", "share", "explore"],
+  "optionalPlugins": ["home", "share", "explore", "contextProvider"],
   "requiredBundles": [
     "home",
     "opensearchDashboardsUtils",

src/plugins/discover/public/application/view_components/context/index.tsx

@@ -14,6 +14,10 @@ import { useSearch, SearchContextValue } from '../utils/use_search';
 
 const SearchContext = React.createContext<SearchContextValue>({} as SearchContextValue);
 
+// NOOP fallback used when the contextProvider plugin is not available
+// (e.g. AI features disabled). Keeps the hook call unconditional.
+const NOOP_PAGE_CONTEXT_HOOK = (_options?: any): string => '';
+
 // eslint-disable-next-line import/no-default-export
 export default function DiscoverContext({ children }: React.PropsWithChildren<ViewProps>) {
   const { services: deServices } = useOpenSearchDashboards<DataExplorerServices>();
@@ -23,6 +27,24 @@ export default function DiscoverContext({ children }: React.PropsWithChildren<Vi
     ...services,
   });
 
+  // Register page context so the AI chatbot is aware of the current query,
+  // language, dataset and time range in classic Discover. Classic Discover
+  // stores the query one level deeper under `_q.query` (vs `_q` in Explore).
+  const usePageContext = services.contextProvider?.hooks?.usePageContext || NOOP_PAGE_CONTEXT_HOOK;
+  usePageContext({
+    description: 'Discover application page context',
+    convert: (urlState: any) => ({
+      appId: 'discover',
+      timeRange: urlState?._g?.time,
+      query: {
+        query: urlState?._q?.query?.query || '',
+        language: urlState?._q?.query?.language || 'kuery',
+      },
+      dataset: urlState?._q?.query?.dataset,
+    }),
+    categories: ['page', 'static'],
+  });
+
   return (
     <OpenSearchDashboardsContextProvider services={services}>
       <SearchContext.Provider value={searchParams}>{children}</SearchContext.Provider>

src/plugins/discover/public/build_services.ts

@@ -60,6 +60,7 @@ import { UrlForwardingStart } from '../../url_forwarding/public';
 import { NavigationPublicPluginStart } from '../../navigation/public';
 import { DataExplorerServices } from '../../data_explorer/public';
 import { Storage } from '../../opensearch_dashboards_utils/public';
+import { ContextProviderStart } from '../../context_provider/public';
 
 export interface DiscoverServices {
   addBasePath: (path: string) => string;
@@ -86,6 +87,7 @@ export interface DiscoverServices {
   visualizations: VisualizationsStart;
   storage: Storage;
   uiActions: UiActionsStart;
+  contextProvider?: ContextProviderStart;
 }
 
 export function buildServices(
@@ -130,6 +132,7 @@ export function buildServices(
     visualizations: plugins.visualizations,
     storage,
     uiActions: plugins.uiActions,
+    contextProvider: plugins.contextProvider,
   };
 }
 

src/plugins/discover/public/plugin.ts

@@ -79,6 +79,7 @@ declare module '../../share/public' {
 }
 import { UsageCollectionSetup } from '../../usage_collection/public';
 import { ExplorePluginSetup } from '../../explore/public';
+import { ContextProviderStart } from '../../context_provider/public';
 
 /**
  * @public
@@ -150,6 +151,7 @@ export interface DiscoverStartPlugins {
   urlForwarding: UrlForwardingStart;
   inspector: InspectorPublicPluginStart;
   visualizations: VisualizationsStart;
+  contextProvider?: ContextProviderStart;
 }
 
 /**