mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 02:20:41 +08:00
61 lines
1.5 KiB
Ruby
61 lines
1.5 KiB
Ruby
class EmailToken < ActiveRecord::Base
|
|
belongs_to :user
|
|
|
|
validates_presence_of :token
|
|
validates_presence_of :user_id
|
|
validates_presence_of :email
|
|
|
|
before_validation(on: :create) do
|
|
self.token = EmailToken.generate_token
|
|
end
|
|
|
|
after_create do
|
|
# Expire the previous tokens
|
|
EmailToken.update_all 'expired = true', ['user_id = ? and id != ?', self.user_id, self.id]
|
|
end
|
|
|
|
def self.token_length
|
|
16
|
|
end
|
|
|
|
def self.valid_after
|
|
1.week.ago
|
|
end
|
|
|
|
def self.unconfirmed
|
|
where(confirmed: false)
|
|
end
|
|
|
|
def self.active
|
|
where(expired: false).where('created_at > ?', valid_after)
|
|
end
|
|
|
|
def self.generate_token
|
|
SecureRandom.hex(EmailToken.token_length)
|
|
end
|
|
|
|
def self.confirm(token)
|
|
return unless token.present?
|
|
return unless token.length/2 == EmailToken.token_length
|
|
|
|
email_token = EmailToken.where("token = ? and expired = FALSE and created_at >= ?", token, EmailToken.valid_after).includes(:user).first
|
|
return if email_token.blank?
|
|
|
|
user = email_token.user
|
|
User.transaction do
|
|
row_count = EmailToken.update_all 'confirmed = true', id: email_token.id, expired: false
|
|
if row_count == 1
|
|
# If we are activating the user, send the welcome message
|
|
user.send_welcome_message = !user.active?
|
|
|
|
user.active = true
|
|
user.email = email_token.email
|
|
user.save!
|
|
end
|
|
end
|
|
user
|
|
rescue ActiveRecord::RecordInvalid
|
|
# If the user's email is already taken, just return nil (failure)
|
|
end
|
|
end
|