Improve types flushed out by CSF changes

storybookjs · Sep 7, 2021 · a504ad3 · a504ad3
1 parent 4702cf0
commit a504ad3
Show file tree

Hide file tree

Showing 14 changed files with 15 additions and 14 deletions.
diff --git a/addons/backgrounds/src/decorators/withBackground.ts b/addons/backgrounds/src/decorators/withBackground.ts
@@ -1,5 +1,5 @@
 import { useMemo, useEffect } from '@storybook/addons';
-import { AnyFramework, StoryFn as StoryFunction, StoryContext } from '@storybook/csf';
+import { AnyFramework, PartialStoryFn as StoryFunction, StoryContext } from '@storybook/csf';
 
 import { PARAM_KEY as BACKGROUNDS_PARAM_KEY } from '../constants';
 import {

diff --git a/addons/backgrounds/src/decorators/withGrid.ts b/addons/backgrounds/src/decorators/withGrid.ts
@@ -1,7 +1,7 @@
 import dedent from 'ts-dedent';
 import deprecate from 'util-deprecate';
 import { useMemo, useEffect } from '@storybook/addons';
-import { AnyFramework, StoryFn as StoryFunction, StoryContext } from '@storybook/csf';
+import { AnyFramework, PartialStoryFn as StoryFunction, StoryContext } from '@storybook/csf';
 
 import { clearStyles, addGridStyle } from '../helpers';
 import { PARAM_KEY as BACKGROUNDS_PARAM_KEY } from '../constants';

diff --git a/addons/measure/src/withMeasure.ts b/addons/measure/src/withMeasure.ts
@@ -1,6 +1,6 @@
 /* eslint-env browser */
 import { useEffect } from '@storybook/addons';
-import { AnyFramework, StoryFn as StoryFunction, StoryContext } from '@storybook/csf';
+import { AnyFramework, PartialStoryFn as StoryFunction, StoryContext } from '@storybook/csf';
 import { drawSelectedElement } from './box-model/visualizer';
 import { init, rescale, destroy } from './box-model/canvas';
 import { deepElementFromPoint } from './util';

diff --git a/addons/outline/src/withOutline.ts b/addons/outline/src/withOutline.ts
@@ -1,5 +1,5 @@
 import { useMemo, useEffect } from '@storybook/addons';
-import { AnyFramework, StoryFn as StoryFunction, StoryContext } from '@storybook/csf';
+import { AnyFramework, PartialStoryFn as StoryFunction, StoryContext } from '@storybook/csf';
 
 import { clearStyles, addOutlineStyles } from './helpers';
 import { PARAM_KEY } from './constants';

diff --git a/app/angular/src/client/preview/types.ts b/app/angular/src/client/preview/types.ts
@@ -11,7 +11,7 @@ export interface ICollection {
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/html/src/client/preview/types.ts b/app/html/src/client/preview/types.ts
@@ -4,7 +4,7 @@ export type StoryFnHtmlReturnType = string | Node;
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/preact/src/client/preview/types.ts b/app/preact/src/client/preview/types.ts
@@ -9,7 +9,7 @@ export interface ShowErrorArgs {
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/react/src/client/preview/types.ts b/app/react/src/client/preview/types.ts
@@ -12,7 +12,7 @@ export type StoryFnReactReturnType = ReactElement<unknown>;
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/server/src/client/preview/types.ts b/app/server/src/client/preview/types.ts
@@ -18,7 +18,7 @@ export type FetchStoryHtmlType = (
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/vue/src/client/preview/types.ts b/app/vue/src/client/preview/types.ts
@@ -12,7 +12,7 @@ export type StoryFnVueReturnType = string | Component;
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/vue3/src/client/preview/types.ts b/app/vue3/src/client/preview/types.ts
@@ -11,7 +11,7 @@ export type StoryFnVueReturnType = ConcreteComponent<any>;
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/app/web-components/src/client/preview/types.ts b/app/web-components/src/client/preview/types.ts
@@ -7,7 +7,7 @@ export type StoryFnHtmlReturnType = string | Node | TemplateResult | SVGTemplate
 
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/lib/addons/src/types.ts b/lib/addons/src/types.ts
@@ -140,7 +140,7 @@ export interface ClientApiAddons<StoryFnReturnType> {
 // Old types for getStorybook()
 export interface IStorybookStory {
   name: string;
-  render: () => any;
+  render: (context: any) => any;
 }
 
 export interface IStorybookSection {

diff --git a/lib/client-api/src/ClientApi.ts b/lib/client-api/src/ClientApi.ts
@@ -15,6 +15,7 @@ import {
   ComponentTitle,
   Globals,
   GlobalTypes,
+  LegacyStoryFn,
 } from '@storybook/csf';
 import {
   NormalizedComponentAnnotations,
@@ -32,7 +33,7 @@ const { FEATURES } = global;
 
 export interface GetStorybookStory<TFramework extends AnyFramework> {
   name: string;
-  render: StoryFn<TFramework>;
+  render: LegacyStoryFn<TFramework>;
 }
 
 export interface GetStorybookKind<TFramework extends AnyFramework> {