Merge pull request #34 from kirillzyusko/feature/strictly-typed-optim…

…ize-component [RNBS-024] - strictly typed `optimized` component
kirillzyusko · Feb 7, 2022 · 8141ce0 · 8141ce0
2 parents b0e3ee6 + 65d50a7
commit 8141ce0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,3 +1,5 @@
+import * as React from 'react';
+
 import optimized from './optimized';
 import { mapLoadable } from './bundler';
 import { Component, EnhancedPreLoadable, PreLoadable } from './interface';
@@ -11,7 +13,7 @@ const defaultPreLoadable: EnhancedPreLoadable = {
 
 let i = 0;
 
-const register = (component: PreLoadable & Partial<EnhancedPreLoadable>) => {
+const register = <P extends {}>(component: PreLoadable & Partial<EnhancedPreLoadable>) => {
     const enhancedComponent: Component = {
         name: `Component${i++}`,
         ...defaultPreLoadable,
@@ -25,7 +27,7 @@ const register = (component: PreLoadable & Partial<EnhancedPreLoadable>) => {
 
     mapLoadable[name] = enhancedComponent;
 
-    return optimized(name);
+    return optimized(name) as React.ComponentClass<P>;
 };
 
 const component = (name: string) => getComponent(name);