Merge pull request #30 from kadirahq/require-context

Add support for require.context

src/cli.js

@@ -9,6 +9,8 @@ import chokidar from 'chokidar';
 import EventEmitter from 'events';
 import loadBabelConfig from '@kadira/storybook/dist/server/babel_config';
 import { filterStorybook } from './util';
+import runWithRequireContext from './require_context';
+const babel = require('babel-core');
 
 program
   .option('-c, --config-dir [dir-name]',
@@ -31,7 +33,7 @@ const {
   exclude,
 } = program;
 
-const configPath = path.resolve(`${configDir}`, 'config');
+const configPath = path.resolve(configDir, 'config.js');
 
 const babelConfig = loadBabelConfig(configDir);
 
@@ -64,7 +66,12 @@ const runner = new Runner(program);
 
 async function main() {
   try {
-    require(configPath);
+    const content = babel.transformFileSync(configPath, babelConfig).code;
+    const contextOpts = {
+      filename: configPath,
+      dirname: path.resolve(configDir),
+    };
+    runWithRequireContext(content, contextOpts);
     const storybook = require('@kadira/storybook').getStorybook();
     const addons = require('@kadira/storybook-addons').default;
 

src/require_context.js

@@ -0,0 +1,44 @@
+import vm from 'vm';
+import fs from 'fs';
+import path from 'path';
+import moduleSystem from 'module';
+
+function requireModules(keys, root, directory, regExp, recursive) {
+  const files = fs.readdirSync(path.join(root, directory));
+
+  files.forEach((filename) => {
+    if (regExp.test(filename)) {
+      // webpack adds a './' to the begining of the key
+      // TODO: Check this in windows
+      const entryKey = `./${path.join(directory, filename)}`;
+      // eslint-disable-next-line no-param-reassign, global-require, import/no-dynamic-require
+      keys[entryKey] = require(path.join(root, directory, filename));
+      return;
+    }
+
+    if (!recursive) {
+      return;
+    }
+
+    if (fs.statSync(path.join(root, directory, filename)).isDirectory()) {
+      requireModules(keys, root, path.join(directory, filename), regExp, recursive);
+    }
+  });
+}
+
+export default function runWithRequireContext(content, options) {
+  const { filename, dirname } = options;
+
+  require.context = (directory, useSubdirectories = false, regExp = /^\.\//) => {
+    const fullPath = path.resolve(dirname, directory);
+    const keys = {};
+    requireModules(keys, fullPath, '.', regExp, useSubdirectories);
+
+    const req = f => (keys[f]);
+    req.keys = () => (Object.keys(keys));
+    return req;
+  };
+
+  const compiledModule = vm.runInThisContext(moduleSystem.wrap(content));
+  compiledModule(module.exports, require, module, filename, dirname);
+}