Merge pull request #349 from DigitalWPI/pdfviewer_fix_10_26_21

Pdfviewer fix 10 26 21
DigitalWPI · Oct 28, 2021 · 88b4a05 · 88b4a05
2 parents cca56b7 + 3cc31b7
commit 88b4a05
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, alert: "Document not found"
+    end
   end
 end