From 91f8644f17651c5e780aff0f5f33b7c4d8d67e5d Mon Sep 17 00:00:00 2001
From: Xavier Serrano <xavier.serrano@lansweeper.com>
Date: Wed, 27 Oct 2021 19:15:13 +0200
Subject: [PATCH] Add auth token to KafkaBackendClient via IdentityApi

Signed-off-by: Xavier Serrano <zombispormedio007@gmail.com>
Signed-off-by: Xavier Serrano <xavier.serrano@lansweeper.com>
---
 .changeset/long-bugs-kiss.md                |  5 +++++
 plugins/kafka/src/api/KafkaBackendClient.ts | 11 +++++++++--
 plugins/kafka/src/plugin.ts                 |  6 ++++--
 3 files changed, 18 insertions(+), 4 deletions(-)
 create mode 100644 .changeset/long-bugs-kiss.md

diff --git a/.changeset/long-bugs-kiss.md b/.changeset/long-bugs-kiss.md
new file mode 100644
index 0000000000..49e0edae72
--- /dev/null
+++ b/.changeset/long-bugs-kiss.md
@@ -0,0 +1,5 @@
+---
+'@backstage/plugin-kafka': patch
+---
+
+Use IdentityApi to provide Auth Token for KafkaBackendClient Api calls
diff --git a/plugins/kafka/src/api/KafkaBackendClient.ts b/plugins/kafka/src/api/KafkaBackendClient.ts
index 988939ff1a..aea5736aad 100644
--- a/plugins/kafka/src/api/KafkaBackendClient.ts
+++ b/plugins/kafka/src/api/KafkaBackendClient.ts
@@ -15,21 +15,28 @@
  */
 
 import { KafkaApi, ConsumerGroupOffsetsResponse } from './types';
-import { DiscoveryApi } from '@backstage/core-plugin-api';
+import { DiscoveryApi, IdentityApi } from '@backstage/core-plugin-api';
 
 export class KafkaBackendClient implements KafkaApi {
   private readonly discoveryApi: DiscoveryApi;
+  private readonly identityApi: IdentityApi;
 
-  constructor(options: { discoveryApi: DiscoveryApi }) {
+  constructor(options: {
+    discoveryApi: DiscoveryApi;
+    identityApi: IdentityApi;
+  }) {
     this.discoveryApi = options.discoveryApi;
+    this.identityApi = options.identityApi;
   }
 
   private async internalGet(path: string): Promise<any> {
     const url = `${await this.discoveryApi.getBaseUrl('kafka')}${path}`;
+    const idToken = await this.identityApi.getIdToken();
     const response = await fetch(url, {
       method: 'GET',
       headers: {
         'Content-Type': 'application/json',
+        ...(idToken && { Authorization: `Bearer ${idToken}` }),
       },
     });
 
diff --git a/plugins/kafka/src/plugin.ts b/plugins/kafka/src/plugin.ts
index e389e5e27a..539b0a0fda 100644
--- a/plugins/kafka/src/plugin.ts
+++ b/plugins/kafka/src/plugin.ts
@@ -21,6 +21,7 @@ import {
   createRoutableExtension,
   createRouteRef,
   discoveryApiRef,
+  identityApiRef,
 } from '@backstage/core-plugin-api';
 
 export const rootCatalogKafkaRouteRef = createRouteRef({
@@ -33,8 +34,9 @@ export const kafkaPlugin = createPlugin({
   apis: [
     createApiFactory({
       api: kafkaApiRef,
-      deps: { discoveryApi: discoveryApiRef },
-      factory: ({ discoveryApi }) => new KafkaBackendClient({ discoveryApi }),
+      deps: { discoveryApi: discoveryApiRef, identityApi: identityApiRef },
+      factory: ({ discoveryApi, identityApi }) =>
+        new KafkaBackendClient({ discoveryApi, identityApi }),
     }),
   ],
   routes: {