From c8ead965cfa5d4759ab8ad7a67cc9eaef074cd46 Mon Sep 17 00:00:00 2001
From: Gabriel de Oliveira Rohden <gabriel.rohden@outlook.com>
Date: Sun, 18 Aug 2019 20:21:36 -0300
Subject: [PATCH] Add generic types to improve type safety

---
 src/Modalize.tsx | 5 +++--
 src/Options.ts   | 6 +++---
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/Modalize.tsx b/src/Modalize.tsx
index 33527c74..edd92f4b 100644
--- a/src/Modalize.tsx
+++ b/src/Modalize.tsx
@@ -13,7 +13,8 @@ const AnimatedFlatList = Animated.createAnimatedComponent(FlatList);
 const AnimatedSectionList = Animated.createAnimatedComponent(SectionList);
 const THRESHOLD = 150;
 
-export default class Modalize extends React.Component<IProps, IState> {
+export default class Modalize<FlatListItem = any, SectionListItem = any>
+  extends React.Component<IProps<FlatListItem, SectionListItem>, IState> {
 
   static defaultProps = {
     handlePosition: 'outside',
@@ -47,7 +48,7 @@ export default class Modalize extends React.Component<IProps, IState> {
   private modalOverlayTap: React.RefObject<TapGestureHandler> = React.createRef();
   private willCloseModalize: boolean = false;
 
-  constructor(props: IProps) {
+  constructor(props: IProps<FlatListItem, SectionListItem>) {
     super(props);
 
     const fullHeight = isIos() ? screenHeight : screenHeight - 10;
diff --git a/src/Options.ts b/src/Options.ts
index 5781673e..a44522a2 100644
--- a/src/Options.ts
+++ b/src/Options.ts
@@ -19,7 +19,7 @@ export interface IConfigProps {
   spring: ISpringProps;
 }
 
-export interface IProps {
+export interface IProps<FlatListItem = any, SectionListItem = any> {
   /**
    * A React component that will define the content of the modal.
    */
@@ -117,12 +117,12 @@ export interface IProps {
   /*
    * An object to pass any of the react-native FlatList's props.
    */
-  flatListProps?: FlatListProps<any>;
+  flatListProps?: FlatListProps<FlatListItem>;
 
   /*
    * An object to pass any of the react-native SectionList's props.
    */
-  sectionListProps?: SectionListProps<any>;
+  sectionListProps?: SectionListProps<SectionListItem>;
 
   /**
    * A header component outside of the ScrollView, on top of the modal.