lib/generators/honeycomb/honeycomb_generator.rb in honeycomb-beeline-1.0.0.pre.beta vs lib/generators/honeycomb/honeycomb_generator.rb in honeycomb-beeline-1.0.0.pre.beta1

@@ -8,19 +8,21 @@
 class HoneycombGenerator < Rails::Generators::Base
   source_root File.expand_path("templates", __dir__)
 
   argument :write_key, required: true, desc: "required"
 
+  class_option :dataset, type: :string, default: "rails"
+
   gem "honeycomb-beeline"
 
   desc "Configures honeycomb with your write key"
 
   def create_initializer_file
     initializer "honeycomb.rb" do
       <<-RUBY.strip_heredoc
         Honeycomb.configure do |config|
           config.write_key = #{write_key.inspect}
-          config.dataset = "rails"
+          config.dataset = #{options['dataset'].inspect}
           config.notification_events = %w[
             sql.active_record
             render_template.action_view
             render_partial.action_view
             render_collection.action_view