From 0cdbbf3f705f6d56fa1d4868b4cac8f9696abfd8 Mon Sep 17 00:00:00 2001
From: Pablo Neves Machado <pablo.nevesmachado@elastic.co>
Date: Fri, 19 Nov 2021 14:32:27 +0100
Subject: [PATCH] Fix HostsRiskScore interface to match new transform version

---
 .../security_solution/hosts/risk_score/index.ts        | 10 +++++++++-
 .../risky_hosts_enabled_module.tsx                     |  4 ++--
 .../overview_risky_host_links/use_hosts_risk_score.ts  |  4 ++--
 3 files changed, 13 insertions(+), 5 deletions(-)

diff --git a/x-pack/plugins/security_solution/common/search_strategy/security_solution/hosts/risk_score/index.ts b/x-pack/plugins/security_solution/common/search_strategy/security_solution/hosts/risk_score/index.ts
index 39f648eab8cd0..378ad41132101 100644
--- a/x-pack/plugins/security_solution/common/search_strategy/security_solution/hosts/risk_score/index.ts
+++ b/x-pack/plugins/security_solution/common/search_strategy/security_solution/hosts/risk_score/index.ts
@@ -27,6 +27,14 @@ export interface HostsRiskScore {
   host: {
     name: string;
   };
-  risk_score: number;
   risk: string;
+  risk_stats: {
+    rule_risks: RuleRisk[];
+    risk_score: number;
+  };
+}
+
+export interface RuleRisk {
+  rule_name: string;
+  rule_risk: string;
 }
diff --git a/x-pack/plugins/security_solution/public/overview/components/overview_risky_host_links/risky_hosts_enabled_module.tsx b/x-pack/plugins/security_solution/public/overview/components/overview_risky_host_links/risky_hosts_enabled_module.tsx
index 4db6f67acb265..a6c38f24d5fc6 100644
--- a/x-pack/plugins/security_solution/public/overview/components/overview_risky_host_links/risky_hosts_enabled_module.tsx
+++ b/x-pack/plugins/security_solution/public/overview/components/overview_risky_host_links/risky_hosts_enabled_module.tsx
@@ -14,9 +14,9 @@ import { HostRisk } from '../../containers/overview_risky_host_links/use_hosts_r
 import { HostsRiskScore } from '../../../../common';
 
 const getListItemsFromHits = (items: HostsRiskScore[]): LinkPanelListItem[] => {
-  return items.map(({ host, risk_score: count, risk: copy }) => ({
+  return items.map(({ host, risk_stats: riskStats, risk: copy }) => ({
     title: host.name,
-    count,
+    count: riskStats.risk_score,
     copy,
     path: '',
   }));
diff --git a/x-pack/plugins/security_solution/public/overview/containers/overview_risky_host_links/use_hosts_risk_score.ts b/x-pack/plugins/security_solution/public/overview/containers/overview_risky_host_links/use_hosts_risk_score.ts
index 15cb7ef7b1c46..9a8795b2a6d07 100644
--- a/x-pack/plugins/security_solution/public/overview/containers/overview_risky_host_links/use_hosts_risk_score.ts
+++ b/x-pack/plugins/security_solution/public/overview/containers/overview_risky_host_links/use_hosts_risk_score.ts
@@ -24,11 +24,11 @@ const noop = () => {};
 const isRecord = (item: unknown): item is Record<string, unknown> =>
   typeof item === 'object' && !!item;
 
-const isHostsRiskScoreHit = (item: unknown): item is HostsRiskScore =>
+const isHostsRiskScoreHit = (item: Partial<HostsRiskScore>): item is HostsRiskScore =>
   isRecord(item) &&
   isRecord(item.host) &&
   typeof item.host.name === 'string' &&
-  typeof item.risk_score === 'number' &&
+  typeof item.risk_stats?.risk_score === 'number' &&
   typeof item.risk === 'string';
 
 export interface HostRisk {