Adding a check if a document exists, redirecting to homepage with not…

…ice if it doesn't
DigitalWPI · Oct 26, 2021 · b148ff3 · b148ff3
1 parent c34428d
commit b148ff3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/app/controllers/pdfviewer_controller.rb b/app/controllers/pdfviewer_controller.rb
@@ -1,7 +1,11 @@
 class PdfviewerController < ApplicationController
   layout "pdfviewer"
   def index
-    @id = params[:id]
-    @parent_id = FileSet.find(@id).parent_id
+    if FileSet.exists?(params[:id])
+      @id = params[:id]
+      @parent_id = FileSet.find(@id).parent_id
+    else
+      redirect_to root_path, notice: "Document not found"
+    end
   end
 end