From 626b7107111279a3981fea928047bcddad22940a Mon Sep 17 00:00:00 2001
From: Nicolas Chaulet <nicolas.chaulet@elastic.co>
Date: Thu, 28 May 2020 15:50:10 -0400
Subject: [PATCH] Fix ts errors for beats management

---
 .../beats/elasticsearch_beats_adapter.ts      | 25 ++++++++++++-------
 ...asticsearch_configuration_block_adapter.ts |  5 ++--
 2 files changed, 19 insertions(+), 11 deletions(-)

diff --git a/x-pack/legacy/plugins/beats_management/server/lib/adapters/beats/elasticsearch_beats_adapter.ts b/x-pack/legacy/plugins/beats_management/server/lib/adapters/beats/elasticsearch_beats_adapter.ts
index f76cd46a604403..0600ed8e3fbf69 100644
--- a/x-pack/legacy/plugins/beats_management/server/lib/adapters/beats/elasticsearch_beats_adapter.ts
+++ b/x-pack/legacy/plugins/beats_management/server/lib/adapters/beats/elasticsearch_beats_adapter.ts
@@ -11,6 +11,15 @@ import { DatabaseAdapter } from '../database/adapter_types';
 import { FrameworkUser } from '../framework/adapter_types';
 import { BeatsTagAssignment, CMBeatsAdapter } from './adapter_types';
 
+function formatWithTags(beat: CMBeat) {
+  const { tags, ...rest } = beat;
+
+  return {
+    tags: tags || [],
+    ...rest,
+  };
+}
+
 export class ElasticsearchBeatsAdapter implements CMBeatsAdapter {
   private database: DatabaseAdapter;
 
@@ -97,10 +106,9 @@ export class ElasticsearchBeatsAdapter implements CMBeatsAdapter {
     if (beats.length === 0) {
       return [];
     }
-    return beats.map((beat: any) => ({
-      tags: [] as string[],
-      ...(omit(beat._source.beat as CMBeat, ['access_token']) as CMBeat),
-    }));
+    return beats.map((beat: any) =>
+      formatWithTags(omit(beat._source.beat as CMBeat, ['access_token']) as CMBeat)
+    );
   }
 
   public async getBeatWithToken(
@@ -124,7 +132,7 @@ export class ElasticsearchBeatsAdapter implements CMBeatsAdapter {
     if (beats.length === 0) {
       return null;
     }
-    return omit<CMBeat, {}>(_get<CMBeat>({ tags: [], ...beats[0] }, '_source.beat'), [
+    return omit<CMBeat, {}>(_get<CMBeat>(formatWithTags(beats[0]), '_source.beat'), [
       'access_token',
     ]);
   }
@@ -165,10 +173,9 @@ export class ElasticsearchBeatsAdapter implements CMBeatsAdapter {
     }
     const beats = _get<any>(response, 'hits.hits', []);
 
-    return beats.map((beat: any) => ({
-      tags: [] as string[],
-      ...(omit(beat._source.beat as CMBeat, ['access_token']) as CMBeat),
-    }));
+    return beats.map((beat: any) =>
+      formatWithTags(omit(beat._source.beat as CMBeat, ['access_token']) as CMBeat)
+    );
   }
 
   public async removeTagsFromBeats(
diff --git a/x-pack/legacy/plugins/beats_management/server/lib/adapters/configuration_blocks/elasticsearch_configuration_block_adapter.ts b/x-pack/legacy/plugins/beats_management/server/lib/adapters/configuration_blocks/elasticsearch_configuration_block_adapter.ts
index f60b5c8b4d0f61..2bc6f187564472 100644
--- a/x-pack/legacy/plugins/beats_management/server/lib/adapters/configuration_blocks/elasticsearch_configuration_block_adapter.ts
+++ b/x-pack/legacy/plugins/beats_management/server/lib/adapters/configuration_blocks/elasticsearch_configuration_block_adapter.ts
@@ -137,12 +137,13 @@ export class ElasticsearchConfigurationBlockAdapter implements ConfigurationBloc
   public async create(user: FrameworkUser, configs: ConfigurationBlock[]): Promise<string[]> {
     const body = flatten(
       configs.map((config) => {
-        const id = config.id || uuidv4();
+        const { id: configId, ...configWithoutId } = config;
+        const id = configId || uuidv4();
         return [
           { index: { _id: `configuration_block:${id}` } },
           {
             type: 'configuration_block',
-            configuration_block: { id, ...config, config: JSON.stringify(config.config) },
+            configuration_block: { id, ...configWithoutId, config: JSON.stringify(config.config) },
           },
         ];
       })