From 9d3cae6e1b1cd087b9a7ba82bba4df5ff925d1be Mon Sep 17 00:00:00 2001
From: Andrew <awcrotwell@gmail.com>
Date: Fri, 21 Aug 2020 17:32:51 -0400
Subject: [PATCH] Add support for multiple directories

---
 spec/features/file_tree_template_spec.cr | 2 +-
 src/lib/file_tree.cr                     | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/spec/features/file_tree_template_spec.cr b/spec/features/file_tree_template_spec.cr
index d89e6c1..88e9006 100644
--- a/spec/features/file_tree_template_spec.cr
+++ b/spec/features/file_tree_template_spec.cr
@@ -4,7 +4,7 @@ module TeeplateFileTreeTemplateFeature
   extend HaveFiles::Spec::Dsl
 
   class Template < Teeplate::FileTree
-    directory "#{__DIR__}/../../test/file_tree_template/template"
+    directories ["#{__DIR__}/../../test/file_tree_template/template"]
 
     @file : String
     @class : String
diff --git a/src/lib/file_tree.cr b/src/lib/file_tree.cr
index a4bf0c7..127dca7 100644
--- a/src/lib/file_tree.cr
+++ b/src/lib/file_tree.cr
@@ -8,6 +8,12 @@ module Teeplate
       {{ run(__DIR__ + "/file_tree/macros/directory", dir.id) }}
     end
 
+    macro directories(list_of_dirs)
+      {% for dir in list_of_dirs %}
+        {{ run(__DIR__ + "/file_tree/macros/directory", dir.id) }}
+      {% end %}
+    end
+
     @file_entries : Array(AsDataEntry)?
     # Returns collected file entries.
     def file_entries : Array(AsDataEntry)