app/controllers/users/emails_controller.rb in shieldify-0.2.2.pre.alpha vs app/controllers/users/emails_controller.rb in shieldify-0.2.3.pre.alpha

@@ -2,17 +2,21 @@
   class EmailsController < ActionController::Base
     def show
       token = params[:token]
       user = User.confirm_email_by_token(token)
 
-      if user.errors.blank?
-        response.headers['X-Shfy-Message'] = I18n.t("shieldify.controllers.emails.confirmation.success_messages")
-        response.headers['X-Shfy-Status'] = 'success'
-      else
-        response.headers['X-Shfy-Message'] = user.errors.full_messages.last
-        response.headers['X-Shfy-Status'] = 'error'
-      end
+      message = user.errors.blank? ? I18n.t("shieldify.controllers.emails.confirmation.success_messages") : user.errors.full_messages.last
+      status = user.errors.blank? ? 'success' : 'error'
 
+      set_cookie('shfy_message', message)
+      set_cookie('shfy_status', status)
+
       redirect_to(Shieldify::Configuration.before_confirmation_url, allow_other_host: true)
     end
+
+    private
+
+    def set_cookie(name, value)
+      response.set_cookie(name, { value: value, expires: 1.hour.from_now })
+    end
   end
-end
\ No newline at end of file
+end