fix: fixed missing type for set image

coderdiaz · Jul 24, 2023 · 8c4267d · 8c4267d
1 parent 8427d82
commit 8c4267d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/src/images/index.ts b/src/images/index.ts
@@ -7,6 +7,7 @@ import unionpay from './unionpay.js';
 import mastercard from './mastercard.js';
 import placeholder from './placeholder.js';
 import visa from './visa.js';
+import { CardType } from '../utils/validation.js';
 
 export default {
   amex,
@@ -18,4 +19,4 @@ export default {
   mastercard,
   placeholder,
   visa,
-};
+} as { [key in CardType]: React.ReactNode | JSX.Element };
diff --git a/src/useCardBrand.ts b/src/useCardBrand.ts
@@ -5,19 +5,19 @@ interface UseCardBrandProps {
   images?: {
     [key: string]: React.ReactNode | JSX.Element;
   };
-  cardNumber?: string;
+  type?: string;
   ariaLabel?: string;
 }
 
 export default function useCardBrand() {
   const getSvgProps = React.useCallback((props: UseCardBrandProps = {}) => {
     const images = props.images ?? {};
-    const type = 'placeholder';
+    const type = props.type ?? 'placeholder';
 
     return React.useMemo(
       () => ({
         'aria-label': props.ariaLabel ?? 'Placeholder card',
-        children: images[type] || images,
+        children: images[type] || null,
         width: '1.5em',
         height: '1em',
         viewBox: '0 0 24 16',

diff --git a/src/utils/validation.ts b/src/utils/validation.ts
@@ -1,4 +1,4 @@
-enum CardType {
+export enum CardType {
   Amex = 'amex',
   Diners = 'dinersclub',
   Discover = 'discover',