[Lens] visualize in maps button

x-pack/plugins/lens/public/indexpattern_datasource/datapanel.tsx

@@ -41,6 +41,8 @@ import { fieldExists } from './pure_helpers';
 import { Loader } from '../loader';
 import { esQuery, IIndexPattern } from '../../../../../src/plugins/data/public';
 import { IndexPatternFieldEditorStart } from '../../../../../src/plugins/index_pattern_field_editor/public';
+import { getUiActions } from '../kibana_services';
+import { VISUALIZE_GEO_FIELD_TRIGGER } from '../../../../../src/plugins/ui_actions/public';
 
 export type Props = Omit<DatasourceDataPanelProps<IndexPatternPrivateState>, 'core'> & {
   data: DataPublicPluginStart;
@@ -85,6 +87,18 @@ const fieldTypeNames: Record<DataType, string> = {
   histogram: i18n.translate('xpack.lens.datatypes.histogram', { defaultMessage: 'histogram' }),
 };
 
+const visualizeGeoFieldTrigger = getUiActions().getTrigger(VISUALIZE_GEO_FIELD_TRIGGER);
+if (visualizeGeoFieldTrigger) {
+  fieldTypeNames.geo_point = i18n.translate('xpack.lens.datatypes.geoPoint', {
+    defaultMessage: 'geo_point',
+  });
+  fieldTypeNames.geo_shape = i18n.translate('xpack.lens.datatypes.geoShape', {
+    defaultMessage: 'geo_shape',
+  });
+  supportedFieldTypes.add('geo_point');
+  supportedFieldTypes.add('geo_shape');
+}
+
 // Wrapper around esQuery.buildEsQuery, handling errors (e.g. because a query can't be parsed) by
 // returning a query dsl object not matching anything
 function buildSafeEsQuery(

x-pack/plugins/lens/public/indexpattern_datasource/field_item.tsx

@@ -54,6 +54,7 @@ import { BucketedAggregation, FieldStatsResponse } from '../../common';
 import { IndexPattern, IndexPatternField, DraggedField } from './types';
 import { LensFieldIcon } from './lens_field_icon';
 import { trackUiEvent } from '../lens_ui_telemetry';
+import { VisualizeGeoFieldButton } from './visualize_geo_field_button';
 
 import { debouncedComponent } from '../debounced_component';
 
@@ -149,7 +150,13 @@ export const InnerFieldItem = function InnerFieldItem(props: FieldItemProps) {
 
   function fetchData() {
     // Range types don't have any useful stats we can show
-    if (state.isLoading || field.type === 'document' || field.type.includes('range')) {
+    if (
+      state.isLoading ||
+      field.type === 'document' ||
+      field.type.includes('range') ||
+      field.type === 'geo_point' ||
+      field.type === 'geo_shape'
+    ) {
       return;
     }
 
@@ -467,6 +474,21 @@ function FieldItemPopoverContents(props: State & FieldItemProps) {
         </EuiText>
       </>
     );
+  } else if (field.type === 'geo_point' || field.type === 'geo_shape') {
+    return (
+      <>
+        <EuiPopoverTitle>{panelHeader}</EuiPopoverTitle>
+
+        <EuiText size="s">
+          {i18n.translate('xpack.lens.indexPattern.fieldStatsLimited', {
+            defaultMessage: `Lens cannot visualize {fieldType} fields.`,
+            values: { fieldType: field.type },
+          })}
+        </EuiText>
+
+        <VisualizeGeoFieldButton indexPatternId={indexPattern.id} fieldName={field.name} />
+      </>
+    );
   } else if (
     (!props.histogram || props.histogram.buckets.length === 0) &&
     (!props.topValues || props.topValues.buckets.length === 0)

x-pack/plugins/lens/public/indexpattern_datasource/visualize_geo_field_button.tsx

@@ -0,0 +1,93 @@
+/*
+ * 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, { Component, MouseEvent } from 'react';
+import { EuiButton } from '@elastic/eui';
+import { FormattedMessage } from '@kbn/i18n/react';
+import { getUiActions } from '../kibana_services';
+import {
+  visualizeGeoFieldTrigger,
+  VISUALIZE_GEO_FIELD_TRIGGER,
+} from '../../../../../src/plugins/ui_actions/public';
+
+interface Props {
+  indexPatternId: string;
+  fieldName: string;
+}
+
+interface State {
+  hasLoadedHref: boolean;
+  href: string | null;
+}
+
+export class VisualizeGeoFieldButton extends Component<Props, State> {
+  private _isMounted = false;
+  state: State = {
+    hasLoadedHref: false,
+    href: null,
+  };
+
+  componentWillUnmount() {
+    this._isMounted = false;
+  }
+
+  componentDidMount() {
+    this._isMounted = true;
+    this.loadHref();
+  }
+
+  async loadHref() {
+    const actions = await getUiActions().getTriggerCompatibleActions(VISUALIZE_GEO_FIELD_TRIGGER, {
+      indexPatternId: this.props.indexPatternId,
+      fieldName: this.props.fieldName,
+    });
+    if (!this._isMounted) {
+      return;
+    }
+
+    const href = actions.length
+      ? await actions[0].getHref?.({
+          indexPatternId: this.props.indexPatternId,
+          fieldName: this.props.fieldName,
+          trigger: visualizeGeoFieldTrigger,
+        })
+      : null;
+
+    this.setState({
+      hasLoadedHref: true,
+      href,
+    });
+  }
+
+  onClick = (event: MouseEvent<HTMLAnchorElement, MouseEvent>) => {
+    getUiActions().getTrigger(visualizeGeoFieldTrigger).exec({
+      indexPatternId: this.props.indexPatternId,
+      fieldName: this.props.fieldName,
+    });
+  };
+
+  render() {
+    return (
+      <>
+        {/* eslint-disable-next-line @elastic/eui/href-or-on-click */}
+        <EuiButton
+          onClick={this.onClick}
+          href={this.state.href}
+          size="s"
+          isLoading={!this.state.hasLoadedHref}
+          isDisabled={this.state.href === null}
+          data-test-subj={`lensGeoFieldVisualize-${this.props.fieldName}`}
+        >
+          <FormattedMessage
+            id="xpack.lens.indexPattern.fieldItem.visualizeGeoFieldLinkText"
+            defaultMessage="Visualize in Maps"
+          />
+        </EuiButton>
+      </>
+    );
+  }
+}

x-pack/plugins/lens/public/kibana_services.ts

@@ -0,0 +1,18 @@
+/*
+ * 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 { CoreStart } from 'kibana/public';
+import type { LensPluginStartDependencies } from './plugin';
+
+let coreStart: CoreStart;
+let pluginsStart: LensPluginStartDependencies;
+export function setStartServices(core: CoreStart, plugins: LensPluginStartDependencies) {
+  coreStart = core;
+  pluginsStart = plugins;
+}
+
+export const getUiActions = () => pluginsStart.uiActions;

x-pack/plugins/lens/public/plugin.ts

@@ -54,6 +54,7 @@ import {
   EmbeddableComponentProps,
   getEmbeddableComponent,
 } from './editor_frame_service/embeddable/embeddable_component';
+import { setStartServices } from './kibana_services';
 
 export interface LensPluginSetupDependencies {
   urlForwarding: UrlForwardingSetup;
@@ -234,6 +235,7 @@ export class LensPlugin {
   }
 
   start(core: CoreStart, startDependencies: LensPluginStartDependencies): LensPublicStart {
+    setStartServices(core, startDependencies);
     const frameStart = this.editorFrameService.start(core, startDependencies);
     this.createEditorFrame = frameStart.createInstance;
     // unregisters the Visualize action and registers the lens one