diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/zero/renderer.rb | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/lib/zero/renderer.rb b/lib/zero/renderer.rb index 8ca544d..5bbb875 100644 --- a/lib/zero/renderer.rb +++ b/lib/zero/renderer.rb @@ -61,12 +61,16 @@ module Zero search_files.each do |file| parts = file.gsub(/#{template_path}/, '').split('.') @templates[parts[0]] ||= {} + + # Set default value + types = 'default' + # Overwrite default value, if it's set in template path if parts.count > 2 then - read_type(parts[1]).each do |type| - @templates[parts[0]][type] = file - end - else - @templates[parts[0]][''] = file + types = parts[1] + end + + read_type(types).each do |type| + @templates[parts[0]][type] = file end end end |