lib/rails/generators/crashlog/crashlog_generator.rb in crashlog-1.0.0.rc1 vs lib/rails/generators/crashlog/crashlog_generator.rb in crashlog-1.0.0.rc2

@@ -1,10 +1,11 @@
 require 'rails/generators'
 
 class CrashlogGenerator < Rails::Generators::Base
 
   class_option :api_key, :aliases => "-k", :type => :string, :desc => "Your CrashLog API key"
+  class_option :secret, :aliases => "-s", :type => :string, :desc => "Your CrashLog API secret"
 
   def self.source_root
     @_crashlog_source_root ||= File.expand_path("../../../../../generators/crashlog/templates", __FILE__)
   end
 
@@ -15,17 +16,23 @@
   end
 
   private
 
   def ensure_api_key_was_configured
-    if !options[:api_key] && !api_key_configured?
-      puts "Must pass --api-key or create config/initializers/crashlog.rb"
+    if !options[:api_key] &&
+       !options[:secret] &&
+       !api_key_configured?
+      puts "Must pass --api-key and --secret or create config/initializers/crashlog.rb"
       exit
     end
   end
 
   def api_key_expression
     "'#{options[:api_key]}'"
+  end
+
+  def secret_expression
+    "'#{options[:secret]}'"
   end
 
   def generate_initializer
     template 'initializer.rb', 'config/initializers/crashlog.rb'
   end