Use Of type in useOf argument

storybookjs · Mar 7, 2023 · dc44ac0 · dc44ac0
1 parent fa83aa2
commit dc44ac0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/code/ui/blocks/src/blocks/Story.tsx b/code/ui/blocks/src/blocks/Story.tsx
@@ -115,6 +115,7 @@ export const getStoryId = (props: StoryProps, context: DocsContextProps): StoryI
       Please refer to the migration guide: https://github.com/storybookjs/storybook/blob/next/MIGRATION.md#story-block'`);
   }
 
+  console.log(props, 'of' in props);
   if (meta) context.referenceMeta(meta, false);
   const resolved = context.resolveOf(of || story || 'story', ['story']);
   return resolved.story.id;

diff --git a/code/ui/blocks/src/blocks/useOf.ts b/code/ui/blocks/src/blocks/useOf.ts
@@ -1,6 +1,5 @@
 import type {
   DocsContextProps,
-  ModuleExport,
   ResolvedModuleExportType,
   ResolvedModuleExportFromType,
 } from '@storybook/types';
@@ -16,7 +15,7 @@ export type Of = Parameters<DocsContextProps['resolveOf']>[0];
  * if the resolved module is a component it will include the project annotations
  */
 export const useOf = <TType extends ResolvedModuleExportType>(
-  moduleExportOrType: ModuleExport | TType,
+  moduleExportOrType: Of,
   validTypes?: TType[]
 ): ResolvedModuleExportFromType<TType> => {
   const context = useContext(DocsContext);