Provide way to cast useFormContext

heyman333 · Aug 15, 2019 · a8847d8 · a8847d8
1 parent f67837f
commit a8847d8
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/useFormContext.tsx b/src/useFormContext.tsx
@@ -5,14 +5,15 @@ const FormGlobalContext = React.createContext<FormContextValues<
   DataType
 > | null>(null);
 
-export function useFormContext() {
-  return React.useContext(FormGlobalContext)!;
+export function useFormContext<T extends DataType>(): FormContextValues<T> {
+  // @ts-ignore
+  return React.useContext(FormGlobalContext);
 }
 
-export function FormContext<T extends DataType>(props: FormProps<T>) {
+export function FormContext(props: FormProps<DataType>) {
   const { children, ...rest } = props;
   return (
-    <FormGlobalContext.Provider value={rest as FormContextValues}>
+    <FormGlobalContext.Provider value={rest}>
       {children}
     </FormGlobalContext.Provider>
   );