feat(overmind-react): use internal hook to reference component in hook

Author: christianalfoni

packages/node_modules/overmind-react/src/index.ts

@@ -1,7 +1,13 @@
 import { BaseApp, EventType, Overmind, Configuration } from 'overmind'
 import * as React from 'react'
-// @ts-ignore
-import { useEffect, useState } from 'react'
+import {
+  // @ts-ignore
+  useEffect,
+  // @ts-ignore
+  useState,
+  // @ts-ignore
+  __SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED,
+} from 'react'
 
 export type IReactComponent<P = any> =
   | React.StatelessComponent<P>
@@ -29,21 +35,18 @@ export type TConnect<App extends BaseApp> = {
 
 let nextComponentId = 0
 
-type FunctionComponent = {
-  (...args: any[]): React.ReactChildren
-  __componentId: number
-  displayName: string
-}
-
 export const createHook = <App extends Overmind<Configuration>>(app: App) => {
   let componentInstanceId = 0
-  return (component: FunctionComponent) => {
-    if (!component) {
-      throw new Error(
-        'Overmind - You have to the Overmind hook the component reference, for debugging purposes'
-      )
-    }
-    const name = component.name || component.displayName
+  const {
+    ReactCurrentOwner,
+  } = __SECRET_INTERNALS_DO_NOT_USE_OR_YOU_WILL_BE_FIRED
+  const useCurrentComponent = () => {
+    return ReactCurrentOwner.current.elementType
+  }
+
+  return () => {
+    const component = useCurrentComponent()
+    const name = component.name
     component.__componentId =
       typeof component.__componentId === 'undefined'
         ? nextComponentId++