lib/ar2gostruct/gostruct.rb in ar2gostruct-0.2.1 vs lib/ar2gostruct/gostruct.rb in ar2gostruct-0.2.2

@@ -1,10 +1,11 @@
 module Ar2gostruct
   class Gostruct
     def initialize(model_dir)
       @model_dir = model_dir
       get_models
+      puts "package #{ENV["package"]}\n\n" unless ENV["package"].blank?
     end
     attr_accessor :model_dir, :models
 
     def convert!
       self.models.each do |m|
@@ -35,6 +36,6 @@
         puts "// #{model_file_name}"
         puts info
       end
 
   end
-end
\ No newline at end of file
+end