lib/soaspec/exe_helpers.rb in soaspec-0.2.2 vs lib/soaspec/exe_helpers.rb in soaspec-0.2.3
- old
+ new
@@ -32,10 +32,14 @@
# Retrieve default file contents based on filename
# @param [String] filename Filename within 'lib/generator' to file retrieve contents from
# @param [Boolean] erb Whether to process file with ERB
def retrieve_contents(filename, erb = true)
- default_file = File.join(File.dirname(__FILE__), 'generator', filename + (erb ? '.erb' : ''))
+ default_file = if filename.start_with?('../')
+ File.join(File.dirname(__FILE__), filename[3..-1] + (erb ? '.erb' : ''))
+ else
+ File.join(File.dirname(__FILE__), 'generator', filename + (erb ? '.erb' : ''))
+ end
contents = File.read(default_file)
erb ? ERB.new(contents).result(binding) : contents
end
# @param [String] filename Name of the file to create
\ No newline at end of file