[8.19] [Discover] Support setting columns in Discover session embeddable (#222858)

src/platform/packages/shared/kbn-saved-search-component/src/components/saved_search.tsx

@@ -16,6 +16,7 @@ import type {
 } from '@kbn/discover-plugin/public';
 import { SerializedPanelState } from '@kbn/presentation-publishing';
 import { css } from '@emotion/react';
+import { SavedSearchAttributes } from '@kbn/saved-search-plugin/common';
 import { SavedSearchComponentProps } from '../types';
 import { SavedSearchComponentErrorContent } from './error';
 
@@ -36,6 +37,7 @@ export const SavedSearchComponent: React.FC<SavedSearchComponentProps> = (props)
     filters,
     index,
     timestampField,
+    columns,
     height,
   } = props;
 
@@ -65,10 +67,11 @@ export const SavedSearchComponent: React.FC<SavedSearchComponentProps> = (props)
           searchSource.setField('filter', filters);
           const { searchSourceJSON, references } = searchSource.serialize();
           // By-value saved object structure
-          const attributes = {
+          const attributes: Partial<SavedSearchAttributes> = {
             kibanaSavedObjectMeta: {
               searchSourceJSON,
             },
+            columns,
           };
           setInitialSerializedState({
             rawState: {
@@ -94,6 +97,7 @@ export const SavedSearchComponent: React.FC<SavedSearchComponentProps> = (props)
       abortController.abort();
     };
   }, [
+    columns,
     dataViews,
     documentViewerEnabled,
     filters,
@@ -137,6 +141,7 @@ const SavedSearchComponentTable: React.FC<
     timeRange,
     timestampField,
     index,
+    columns,
   } = props;
   const embeddableApi = useRef<SearchEmbeddableApi | undefined>(undefined);
 
@@ -198,6 +203,14 @@ const SavedSearchComponentTable: React.FC<
     [timeRange]
   );
 
+  useEffect(
+    function syncColumns() {
+      if (!embeddableApi.current) return;
+      embeddableApi.current.setColumns(columns);
+    },
+    [columns]
+  );
+
   return (
     <EmbeddableRenderer<SearchEmbeddableSerializedState, SearchEmbeddableApi>
       maybeId={undefined}

src/platform/packages/shared/kbn-saved-search-component/src/types.ts

@@ -26,6 +26,7 @@ export interface SavedSearchComponentProps {
   query?: Query;
   filters?: Filter[];
   timestampField?: string;
+  columns?: string[];
   height?: CSSProperties['height'];
   displayOptions?: NonPersistedDisplayOptions;
 }

src/platform/packages/shared/kbn-saved-search-component/tsconfig.json

@@ -25,5 +25,6 @@
     "@kbn/discover-plugin",
     "@kbn/i18n",
     "@kbn/presentation-publishing",
+    "@kbn/saved-search-plugin",
   ]
 }

src/platform/plugins/shared/discover/public/embeddable/initialize_search_embeddable_api.tsx

@@ -29,7 +29,7 @@ import type { DiscoverServices } from '../build_services';
 import { EDITABLE_SAVED_SEARCH_KEYS } from './constants';
 import { getSearchEmbeddableDefaults } from './get_search_embeddable_defaults';
 import type {
-  PublishesSavedSearch,
+  PublishesWritableSavedSearch,
   SearchEmbeddableRuntimeState,
   SearchEmbeddableSerializedAttributes,
   SearchEmbeddableSerializedState,
@@ -73,7 +73,9 @@ export const initializeSearchEmbeddableApi = async (
     discoverServices: DiscoverServices;
   }
 ): Promise<{
-  api: PublishesSavedSearch & PublishesWritableDataViews & Partial<PublishesWritableUnifiedSearch>;
+  api: PublishesWritableSavedSearch &
+    PublishesWritableDataViews &
+    Partial<PublishesWritableUnifiedSearch>;
   stateManager: SearchEmbeddableStateManager;
   anyStateChange$: Observable<void>;
   comparators: StateComparators<SearchEmbeddableSerializedAttributes>;
@@ -168,6 +170,10 @@ export const initializeSearchEmbeddableApi = async (
     searchSource$.next(searchSource);
   };
 
+  const setColumns = (columns: string[] | undefined) => {
+    stateManager.columns.next(columns);
+  };
+
   /** Keep the saved search in sync with any state changes */
   const syncSavedSearch = combineLatest([onAnyStateChange, searchSource$])
     .pipe(
@@ -194,6 +200,7 @@ export const initializeSearchEmbeddableApi = async (
       setFilters,
       query$,
       setQuery,
+      setColumns,
     },
     stateManager,
     anyStateChange$: onAnyStateChange.pipe(map(() => undefined)),

src/platform/plugins/shared/discover/public/embeddable/types.ts

@@ -105,7 +105,7 @@ export type SearchEmbeddableApi = DefaultEmbeddableApi<SearchEmbeddableSerialize
   PublishesBlockingError &
   Required<PublishesWritableTitle> &
   Required<PublishesDescription> &
-  PublishesSavedSearch &
+  PublishesWritableSavedSearch &
   PublishesWritableDataViews &
   PublishesWritableUnifiedSearch &
   HasLibraryTransforms &
@@ -119,6 +119,10 @@ export interface PublishesSavedSearch {
   savedSearch$: PublishingSubject<SavedSearch>;
 }
 
+export interface PublishesWritableSavedSearch extends PublishesSavedSearch {
+  setColumns: (columns: string[] | undefined) => void;
+}
+
 export const apiPublishesSavedSearch = (
   api: EmbeddableApiContext['embeddable']
 ): api is PublishesSavedSearch => {

src/platform/plugins/shared/discover/public/index.ts

@@ -33,6 +33,7 @@ export {
   SEARCH_EMBEDDABLE_CELL_ACTIONS_TRIGGER_ID,
   apiPublishesSavedSearch,
   type PublishesSavedSearch,
+  type PublishesWritableSavedSearch,
   type HasTimeRange,
   type SearchEmbeddableSerializedState,
   type SearchEmbeddableRuntimeState,