From d92b685032d213df33faa84215651834927e66a9 Mon Sep 17 00:00:00 2001
From: Robert Jaszczurek <92210485+rbrtj@users.noreply.github.com>
Date: Thu, 31 Oct 2024 12:56:50 +0100
Subject: [PATCH] [ML] Fixes link to anomaly explorer from anomaly embeddables
 when viewing by job group (#198256)

## Summary

Fix for: [#196509](https://github.com/elastic/kibana/issues/196509)

After fix:

https://github.com/user-attachments/assets/9d731676-fd47-41fe-8843-8b8e6c6e153d
(cherry picked from commit 9eb8b885148ba8fb5f22a1c64c02452eba5dfca9)
---
 .../job_selector/use_job_selection.ts         | 22 +++++++++++++++++--
 1 file changed, 20 insertions(+), 2 deletions(-)

diff --git a/x-pack/plugins/ml/public/application/components/job_selector/use_job_selection.ts b/x-pack/plugins/ml/public/application/components/job_selector/use_job_selection.ts
index 7fcc1e71e1808..51d1882084d3e 100644
--- a/x-pack/plugins/ml/public/application/components/job_selector/use_job_selection.ts
+++ b/x-pack/plugins/ml/public/application/components/job_selector/use_job_selection.ts
@@ -25,6 +25,24 @@ function getInvalidJobIds(jobs: MlJobWithTimeRange[], ids: string[]) {
   });
 }
 
+// This is useful when redirecting from dashboards where groupIds are treated as jobIds
+const getJobIdsFromGroups = (jobIds: string[], jobs: MlJobWithTimeRange[]) => {
+  const result = new Set<string>();
+
+  jobIds.forEach((id) => {
+    const jobsInGroup = jobs.filter((job) => job.groups?.includes(id));
+
+    if (jobsInGroup.length > 0) {
+      jobsInGroup.forEach((job) => result.add(job.job_id));
+    } else {
+      // If it's not a group ID, keep it (regardless of whether it's valid or not)
+      result.add(id);
+    }
+  });
+
+  return Array.from(result);
+};
+
 export interface JobSelection {
   jobIds: string[];
   selectedGroups: string[];
@@ -37,9 +55,9 @@ export const useJobSelection = (jobs: MlJobWithTimeRange[]) => {
   const getJobSelection = useJobSelectionFlyout();
 
   const tmpIds = useMemo(() => {
-    const ids = globalState?.ml?.jobIds || [];
+    const ids = getJobIdsFromGroups(globalState?.ml?.jobIds || [], jobs);
     return (typeof ids === 'string' ? [ids] : ids).map((id: string) => String(id));
-  }, [globalState?.ml?.jobIds]);
+  }, [globalState?.ml?.jobIds, jobs]);
 
   const invalidIds = useMemo(() => {
     return getInvalidJobIds(jobs, tmpIds);