feat: add close modal

captableinc · Oct 7, 2024 · 63fc004 · 63fc004
1 parent f117e7f
commit 63fc004
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/components/modals/safe.tsx b/src/components/modals/safe.tsx
@@ -12,6 +12,7 @@ type SafeModalProps = {
 export const SafeModal: React.FC<SafeModalProps> = ({ type }) => {
   const [open, setOpen] = useState(true);
 
+  const handleClose = () => setOpen(false);
   return (
     <Modal
       size="4xl"
@@ -80,7 +81,7 @@ export const SafeModal: React.FC<SafeModalProps> = ({ type }) => {
         },
       }}
     >
-      <SafeForm type={type} />
+      <SafeForm handleClose={handleClose} type={type} />
     </Modal>
   );
 };
diff --git a/src/components/safe/form/index.tsx b/src/components/safe/form/index.tsx
@@ -33,6 +33,7 @@ import { useSession } from "next-auth/react";
 
 type SafeFormProps = {
   type: "create" | "import";
+  handleClose: () => void;
 };
 
 const SafeFormSchema = z.object({
@@ -53,7 +54,7 @@ type SafeFormType = z.infer<typeof SafeFormSchema>;
 
 const placeholder = "________________________________________";
 
-export const SafeForm: React.FC<SafeFormProps> = ({ type }) => {
+export const SafeForm: React.FC<SafeFormProps> = ({ type, handleClose }) => {
   const { data: session } = useSession();
   const { mutate } = useMutation(createSafe);
 
@@ -75,6 +76,8 @@ export const SafeForm: React.FC<SafeFormProps> = ({ type }) => {
       json: { ...data },
       urlParams: { companyId: session?.user.companyId ?? "" },
     });
+
+    handleClose();
   };
 
   return (