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 ; @@ -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,