[Saved Searches] Add support for saved searches by value

src/plugins/discover/common/embeddable/index.ts

@@ -0,0 +1,9 @@
+/*
+ * 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 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+export { inject, extract } from './search_inject_extract';

src/plugins/discover/common/embeddable/search_inject_extract.test.ts

@@ -0,0 +1,76 @@
+/*
+ * 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 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import { extract, inject } from './search_inject_extract';
+
+describe('search inject extract', () => {
+  describe('inject', () => {
+    it('should not inject references if state does not have attributes', () => {
+      const state = { type: 'type', id: 'id' };
+      const injectedReferences = [{ name: 'name', type: 'type', id: 'id' }];
+      expect(inject(state, injectedReferences)).toEqual(state);
+    });
+
+    it('should inject references if state has references with the same name', () => {
+      const state = {
+        type: 'type',
+        id: 'id',
+        attributes: {
+          references: [{ name: 'name', type: 'type', id: '1' }],
+        },
+      };
+      const injectedReferences = [{ name: 'name', type: 'type', id: '2' }];
+      expect(inject(state, injectedReferences)).toEqual({
+        ...state,
+        attributes: {
+          ...state.attributes,
+          references: injectedReferences,
+        },
+      });
+    });
+
+    it('should clear references if state has no references with the same name', () => {
+      const state = {
+        type: 'type',
+        id: 'id',
+        attributes: {
+          references: [{ name: 'name', type: 'type', id: '1' }],
+        },
+      };
+      const injectedReferences = [{ name: 'other', type: 'type', id: '2' }];
+      expect(inject(state, injectedReferences)).toEqual({
+        ...state,
+        attributes: {
+          ...state.attributes,
+          references: [],
+        },
+      });
+    });
+  });
+
+  describe('extract', () => {
+    it('should not extract references if state does not have attributes', () => {
+      const state = { type: 'type', id: 'id' };
+      expect(extract(state)).toEqual({ state, references: [] });
+    });
+
+    it('should extract references if state has references', () => {
+      const state = {
+        type: 'type',
+        id: 'id',
+        attributes: {
+          references: [{ name: 'name', type: 'type', id: '1' }],
+        },
+      };
+      expect(extract(state)).toEqual({
+        state,
+        references: [{ name: 'name', type: 'type', id: '1' }],
+      });
+    });
+  });
+});

src/plugins/discover/common/embeddable/search_inject_extract.ts

@@ -0,0 +1,52 @@
+/*
+ * 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 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import type { SavedObjectReference } from '@kbn/core-saved-objects-server';
+import type { EmbeddableStateWithType } from '@kbn/embeddable-plugin/common';
+import type { SearchByValueInput } from '@kbn/saved-search-plugin/public';
+
+export const inject = (
+  state: EmbeddableStateWithType,
+  injectedReferences: SavedObjectReference[]
+): EmbeddableStateWithType => {
+  if (hasAttributes(state)) {
+    // Filter out references that are not in the state
+    // https://github.com/elastic/kibana/pull/119079
+    const references = state.attributes.references
+      .map((stateRef) =>
+        injectedReferences.find((injectedRef) => injectedRef.name === stateRef.name)
+      )
+      .filter(Boolean);
+
+    state = {
+      ...state,
+      attributes: {
+        ...state.attributes,
+        references,
+      },
+    } as EmbeddableStateWithType;
+  }
+
+  return state;
+};
+
+export const extract = (
+  state: EmbeddableStateWithType
+): { state: EmbeddableStateWithType; references: SavedObjectReference[] } => {
+  let references: SavedObjectReference[] = [];
+
+  if (hasAttributes(state)) {
+    references = state.attributes.references;
+  }
+
+  return { state, references };
+};
+
+const hasAttributes = (
+  state: EmbeddableStateWithType
+): state is EmbeddableStateWithType & SearchByValueInput => 'attributes' in state;

src/plugins/discover/public/embeddable/get_discover_locator_params.test.ts

@@ -0,0 +1,37 @@
+/*
+ * 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 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import { savedSearchMock } from '../__mocks__/saved_search';
+import { getDiscoverLocatorParams } from './get_discover_locator_params';
+import type { SearchInput } from './types';
+
+describe('getDiscoverLocatorParams', () => {
+  it('should return saved search id if input has savedObjectId', () => {
+    const input = { savedObjectId: 'savedObjectId' } as SearchInput;
+    expect(getDiscoverLocatorParams({ input, savedSearch: savedSearchMock })).toEqual({
+      savedSearchId: 'savedObjectId',
+    });
+  });
+
+  it('should return Discover params if input has no savedObjectId', () => {
+    const input = {} as SearchInput;
+    expect(getDiscoverLocatorParams({ input, savedSearch: savedSearchMock })).toEqual({
+      dataViewId: savedSearchMock.searchSource.getField('index')?.id,
+      dataViewSpec: savedSearchMock.searchSource.getField('index')?.toMinimalSpec(),
+      timeRange: savedSearchMock.timeRange,
+      refreshInterval: savedSearchMock.refreshInterval,
+      filters: savedSearchMock.searchSource.getField('filter'),
+      query: savedSearchMock.searchSource.getField('query'),
+      columns: savedSearchMock.columns,
+      sort: savedSearchMock.sort,
+      viewMode: savedSearchMock.viewMode,
+      hideAggregatedPreview: savedSearchMock.hideAggregatedPreview,
+      breakdownField: savedSearchMock.breakdownField,
+    });
+  });
+});

src/plugins/discover/public/embeddable/get_discover_locator_params.ts

@@ -0,0 +1,41 @@
+/*
+ * 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 and the Server Side Public License, v 1; you may not use this file except
+ * in compliance with, at your election, the Elastic License 2.0 or the Server
+ * Side Public License, v 1.
+ */
+
+import type { Filter } from '@kbn/es-query';
+import type { SavedSearch } from '@kbn/saved-search-plugin/common';
+import type { SearchByReferenceInput } from '@kbn/saved-search-plugin/public';
+import type { DiscoverAppLocatorParams } from '../../common';
+import type { SearchInput } from './types';
+
+export const getDiscoverLocatorParams = ({
+  input,
+  savedSearch,
+}: {
+  input: SearchInput;
+  savedSearch: SavedSearch;
+}) => {
+  const dataView = savedSearch.searchSource.getField('index');
+  const savedObjectId = (input as SearchByReferenceInput).savedObjectId;
+  const locatorParams: DiscoverAppLocatorParams = savedObjectId
+    ? { savedSearchId: savedObjectId }
+    : {
+        dataViewId: dataView?.id,
+        dataViewSpec: dataView?.toMinimalSpec(),
+        timeRange: savedSearch.timeRange,
+        refreshInterval: savedSearch.refreshInterval,
+        filters: savedSearch.searchSource.getField('filter') as Filter[],
+        query: savedSearch.searchSource.getField('query'),
+        columns: savedSearch.columns,
+        sort: savedSearch.sort,
+        viewMode: savedSearch.viewMode,
+        hideAggregatedPreview: savedSearch.hideAggregatedPreview,
+        breakdownField: savedSearch.breakdownField,
+      };
+
+  return locatorParams;
+};