lib/generators/authtrail/install_generator.rb in authtrail-0.3.0 vs lib/generators/authtrail/install_generator.rb in authtrail-0.3.1

@@ -4,23 +4,49 @@
   module Generators
     class InstallGenerator < Rails::Generators::Base
       include ActiveRecord::Generators::Migration
       source_root File.join(__dir__, "templates")
 
+      class_option :lockbox, type: :boolean
+
       def copy_migration
         migration_template "login_activities_migration.rb", "db/migrate/create_login_activities.rb", migration_version: migration_version
       end
 
       def copy_templates
         template "initializer.rb", "config/initializers/authtrail.rb"
       end
 
       def generate_model
-        template "login_activity_model.rb", "app/models/login_activity.rb"
+        if lockbox?
+          template "model_lockbox.rb", "app/models/login_activity.rb"
+        else
+          template "model.rb", "app/models/login_activity.rb"
+        end
       end
 
       def migration_version
         "[#{ActiveRecord::VERSION::MAJOR}.#{ActiveRecord::VERSION::MINOR}]"
+      end
+
+      def identity_column
+        if lockbox?
+          "t.text :identity_ciphertext\n      t.string :identity_bidx, index: true"
+        else
+          "t.string :identity, index: true"
+        end
+      end
+
+      def ip_column
+        if lockbox?
+          "t.text :ip_ciphertext\n      t.string :ip_bidx, index: true"
+        else
+          "t.string :ip, index: true"
+        end
+      end
+
+      def lockbox?
+        options[:lockbox]
       end
     end
   end
 end