Shim ui/autocomplete_providers into legacy data plugin (#46138)

* Shim ui/autocomplete_providers into legacy data plugin

Fix: #44949

* Fix PR commnets

* fix CI

* Add comment

src/legacy/core_plugins/data/public/query/query_bar/components/query_bar_input.test.mocks.ts

@@ -41,10 +41,6 @@ export const mockPersistedLogFactory = jest.fn<jest.Mocked<typeof mockPersistedL
   return mockPersistedLog;
 });
 
-export const mockGetAutocompleteSuggestions = jest.fn(() => Promise.resolve([]));
-const mockAutocompleteProvider = jest.fn(() => mockGetAutocompleteSuggestions);
-export const mockGetAutocompleteProvider = jest.fn(() => mockAutocompleteProvider);
-
 export const mockFetchIndexPatterns = jest
   .fn()
   .mockReturnValue(Promise.resolve([mockIndexPattern]));

src/legacy/core_plugins/data/public/query/query_bar/components/query_bar_input.tsx

@@ -23,19 +23,19 @@ import React from 'react';
 import { EuiFieldText, EuiOutsideClickDetector, PopoverAnchorPosition } from '@elastic/eui';
 
 import { InjectedIntl, injectI18n } from '@kbn/i18n/react';
-import {
-  AutocompleteSuggestion,
-  AutocompleteSuggestionType,
-  getAutocompleteProvider,
-} from 'ui/autocomplete_providers';
 import { debounce, compact, isEqual, omit } from 'lodash';
 import { PersistedLog } from 'ui/persisted_log';
 import { Storage } from 'ui/storage';
+import { npStart } from 'ui/new_platform';
 import {
   UiSettingsClientContract,
   SavedObjectsClientContract,
   HttpServiceBase,
 } from 'src/core/public';
+import {
+  AutocompleteSuggestion,
+  AutocompleteSuggestionType,
+} from '../../../../../../../plugins/data/public';
 import { IndexPattern, StaticIndexPattern } from '../../../index_patterns';
 import { Query } from '../index';
 import { fromUser, matchPairs, toUser } from '../lib';
@@ -44,6 +44,11 @@ import { SuggestionsComponent } from './typeahead/suggestions_component';
 import { getQueryLog } from '../lib/get_query_log';
 import { fetchIndexPatterns } from '../lib/fetch_index_patterns';
 
+// todo: related to https://github.com/elastic/kibana/pull/45762/files
+// Will be refactored after merge of related PR
+const getAutocompleteProvider = (language: string) =>
+  npStart.plugins.data.autocomplete.getProvider(language);
+
 interface Props {
   uiSettings: UiSettingsClientContract;
   indexPatterns: Array<IndexPattern | string>;

src/legacy/core_plugins/data/public/query/query_bar/components/typeahead/suggestion_component.test.tsx

@@ -19,7 +19,7 @@
 
 import { mount, shallow } from 'enzyme';
 import React from 'react';
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../../plugins/data/public';
 import { SuggestionComponent } from './suggestion_component';
 
 const noop = () => {

src/legacy/core_plugins/data/public/query/query_bar/components/typeahead/suggestion_component.tsx

@@ -20,7 +20,7 @@
 import { EuiIcon } from '@elastic/eui';
 import classNames from 'classnames';
 import React, { FunctionComponent } from 'react';
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../../plugins/data/public';
 
 function getEuiIconType(type: string) {
   switch (type) {

src/legacy/core_plugins/data/public/query/query_bar/components/typeahead/suggestions_component.test.tsx

@@ -19,7 +19,7 @@
 
 import { mount, shallow } from 'enzyme';
 import React from 'react';
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../../plugins/data/public';
 import { SuggestionComponent } from './suggestion_component';
 import { SuggestionsComponent } from './suggestions_component';
 

src/legacy/core_plugins/data/public/query/query_bar/components/typeahead/suggestions_component.tsx

@@ -19,7 +19,7 @@
 
 import { isEmpty } from 'lodash';
 import React, { Component } from 'react';
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../../plugins/data/public';
 import { SuggestionComponent } from './suggestion_component';
 
 interface Props {

src/legacy/core_plugins/kibana/public/kibana.js

@@ -43,7 +43,6 @@ import 'uiExports/embeddableFactories';
 import 'uiExports/embeddableActions';
 import 'uiExports/inspectorViews';
 import 'uiExports/search';
-import 'uiExports/autocompleteProviders';
 import 'uiExports/shareContextMenuExtensions';
 
 import 'ui/autoload/all';

src/legacy/core_plugins/kibana/public/visualize/embeddable/visualize_embeddable_factory.tsx

@@ -18,7 +18,6 @@
  */
 
 import 'ui/registry/field_formats';
-import 'uiExports/autocompleteProviders';
 import 'uiExports/contextMenuActions';
 import 'uiExports/devTools';
 import 'uiExports/docViews';

src/legacy/ui/ui_exports/ui_export_types/index.js

@@ -63,7 +63,6 @@ export {
   aliases,
   visualize,
   search,
-  autocompleteProviders,
   shareContextMenuExtensions,
 } from './ui_app_extensions';
 

src/legacy/ui/ui_exports/ui_export_types/ui_app_extensions.js

@@ -36,7 +36,6 @@ const appExtension = wrap(
 export const visTypes = appExtension;
 
 export const visEditorTypes = appExtension;
-export const autocompleteProviders = appExtension;
 export const savedObjectTypes = appExtension;
 export const embeddableActions = appExtension;
 export const embeddableFactories = appExtension;

src/plugins/data/common/index.ts

@@ -17,6 +17,5 @@
  * under the License.
  */
 
-export * from './timefilter/types';
-export * from './query/types';
+export * from './query';
 export * from './field_formats';

src/plugins/data/common/types.ts

@@ -17,4 +17,6 @@
  * under the License.
  */
 
-export * from '.';
+export * from './field_formats/types';
+export * from './timefilter/types';
+export * from './query/types';

src/legacy/ui/public/autocomplete_providers/index.js → src/plugins/data/public/autocomplete_provider/index.ts

@@ -16,13 +16,25 @@
  * specific language governing permissions and limitations
  * under the License.
  */
+import { AutocompleteProvider } from './types';
 
-const autocompleteProviders = new Map();
+export class AutocompleteProviderRegister {
+  private readonly registeredProviders: Map<string, AutocompleteProvider> = new Map();
 
-export function addAutocompleteProvider(language, provider) {
-  autocompleteProviders.set(language, provider);
-}
+  /** @public **/
+  public addProvider(language: string, provider: AutocompleteProvider): void {
+    if (language && provider) {
+      this.registeredProviders.set(language, provider);
+    }
+  }
+
+  /** @public **/
+  public getProvider(language: string): AutocompleteProvider | undefined {
+    return this.registeredProviders.get(language);
+  }
 
-export function getAutocompleteProvider(language) {
-  return autocompleteProviders.get(language);
+  /** @internal **/
+  public clearProviders(): void {
+    this.registeredProviders.clear();
+  }
 }

src/legacy/ui/public/autocomplete_providers/index.d.ts → src/plugins/data/public/autocomplete_provider/types.ts

@@ -17,11 +17,9 @@
  * under the License.
  */
 
-/**
- * WARNING: these typings are incomplete
- */
 import { StaticIndexPattern } from 'ui/index_patterns';
 
+/** @public **/
 export type AutocompleteProvider = (args: {
   config: {
     get(configKey: string): any;
@@ -30,27 +28,26 @@ export type AutocompleteProvider = (args: {
   boolFilter?: any;
 }) => GetSuggestions;
 
+/** @public **/
 export type GetSuggestions = (args: {
   query: string;
   selectionStart: number;
   selectionEnd: number;
 }) => Promise<AutocompleteSuggestion[]>;
 
+/** @public **/
 export type AutocompleteSuggestionType =
   | 'field'
   | 'value'
   | 'operator'
   | 'conjunction'
   | 'recentSearch';
 
+/** @public **/
 export interface AutocompleteSuggestion {
   description?: string;
   end: number;
   start: number;
   text: string;
   type: AutocompleteSuggestionType;
 }
-
-export function addAutocompleteProvider(language: string, provider: AutocompleteProvider): void;
-
-export function getAutocompleteProvider(language: string): AutocompleteProvider | undefined;

src/plugins/data/public/index.ts

@@ -20,12 +20,13 @@
 import { PluginInitializerContext } from '../../../core/public';
 import { DataPublicPlugin } from './plugin';
 
-export { RefreshInterval, TimeRange, Query } from './types';
-
 export function plugin(initializerContext: PluginInitializerContext) {
   return new DataPublicPlugin(initializerContext);
 }
 
 export { DataPublicPlugin as Plugin };
 
 export * from '../common';
+
+export * from './autocomplete_provider';
+export * from './types';

src/plugins/data/public/mocks.ts

@@ -21,13 +21,24 @@ import { Plugin } from '.';
 export type Setup = jest.Mocked<ReturnType<Plugin['setup']>>;
 export type Start = jest.Mocked<ReturnType<Plugin['start']>>;
 
+const autocompleteMock: any = {
+  addProvider: jest.fn(),
+  getProvider: jest.fn(),
+  clearProviders: jest.fn(),
+};
+
 const createSetupContract = (): Setup => {
-  const setupContract: Setup = {};
+  const setupContract: Setup = {
+    autocomplete: autocompleteMock as Setup['autocomplete'],
+  };
+
   return setupContract;
 };
 
 const createStartContract = (): Start => {
-  const startContract: Start = undefined;
+  const startContract: Start = {
+    autocomplete: autocompleteMock as Start['autocomplete'],
+  };
   return startContract;
 };
 

src/plugins/data/public/plugin.ts

@@ -18,14 +18,34 @@
  */
 
 import { PluginInitializerContext, CoreSetup, CoreStart, Plugin } from '../../../core/public';
+import { AutocompleteProviderRegister } from './autocomplete_provider';
+
+export interface DataPublicPluginSetup {
+  autocomplete: Pick<AutocompleteProviderRegister, 'addProvider' | 'getProvider'>;
+}
+
+export interface DataPublicPluginStart {
+  autocomplete: Pick<AutocompleteProviderRegister, 'getProvider'>;
+}
+
+export class DataPublicPlugin implements Plugin<DataPublicPluginSetup, DataPublicPluginStart> {
+  private readonly autocomplete = new AutocompleteProviderRegister();
 
-export class DataPublicPlugin implements Plugin<{}> {
   constructor(initializerContext: PluginInitializerContext) {}
 
-  public setup(core: CoreSetup) {
-    return {};
+  public setup(core: CoreSetup): DataPublicPluginSetup {
+    return {
+      autocomplete: this.autocomplete,
+    };
+  }
+
+  public start(core: CoreStart): DataPublicPluginStart {
+    return {
+      autocomplete: this.autocomplete,
+    };
   }
 
-  public start(core: CoreStart) {}
-  public stop() {}
+  public stop() {
+    this.autocomplete.clearProviders();
+  }
 }

src/plugins/data/public/types.ts

@@ -18,3 +18,4 @@
  */
 
 export * from '../common/types';
+export * from './autocomplete_provider/types';

x-pack/index.js

@@ -26,7 +26,7 @@ import { indexManagement } from './legacy/plugins/index_management';
 import { indexLifecycleManagement } from './legacy/plugins/index_lifecycle_management';
 import { consoleExtensions } from './legacy/plugins/console_extensions';
 import { spaces } from './legacy/plugins/spaces';
-import { kueryAutocomplete } from './legacy/plugins/kuery_autocomplete';
+import { kueryAutocompleteInitializer } from './legacy/plugins/kuery_autocomplete';
 import { canvas } from './legacy/plugins/canvas';
 import { infra } from './legacy/plugins/infra';
 import { taskManager } from './legacy/plugins/task_manager';
@@ -72,7 +72,7 @@ module.exports = function (kibana) {
     indexManagement(kibana),
     consoleExtensions(kibana),
     indexLifecycleManagement(kibana),
-    kueryAutocomplete(kibana),
+    kueryAutocompleteInitializer(kibana),
     infra(kibana),
     taskManager(kibana),
     rollup(kibana),

x-pack/legacy/plugins/apm/public/components/shared/KueryBar/index.tsx

@@ -10,10 +10,7 @@ import { EuiCallOut } from '@elastic/eui';
 import styled from 'styled-components';
 import { FormattedMessage } from '@kbn/i18n/react';
 import { i18n } from '@kbn/i18n';
-import {
-  AutocompleteSuggestion,
-  getAutocompleteProvider
-} from 'ui/autocomplete_providers';
+import { npStart } from 'ui/new_platform';
 import { StaticIndexPattern, getFromSavedObject } from 'ui/index_patterns';
 import { fromKueryExpression, toElasticsearchQuery } from '@kbn/es-query';
 import { fromQuery, toQuery } from '../Links/url_helpers';
@@ -29,11 +26,15 @@ import { useMatchedRoutes } from '../../../hooks/useMatchedRoutes';
 import { RouteName } from '../../app/Main/route_config/route_names';
 import { useKibanaCore } from '../../../../../observability/public';
 import { getAPMIndexPattern } from '../../../services/rest/savedObjects';
+import { AutocompleteSuggestion } from '../../../../../../../../src/plugins/data/public';
 
 const Container = styled.div`
   margin-bottom: 10px;
 `;
 
+const getAutocompleteProvider = (language: string) =>
+  npStart.plugins.data.autocomplete.getProvider(language);
+
 interface State {
   indexPattern: StaticIndexPattern | null;
   suggestions: AutocompleteSuggestion[];

x-pack/legacy/plugins/apm/public/index.tsx

@@ -13,7 +13,6 @@ import 'ui/autoload/styles';
 import chrome from 'ui/chrome';
 // @ts-ignore
 import { uiModules } from 'ui/modules';
-import 'uiExports/autocompleteProviders';
 import { GlobalHelpExtension } from './components/app/GlobalHelpExtension';
 import { plugin } from './new-platform';
 import { REACT_APP_ROOT_ID } from './new-platform/plugin';

x-pack/legacy/plugins/beats_management/public/components/autocomplete_field/index.tsx

@@ -13,8 +13,7 @@ import {
 import React from 'react';
 import styled from 'styled-components';
 
-// @ts-ignore
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../src/plugins/data/public';
 
 import { composeStateUpdaters } from '../../utils/typed_react';
 import { SuggestionItem } from './suggestion_item';

x-pack/legacy/plugins/beats_management/public/components/autocomplete_field/suggestion_item.tsx

@@ -8,8 +8,8 @@ import { EuiIcon } from '@elastic/eui';
 import { tint } from 'polished';
 import React from 'react';
 import styled from 'styled-components';
-// @ts-ignore
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+
+import { AutocompleteSuggestion } from '../../../../../../../src/plugins/data/public';
 
 interface SuggestionItemProps {
   isSelected?: boolean;

x-pack/legacy/plugins/beats_management/public/components/table/table.tsx

@@ -8,7 +8,7 @@ import { EuiBasicTable, EuiFlexGroup, EuiFlexItem, EuiSpacer } from '@elastic/eu
 import { i18n } from '@kbn/i18n';
 import React from 'react';
 import styled from 'styled-components';
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../src/plugins/data/public';
 import { TABLE_CONFIG } from '../../../common/constants';
 import { AutocompleteField } from '../autocomplete_field/index';
 import { ControlSchema } from './action_schema';

x-pack/legacy/plugins/beats_management/public/containers/with_kuery_autocompletion.tsx

@@ -6,7 +6,7 @@
 
 import React from 'react';
 
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../src/plugins/data/public';
 
 import { FrontendLibs } from '../lib/types';
 import { RendererFunction } from '../utils/typed_react';

x-pack/legacy/plugins/beats_management/public/lib/adapters/elasticsearch/adapter_types.ts

@@ -3,7 +3,7 @@
  * or more contributor license agreements. Licensed under the Elastic License;
  * you may not use this file except in compliance with the Elastic License.
  */
-import { AutocompleteSuggestion } from 'ui/autocomplete_providers';
+import { AutocompleteSuggestion } from '../../../../../../../../src/plugins/data/public';
 
 export interface ElasticsearchAdapter {
   convertKueryToEsQuery: (kuery: string) => Promise<string>;