diff --git a/.storybook/decorators/withStealthKeys.tsx b/.storybook/decorators/withStealthKeys.tsx index c5787e2..c35b182 100644 --- a/.storybook/decorators/withStealthKeys.tsx +++ b/.storybook/decorators/withStealthKeys.tsx @@ -15,6 +15,12 @@ const baseValue: StealthKeysValue = { solanaMetaAddress: null, ckbKeys: null, ckbMetaAddress: null, + isRecoveryMode: false, + isReadOnly: false, + setIsRecoveryMode: noop, + setIsReadOnly: noop, + restoreFromRecoveryKit: noop, + exitRecoveryMode: noop, setEvmKeys: noop, setEvmMetaAddress: noop, setStellarKeys: noop, diff --git a/src/components/Header.tsx b/src/components/Header.tsx index 70d439a..7b1d867 100644 --- a/src/components/Header.tsx +++ b/src/components/Header.tsx @@ -7,6 +7,7 @@ import { LocaleSwitcher } from './LocaleSwitcher'; import { NetworkChip } from './NetworkChip'; import { useTheme } from '@/context/ThemeContext'; import { useNotificationsStore } from '@/stores/notificationsStore'; +import { useStealthKeys } from '@/context/StealthKeysContext'; export function Header() { const location = useLocation(); @@ -14,6 +15,7 @@ export function Header() { const [mobileMenuOpen, setMobileMenuOpen] = useState(false); const { theme, toggleTheme } = useTheme(); const unreadCount = useNotificationsStore((state) => state.unreadCount()); + const { isRecoveryMode } = useStealthKeys(); const navLinks = [ { to: '/send', label: t('nav.send') }, @@ -35,6 +37,11 @@ export function Header() { Demo + {isRecoveryMode && ( + + Recovery Mode + + )}