Merge pull request #15 from ronco/feature/head-component-support

Serialize head component if present
ember-fastboot · Feb 10, 2016 · 57a5ebc · 57a5ebc
2 parents 8ee9faf + 9087bb5
commit 57a5ebc
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/lib/ember-app.js b/lib/ember-app.js
@@ -60,10 +60,15 @@ EmberApp.prototype.visit = function(url) {
   var options = { isBrowser: false, document: doc, rootElement: rootElement };
 
   return this._app.visit(url, options).then(function(instance) {
+    var head;
+    if (doc.head) {
+      head = HTMLSerializer.serialize(doc.head);
+    }
     try {
       return {
         url: instance.getURL(), // TODO: use this to determine whether to 200 or redirect
         title: doc.title,
+        head: head,
         body: HTMLSerializer.serialize(rootElement) // This matches the current code; but we probably want `serializeChildren` here
       };
     } finally {

diff --git a/lib/server.js b/lib/server.js
@@ -32,19 +32,22 @@ FastBootServer.prototype.log = function(statusCode, message, startTime) {
   this.ui.writeLine(chalk.blue(now.toISOString()) + " " + chalk[color](statusCode) + " " + message);
 };
 
-FastBootServer.prototype.insertIntoIndexHTML = function(title, body) {
+FastBootServer.prototype.insertIntoIndexHTML = function(title, body, head) {
   var html = this.html.replace("<!-- EMBER_CLI_FASTBOOT_BODY -->", body);
 
   if (title) {
     html = html.replace("<!-- EMBER_CLI_FASTBOOT_TITLE -->", "<title>" + title + "</title>");
   }
+  if (head) {
+    html = html.replace("<!-- EMBER_CLI_FASTBOOT_HEAD -->", head);
+  }
 
   return html;
 };
 
 FastBootServer.prototype.handleSuccess = function(res, path, result, startTime) {
   this.log(200, 'OK ' + path, startTime);
-  res.send(this.insertIntoIndexHTML(result.title, result.body));
+  res.send(this.insertIntoIndexHTML(result.title, result.body, result.head));
 };
 
 FastBootServer.prototype.handleFailure = function(res, path, error, startTime) {