Scan templates directory for .eco files as well

alexkravets · Feb 20, 2013 · c0cf6ab · c0cf6ab
1 parent db03931
commit c0cf6ab
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/eco-plus.coffee b/eco-plus.coffee
@@ -164,10 +164,14 @@ module.exports =
   templates: (path) ->
     templates = {}
     findit.sync path, (f) ->
-      if f.endsWith('.html')
-        name = f.remove(path + '/').remove('.html')
-        src = fs.readFileSync(f, 'utf-8')
-        templates[name] = src
+      get_templates = (extension) ->
+        if f.endsWith(".#{extension}")
+          name = f.remove(path + '/').remove(".#{extension}")
+          src = fs.readFileSync(f, 'utf-8')
+          templates[name] = src
+
+      get_templates 'html'
+      get_templates 'eco'
     templates
 
   render_js: (templates, context) ->