fix: missing component check

ccontrols · Aug 29, 2020 · deb1653 · deb1653
1 parent d76dc83
commit deb1653
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/core/store/src/state/context/components.ts b/core/store/src/state/context/components.ts
@@ -6,7 +6,7 @@ import {
 } from '@component-controls/core';
 
 import { useStore } from './store';
-import { useStory } from './story';
+import { useStory, useCurrentStory } from './story';
 import { useCurrentDocument } from './document';
 
 export interface ComponentInputProps {
@@ -134,13 +134,14 @@ export const useComponent = ({
  */
 
 export const useCurrentPropsCount = (): number => {
-  const components = useComponents({ of: '.' });
   const store = useStore();
+  const story = useCurrentStory();
+  const components = useComponents({ name: story?.id });
   const doc = useCurrentDocument();
   return components && doc
     ? Object.keys(components).reduce((acc, key) => {
         const component = store.components[doc.componentsLookup[key]];
-        return acc + Object.keys(component.info?.props || {}).length;
+        return acc + Object.keys(component?.info?.props || {}).length;
       }, 0)
     : 0;
 };
diff --git a/core/store/src/state/recoil/components.ts b/core/store/src/state/recoil/components.ts
@@ -6,7 +6,7 @@ import {
 } from '@component-controls/core';
 
 import { useStore } from './store';
-import { useStory } from './story';
+import { useStory, useCurrentStory } from './story';
 import { useCurrentDocument } from './document';
 
 export interface ComponentInputProps {
@@ -32,7 +32,9 @@ export const useComponents = ({
   name,
 }: ComponentInputProps): Components => {
   const store = useStore();
-  const story = useStory({ id: name });
+  const story =
+    of === CURRENT_STORY ? useCurrentStory() : useStory({ id: name });
+
   const { component: storyComponentName } = story || {};
   const storyComponent = getComponentName(storyComponentName);
   const doc = story && story.doc ? store.docs[story.doc] : undefined;
@@ -133,8 +135,9 @@ export const useComponent = ({
  */
 
 export const useCurrentPropsCount = (): number => {
-  const components = useComponents({ of: '.' });
   const store = useStore();
+  const story = useCurrentStory();
+  const components = useComponents({ name: story?.id });
   const doc = useCurrentDocument();
   return components && doc
     ? Object.keys(components).reduce((acc, key) => {