From 67492cdd1545232b66cc04ad2f385a71e96cf951 Mon Sep 17 00:00:00 2001
From: sto <puz.github.sto.happiness841@passmail.net>
Date: Thu, 19 Jun 2025 11:30:21 +0200
Subject: [PATCH] Update completion -> back to contestant

---
 app/controllers/completions_controller.rb | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/app/controllers/completions_controller.rb b/app/controllers/completions_controller.rb
index d97fd40..a9de7af 100644
--- a/app/controllers/completions_controller.rb
+++ b/app/controllers/completions_controller.rb
@@ -52,7 +52,11 @@ class CompletionsController < ApplicationController
     @completion.contestant_id = params[:contestant_id] if params[:contestant_id]
     if @completion.update(completion_params)
       extend_completions!(@completion.contestant)
-      redirect_to @contest
+      if @contestant
+        redirect_to edit_contest_contestant_path(@contest, @contestant)
+      else
+        redirect_to @contest
+      end
     else
       if @contestant
         @action_name = t("helpers.buttons.back_to_contestant")
@@ -82,8 +86,8 @@ class CompletionsController < ApplicationController
   def set_contestant
     if params.key?(:contestant_id)
       @contestant = Contestant.find(params[:contestant_id])
-    else
-      @contetant = nil
+    elsif params[:completion].key?(:contestant_id)
+      @contestant = Contestant.find(params[:completion][:contestant_id])
     end
   end