feat: [useComponents] add hasComponent method

feat: [useComponents] add `hasComponent` method
alibaba · Mar 9, 2021 · d933955 · d933955
1 parent c37afe1
commit d933955
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/src/components/ComponentsProvider/ComponentsContext.ts b/src/components/ComponentsProvider/ComponentsContext.ts
@@ -3,5 +3,6 @@ import { ComponentsContextInterface } from './interface';
 
 export const ComponentsContext = React.createContext<ComponentsContextInterface>({
   addComponent: () => {},
+  hasComponent: () => false,
   getComponent: () => null,
 });
diff --git a/src/components/ComponentsProvider/index.tsx b/src/components/ComponentsProvider/index.tsx
@@ -20,6 +20,10 @@ export const ComponentsProvider: React.FC<ComponentsProviderProps> = (props) =>
     componentsRef.current[code] = val;
   }
 
+  function hasComponent(code: string) {
+    return componentsRef.current.hasOwnProperty(code);
+  }
+
   function getComponent(code: string, callback: GetComponentCallback = () => {}) {
     const comp = componentsRef.current[code];
 
@@ -71,7 +75,7 @@ export const ComponentsProvider: React.FC<ComponentsProviderProps> = (props) =>
   }
 
   return (
-    <ComponentsContext.Provider value={{ addComponent, getComponent }}>
+    <ComponentsContext.Provider value={{ addComponent, hasComponent, getComponent }}>
       {children}
     </ComponentsContext.Provider>
   );

diff --git a/src/components/ComponentsProvider/interface.ts b/src/components/ComponentsProvider/interface.ts
@@ -38,6 +38,7 @@ export interface GetComponentCallback {
 
 export interface ComponentsContextInterface {
   addComponent: (code: string, value: ComponentInterface) => void;
+  hasComponent: (code: string) => boolean;
   getComponent: (
     code: string,
     callback?: GetComponentCallback,