From 8d397ef27a481599c850bc0565f354f1f66ca4da Mon Sep 17 00:00:00 2001
From: Andre Wanlin <awanlin@rapidrtc.com>
Date: Sat, 25 Sep 2021 10:27:57 -0500
Subject: [PATCH] Add check for Kubernetes annotation on entity

Signed-off-by: Andre Wanlin <awanlin@rapidrtc.com>
---
 .changeset/smart-cooks-drop.md    | 5 +++++
 plugins/kubernetes/src/Router.tsx | 6 ++++++
 plugins/kubernetes/src/index.ts   | 2 +-
 3 files changed, 12 insertions(+), 1 deletion(-)
 create mode 100644 .changeset/smart-cooks-drop.md

diff --git a/.changeset/smart-cooks-drop.md b/.changeset/smart-cooks-drop.md
new file mode 100644
index 0000000000..5ce59f2837
--- /dev/null
+++ b/.changeset/smart-cooks-drop.md
@@ -0,0 +1,5 @@
+---
+'@backstage/plugin-kubernetes': patch
+---
+
+Added a check for the Kubernetes annotation on the entity
diff --git a/plugins/kubernetes/src/Router.tsx b/plugins/kubernetes/src/Router.tsx
index 11c9b488fb..f98d3ca45e 100644
--- a/plugins/kubernetes/src/Router.tsx
+++ b/plugins/kubernetes/src/Router.tsx
@@ -27,6 +27,12 @@ const KUBERNETES_ANNOTATION = 'backstage.io/kubernetes-id';
 const KUBERNETES_LABEL_SELECTOR_QUERY_ANNOTATION =
   'backstage.io/kubernetes-label-selector';
 
+export const isKubernetesAvailable = (entity: Entity) =>
+  Boolean(entity.metadata.annotations?.[KUBERNETES_ANNOTATION]) ||
+  Boolean(
+    entity.metadata.annotations?.[KUBERNETES_LABEL_SELECTOR_QUERY_ANNOTATION],
+  );
+
 type Props = {
   /** @deprecated The entity is now grabbed from context instead */
   entity?: Entity;
diff --git a/plugins/kubernetes/src/index.ts b/plugins/kubernetes/src/index.ts
index 63f68f0af1..08af757d47 100644
--- a/plugins/kubernetes/src/index.ts
+++ b/plugins/kubernetes/src/index.ts
@@ -25,6 +25,6 @@ export {
   kubernetesPlugin as plugin,
   EntityKubernetesContent,
 } from './plugin';
-export { Router } from './Router';
+export { Router, isKubernetesAvailable } from './Router';
 export * from './kubernetes-auth-provider';
 export * from './utils/clusterLinks';