lib/generators/ixtlan/audit_base.rb in ixtlan-generators-0.1.3 vs lib/generators/ixtlan/audit_base.rb in ixtlan-generators-0.1.4

@@ -9,15 +9,19 @@
           args << ARGV.shift
         else
           args << "audit"
         end
 
-        args << 'login:string message:string'
+        args += ['login:string', 'message:string']
         args += ARGV
         
-        args << "--read-only --timestamps"
+        args += ["--read-only", "--timestamps"]
 
         generate generator_name, *args
+      end
+
+      def add_gem
+        gem 'ixtlan-audit' unless File.read('Gemfile') =~ /gem\s['"]ixtlan-audit['"]/
       end
     end
   end
 end