Move logic to image_service.

1 files changed, 39 insertions, 0 deletions

diff --git a/app/services/projects/image_service.rb b/app/services/projects/image_service.rb
new file mode 100644
index 00000000000..c79ddddd972
--- /dev/null
+++ b/app/services/projects/image_service.rb
@@ -0,0 +1,39 @@
+module Projects
+  class ImageService < BaseService
+    include Rails.application.routes.url_helpers
+    def initialize(repository, params, root_url)
+      @repository, @params, @root_url = repository, params.dup, root_url
+    end
+
+    def execute
+      uploader = FileUploader.new('uploads', upload_path, accepted_images)
+      image = @params['markdown_img']
+
+      if image && correct_mime_type?(image)
+        alt = image.original_filename
+        uploader.store!(image)
+        link = {
+                 'alt' => File.basename(alt, '.*'),
+                 'url' => File.join(@root_url, uploader.url) 
+               }
+      else
+        link = nil
+      end
+    end
+
+  protected
+
+    def upload_path
+      base_dir = FileUploader.generate_dir
+      File.join(@repository.path_with_namespace, base_dir)
+    end
+
+    def accepted_images
+      %w(png jpg jpeg gif)
+    end
+
+    def correct_mime_type?(image)
+      accepted_images.map{ |format| image.content_type.include? format }.any?
+    end
+  end
+end