safer base64 handling in story

criipto · May 29, 2024 · 4f3930d · 4f3930d
1 parent a587f13
commit 4f3930d
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/src/components/AuthMethodButton/AuthMethodButton.stories.tsx b/src/components/AuthMethodButton/AuthMethodButton.stories.tsx
@@ -42,11 +42,22 @@ export default {
   }
 } as ComponentMeta<typeof AuthMethodButton>;
 
+function tryBase64Decode(input?: string) {
+  if (!input) return null;
+
+  try {
+    const decoded = atob(input);
+    if (btoa(decoded) === input) return decoded;
+    return null;
+  } catch (err) {
+    return null;
+  }
+}
+
+
 // More on component templates: https://storybook.js.org/docs/react/writing-stories/introduction#using-args
 const Template: ComponentStory<typeof AuthMethodButton> = (args, {globals}) => {
-  const loginHint =
-    (args as any).loginHint && btoa(atob((args as any).loginHint)) === (args as any).loginHint ? atob((args as any).loginHint) :
-    (args as any).loginHint;
+  const loginHint = tryBase64Decode((args as any).loginHint);
 
   return (
     <CriiptoVerifyProvider