Back buttons

app/controllers/completions_controller.rb

@@ -2,16 +2,27 @@ class CompletionsController < ApplicationController
   include CompletionsConcern
 
   before_action :set_contest
+  before_action :set_contestant
   before_action :set_data, only: %i[ create edit new update ]
   before_action :set_completion, only: %i[ destroy edit update ]
 
   def edit
     authorize @contest
+
+    if @contestant
+      @action_name = t("helpers.buttons.back_to_contestant")
+      @action_path = edit_contest_contestant_path(@contest, @contestant)
+    end
   end
 
   def new
     authorize @contest
 
+    if @contestant
+      @action_name = t("helpers.buttons.back_to_contestant")
+      @action_path = edit_contest_contestant_path(@contest, @contestant)
+    end
+
     @completion = Completion.new
     if params[:contestant_id]
       @completion.contestant_id = params[:contestant_id]
@@ -27,6 +38,10 @@ class CompletionsController < ApplicationController
       extend_completions!(@completion.contestant)
       redirect_to contest_path(@contest)
     else
+      if @contestant
+        @action_name = t("helpers.buttons.back_to_contestant")
+        @action_path = edit_contest_contestant_path(@contest, @contestant)
+      end
       render :new, status: :unprocessable_entity
     end
   end
@@ -39,6 +54,10 @@ class CompletionsController < ApplicationController
       extend_completions!(@completion.contestant)
       redirect_to @contest
     else
+      if @contestant
+        @action_name = t("helpers.buttons.back_to_contestant")
+        @action_path = edit_contest_contestant_path(@contest, @contestant)
+      end
       render :edit, status: :unprocessable_entity
     end
   end
@@ -60,6 +79,14 @@ class CompletionsController < ApplicationController
     @contest = Contest.find(params[:contest_id])
   end
 
+  def set_contestant
+    if params.key?(:contestant_id)
+      @contestant = Contestant.find(params[:contestant_id])
+    else
+      @contetant = nil
+    end
+  end
+
   def set_data
     @contestants = @contest.contestants
     @puzzles = @contest.puzzles