Files controller handle redirect to remote storage. Added config block to carrierwave init

2 files changed, 13 insertions, 9 deletions

diff --git a/app/controllers/files_controller.rb b/app/controllers/files_controller.rb
index 3cd2e77322c..bf30de565ed 100644
--- a/app/controllers/files_controller.rb
+++ b/app/controllers/files_controller.rb
@@ -1,12 +1,16 @@
 class FilesController < ApplicationController
   def download
     note = Note.find(params[:id])
+    uploader = note.attachment
 
-    if can?(current_user, :read_project, note.project)
-      uploader = note.attachment
-      send_file uploader.file.path, disposition: 'attachment'
+    if uploader.file_storage?
+      if can?(current_user, :read_project, note.project)
+        send_file uploader.file.path, disposition: 'attachment'
+      else
+        not_found!
+      end
     else
-      not_found!
+      redirect_to uploader.url
     end
   end
 end
diff --git a/app/uploaders/attachment_uploader.rb b/app/uploaders/attachment_uploader.rb
index 960e71ef640..c0afe9686f1 100644
--- a/app/uploaders/attachment_uploader.rb
+++ b/app/uploaders/attachment_uploader.rb
@@ -21,10 +21,10 @@ class AttachmentUploader < CarrierWave::Uploader::Base
   end
 
   def secure_url
-    if self.class.storage == CarrierWave::Storage::File
-      "/files/#{model.class.to_s.underscore}/#{model.id}/#{file.filename}"
-    else
-      url
-    end
+    "/files/#{model.class.to_s.underscore}/#{model.id}/#{file.filename}"
+  end
+
+  def file_storage?
+    self.class.storage == CarrierWave::Storage::File
   end
 end