Handle file not found exception from storage as 404

Author: gabsource

src/Http/Controllers/DownloadController.php

@@ -3,32 +3,40 @@
 namespace Bnb\Laravel\Attachments\Http\Controllers;
 
 use Bnb\Laravel\Attachments\Contracts\AttachmentContract;
+use Illuminate\Contracts\Filesystem\FileNotFoundException;
 use Illuminate\Http\Request;
 use Illuminate\Routing\Controller;
 use Lang;
 
 class DownloadController extends Controller
 {
+
     /**
      * Attachment model
      *
      * @var AttachmentContract
      */
     protected $model;
 
+
     public function __construct(AttachmentContract $model)
     {
         $this->model = $model;
     }
 
+
     public function download($id, Request $request)
     {
         $disposition = ($disposition = $request->input('disposition')) === 'inline' ? $disposition : 'attachment';
 
         if ($file = $this->model->where('uuid', $id)->first()) {
-            /** @var \Bnb\Laravel\Attachments\Attachment $file */
-            if ( ! $file->output($disposition)) {
-                abort(403, Lang::get('attachments::messages.errors.access_denied'));
+            try {
+                /** @var \Bnb\Laravel\Attachments\Attachment $file */
+                if ( ! $file->output($disposition)) {
+                    abort(403, Lang::get('attachments::messages.errors.access_denied'));
+                }
+            } catch (FileNotFoundException $e) {
+                abort(404, Lang::get('attachments::messages.errors.file_not_found'));
             }
         }
 

src/Http/Controllers/ShareController.php

@@ -6,6 +6,7 @@
 use Carbon\Carbon;
 use Crypt;
 use Illuminate\Contracts\Encryption\DecryptException;
+use Illuminate\Contracts\Filesystem\FileNotFoundException;
 use Illuminate\Http\Request;
 use Illuminate\Routing\Controller;
 use Lang;
@@ -46,10 +47,14 @@ public function download($token, Request $request)
         }
 
         if ($file = $this->model->where('uuid', $id)->first()) {
+            try {
             /** @var AttachmentContract $file */
             if ( ! $file->output($disposition)) {
                 abort(403, Lang::get('attachments::messages.errors.access_denied'));
             }
+            } catch (FileNotFoundException $e) {
+                abort(404, Lang::get('attachments::messages.errors.file_not_found'));
+            }
         }
 
         abort(404, Lang::get('attachments::messages.errors.file_not_found'));