12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- #---
- # Excerpted from "Agile Web Development with Rails",
- # published by The Pragmatic Bookshelf.
- # Copyrights apply to this code. It may not be used to create training material,
- # courses, books, articles, and the like. Contact us if you are in doubt.
- # We make no guarantees that this code is fit for any purpose.
- # Visit http://www.pragmaticprogrammer.com/titles/rails4 for more book information.
- #---
- #---
- # Excerpted from "Agile Web Development with Rails, 4rd Ed.",
- # published by The Pragmatic Bookshelf.
- # Copyrights apply to this code. It may not be used to create training material,
- # courses, books, articles, and the like. Contact us if you are in doubt.
- # We make no guarantees that this code is fit for any purpose.
- # Visit http://www.pragmaticprogrammer.com/titles/rails4 for more book information.
- #---
- require 'digest/sha2'
- class User < ActiveRecord::Base
- validates :name, :presence => true, :uniqueness => true
-
- validates :password, :confirmation => true
- attr_accessor :password_confirmation
- attr_reader :password
- validate :password_must_be_present
-
- def User.authenticate(name, password)
- if user = find_by_name(name)
- if user.hashed_password == encrypt_password(password, user.salt)
- user
- end
- end
- end
- def User.encrypt_password(password, salt)
- Digest::SHA2.hexdigest(password + "wibble" + salt)
- end
-
- # 'password' is a virtual attribute
- def password=(password)
- @password = password
- if password.present?
- generate_salt
- self.hashed_password = self.class.encrypt_password(password, salt)
- end
- end
-
- after_destroy :ensure_an_admin_remains
- def ensure_an_admin_remains
- if User.count.zero?
- raise "Can't delete last user"
- end
- end
- private
- def password_must_be_present
- errors.add(:password, "Missing password") unless hashed_password.present?
- end
-
- def generate_salt
- self.salt = self.object_id.to_s + rand.to_s
- end
- end
|