ログイン時にアクセスURLへリダイレクト #18

app/controllers/application_controller.rb

@@ -6,10 +6,15 @@ class ApplicationController < ActionController::Base
   def require_login
     unless user_signed_in?
       flash[:alert] = 'You need Login!'
+      session[:login_redirect_to] = request.url
       redirect_to root_path
     end
   end
 
+  def after_sign_in_path_for(resource)
+    session[:login_redirect_to] || request.env['omniauth.origin'] || stored_location_for(resource) || root_path
+  end
+
   rescue_from(ActionController::ParameterMissing) do |parameter_missing_exception|
     render text: "Required parameter missing: #{parameter_missing_exception.param}", status: :bad_request
   end

app/controllers/users/omniauth_callbacks_controller.rb

@@ -3,7 +3,7 @@ class Users::OmniauthCallbacksController < Devise::OmniauthCallbacksController
 
     email = request.env['omniauth.auth'].info['email']
 
-    # reject if email is not zigexn nor ventura
+    # reject if email is not zigexn nor ventura.
     if email !~ /@zigexn\.co\.jp$/ && email !~ /@zigexn\.vn$/
       redirect_to root_path, flash: { alert: 'Your email address is not permitted.' }
       return

spec/controllers/application_controller_spec.rb

+require 'spec_helper'
+
+describe ApplicationController do
+
+  pending '#require_login'
+
+  pending '#after_sign_in_path_for'
+
+  pending '#rescue_from'
+
+end