feat: save file name in store

ccontrols · Feb 23, 2020 · 5de8627 · 5de8627
1 parent 9f8a5a9
commit 5de8627
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/core/loader/src/loader.ts b/core/loader/src/loader.ts
@@ -21,7 +21,7 @@ module.exports.default = async function(source: string) {
     const time = new Date();
     const fileName = path.join(__dirname, 'story-store-data.js');
     fs.utimesSync(fileName, time, time);
-    addStoriesKind(stories);
+    addStoriesKind(stories, this);
   }
   return source;
 };
diff --git a/core/loader/src/store.ts b/core/loader/src/store.ts
@@ -1,3 +1,4 @@
+import { loader } from 'webpack';
 import { toId, storyNameFromExport } from '@storybook/csf';
 import { StoriesGroup, Story } from '@component-controls/specification';
 import { StoryStore, StoriesKind } from './types';
@@ -7,10 +8,14 @@ const store: StoryStore = {
   stories: {},
 };
 
-export const addStoriesKind = (kind: StoriesGroup) => {
+export const addStoriesKind = (
+  kind: StoriesGroup,
+  context: loader.LoaderContext,
+) => {
   let stories: StoriesKind;
   if (kind.title) {
     stories = {
+      fileName: context.resourcePath,
       title: kind.title,
       stories: [],
       source: kind.source,

diff --git a/core/loader/src/types.ts b/core/loader/src/types.ts
@@ -5,6 +5,11 @@ export interface LoaderOptions {
 }
 
 export interface StoriesKind {
+  /**
+   * file name of the file of stories
+   */
+  fileName?: string;
+
   title: string;
   stories: string[];
   source?: string;