lib/rake_ext/project.rb in ruby_ext-0.4.17 vs lib/rake_ext/project.rb in ruby_ext-0.4.18

@@ -1,5 +1,9 @@
+require 'yaml'
+require 'psych'
+YAML::ENGINE.yamler = 'syck'
+
 # 
 # parse_dependencies_for
 # 
 def parse_project_gemfile
   required_gems, required_fake_gems = [], []
@@ -84,10 +88,14 @@
 
         s.name = name
         s.platform = options.delete(:platform) || Gem::Platform::RUBY
         s.has_rdoc = options.delete(:has_rdoc) == nil ? false : true
         s.require_path = options.delete(:lib) || "lib"
-        s.files = options.delete(:files) || (%w{Rakefile readme.md} + Dir.glob("{lib,spec}/**/*"))
+        s.files = options.delete(:files) || (
+          %w{Rakefile readme.md} + 
+          Dir.glob("{lib,spec}/**/*") + 
+          ((options[:dirs] && Array(options.delete(:dirs)).collect{|d| Dir["#{d}/**/*"]}) || [])
+        )
         s.bindir = options.delete(:bin) if options.include? :bin
 
         s.version = options.delete(:version) || GemHelper.next_version(name)
 
         # s.name = options[:name)
\ No newline at end of file