(#1736) Expose forceReRender over channel

storybookjs · Dec 28, 2017 · 146229d · 146229d
1 parent 4b5622a
commit 146229d
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/app/angular/src/client/preview/index.js b/app/angular/src/client/preview/index.js
@@ -23,9 +23,10 @@ const storyStore = new StoryStore();
 const reduxStore = createStore(reducer);
 const context = { storyStore, reduxStore };
 
+let channel;
 if (isBrowser) {
   const queryParams = qs.parse(window.location.search.substring(1));
-  const channel = createChannel({ page: 'preview' });
+  channel = createChannel({ page: 'preview' });
   channel.on('setCurrentStory', data => {
     reduxStore.dispatch(selectStory(data.kind, data.story));
   });
@@ -55,3 +56,10 @@ const renderUI = () => {
 reduxStore.subscribe(renderUI);
 
 export const forceReRender = () => render(context, true);
+
+if (isBrowser) {
+  channel.on('forceReRender', () => {
+    console.log('force ReRender'); // Remove before merging
+    forceReRender();
+  });
+}
diff --git a/app/react/src/client/preview/index.js b/app/react/src/client/preview/index.js
@@ -26,9 +26,10 @@ const reduxStore = createStore(reducer);
 /* eslint-enable */
 const context = { storyStore, reduxStore };
 
+let channel;
 if (isBrowser) {
   const queryParams = qs.parse(window.location.search.substring(1));
-  const channel = createChannel({ page: 'preview' });
+  channel = createChannel({ page: 'preview' });
   channel.on('setCurrentStory', data => {
     reduxStore.dispatch(selectStory(data.kind, data.story));
   });
@@ -58,3 +59,10 @@ const renderUI = () => {
 reduxStore.subscribe(renderUI);
 
 export const forceReRender = () => render(context, true);
+
+if (isBrowser) {
+  channel.on('forceReRender', () => {
+    console.log('force ReRender'); // Remove before merging
+    forceReRender();
+  });
+}
diff --git a/app/vue/src/client/preview/index.js b/app/vue/src/client/preview/index.js
@@ -24,9 +24,10 @@ const storyStore = new StoryStore();
 const reduxStore = createStore(reducer);
 const context = { storyStore, reduxStore };
 
+let channel;
 if (isBrowser) {
   const queryParams = qs.parse(window.location.search.substring(1));
-  const channel = createChannel({ page: 'preview' });
+  channel = createChannel({ page: 'preview' });
   channel.on('setCurrentStory', data => {
     reduxStore.dispatch(selectStory(data.kind, data.story));
   });
@@ -56,3 +57,10 @@ const renderUI = () => {
 reduxStore.subscribe(renderUI);
 
 export const forceReRender = () => render(context, true);
+
+if (isBrowser) {
+  channel.on('forceReRender', () => {
+    console.log('force ReRender'); // Remove before merging
+    forceReRender();
+  });
+}