Fix build

hop-dev · Sep 17, 2024 · a8b96d8 · a8b96d8
1 parent 1b94ce7
commit a8b96d8
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 47 deletions.
diff --git a/x-pack/plugins/security_solution/server/lib/entity_analytics/entity_store/definition.ts b/x-pack/plugins/security_solution/server/lib/entity_analytics/entity_store/definition.ts
@@ -9,48 +9,50 @@ import { entityDefinitionSchema, type EntityDefinition } from '@kbn/entities-sch
 import { ENTITY_STORE_DEFAULT_SOURCE_INDICES } from './constants';
 import { getEntityDefinitionId } from './utils/utils';
 
-export const HOST_ENTITY_DEFINITION: EntityDefinition = entityDefinitionSchema.parse({
-  id: getEntityDefinitionId('host'),
-  name: 'EA Host Store',
-  type: 'host',
-  indexPatterns: ENTITY_STORE_DEFAULT_SOURCE_INDICES,
-  identityFields: ['host.name'],
-  displayNameTemplate: '{{host.name}}',
-  metadata: [
-    'host.domain',
-    'host.hostname',
-    'host.id',
-    'host.ip',
-    'host.mac',
-    'host.name',
-    'host.type',
-    'host.architecture',
-  ],
-  history: {
-    timestampField: '@timestamp',
-    interval: '1m',
-  },
-  version: '1.0.0',
-});
+export const buildHostEntityDefinition = (): EntityDefinition =>
+  entityDefinitionSchema.parse({
+    id: getEntityDefinitionId('host'),
+    name: 'EA Host Store',
+    type: 'host',
+    indexPatterns: ENTITY_STORE_DEFAULT_SOURCE_INDICES,
+    identityFields: ['host.name'],
+    displayNameTemplate: '{{host.name}}',
+    metadata: [
+      'host.domain',
+      'host.hostname',
+      'host.id',
+      'host.ip',
+      'host.mac',
+      'host.name',
+      'host.type',
+      'host.architecture',
+    ],
+    history: {
+      timestampField: '@timestamp',
+      interval: '1m',
+    },
+    version: '1.0.0',
+  });
 
-export const USER_ENTITY_DEFINITION: EntityDefinition = entityDefinitionSchema.parse({
-  id: getEntityDefinitionId('user'),
-  name: 'EA User Store',
-  indexPatterns: ENTITY_STORE_DEFAULT_SOURCE_INDICES,
-  identityFields: ['user.name'],
-  displayNameTemplate: '{{user.name}}',
-  metadata: [
-    'user.domain',
-    'user.email',
-    'user.full_name',
-    'user.hash',
-    'user.id',
-    'user.name',
-    'user.roles',
-  ],
-  history: {
-    timestampField: '@timestamp',
-    interval: '1m',
-  },
-  version: '1.0.0',
-});
+export const buildUserEntityDefinition = (): EntityDefinition =>
+  entityDefinitionSchema.parse({
+    id: getEntityDefinitionId('user'),
+    name: 'EA User Store',
+    indexPatterns: ENTITY_STORE_DEFAULT_SOURCE_INDICES,
+    identityFields: ['user.name'],
+    displayNameTemplate: '{{user.name}}',
+    metadata: [
+      'user.domain',
+      'user.email',
+      'user.full_name',
+      'user.hash',
+      'user.id',
+      'user.name',
+      'user.roles',
+    ],
+    history: {
+      timestampField: '@timestamp',
+      interval: '1m',
+    },
+    version: '1.0.0',
+  });
diff --git a/x-pack/plugins/security_solution/server/lib/entity_analytics/entity_store/utils/utils.ts b/x-pack/plugins/security_solution/server/lib/entity_analytics/entity_store/utils/utils.ts
@@ -11,12 +11,12 @@ import type {
   EngineDescriptor,
   EntityType,
 } from '../../../../../common/api/entity_analytics/entity_store/common.gen';
-import { HOST_ENTITY_DEFINITION, USER_ENTITY_DEFINITION } from '../definition';
+import { buildHostEntityDefinition, buildUserEntityDefinition } from '../definition';
 import { entityEngineDescriptorTypeName } from '../saved_object';
 
 export const getEntityDefinition = (entityType: EntityType) => {
-  if (entityType === 'host') return HOST_ENTITY_DEFINITION;
-  if (entityType === 'user') return USER_ENTITY_DEFINITION;
+  if (entityType === 'host') return buildHostEntityDefinition();
+  if (entityType === 'user') return buildUserEntityDefinition();
 
   throw new Error(`Unsupported entity type: ${entityType}`);
 };