From ae65919f92c1b41ecd10f18500d0b8419a5a0dd9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gast=C3=B3n=20Fournier?= <gaston@getunleash.io>
Date: Tue, 4 Feb 2025 16:14:47 +0100
Subject: [PATCH] fix: license banner using message type (#9209)

## About the changes
Actually uses messageType and maps from warn to warning.
---
 .../banners/internalBanners/LicenseBanner.tsx     | 15 +++++++++++++--
 .../hooks/api/getters/useLicense/useLicense.ts    |  1 +
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/frontend/src/component/banners/internalBanners/LicenseBanner.tsx b/frontend/src/component/banners/internalBanners/LicenseBanner.tsx
index 8230c4e381..ae56e53acc 100644
--- a/frontend/src/component/banners/internalBanners/LicenseBanner.tsx
+++ b/frontend/src/component/banners/internalBanners/LicenseBanner.tsx
@@ -32,8 +32,7 @@ export const LicenseBanner = () => {
             if (!license.loading && !license.error && licenseInfo.message) {
                 const banner = {
                     message: licenseInfo.message,
-                    variant:
-                        licenseInfo.messageType ?? ('warning' as BannerVariant),
+                    variant: mapToVariant(licenseInfo.messageType),
                     sticky: true,
                 };
                 return <Banner key={banner.message} banner={banner} />;
@@ -43,3 +42,15 @@ export const LicenseBanner = () => {
 
     return null;
 };
+function mapToVariant(
+    messageType: string | undefined,
+): BannerVariant | undefined {
+    if (messageType) {
+        switch (messageType) {
+            case 'warn':
+                return 'warning';
+            default:
+                return messageType as BannerVariant;
+        }
+    }
+}
diff --git a/frontend/src/hooks/api/getters/useLicense/useLicense.ts b/frontend/src/hooks/api/getters/useLicense/useLicense.ts
index 9eea990447..0409cbb37f 100644
--- a/frontend/src/hooks/api/getters/useLicense/useLicense.ts
+++ b/frontend/src/hooks/api/getters/useLicense/useLicense.ts
@@ -43,6 +43,7 @@ export const useLicenseCheck = (): LicenseInfo => {
     return {
         isValid: data?.isValid,
         message: data?.message,
+        messageType: data?.messageType,
         loading: !error && !data,
         reCheckLicense: () => mutate(),
         error,