Core: Add viewMode to StoryContext

lib/addons/src/types.ts

@@ -59,6 +59,7 @@ export type StoryContext = StoryIdentifier & {
   argTypes: ArgTypes;
   globals: Args;
   hooks?: HooksContext;
+  viewMode?: ViewMode;
 };
 
 export interface WrapperSettings {

lib/client-api/README.md

@@ -12,6 +12,7 @@ Each story is loaded via the `.add()` API and contains the follow attributes, wh
 - `parameters` - static data about the story, see below.
 - `args` - dynamic inputs to the story, see below.
 - `hooks` - listeners that will rerun when the story changes or is unmounted, see `@storybook/addons`.
+- `viewMode` - property that tells if the story is being rendered in Canvas or Docs tab. Values are `story` for canvas and `docs` for docs.
 
 ## Parameters
 

lib/client-api/src/decorators.test.ts

@@ -7,9 +7,10 @@ function makeContext(input: Record<string, any>): StoryContext {
     id: 'id',
     kind: 'kind',
     name: 'name',
+    viewMode: 'story',
     parameters: {},
     ...input,
-  };
+  } as StoryContext;
 }
 
 describe('client-api.decorators', () => {

lib/client-api/src/story_store.ts

@@ -406,6 +406,7 @@ export default class StoryStore {
         args: _stories[id].args,
         argTypes,
         globals: this._globals,
+        viewMode: this._selection?.viewMode,
       });
 
     // Pull out parameters.args.$ || .argTypes.$.defaultValue into initialArgs