feat(di): add hooks for registries and registryComponent

bem · Mar 20, 2019 · c512dc2 · c512dc2
1 parent 60a8984
commit c512dc2
Show file tree

Hide file tree

Showing 3 changed files with 236 additions and 164 deletions.
diff --git a/packages/di/di.tsx b/packages/di/di.tsx
@@ -1,4 +1,11 @@
-import React, { StatelessComponent, ComponentType, createContext } from 'react';
+import React, {
+    ReactNode,
+    FunctionComponent,
+    StatelessComponent,
+    ComponentType,
+    createContext,
+    useContext,
+} from 'react';
 
 export type GetNonDefaultProps<T> = keyof T extends never ? never : T;
 export type RegistryContext = Record<string, Registry>;
@@ -40,12 +47,12 @@ export function withRegistry(...registries: Registry[]) {
     };
 }
 
-export interface IComponentRegistryConsumer {
+export interface IComponentRegistryConsumerProps {
     id: string;
-    children: (registry: any) => React.ReactNode;
+    children: (registry: any) => ReactNode;
 }
 
-export const ComponentRegistryConsumer: React.SFC<IComponentRegistryConsumer> = props => (
+export const ComponentRegistryConsumer: FunctionComponent<IComponentRegistryConsumerProps> = props => (
     <RegistryConsumer>
         {registries => {
             if (__DEV__) {
@@ -59,6 +66,16 @@ export const ComponentRegistryConsumer: React.SFC<IComponentRegistryConsumer> =
     </RegistryConsumer>
 );
 
+export const useRegistries = () => {
+    return useContext(registryContext);
+};
+
+export const useComponentRegistry = <T extends {}>(id: string) => {
+    const registries = useRegistries();
+
+    return registries[id].snapshot<T>();
+};
+
 export interface IRegistryOptions {
     id: string;
     inverted?: boolean;

diff --git a/packages/di/package.json b/packages/di/package.json
@@ -30,7 +30,7 @@
     "test": "../../node_modules/.bin/nyc npm run unit test/*.test.tsx"
   },
   "peerDependencies": {
-    "react": "^16.0.0"
+    "react": "^16.8.0"
   },
   "nyc": {
     "include": [