refactor upload service

app/logical/upload_service/replacer.rb

@@ -0,0 +1,141 @@
+class UploadService
+  class Replacer
+    attr_reader :post, :replacement
+
+    def initialize(post:, replacement:)
+      @post = post
+      @replacement = replacement
+    end
+
+    def comment_replacement_message(post, replacement)
+      %("#{replacement.creator.name}":[/users/#{replacement.creator.id}] replaced this post with a new image:\n\n#{replacement_message(post, replacement)})
+    end
+
+    def replacement_message(post, replacement)
+      linked_source = linked_source(replacement.replacement_url)
+      linked_source_was = linked_source(post.source_was)
+
+      <<-EOS.strip_heredoc
+        [table]
+          [tbody]
+            [tr]
+              [th]Old[/th]
+              [td]#{linked_source_was}[/td]
+              [td]#{post.md5_was}[/td]
+              [td]#{post.file_ext_was}[/td]
+              [td]#{post.image_width_was} x #{post.image_height_was}[/td]
+              [td]#{post.file_size_was.to_s(:human_size, precision: 4)}[/td]
+            [/tr]
+            [tr]
+              [th]New[/th]
+              [td]#{linked_source}[/td]
+              [td]#{post.md5}[/td]
+              [td]#{post.file_ext}[/td]
+              [td]#{post.image_width} x #{post.image_height}[/td]
+              [td]#{post.file_size.to_s(:human_size, precision: 4)}[/td]
+            [/tr]
+          [/tbody]
+        [/table]
+      EOS
+    end
+
+    def linked_source(source)
+      return nil if source.nil?
+
+      # truncate long sources in the middle: "www.pixiv.net...lust_id=23264933"
+      truncated_source = source.gsub(%r{\Ahttps?://}, "").truncate(64, omission: "...#{source.last(32)}")
+
+      if source =~ %r{\Ahttps?://}i
+        %("#{truncated_source}":[#{source}])
+      else
+        truncated_source
+      end
+    end
+
+    def undo!
+      undo_replacement = post.replacements.create(replacement_url: replacement.original_url)
+      undoer = Replacer.new(post: post, replacement: undo_replacement)
+      undoer.process!
+    end
+
+    def process!
+      preprocessor = Preprocessor.new(
+        rating: post.rating,
+        tag_string: replacement.tags,
+        source: replacement.replacement_url,
+        file: replacement.replacement_file,
+        original_post_id: post.id
+      )
+      upload = preprocessor.start!
+      return if upload.is_errored?
+      upload = preprocessor.finish!(upload)
+      return if upload.is_errored?
+      md5_changed = upload.md5 != post.md5
+      
+      if replacement.replacement_file.present?
+        replacement.replacement_url = "file://#{replacement.replacement_file.original_filename}"
+      elsif upload.downloaded_source.present?
+        replacement.replacement_url = upload.downloaded_source
+      end
+
+      if md5_changed
+        post.queue_delete_files(PostReplacement::DELETION_GRACE_PERIOD)
+      end
+
+      replacement.file_ext = upload.file_ext
+      replacement.file_size = upload.file_size
+      replacement.image_height = upload.image_height
+      replacement.image_width = upload.image_width
+      replacement.md5 = upload.md5
+
+      post.md5 = upload.md5
+      post.file_ext = upload.file_ext
+      post.image_width = upload.image_width
+      post.image_height = upload.image_height
+      post.file_size = upload.file_size
+      post.source = upload.downloaded_source || upload.source
+      post.tag_string = upload.tag_string
+
+      update_ugoira_frame_data(post, upload)
+
+      if md5_changed
+        CurrentUser.as(User.system) do
+          post.comments.create!(body: comment_replacement_message(post, replacement), do_not_bump_post: true)
+        end
+      end
+
+      if replacement.final_source.present?
+        post.update(source: replacement.final_source)
+      end
+
+      replacement.save!
+      post.save!
+
+      rescale_notes(post)
+    end
+
+    def rescale_notes(post)
+      x_scale = post.image_width.to_f  / post.image_width_before_last_save.to_f
+      y_scale = post.image_height.to_f / post.image_height_before_last_save.to_f
+
+      post.notes.each do |note|
+        note.reload
+        note.rescale!(x_scale, y_scale)
+      end
+    end
+
+    def update_ugoira_frame_data(post, upload)
+      post.pixiv_ugoira_frame_data.destroy if post.pixiv_ugoira_frame_data.present?
+
+      unless post.is_ugoira?
+        return
+      end
+
+      PixivUgoiraFrameData.create(
+        post_id: post.id,
+        data: upload.context["ugoira"]["frame_data"],
+        content_type: upload.context["ugoira"]["content_type"]
+      )
+    end
+  end
+end