From 4c89da2b45b835d41d9200cd1b3b78c84304c7ca Mon Sep 17 00:00:00 2001
From: James Rucker <james.rucker@elastic.co>
Date: Mon, 29 Mar 2021 18:49:16 -0700
Subject: [PATCH] [Workplace Search] Send kibana_host when calling
 /reauth_prepare (#95529)

* WIP almost there?

* Fix server route validation

* fix account route

* Fix test expectations

Co-authored-by: scottybollinger <scotty.bollinger@elastic.co>
Co-authored-by: Kibana Machine <42973632+kibanamachine@users.noreply.github.com>
---
 .../components/add_source/add_source_logic.test.ts | 14 ++++++++++++--
 .../components/add_source/add_source_logic.ts      |  6 +++++-
 .../server/routes/workplace_search/sources.ts      |  6 ++++++
 3 files changed, 23 insertions(+), 3 deletions(-)

diff --git a/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.test.ts b/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.test.ts
index 6c60cd74a9c9..8ced90e7d772 100644
--- a/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.test.ts
+++ b/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.test.ts
@@ -458,7 +458,12 @@ describe('AddSourceLogic', () => {
           AddSourceLogic.actions.getSourceReConnectData('github');
 
           expect(http.get).toHaveBeenCalledWith(
-            '/api/workplace_search/org/sources/github/reauth_prepare'
+            '/api/workplace_search/org/sources/github/reauth_prepare',
+            {
+              query: {
+                kibana_host: '',
+              },
+            }
           );
           await nextTick();
           expect(setSourceConnectDataSpy).toHaveBeenCalledWith(sourceConnectData);
@@ -648,7 +653,12 @@ describe('AddSourceLogic', () => {
         AddSourceLogic.actions.getSourceReConnectData('123');
 
         expect(http.get).toHaveBeenCalledWith(
-          '/api/workplace_search/account/sources/123/reauth_prepare'
+          '/api/workplace_search/account/sources/123/reauth_prepare',
+          {
+            query: {
+              kibana_host: '',
+            },
+          }
         );
       });
 
diff --git a/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.ts b/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.ts
index ed63f82764f7..6ca7f6fa72e2 100644
--- a/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.ts
+++ b/x-pack/plugins/enterprise_search/public/applications/workplace_search/views/content_sources/components/add_source/add_source_logic.ts
@@ -418,8 +418,12 @@ export const AddSourceLogic = kea<MakeLogicType<AddSourceValues, AddSourceAction
         ? `/api/workplace_search/org/sources/${sourceId}/reauth_prepare`
         : `/api/workplace_search/account/sources/${sourceId}/reauth_prepare`;
 
+      const query = {
+        kibana_host: kibanaHost,
+      } as HttpFetchQuery;
+
       try {
-        const response = await HttpLogic.values.http.get(route);
+        const response = await HttpLogic.values.http.get(route, { query });
         actions.setSourceConnectData(response);
       } catch (e) {
         flashAPIErrors(e);
diff --git a/x-pack/plugins/enterprise_search/server/routes/workplace_search/sources.ts b/x-pack/plugins/enterprise_search/server/routes/workplace_search/sources.ts
index 6638aa82b32a..d50ff6564b02 100644
--- a/x-pack/plugins/enterprise_search/server/routes/workplace_search/sources.ts
+++ b/x-pack/plugins/enterprise_search/server/routes/workplace_search/sources.ts
@@ -194,6 +194,9 @@ export function registerAccountSourceReauthPrepareRoute({
         params: schema.object({
           id: schema.string(),
         }),
+        query: schema.object({
+          kibana_host: schema.string(),
+        }),
       },
     },
     enterpriseSearchRequestHandler.createRequest({
@@ -539,6 +542,9 @@ export function registerOrgSourceReauthPrepareRoute({
         params: schema.object({
           id: schema.string(),
         }),
+        query: schema.object({
+          kibana_host: schema.string(),
+        }),
       },
     },
     enterpriseSearchRequestHandler.createRequest({