don't redirect back to deleted files or models

manyfold3d · Mar 4, 2024 · cf22ebf · cf22ebf
1 parent aa87b6c
commit cf22ebf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/app/controllers/model_files_controller.rb b/app/controllers/model_files_controller.rb
@@ -51,7 +51,12 @@ def bulk_update
   def destroy
     authorize @file
     @file.delete_from_disk_and_destroy
-    redirect_back_or_to library_model_path(@library, @model), notice: t(".success")
+    if URI.parse(request.referer).path == library_model_model_file_path(@library, @model, @file)
+      # If we're coming from the file page itself, we can't go back there
+      redirect_to library_model_path(@library, @model), notice: t(".success")
+    else
+      redirect_back_or_to library_model_path(@library, @model), notice: t(".success")
+    end
   end
 
   private

diff --git a/app/controllers/models_controller.rb b/app/controllers/models_controller.rb
@@ -100,7 +100,12 @@ def bulk_update
   def destroy
     authorize @model
     @model.delete_from_disk_and_destroy
-    redirect_back_or_to library_path(@library), notice: t(".success")
+    if URI.parse(request.referer).path == library_model_path(@library, @model)
+      # If we're coming from the model page itself, we can't go back there
+      redirect_to library_path(@library), notice: t(".success")
+    else
+      redirect_back_or_to library_path(@library), notice: t(".success")
+    end
   end
 
   private