Adding missing type to lens field icon

x-pack/legacy/plugins/lens/public/indexpattern_plugin/field_item.tsx

@@ -37,13 +37,12 @@ import { i18n } from '@kbn/i18n';
 import { toElasticsearchQuery } from '@kbn/es-query';
 import { Query } from 'src/plugins/data/common';
 // @ts-ignore
-import classNames from 'classnames';
 import { fieldFormats } from '../../../../../../src/legacy/ui/public/registry/field_formats';
 import { IndexPattern, IndexPatternField, DraggedField } from './indexpattern';
 import { DragDrop } from '../drag_drop';
 import { DatasourceDataPanelProps, DataType } from '../types';
 import { BucketedAggregation, FieldStatsResponse } from '../../common';
-import { LensFieldIcon } from './lens_field_icon';
+import { LensFieldIcon, getColorForDataType } from './lens_field_icon';
 
 export interface FieldItemProps {
   core: DatasourceDataPanelProps['core'];
@@ -362,6 +361,7 @@ function FieldItemPopoverContents(props: State & FieldItemProps) {
       colorValues: [],
       specId,
     };
+    const expectedColor = getColorForDataType(field.type);
 
     const seriesColors = new Map([[colors, expectedColor]]);
 

x-pack/legacy/plugins/lens/public/indexpattern_plugin/lens_field_icon.tsx

@@ -5,10 +5,32 @@
  */
 
 import React from 'react';
+import { ICON_TYPES, palettes } from '@elastic/eui';
 import classNames from 'classnames';
 import { FieldIcon } from '../../../../../../src/plugins/kibana_react/public';
 import { DataType } from '../types';
 
+function stringToNum(s: string) {
+  return Array.from(s).reduce((acc, ch) => acc + ch.charCodeAt(0), 1);
+}
+
+function getIconForDataType(dataType: string) {
+  const icons: Partial<Record<string, UnwrapArray<typeof ICON_TYPES>>> = {
+    boolean: 'invert',
+    date: 'calendar',
+    geo_point: 'globe',
+    ip: 'storage',
+  };
+  return icons[dataType] || ICON_TYPES.find(t => t === dataType) || 'document';
+}
+
+export function getColorForDataType(type: string) {
+  const iconType = getIconForDataType(type);
+  const { colors } = palettes.euiPaletteColorBlind;
+  const colorIndex = stringToNum(iconType) % colors.length;
+  return colors[colorIndex];
+}
+
 export function LensFieldIcon({ type }: { type: DataType }) {
   const classes = classNames(
     'lnsFieldListPanel__fieldIcon',