mirror of
				https://github.com/lingble/chatwoot.git
				synced 2025-11-03 20:48:07 +00:00 
			
		
		
		
	## Linear: - https://github.com/chatwoot/chatwoot/issues/486 ## Description This PR implements Multi-Factor Authentication (MFA) support for user accounts, enhancing security by requiring a second form of verification during login. The feature adds TOTP (Time-based One-Time Password) authentication with QR code generation and backup codes for account recovery. ## Type of change - [ ] New feature (non-breaking change which adds functionality) ## How Has This Been Tested? - Added comprehensive RSpec tests for MFA controller functionality - Tested MFA setup flow with QR code generation - Verified OTP validation and backup code generation - Tested login flow with MFA enabled/disabled ## Checklist: - [ ] My code follows the style guidelines of this project - [ ] I have performed a self-review of my code - [ ] I have commented on my code, particularly in hard-to-understand areas - [ ] I have made corresponding changes to the documentation - [ ] My changes generate no new warnings - [ ] I have added tests that prove my fix is effective or that my feature works - [ ] New and existing unit tests pass locally with my changes - [ ] Any dependent changes have been merged and published in downstream modules --------- Co-authored-by: Pranav <pranav@chatwoot.com> Co-authored-by: Sojan Jose <sojan@pepalo.com> Co-authored-by: Muhsin Keloth <muhsinkeramam@gmail.com>
		
			
				
	
	
		
			202 lines
		
	
	
		
			6.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
			
		
		
	
	
			202 lines
		
	
	
		
			6.7 KiB
		
	
	
	
		
			Ruby
		
	
	
	
	
	
# == Schema Information
 | 
						|
#
 | 
						|
# Table name: users
 | 
						|
#
 | 
						|
#  id                     :integer          not null, primary key
 | 
						|
#  availability           :integer          default("online")
 | 
						|
#  confirmation_sent_at   :datetime
 | 
						|
#  confirmation_token     :string
 | 
						|
#  confirmed_at           :datetime
 | 
						|
#  consumed_timestep      :integer
 | 
						|
#  current_sign_in_at     :datetime
 | 
						|
#  current_sign_in_ip     :string
 | 
						|
#  custom_attributes      :jsonb
 | 
						|
#  display_name           :string
 | 
						|
#  email                  :string
 | 
						|
#  encrypted_password     :string           default(""), not null
 | 
						|
#  last_sign_in_at        :datetime
 | 
						|
#  last_sign_in_ip        :string
 | 
						|
#  message_signature      :text
 | 
						|
#  name                   :string           not null
 | 
						|
#  otp_backup_codes       :text
 | 
						|
#  otp_required_for_login :boolean          default(FALSE), not null
 | 
						|
#  otp_secret             :string
 | 
						|
#  provider               :string           default("email"), not null
 | 
						|
#  pubsub_token           :string
 | 
						|
#  remember_created_at    :datetime
 | 
						|
#  reset_password_sent_at :datetime
 | 
						|
#  reset_password_token   :string
 | 
						|
#  sign_in_count          :integer          default(0), not null
 | 
						|
#  tokens                 :json
 | 
						|
#  type                   :string
 | 
						|
#  ui_settings            :jsonb
 | 
						|
#  uid                    :string           default(""), not null
 | 
						|
#  unconfirmed_email      :string
 | 
						|
#  created_at             :datetime         not null
 | 
						|
#  updated_at             :datetime         not null
 | 
						|
#
 | 
						|
# Indexes
 | 
						|
#
 | 
						|
#  index_users_on_email                   (email)
 | 
						|
#  index_users_on_otp_required_for_login  (otp_required_for_login)
 | 
						|
#  index_users_on_otp_secret              (otp_secret) UNIQUE
 | 
						|
#  index_users_on_pubsub_token            (pubsub_token) UNIQUE
 | 
						|
#  index_users_on_reset_password_token    (reset_password_token) UNIQUE
 | 
						|
#  index_users_on_uid_and_provider        (uid,provider) UNIQUE
 | 
						|
#
 | 
						|
 | 
						|
class User < ApplicationRecord
 | 
						|
  include AccessTokenable
 | 
						|
  include Avatarable
 | 
						|
  # Include default devise modules.
 | 
						|
  include DeviseTokenAuth::Concerns::User
 | 
						|
  include Pubsubable
 | 
						|
  include Rails.application.routes.url_helpers
 | 
						|
  include Reportable
 | 
						|
  include SsoAuthenticatable
 | 
						|
  include UserAttributeHelpers
 | 
						|
 | 
						|
  devise :database_authenticatable,
 | 
						|
         :registerable,
 | 
						|
         :recoverable,
 | 
						|
         :rememberable,
 | 
						|
         :trackable,
 | 
						|
         :validatable,
 | 
						|
         :confirmable,
 | 
						|
         :password_has_required_content,
 | 
						|
         :two_factor_authenticatable,
 | 
						|
         :omniauthable, omniauth_providers: [:google_oauth2, :saml]
 | 
						|
 | 
						|
  # TODO: remove in a future version once online status is moved to account users
 | 
						|
  # remove the column availability from users
 | 
						|
  enum availability: { online: 0, offline: 1, busy: 2 }
 | 
						|
 | 
						|
  # The validation below has been commented out as it does not
 | 
						|
  # work because :validatable in devise overrides this.
 | 
						|
  # validates_uniqueness_of :email, scope: :account_id
 | 
						|
 | 
						|
  validates :email, presence: true
 | 
						|
 | 
						|
  serialize :otp_backup_codes, type: Array
 | 
						|
 | 
						|
  # Encrypt sensitive MFA fields
 | 
						|
  encrypts :otp_secret, deterministic: true
 | 
						|
  encrypts :otp_backup_codes
 | 
						|
 | 
						|
  has_many :account_users, dependent: :destroy_async
 | 
						|
  has_many :accounts, through: :account_users
 | 
						|
  accepts_nested_attributes_for :account_users
 | 
						|
 | 
						|
  has_many :assigned_conversations, foreign_key: 'assignee_id', class_name: 'Conversation', dependent: :nullify, inverse_of: :assignee
 | 
						|
  alias_attribute :conversations, :assigned_conversations
 | 
						|
  has_many :csat_survey_responses, foreign_key: 'assigned_agent_id', dependent: :nullify, inverse_of: :assigned_agent
 | 
						|
  has_many :conversation_participants, dependent: :destroy_async
 | 
						|
  has_many :participating_conversations, through: :conversation_participants, source: :conversation
 | 
						|
 | 
						|
  has_many :inbox_members, dependent: :destroy_async
 | 
						|
  has_many :inboxes, through: :inbox_members, source: :inbox
 | 
						|
  has_many :messages, as: :sender, dependent: :nullify
 | 
						|
  has_many :invitees, through: :account_users, class_name: 'User', foreign_key: 'inviter_id', source: :inviter, dependent: :nullify
 | 
						|
 | 
						|
  has_many :custom_filters, dependent: :destroy_async
 | 
						|
  has_many :dashboard_apps, dependent: :nullify
 | 
						|
  has_many :mentions, dependent: :destroy_async
 | 
						|
  has_many :notes, dependent: :nullify
 | 
						|
  has_many :notification_settings, dependent: :destroy_async
 | 
						|
  has_many :notification_subscriptions, dependent: :destroy_async
 | 
						|
  has_many :notifications, dependent: :destroy_async
 | 
						|
  has_many :team_members, dependent: :destroy_async
 | 
						|
  has_many :teams, through: :team_members
 | 
						|
  has_many :articles, foreign_key: 'author_id', dependent: :nullify, inverse_of: :author
 | 
						|
  # rubocop:disable Rails/HasManyOrHasOneDependent
 | 
						|
  # we are handling this in `remove_macros` callback
 | 
						|
  has_many :macros, foreign_key: 'created_by_id', inverse_of: :created_by
 | 
						|
  # rubocop:enable Rails/HasManyOrHasOneDependent
 | 
						|
 | 
						|
  before_validation :set_password_and_uid, on: :create
 | 
						|
  after_destroy :remove_macros
 | 
						|
 | 
						|
  scope :order_by_full_name, -> { order('lower(name) ASC') }
 | 
						|
 | 
						|
  before_validation do
 | 
						|
    self.email = email.try(:downcase)
 | 
						|
  end
 | 
						|
 | 
						|
  def send_devise_notification(notification, *)
 | 
						|
    devise_mailer.with(account: Current.account).send(notification, self, *).deliver_later
 | 
						|
  end
 | 
						|
 | 
						|
  def set_password_and_uid
 | 
						|
    self.uid = email
 | 
						|
  end
 | 
						|
 | 
						|
  def assigned_inboxes
 | 
						|
    administrator? ? Current.account.inboxes : inboxes.where(account_id: Current.account.id)
 | 
						|
  end
 | 
						|
 | 
						|
  def serializable_hash(options = nil)
 | 
						|
    super(options).merge(confirmed: confirmed?)
 | 
						|
  end
 | 
						|
 | 
						|
  def push_event_data
 | 
						|
    {
 | 
						|
      id: id,
 | 
						|
      name: name,
 | 
						|
      available_name: available_name,
 | 
						|
      avatar_url: avatar_url,
 | 
						|
      type: 'user',
 | 
						|
      availability_status: availability_status,
 | 
						|
      thumbnail: avatar_url
 | 
						|
    }
 | 
						|
  end
 | 
						|
 | 
						|
  def webhook_data
 | 
						|
    {
 | 
						|
      id: id,
 | 
						|
      name: name,
 | 
						|
      email: email,
 | 
						|
      type: 'user'
 | 
						|
    }
 | 
						|
  end
 | 
						|
 | 
						|
  # https://github.com/lynndylanhurley/devise_token_auth/blob/6d7780ee0b9750687e7e2871b9a1c6368f2085a9/app/models/devise_token_auth/concerns/user.rb#L45
 | 
						|
  # Since this method is overriden in devise_token_auth it breaks the email reconfirmation flow.
 | 
						|
  def will_save_change_to_email?
 | 
						|
    mutations_from_database.changed?('email')
 | 
						|
  end
 | 
						|
 | 
						|
  def self.from_email(email)
 | 
						|
    find_by(email: email&.downcase)
 | 
						|
  end
 | 
						|
 | 
						|
  # 2FA/MFA Methods
 | 
						|
  # Delegated to Mfa::ManagementService for better separation of concerns
 | 
						|
  def mfa_service
 | 
						|
    @mfa_service ||= Mfa::ManagementService.new(user: self)
 | 
						|
  end
 | 
						|
 | 
						|
  delegate :two_factor_provisioning_uri, to: :mfa_service
 | 
						|
  delegate :backup_codes_generated?, to: :mfa_service
 | 
						|
  delegate :enable_two_factor!, to: :mfa_service
 | 
						|
  delegate :disable_two_factor!, to: :mfa_service
 | 
						|
  delegate :generate_backup_codes!, to: :mfa_service
 | 
						|
  delegate :validate_backup_code!, to: :mfa_service
 | 
						|
 | 
						|
  def mfa_enabled?
 | 
						|
    otp_required_for_login?
 | 
						|
  end
 | 
						|
 | 
						|
  def mfa_feature_available?
 | 
						|
    Chatwoot.mfa_enabled?
 | 
						|
  end
 | 
						|
 | 
						|
  private
 | 
						|
 | 
						|
  def remove_macros
 | 
						|
    macros.personal.destroy_all
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
User.include_mod_with('Audit::User')
 | 
						|
User.include_mod_with('Concerns::User')
 |