lib/generators/jt/user/templates/controllers/concerns/current_user.rb in jt-rails-generator-user-1.2.1 vs lib/generators/jt/user/templates/controllers/concerns/current_user.rb in jt-rails-generator-user-1.2.2

@@ -13,12 +13,16 @@
 	def current_user
 		@current_user ||= User.find_by_id(session[:user_id]) if session[:user_id]
 		@current_user
 	end
 
-	def store_location
-		session[:return_to] = request.fullpath if request.get? && !request.xhr?
-	end
+  def store_location(url = nil)
+    if url
+      session[:return_to] = url
+    elsif request.get? && !request.xhr?
+      session[:return_to] = request.fullpath
+    end
+  end
 
 	def require_user
 		if !current_user
 			store_location
 			redirect_to login_url
\ No newline at end of file