feat: add boolean for storing two factor skip state

db/migrate/20220502180100_add_two_factor_required_to_sessions.rb

@@ -0,0 +1,7 @@
+# frozen_string_literal: true
+
+class AddTwoFactorRequiredToSessions < ActiveRecord::Migration[4.2]
+  def change
+    add_column :authie_sessions, :skip_two_factor, :boolean, default: false
+  end
+end

lib/authie/session.rb

@@ -114,9 +114,10 @@ def see_password
     #
     # @raises [ActiveRecord::RecordInvalid]
     # @return [Authie::Session]
-    def mark_as_two_factored
+    def mark_as_two_factored(skip: false)
       @session.two_factored_at = Time.now
       @session.two_factored_ip = @controller.request.ip
+      @session.skip_two_factor = skip
       @session.save!
       Authie.config.events.dispatch(:marked_as_two_factor, self)
       self
@@ -304,6 +305,7 @@ def get_session(controller)
     delegate :two_factored_at, to: :session
     delegate :two_factored_ip, to: :session
     delegate :two_factored?, to: :session
+    delegate :skip_two_factor, to: :session
     delegate :update, to: :session
     delegate :update!, to: :session
     delegate :user_agent, to: :session

spec/lib/session_spec.rb

@@ -240,6 +240,16 @@
       expect(Authie.config.events).to receive(:dispatch).with(:marked_as_two_factor, session)
       session.mark_as_two_factored
     end
+
+    it 'can set the skip two factor boolean to true if provided' do
+      session.mark_as_two_factored(skip: true)
+      expect(session.skip_two_factor).to eq true
+    end
+
+    it 'does not set the skip two factor boolean if not set' do
+      session.mark_as_two_factored
+      expect(session.skip_two_factor).to eq false
+    end
   end
 
   describe '#reset_token' do