google_authenticator.rake in google-authenticator-rails-3.0.0

- old
+ new

@@ -5,13 +5,13 @@
                     args.extras.unshift(args[:optional_model_list])
                   else
                     # Adapted from https://stackoverflow.com/a/8248849/7478194
                     Dir[Rails.root.join('app/models/*.rb').to_s].map { |filename| File.basename(filename, '.rb').camelize }
                   end
-    
+
     ActiveRecord::Base.transaction do
-      match_op = " = #{already_encrypted ? 138 : 16}"
+      match_op = already_encrypted ? " in (138,162)" : "in (16,32)"
       model_names.each do |model_name|
         klass = model_name.constantize
         next unless klass.ancestors.include?(ActiveRecord::Base) && klass.try(:google_secrets_encrypted)
         print "#{op_name}ing model #{klass.name.inspect} (table #{klass.table_name.inspect}): "
         count = 0
@@ -21,11 +21,11 @@
         end
         puts "#{count} #{'secret'.pluralize(count)} #{op_name}ed"
       end
     end
   end
-  
+
   desc 'Encrypt all secret columns (add the :encrypt_secrets options *before* running)'
   task :encrypt_secrets, [:optional_model_list] => :environment do |_t, args|
     do_encrypt(args, false, 'Encrypt') { |record| record.encrypt_google_secret! }
   end
 
@@ -44,12 +44,12 @@
         GoogleAuthenticatorRails.secret_encryptor = secret_encryptor
         record.send(:change_google_secret_to!, plain_secret)
       end
     end
   end
-  
+
   desc 'Decrypt all secret columns (remove the :encrypt_secrets options *after* running)'
   task :decrypt_secrets, [:optional_model_list] => :environment do |_t, args|
     do_encrypt(args, true, 'Decrypt') { |record| record.send(:change_google_secret_to!, record.google_secret_value, false) }
-  end  
+  end
 
 end