storybookjs · arunoda · Apr 11, 2016 · Apr 7, 2016 · Apr 8, 2016 · Apr 8, 2016
diff --git a/dist/server/get_head_html.js b/dist/server/get_head_html.js
@@ -0,0 +1,25 @@
+'use strict';
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+
+exports.default = function (configDirPath) {
+  var headHtmlPath = _path2.default.resolve(configDirPath, 'head.html');
+  var headHtml = '';
+  if (_fs2.default.existsSync(headHtmlPath)) {
+    headHtml = _fs2.default.readFileSync(headHtmlPath, 'utf8');
+  }
+
+  return headHtml;
+};
+
+var _path = require('path');
+
+var _path2 = _interopRequireDefault(_path);
+
+var _fs = require('fs');
+
+var _fs2 = _interopRequireDefault(_fs);
+
+function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
diff --git a/dist/server/iframe.html.js b/dist/server/iframe.html.js
@@ -4,6 +4,6 @@ Object.defineProperty(exports, "__esModule", {
   value: true
 });
 
-exports.default = function () {
-  return "\n    <!DOCTYPE html>\n    <html>\n      <head>\n        <title>React Storybook</title>\n      </head>\n      <body>\n        <div id=\"root\" />\n        <script src=\"/static/preview.bundle.js\"></script>\n      </body>\n    </html>\n  ";
+exports.default = function (headHtml) {
+  return "\n    <!DOCTYPE html>\n    <html>\n      <head>\n        <title>React Storybook</title>\n        " + headHtml + "\n      </head>\n      <body>\n        <div id=\"root\" />\n        <script src=\"/static/preview.bundle.js\"></script>\n      </body>\n    </html>\n  ";
 };
diff --git a/dist/server/index.js b/dist/server/index.js
@@ -53,6 +53,10 @@ var _fs = require('fs');
 
 var _fs2 = _interopRequireDefault(_fs);
 
+var _get_head_html = require('./get_head_html');
+
+var _get_head_html2 = _interopRequireDefault(_get_head_html);
+
 function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
 
 process.env.NODE_ENV = 'production';
@@ -133,8 +137,9 @@ app.get('/', function (req, res) {
   res.send((0, _index2.default)());
 });
 
+var headHtml = (0, _get_head_html2.default)(configDirPath);
 app.get('/iframe', function (req, res) {
-  res.send((0, _iframe2.default)());
+  res.send((0, _iframe2.default)(headHtml));
 });
 
 app.listen(_commander2.default.port, function (error) {

diff --git a/package.json b/package.json
@@ -56,6 +56,7 @@
     "eslint-plugin-babel": "^3.1.0",
     "eslint-plugin-react": "^4.2.3",
     "mocha": "^2.4.5",
+    "mock-fs": "^3.8.0",
     "nodemon": "^1.9.1",
     "react": "^0.14.8",
     "react-dom": "^0.14.8",

diff --git a/src/server/__tests__/get_head_html.js b/src/server/__tests__/get_head_html.js
@@ -0,0 +1,44 @@
+const { describe, it, beforeEach, afterEach } = global;
+import { expect } from 'chai';
+import getHeadHtml from '../get_head_html';
+import mock from 'mock-fs';
+
+const HEAD_HTML_CONTENTS = '<script>console.log("custom script!");</script>';
+
+describe('server.getHeadHtml', () => {
+  describe('when .storybook/head.html does not exist', () => {
+    beforeEach(() => {
+      mock({
+        config: {},
+      });
+    });
+
+    afterEach(() => {
+      mock.restore();
+    });
+
+    it('return an empty string', () => {
+      const result = getHeadHtml('./config');
+      expect(result).to.be.equal('');
+    });
+  });
+
+  describe('when .storybook/head.html exists', () => {
+    beforeEach(() => {
+      mock({
+        config: {
+          'head.html': HEAD_HTML_CONTENTS,
+        },
+      });
+    });
+
+    afterEach(() => {
+      mock.restore();
+    });
+
+    it('return the contents of the file', () => {
+      const result = getHeadHtml('./config');
+      expect(result).to.be.equal(HEAD_HTML_CONTENTS);
+    });
+  });
+});
diff --git a/src/server/get_head_html.js b/src/server/get_head_html.js
@@ -0,0 +1,12 @@
+import path from 'path';
+import fs from 'fs';
+
+export default function (configDirPath) {
+  const headHtmlPath = path.resolve(configDirPath, 'head.html');
+  let headHtml = '';
+  if (fs.existsSync(headHtmlPath)) {
+    headHtml = fs.readFileSync(headHtmlPath, 'utf8');
+  }
+
+  return headHtml;
+}
diff --git a/src/server/iframe.html.js b/src/server/iframe.html.js
@@ -1,9 +1,10 @@
-export default function () {
+export default function (headHtml) {
   return `
     <!DOCTYPE html>
     <html>
       <head>
         <title>React Storybook</title>
+        ${headHtml}
       </head>
       <body>
         <div id="root" />

diff --git a/src/server/index.js b/src/server/index.js
@@ -13,6 +13,7 @@ import packageJson from '../../package.json';
 import config from './webpack.config';
 import path from 'path';
 import fs from 'fs';
+import getHeadHtml from './get_head_html';
 
 const logger = console;
 
@@ -106,8 +107,9 @@ app.get('/', function (req, res) {
   res.send(getIndexHtml());
 });
 
+const headHtml = getHeadHtml(configDirPath);
 app.get('/iframe', function (req, res) {
-  res.send(getIframeHtml());
+  res.send(getIframeHtml(headHtml));
 });
 
 app.listen(program.port, function (error) {