From cab670b1ad94d35481400fa992251a5938d2d05b Mon Sep 17 00:00:00 2001
From: Frederic Merizen <frederic.merizen@gmail.com>
Date: Thu, 22 Mar 2018 09:31:07 +0100
Subject: [PATCH] [#1677] Simplify code

---
 app/controllers/invites_controller.rb | 19 ++++++++-----------
 1 file changed, 8 insertions(+), 11 deletions(-)

diff --git a/app/controllers/invites_controller.rb b/app/controllers/invites_controller.rb
index 1e029e731..8e5518d7f 100644
--- a/app/controllers/invites_controller.rb
+++ b/app/controllers/invites_controller.rb
@@ -1,16 +1,15 @@
 class InvitesController < ApplicationController
-  before_action :gestionnaire_or_user?
+  before_action :ensure_user_signed_in
 
   def create
-    email_sender = @current_devise_profil.email
-
-    class_var = @current_devise_profil.class == User ? InviteUser : InviteGestionnaire
-    dossier = @current_devise_profil.dossiers.find(params[:dossier_id])
-
     email = params[:email].downcase
 
-    user = User.find_by(email: email)
-    invite = class_var.create(dossier: dossier, user: user, email: email, email_sender: email_sender)
+    invite = InviteUser.create(
+      dossier: current_user.dossiers.find(params[:dossier_id]),
+      user: User.find_by(email: email),
+      email: email,
+      email_sender: current_user.email
+    )
 
     if invite.valid?
       if invite.user.present?
@@ -29,11 +28,9 @@ class InvitesController < ApplicationController
 
   private
 
-  def gestionnaire_or_user?
+  def ensure_user_signed_in
     if !user_signed_in?
       return redirect_to root_path
     end
-
-    @current_devise_profil = current_user if user_signed_in?
   end
 end