lib/generators/ahoy/clicks_generator.rb in ahoy_email-2.0.1 vs lib/generators/ahoy/clicks_generator.rb in ahoy_email-2.0.2

@@ -2,10 +2,36 @@
 
 module Ahoy
   module Generators
     class ClicksGenerator < Rails::Generators::Base
       def copy_templates
-        invoke "ahoy:clicks:activerecord"
+        activerecord = defined?(ActiveRecord)
+        mongoid = defined?(Mongoid)
+
+        selection =
+          if activerecord && mongoid
+            puts <<-MSG
+
+Which data store would you like to use?
+ 1. Active Record (default)
+ 2. Mongoid
+            MSG
+
+            ask(">")
+          elsif activerecord
+            "1"
+          else
+            "2"
+          end
+
+        case selection
+        when "", "1"
+          invoke "ahoy:clicks:activerecord"
+        when "2"
+          invoke "ahoy:clicks:mongoid"
+        else
+          abort "Error: must enter a number [1-2]"
+        end
       end
     end
   end
 end