mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 17:15:32 +08:00
beb91e7eff
The ROTP gem is only used in a very small amount of places in the app, we don't need to globally require it. Also set the Addressable gem to not have a specific version range, as it has not been a problem yet. Some slight refactoring of UserSecondFactor here too to use SecondFactorManager to avoid code repetition
55 lines
1.2 KiB
Ruby
55 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class UserSecondFactor < ActiveRecord::Base
|
|
include SecondFactorManager
|
|
belongs_to :user
|
|
|
|
scope :backup_codes, -> do
|
|
where(method: UserSecondFactor.methods[:backup_codes], enabled: true)
|
|
end
|
|
|
|
scope :totps, -> do
|
|
where(method: UserSecondFactor.methods[:totp], enabled: true)
|
|
end
|
|
|
|
scope :all_totps, -> do
|
|
where(method: UserSecondFactor.methods[:totp])
|
|
end
|
|
|
|
def self.methods
|
|
@methods ||= Enum.new(
|
|
totp: 1,
|
|
backup_codes: 2,
|
|
)
|
|
end
|
|
|
|
def totp_object
|
|
get_totp_object(self.data)
|
|
end
|
|
|
|
def totp_provisioning_uri
|
|
totp_object.provisioning_uri(user.email)
|
|
end
|
|
|
|
end
|
|
|
|
# == Schema Information
|
|
#
|
|
# Table name: user_second_factors
|
|
#
|
|
# id :bigint not null, primary key
|
|
# user_id :integer not null
|
|
# method :integer not null
|
|
# data :string not null
|
|
# enabled :boolean default(FALSE), not null
|
|
# last_used :datetime
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
# name :string
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_user_second_factors_on_method_and_enabled (method,enabled)
|
|
# index_user_second_factors_on_user_id (user_id)
|
|
#
|