1 files changed, 36 insertions, 43 deletions

diff --git a/packages/renderer/src/components/locationBar/SecurityLabel.tsx b/packages/renderer/src/components/locationBar/SecurityLabel.tsx
index 0017f89..ac51cff 100644
--- a/packages/renderer/src/components/locationBar/SecurityLabel.tsx
+++ b/packages/renderer/src/components/locationBar/SecurityLabel.tsx
@@ -18,17 +18,17 @@
  * SPDX-License-Identifier: AGPL-3.0-only
  */
 
-import IconHttps from '@mui/icons-material/HttpsOutlined';
-import IconHttp from '@mui/icons-material/NoEncryption';
-import IconGlobe from '@mui/icons-material/Public';
-import IconWarning from '@mui/icons-material/Warning';
+import HttpsOutliedIcon from '@mui/icons-material/HttpsOutlined';
+import NoEncrpytionIcon from '@mui/icons-material/NoEncryption';
+import PublicIcon from '@mui/icons-material/Public';
+import WarningIcon from '@mui/icons-material/Warning';
 import { styled } from '@mui/material/styles';
+import { SecurityLabelKind } from '@sophie/shared';
 import React from 'react';
 import { useTranslation } from 'react-i18next';
 
 import LocationInputAdornment from './LocationInputAdornment';
 import getAlertColor from './getAlertColor';
-import type { SplitResult } from './splitUrl';
 
 const SecurityLabelRoot = styled(LocationInputAdornment, {
   name: 'SecurityLabel',
@@ -53,55 +53,48 @@ const SecurityLabelText = styled('span', {
 }));
 
 export default function SecurityLabel({
-  splitResult,
+  kind,
   changed,
   position,
 }: {
-  splitResult: SplitResult;
+  kind: SecurityLabelKind;
   changed: boolean;
   position: 'start' | 'end';
 }): JSX.Element {
-  const { t } = useTranslation(undefined, {
-    keyPrefix: 'securityLabel',
-  });
+  const { t } = useTranslation();
 
-  const { type } = splitResult;
-  if (changed || type === 'empty') {
+  if (changed || kind === SecurityLabelKind.Empty) {
     return (
       <SecurityLabelRoot alert={false} position={position} aria-hidden>
-        <IconGlobe fontSize="small" />
+        <PublicIcon fontSize="small" />
       </SecurityLabelRoot>
     );
   }
-  switch (type) {
-    case 'valid': {
-      const { secure } = splitResult;
-      return secure ? (
-        <SecurityLabelRoot
-          alert={false}
-          position={position}
-          aria-label={t('secureConnection')}
-        >
-          <IconHttps fontSize="small" />
-        </SecurityLabelRoot>
-      ) : (
-        <SecurityLabelRoot alert position={position}>
-          <IconHttp fontSize="small" />
-          <SecurityLabelText>{t('notSecureConnection')}</SecurityLabelText>
-        </SecurityLabelRoot>
-      );
-    }
-    case 'invalid':
-      return (
-        <SecurityLabelRoot alert position={position}>
-          <IconWarning fontSize="small" />
-          <SecurityLabelText>{t('unknownSite')}</SecurityLabelText>
-        </SecurityLabelRoot>
-      );
-    default:
-      /* eslint-disable-next-line @typescript-eslint/restrict-template-expressions --
-        Error handling for impossible case.
-      */
-      throw new Error(`Unexpected SplitResult: ${type}`);
+  if (kind === SecurityLabelKind.SecureConnection) {
+    return (
+      <SecurityLabelRoot
+        alert={false}
+        position={position}
+        aria-label={t('securityLabel.secureConnection')}
+      >
+        <HttpsOutliedIcon fontSize="small" />
+      </SecurityLabelRoot>
+    );
   }
+  const tslError =
+    kind === SecurityLabelKind.NotSecureConnection ||
+    kind === SecurityLabelKind.CertificateError;
+  const icon = tslError ? (
+    <NoEncrpytionIcon fontSize="small" />
+  ) : (
+    <WarningIcon fontSize="small" />
+  );
+  return (
+    <SecurityLabelRoot alert position={position}>
+      {icon}
+      <SecurityLabelText>
+        {t([`securityLabel.${kind}`, 'securityLabel.unspecific'])}
+      </SecurityLabelText>
+    </SecurityLabelRoot>
+  );
 }