Make block-tag comparison more flexible

Olivine-Labs · Feb 1, 2017 · c9489f8 · c9489f8
1 parent c2e7573
commit c9489f8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/lustache/renderer.lua b/src/lustache/renderer.lua
@@ -24,6 +24,11 @@ local html_escape_characters = {
   ["/"] = "&#x2F;"
 }
 
+local block_tags = {
+  ["#"] = true,
+  ["^"] = true,
+}
+
 local function is_array(array)
   if type(array) ~= "table" then return false end
   local max, n = 0, 0
@@ -89,7 +94,7 @@ local function nest_tokens(tokens)
   local token, section
 
   for i,token in ipairs(tokens) do
-    if token.type == "#" or token.type == "^" then
+    if block_tags[token.type] then
       token.tokens = {}
       sections[#sections+1] = token
       collector[#collector+1] = token