From c7f15738eec06847b3c6b417d809a9b43ca52d74 Mon Sep 17 00:00:00 2001
From: Rich-Harris <richard.a.harris@gmail.com>
Date: Mon, 17 Apr 2017 11:15:42 -0400
Subject: [PATCH] create all blocks in preprocess step

---
 src/generators/dom/preprocess.js                 | 16 +++++++++++++++-
 .../dom/visitors/Component/Component.js          | 10 ++--------
 2 files changed, 17 insertions(+), 9 deletions(-)

diff --git a/src/generators/dom/preprocess.js b/src/generators/dom/preprocess.js
index d3c97d235a..5350642cbf 100644
--- a/src/generators/dom/preprocess.js
+++ b/src/generators/dom/preprocess.js
@@ -106,7 +106,21 @@ const preprocessors = {
 			}
 		});
 
-		preprocessChildren( generator, block, node.children );
+		const isComponent = generator.components.has( node.name ) || node.name === ':Self';
+
+		if ( isComponent ) {
+			const name = block.getUniqueName( ( node.name === ':Self' ? generator.name : node.name ).toLowerCase() );
+
+			node._block = block.child({
+				name: generator.getUniqueName( `create_${name}_yield_fragment` )
+			});
+
+			preprocessChildren( generator, node._block, node.children );
+		}
+
+		else {
+			preprocessChildren( generator, block, node.children );
+		}
 	}
 };
 
diff --git a/src/generators/dom/visitors/Component/Component.js b/src/generators/dom/visitors/Component/Component.js
index 87d9e9d461..fc2729d202 100644
--- a/src/generators/dom/visitors/Component/Component.js
+++ b/src/generators/dom/visitors/Component/Component.js
@@ -6,10 +6,6 @@ import visitEventHandler from './EventHandler.js';
 import visitBinding from './Binding.js';
 import visitRef from './Ref.js';
 
-function capDown ( name ) {
-	return `${name[0].toLowerCase()}${name.slice( 1 )}`;
-}
-
 function stringifyProps ( props ) {
 	if ( !props.length ) return '{}';
 
@@ -38,7 +34,7 @@ const visitors = {
 
 export default function visitComponent ( generator, block, state, node ) {
 	const hasChildren = node.children.length > 0;
-	const name = block.getUniqueName( capDown( node.name === ':Self' ? generator.name : node.name ) );
+	const name = block.getUniqueName( ( node.name === ':Self' ? generator.name : node.name ).toLowerCase() );
 
 	const childState = Object.assign( {}, state, {
 		parentNode: null
@@ -105,9 +101,7 @@ export default function visitComponent ( generator, block, state, node ) {
 	if ( hasChildren ) {
 		const params = block.params.join( ', ' );
 
-		const childBlock = block.child({
-			name: generator.getUniqueName( `create_${name}_yield_fragment` ) // TODO should getUniqueName happen inside Fragment? probably
-		});
+		const childBlock = node._block;
 
 		node.children.forEach( child => {
 			visit( generator, childBlock, childState, child );