index.d.ts

import * as React from 'react';

export type Store<T> = {
  /**
   * @description
   * Object that contains the current state.
   */
  state: T;

  /**
   * @description
   * Set the given state keys. The state parameter is an object with the keys and values to set.
   * This only sets/overwrites the specific keys provided.
   */
  set(nextState: Partial<T>): void;

  /**
   * @description
   * Reset the store to the initial state.
   */
  reset(): void;
};

export type Renderable = React.ComponentType | JSX.Element;
export type RenderFunction = () => Renderable | Renderable[];

export type LegacyStorybookStateCallback<T> = (store: Store<T>) => Renderable | Renderable[];
export function withState<T extends {}>(state: T, callback: LegacyStorybookStateCallback<T>): any // RenderFunction;

export type StorybookStateCallback<T> = (context: {store: Store<T>}) => Renderable | Renderable[];
export function withState<T extends {}>(state: T): (callback: StorybookStateCallback<T>) => any