From cdb8b9d01ddf1a83668b7108bbbcd8a9271a5900 Mon Sep 17 00:00:00 2001
From: Rich-Harris <richard.a.harris@gmail.com>
Date: Sat, 8 Apr 2017 11:59:30 -0400
Subject: [PATCH] get yield blocks working

---
 src/generators/dom/visitors/Component.js | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/src/generators/dom/visitors/Component.js b/src/generators/dom/visitors/Component.js
index 95c3e737b5..1db9f3e665 100644
--- a/src/generators/dom/visitors/Component.js
+++ b/src/generators/dom/visitors/Component.js
@@ -1,5 +1,6 @@
 import deindent from '../../../utils/deindent.js';
 import CodeBuilder from '../../../utils/CodeBuilder.js';
+import getBuilders from '../utils/getBuilders.js';
 import visit from '../visit.js';
 import addComponentAttributes from './attributes/addComponentAttributes.js';
 
@@ -75,14 +76,14 @@ export default function visitComponent ( generator, fragment, node ) {
 
 	// Component has children, put them in a separate {{yield}} block
 	if ( hasChildren ) {
-		const yieldName = generator.getUniqueName( `render_${name}_yield_fragment` );
 		const params = fragment.params.join( ', ' );
 
-		const childFragment = this.current.child({
+		const childFragment = fragment.child({
 			type: 'component',
 			name: generator.getUniqueName( `render_${name}_yield_fragment` ), // TODO should getUniqueName happen inside Fragment? probably
 			target: 'target',
-			localElementDepth: 0
+			localElementDepth: 0,
+			builders: getBuilders()
 		});
 
 		node.children.forEach( child => {
@@ -92,7 +93,7 @@ export default function visitComponent ( generator, fragment, node ) {
 		const yieldFragment = fragment.getUniqueName( `${name}_yield_fragment` );
 
 		fragment.builders.create.addLine(
-			`var ${yieldFragment} = ${yieldName}( ${params}, ${fragment.component} );`
+			`var ${yieldFragment} = ${childFragment.name}( ${params}, ${fragment.component} );`
 		);
 
 		fragment.builders.update.addLine(
@@ -100,6 +101,8 @@ export default function visitComponent ( generator, fragment, node ) {
 		);
 
 		componentInitProperties.push( `_yield: ${yieldFragment}`);
+
+		generator.addRenderer( childFragment );
 	}
 
 	const statements = [];