From b2aed8f86f48b850089e42e21e5c2bbc33618a7c Mon Sep 17 00:00:00 2001
From: Sahab Yazdani <sahab.yazdani+github@gmail.com>
Date: Tue, 4 Oct 2011 12:14:33 -0400
Subject: [PATCH] Add dot notation support for section and inverted section
 tags

---
 mustache.js  | 38 +++++++++++++++++++++-----------------
 test/unit.js |  9 +++++++++
 2 files changed, 30 insertions(+), 17 deletions(-)

diff --git a/mustache.js b/mustache.js
index 04e9b00..1b79cd5 100644
--- a/mustache.js
+++ b/mustache.js
@@ -345,6 +345,23 @@ var Mustache = (function(undefined) {
 		return undefined;
 	}
 	
+	function find_with_dot_notation(name, context) {
+		var name_components = name.split('.'),
+			i = 1, n = name_components.length,
+			value = find_in_stack(name_components[0], context);
+			
+		while (value && i<n) {
+			value = find(name_components[i], value);
+			i++;
+		}
+	
+		if (i!==n && !value) {
+			value = undefined;
+		}
+		
+		return value;
+	}
+	
 	/* END Run Time Helpers */
 
 	function text(state, token) {
@@ -369,26 +386,13 @@ var Mustache = (function(undefined) {
 			implicit_iterator = (state.pragmas['IMPLICIT-ITERATOR'] || {iterator: '.'}).iterator;
 		
 		state.send_code_func((function(variable, escape) { return function(context, send_func) {
-			var variable_components,
-				i, n, value;
+			var value;
 			
 			if ( variable === implicit_iterator ) { // special case for implicit iterator (usually '.')
 				value = {}; value[implicit_iterator] = context[context.length-1];
 				value = find(variable, value);
 			} else {
-				variable_components = variable.split('.');
-				i = 1;
-				n = variable_components.length;
-				
-				value = find_in_stack(variable_components[0], context);
-				while (value && i<n) {
-					value = find(variable_components[i], value);
-					i++;
-				}
-			
-				if (i!==n && !value) {
-					value = undefined;
-				}
+				value = find_with_dot_notation(variable, context);
 			}
 
 			if (value!==undefined) {
@@ -456,14 +460,14 @@ var Mustache = (function(undefined) {
 		
 		if (s.inverted) {
 			state.send_code_func((function(program, variable){ return function(context, send_func) {
-				var value = find_in_stack(variable, context);
+				var value = find_with_dot_notation(variable, context);
 				if (!value || is_array(value) && value.length === 0) { // false or empty list, render it
 					program(context, send_func);
 				}
 			};})(program, s.variable));
 		} else {
 			state.send_code_func((function(program, variable, template, partials){ return function(context, send_func) {
-				var value = find_in_stack(variable, context);			
+				var value = find_with_dot_notation(variable, context);
 				if (is_array(value)) { // Enumerable, Let's loop!
 					for (var i=0, n=value.length; i<n; ++i) {
 						context.push(value[i]);
diff --git a/test/unit.js b/test/unit.js
index 9c9933a..d5c6801 100644
--- a/test/unit.js
+++ b/test/unit.js
@@ -172,6 +172,15 @@ test("Dot Notation", function() {
 		),
 		'5'
 	);	
+
+	equals(
+		Mustache.to_html(
+			'{{#a.b.c}}{{d}}{{/a.b.c}}',
+			{ a: { b: function() { return { c: [ {d: 'a'}, {d: 'b'}, {d: 'c'} ] } } } },
+			{}
+		),
+		'abc'
+	);	
 });