A Rake middleware to specify to OmniAuth where to redirect after successful authentication

application.rb

# config/application.rb

config.middleware.use 'StoreRedirectTo'

# and make sure you autoload the contents of /lib

store_redirect_to.rb

# lib/store_redirect_to.rb

class StoreRedirectTo
  def initialize(app)
    @app = app
  end

  def call(env)
    request = Rack::Request.new(env)
    path = request.path
    signin  = path.match(%r(^/auth/[^/]+$)) && path != '/auth/failure'
    signout = path == '/users/signout'
    if signin || signout
      redirect = request.params['redirect_to'] || request.referer
      request.session['redirect_to'] = redirect if redirect
    end

    @app.call(env)
  end
end

your_sessions_controller.rb

# after successful sign-in or sign-out
redirect_to session[:redirect_to] || root_path
session[:redirect_to] = nil