From 630af7079e5bcc930599393c27d113e296541193 Mon Sep 17 00:00:00 2001
From: Muhammed Thanish <mnmtanish@gmail.com>
Date: Tue, 24 May 2016 19:02:43 +0530
Subject: [PATCH] Export storybook middleware

---
 src/server/index.js      | 30 +++---------------------------
 src/server/middleware.js | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 38 insertions(+), 27 deletions(-)
 create mode 100644 src/server/middleware.js

diff --git a/src/server/index.js b/src/server/index.js
index 2737fceb9a95..e1e6de04027f 100755
--- a/src/server/index.js
+++ b/src/server/index.js
@@ -1,18 +1,11 @@
 #!/usr/bin/env node
 
-import webpack from 'webpack';
-import webpackDevMiddleware from 'webpack-dev-middleware';
-import webpackHotMiddleware from 'webpack-hot-middleware';
-import getIndexHtml from './index.html';
-import getIframeHtml from './iframe.html';
 import express from 'express';
 import program from 'commander';
-import packageJson from '../../package.json';
-import baseConfig from './webpack.config';
-import loadConfig from './config';
 import path from 'path';
 import fs from 'fs';
-import { getHeadHtml } from './utils';
+import storybook from './middleware';
+import packageJson from '../../package.json';
 
 const logger = console;
 
@@ -53,24 +46,7 @@ if (program.staticDir) {
 // Build the webpack configuration using the `baseConfig`
 // custom `.babelrc` file and `webpack.config.js` files
 const configDir = program.configDir || './.storybook';
-const config = loadConfig('DEVELOPMENT', baseConfig, configDir);
-
-const compiler = webpack(config);
-const devMiddlewareOptions = {
-  noInfo: true,
-  publicPath: config.output.publicPath,
-};
-app.use(webpackDevMiddleware(compiler, devMiddlewareOptions));
-app.use(webpackHotMiddleware(compiler));
-
-app.get('/', function (req, res) {
-  res.send(getIndexHtml());
-});
-
-const headHtml = getHeadHtml(configDir);
-app.get('/iframe.html', function (req, res) {
-  res.send(getIframeHtml(headHtml));
-});
+app.use(storybook(configDir));
 
 app.listen(...listenAddr, function (error) {
   if (error) {
diff --git a/src/server/middleware.js b/src/server/middleware.js
new file mode 100644
index 000000000000..8cb8e81d1e48
--- /dev/null
+++ b/src/server/middleware.js
@@ -0,0 +1,35 @@
+import { Router } from 'express';
+import webpack from 'webpack';
+import webpackDevMiddleware from 'webpack-dev-middleware';
+import webpackHotMiddleware from 'webpack-hot-middleware';
+import baseConfig from './webpack.config';
+import loadConfig from './config';
+import getIndexHtml from './index.html';
+import getIframeHtml from './iframe.html';
+import { getHeadHtml } from './utils';
+
+export default function (configDir) {
+  // Build the webpack configuration using the `baseConfig`
+  // custom `.babelrc` file and `webpack.config.js` files
+  const config = loadConfig('DEVELOPMENT', baseConfig, configDir);
+  const compiler = webpack(config);
+  const devMiddlewareOptions = {
+    noInfo: true,
+    publicPath: config.output.publicPath,
+  };
+
+  const router = new Router();
+  router.use(webpackDevMiddleware(compiler, devMiddlewareOptions));
+  router.use(webpackHotMiddleware(compiler));
+
+  router.get('/', function (req, res) {
+    res.send(getIndexHtml());
+  });
+
+  const headHtml = getHeadHtml(configDir);
+  router.get('/iframe.html', function (req, res) {
+    res.send(getIframeHtml(headHtml));
+  });
+
+  return router;
+}