Contents

require 'rails/generators/active_record'

module ActiveRecord
  module Generators
    class DeviseTwoFactorGenerator < ActiveRecord::Generators::Base
      source_root File.expand_path('../templates', __FILE__)

      def copy_devise_migration
        migration_template 'migration.rb', "db/migrate/devise_two_factor_add_to_#{table_name}.rb"
      end

      def inject_field_types
        class_path = if namespaced?
          class_name.to_s.split("::")
        else
          [class_name]
        end

        inject_into_class(model_path, class_path.last, content) if model_exists?
      end

      def content
<<RUBY
  attr_encrypted :otp_auth_secret
  attr_encrypted :otp_recovery_secret
  validates :otp_auth_secret, symmetric_encryption: true
  validates :otp_recovery_secret, symmetric_encryption: true
RUBY
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
devise-2fa-0.4.0	lib/generators/active_record/devise_two_factor_generator.rb
devise-2fa-0.2.1	lib/generators/active_record/devise_two_factor_generator.rb
devise-2fa-0.2.0	lib/generators/active_record/devise_two_factor_generator.rb
devise-2fa-0.1.1	lib/generators/active_record/devise_two_factor_generator.rb
devise-2fa-0.1.0	lib/generators/active_record/devise_two_factor_generator.rb