[maps] Lens choropleth chart

x-pack/plugins/lens/public/editor_frame_service/editor_frame/suggestion_helpers.ts

@@ -149,7 +149,8 @@ export function getSuggestions({
             currentVisualizationState,
             subVisualizationId,
             palette,
-            visualizeTriggerFieldContext && 'isVisualizeAction' in visualizeTriggerFieldContext
+            visualizeTriggerFieldContext && 'isVisualizeAction' in visualizeTriggerFieldContext,
+            activeData
           );
         });
     })
@@ -207,7 +208,8 @@ function getVisualizationSuggestions(
   currentVisualizationState: unknown,
   subVisualizationId?: string,
   mainPalette?: PaletteOutput,
-  isFromContext?: boolean
+  isFromContext?: boolean,
+  activeData?: Record<string, Datatable>
 ) {
   return visualization
     .getSuggestions({
@@ -217,6 +219,7 @@ function getVisualizationSuggestions(
       subVisualizationId,
       mainPalette,
       isFromContext,
+      activeData,
     })
     .map(({ state, ...visualizationSuggestion }) => ({
       ...visualizationSuggestion,

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

@@ -12,7 +12,15 @@ export type {
   TypedLensByValueInput,
 } from './embeddable/embeddable_component';
 export type { XYState } from './xy_visualization/types';
-export type { DataType, OperationMetadata, Visualization } from './types';
+export type {
+  DatasourcePublicAPI,
+  DataType,
+  OperationMetadata,
+  SuggestionRequest,
+  TableSuggestion,
+  Visualization,
+  VisualizationSuggestion,
+} from './types';
 export type {
   AxesSettingsConfig,
   XYLayerConfig,

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

@@ -643,6 +643,7 @@ export interface SuggestionRequest<T = unknown> {
    * Different suggestions can be generated for each subtype of the visualization
    */
   subVisualizationId?: string;
+  activeData?: Record<string, Datatable>;
 }
 
 /**

x-pack/plugins/maps/common/descriptor_types/source_descriptor_types.ts

@@ -158,6 +158,7 @@ export type TiledSingleLayerVectorSourceDescriptor = AbstractSourceDescriptor &
 
 export type InlineFieldDescriptor = {
   name: string;
+  label?: string;
   type: 'string' | 'number';
 };
 

x-pack/plugins/maps/common/index.ts

@@ -22,6 +22,7 @@ export {
   MAX_ZOOM,
   MIN_ZOOM,
   VECTOR_SHAPE_TYPE,
+  VECTOR_STYLES,
 } from './constants';
 
 export type { FieldFormatter } from './constants';

x-pack/plugins/maps/kibana.json

@@ -8,10 +8,12 @@
   "kibanaVersion": "kibana",
   "configPath": ["xpack", "maps"],
   "requiredPlugins": [
+    "lens",
     "licensing",
     "features",
     "inspector",
     "data",
+    "fieldFormats",
     "fileUpload",
     "uiActions",
     "navigation",

x-pack/plugins/maps/public/actions/layer_actions.ts

@@ -17,6 +17,7 @@ import {
   getLayerListRaw,
   getMapColors,
   getMapReady,
+  getMapSettings,
   getSelectedLayerId,
 } from '../selectors/map_selectors';
 import { FLYOUT_STATE } from '../reducers/ui';
@@ -35,12 +36,18 @@ import {
   SET_SELECTED_LAYER,
   SET_WAITING_FOR_READY_HIDDEN_LAYERS,
   TRACK_CURRENT_LAYER_STATE,
+  UPDATE_LAYER,
   UPDATE_LAYER_ORDER,
   UPDATE_LAYER_PROP,
   UPDATE_LAYER_STYLE,
   UPDATE_SOURCE_PROP,
 } from './map_action_constants';
-import { clearDataRequests, syncDataForLayerId, updateStyleMeta } from './data_request_actions';
+import {
+  autoFitToBounds,
+  clearDataRequests,
+  syncDataForLayerId,
+  updateStyleMeta,
+} from './data_request_actions';
 import { updateTooltipStateForLayer } from './tooltip_actions';
 import {
   Attribution,
@@ -117,6 +124,22 @@ export function replaceLayerList(newLayerList: LayerDescriptor[]) {
   };
 }
 
+export function updateLayerById(layerDescriptor: LayerDescriptor) {
+  return async (
+    dispatch: ThunkDispatch<MapStoreState, void, AnyAction>,
+    getState: () => MapStoreState
+  ) => {
+    dispatch({
+      type: UPDATE_LAYER,
+      layer: layerDescriptor,
+    });
+    await dispatch(syncDataForLayerId(layerDescriptor.id, false));
+    if (getMapSettings(getState()).autoFitToDataBounds) {
+      dispatch(autoFitToBounds());
+    }
+  };
+}
+
 export function cloneLayer(layerId: string) {
   return async (
     dispatch: ThunkDispatch<MapStoreState, void, AnyAction>,

x-pack/plugins/maps/public/actions/map_action_constants.ts

@@ -23,6 +23,7 @@ export const LAYER_DATA_LOAD_ERROR = 'LAYER_DATA_LOAD_ERROR';
 export const UPDATE_SOURCE_DATA_REQUEST = 'UPDATE_SOURCE_DATA_REQUEST';
 export const SET_JOINS = 'SET_JOINS';
 export const SET_QUERY = 'SET_QUERY';
+export const UPDATE_LAYER = 'UPDATE_LAYER';
 export const UPDATE_LAYER_PROP = 'UPDATE_LAYER_PROP';
 export const UPDATE_LAYER_STYLE = 'UPDATE_LAYER_STYLE';
 export const SET_LAYER_STYLE_META = 'SET_LAYER_STYLE_META';

x-pack/plugins/maps/public/classes/fields/inline_field.ts

@@ -10,21 +10,25 @@ import { IField, AbstractField } from './field';
 import { IVectorSource } from '../sources/vector_source';
 
 export class InlineField<T extends IVectorSource> extends AbstractField implements IField {
+  private readonly _label?: string;
   private readonly _source: T;
   private readonly _dataType: string;
 
   constructor({
     fieldName,
+    label,
     source,
     origin,
     dataType,
   }: {
     fieldName: string;
+    label?: string;
     source: T;
     origin: FIELD_ORIGIN;
     dataType: string;
   }) {
     super({ fieldName, origin });
+    this._label = label;
     this._source = source;
     this._dataType = dataType;
   }
@@ -42,7 +46,7 @@ export class InlineField<T extends IVectorSource> extends AbstractField implemen
   }
 
   async getLabel(): Promise<string> {
-    return this.getName();
+    return this._label ? this._label : this.getName();
   }
 
   async getDataType(): Promise<string> {

x-pack/plugins/maps/public/classes/layers/vector_layer/geojson_vector_layer/geojson_vector_layer.tsx

@@ -247,7 +247,7 @@ export class GeoJsonVectorLayer extends AbstractVectorLayer {
       }
 
       const joinStates = await this._syncJoins(syncContext, style);
-      performInnerJoins(
+      await performInnerJoins(
         sourceResult,
         joinStates,
         syncContext.updateSourceData,

x-pack/plugins/maps/public/classes/layers/vector_layer/geojson_vector_layer/perform_inner_joins.test.ts

@@ -8,12 +8,8 @@
 import sinon from 'sinon';
 import _ from 'lodash';
 import { FeatureCollection } from 'geojson';
-import { ESTermSourceDescriptor } from '../../../../../common/descriptor_types';
-import {
-  AGG_TYPE,
-  FEATURE_VISIBLE_PROPERTY_NAME,
-  SOURCE_TYPES,
-} from '../../../../../common/constants';
+import { TableSourceDescriptor } from '../../../../../common/descriptor_types';
+import { FEATURE_VISIBLE_PROPERTY_NAME, SOURCE_TYPES } from '../../../../../common/constants';
 import { performInnerJoins } from './perform_inner_joins';
 import { InnerJoin } from '../../../joins/inner_join';
 import { IVectorSource } from '../../../sources/vector_source';
@@ -53,18 +49,21 @@ const featureCollection = {
 const joinDescriptor = {
   leftField: LEFT_FIELD,
   right: {
-    applyGlobalQuery: true,
-    applyGlobalTime: true,
     id: 'd3625663-5b34-4d50-a784-0d743f676a0c',
-    indexPatternId: 'myIndexPattern',
-    metrics: [
+    __rows: [],
+    __columns: [
       {
-        type: AGG_TYPE.COUNT,
+        name: 'rightKey',
+        type: 'string',
+      },
+      {
+        name: COUNT_PROPERTY_NAME,
+        type: 'number',
       },
     ],
     term: 'rightKey',
-    type: SOURCE_TYPES.ES_TERM_SOURCE,
-  } as ESTermSourceDescriptor,
+    type: SOURCE_TYPES.TABLE_SOURCE,
+  } as TableSourceDescriptor,
 };
 const mockVectorSource = {
   getInspectorAdapters: () => {
@@ -75,18 +74,21 @@ const mockVectorSource = {
       getName: () => {
         return LEFT_FIELD;
       },
-    } as IField;
+      getLabel: () => {
+        return LEFT_FIELD;
+      },
+    } as unknown as IField;
   },
 } as unknown as IVectorSource;
 const innerJoin = new InnerJoin(joinDescriptor, mockVectorSource);
 const propertiesMap = new Map<string, Record<string | number, unknown>>();
 propertiesMap.set('alpha', { [COUNT_PROPERTY_NAME]: 1 });
 
-test('should skip join when no state has changed', () => {
+test('should skip join when no state has changed', async () => {
   const updateSourceData = sinon.spy();
   const onJoinError = sinon.spy();
 
-  performInnerJoins(
+  await performInnerJoins(
     {
       refreshed: false,
       featureCollection: _.cloneDeep(featureCollection) as FeatureCollection,
@@ -105,11 +107,11 @@ test('should skip join when no state has changed', () => {
   expect(onJoinError.notCalled);
 });
 
-test('should perform join when features change', () => {
+test('should perform join when features change', async () => {
   const updateSourceData = sinon.spy();
   const onJoinError = sinon.spy();
 
-  performInnerJoins(
+  await performInnerJoins(
     {
       refreshed: true,
       featureCollection: _.cloneDeep(featureCollection) as FeatureCollection,
@@ -128,11 +130,11 @@ test('should perform join when features change', () => {
   expect(onJoinError.notCalled);
 });
 
-test('should perform join when join state changes', () => {
+test('should perform join when join state changes', async () => {
   const updateSourceData = sinon.spy();
   const onJoinError = sinon.spy();
 
-  performInnerJoins(
+  await performInnerJoins(
     {
       refreshed: false,
       featureCollection: _.cloneDeep(featureCollection) as FeatureCollection,
@@ -151,11 +153,11 @@ test('should perform join when join state changes', () => {
   expect(onJoinError.notCalled);
 });
 
-test('should call updateSourceData with feature collection with updated feature visibility and join properties', () => {
+test('should call updateSourceData with feature collection with updated feature visibility and join properties', async () => {
   const updateSourceData = sinon.spy();
   const onJoinError = sinon.spy();
 
-  performInnerJoins(
+  await performInnerJoins(
     {
       refreshed: true,
       featureCollection: _.cloneDeep(featureCollection) as FeatureCollection,
@@ -204,11 +206,11 @@ test('should call updateSourceData with feature collection with updated feature
   expect(onJoinError.notCalled);
 });
 
-test('should call updateSourceData when no results returned from terms aggregation', () => {
+test('should call updateSourceData when no results returned from terms aggregation', async () => {
   const updateSourceData = sinon.spy();
   const onJoinError = sinon.spy();
 
-  performInnerJoins(
+  await performInnerJoins(
     {
       refreshed: false,
       featureCollection: _.cloneDeep(featureCollection) as FeatureCollection,
@@ -256,15 +258,15 @@ test('should call updateSourceData when no results returned from terms aggregati
   expect(onJoinError.notCalled);
 });
 
-test('should call onJoinError when there are no matching features', () => {
+test('should call onJoinError when there are no matching features', async () => {
   const updateSourceData = sinon.spy();
   const onJoinError = sinon.spy();
 
   // instead of returning military alphabet like "alpha" or "bravo", mismatched key returns numbers, like '1'
   const propertiesMapFromMismatchedKey = new Map<string, Record<string | number, unknown>>();
   propertiesMapFromMismatchedKey.set('1', { [COUNT_PROPERTY_NAME]: 1 });
 
-  performInnerJoins(
+  await performInnerJoins(
     {
       refreshed: false,
       featureCollection: _.cloneDeep(featureCollection) as FeatureCollection,

x-pack/plugins/maps/public/classes/layers/vector_layer/geojson_vector_layer/perform_inner_joins.ts

@@ -23,7 +23,7 @@ export interface JoinState {
   propertiesMap?: PropertiesMap;
 }
 
-export function performInnerJoins(
+export async function performInnerJoins(
   sourceResult: SourceResult,
   joinStates: JoinState[],
   updateSourceData: DataRequestContext['updateSourceData'],
@@ -102,7 +102,11 @@ export function performInnerJoins(
     }
 
     const joinStatus = joinStatusesWithoutAnyMatches[0];
-    const leftFieldName = joinStatus.joinState.join.getLeftField().getName();
+    const leftFieldName = await joinStatus.joinState.join.getLeftField().getLabel();
+    const rightFieldName = await joinStatus.joinState.join
+      .getRightJoinSource()
+      .getTermField()
+      .getLabel();
     const reason =
       joinStatus.keys.length === 0
         ? i18n.translate('xpack.maps.vectorLayer.joinError.noLeftFieldValuesMsg', {
@@ -114,10 +118,7 @@ export function performInnerJoins(
             values: {
               leftFieldName,
               leftFieldValues: prettyPrintArray(joinStatus.keys),
-              rightFieldName: joinStatus.joinState.join
-                .getRightJoinSource()
-                .getTermField()
-                .getName(),
+              rightFieldName,
               rightFieldValues: prettyPrintArray(
                 Array.from(joinStatus.joinState.propertiesMap!.keys())
               ),