1 files changed, 54 insertions, 0 deletions

diff --git a/app/controllers/otp_challenges_controller.rb b/app/controllers/otp_challenges_controller.rb
new file mode 100644
index 00000000..892503a8
--- /dev/null
+++ b/app/controllers/otp_challenges_controller.rb
@@ -0,0 +1,54 @@
+# The second half of a two-step login. A pending marker (set by
+# sessions#create after a correct password) plus deadline and attempt
+# counter live in the session; the real user_id is only written after a
+# valid code, through a fresh session.
+class OtpChallengesController < ApplicationController
+
+  layout 'admin'
+
+  MAX_ATTEMPTS = 5
+
+  def new
+    redirect_to login_path unless pending_user
+  end
+
+  def create
+    user = pending_user
+    return redirect_to login_path unless user
+
+    session[:otp_attempts] = session[:otp_attempts].to_i + 1
+    if session[:otp_attempts] > MAX_ATTEMPTS
+      clear_pending
+      flash[:error] = "Too many attempts -- log in again."
+      return redirect_to login_path
+    end
+
+    if user.verify_otp!(params[:code])
+      return_to = session[:return_to]
+      reset_session
+      self.current_user = user
+      flash[:notice] = "Logged in successfully"
+      redirect_to safe_return_to(return_to, :default => admin_path)
+    else
+      flash.now[:error] = "That code did not match."
+      render :new
+    end
+  end
+
+  private
+
+    def pending_user
+      return nil if session[:pending_otp_user_id].blank?
+      if session[:otp_deadline].to_i < Time.now.to_i
+        clear_pending
+        return nil
+      end
+      @pending_user ||= User.find_by(:id => session[:pending_otp_user_id])
+    end
+
+    def clear_pending
+      session.delete(:pending_otp_user_id)
+      session.delete(:otp_deadline)
+      session.delete(:otp_attempts)
+    end
+end