diff --git a/app/models/post_replacement.rb b/app/models/post_replacement.rb index 0351e1d7a..8396daf73 100644 --- a/app/models/post_replacement.rb +++ b/app/models/post_replacement.rb @@ -19,7 +19,14 @@ class PostReplacement < ApplicationRecord def process! transaction do - upload = Upload.create!(file: replacement_file, source: replacement_url, rating: post.rating, tag_string: self.tags) + upload = Upload.create!( + file: replacement_file, + source: replacement_url, + rating: post.rating, + tag_string: self.tags, + replaced_post: post, + ) + upload.process_upload upload.update(status: "completed", post_id: post.id) diff --git a/app/models/upload.rb b/app/models/upload.rb index 828d767e7..d0b9abd8b 100644 --- a/app/models/upload.rb +++ b/app/models/upload.rb @@ -7,7 +7,7 @@ class Upload < ApplicationRecord attr_accessor :file, :image_width, :image_height, :file_ext, :md5, :file_size, :as_pending, :artist_commentary_title, :artist_commentary_desc, :include_artist_commentary, - :referer_url, :downloaded_source + :referer_url, :downloaded_source, :replaced_post belongs_to :uploader, :class_name => "User" belongs_to :post before_validation :initialize_uploader, :on => :create @@ -22,7 +22,7 @@ class Upload < ApplicationRecord :tag_string, :status, :backtrace, :post_id, :md5_confirmation, :parent_id, :server, :artist_commentary_title, :artist_commentary_desc, :include_artist_commentary, - :referer_url + :referer_url, :replaced_post module ValidationMethods def uploader_is_not_limited @@ -46,7 +46,10 @@ class Upload < ApplicationRecord # Because uploads are processed serially, there's no race condition here. def validate_md5_uniqueness md5_post = Post.find_by_md5(md5) - if md5_post + + if md5_post && replaced_post + raise "duplicate: #{md5_post.id}" if replaced_post != md5_post + elsif md5_post raise "duplicate: #{md5_post.id}" end end diff --git a/test/unit/post_replacement_test.rb b/test/unit/post_replacement_test.rb index 619b70a92..5aaae9186 100644 --- a/test/unit/post_replacement_test.rb +++ b/test/unit/post_replacement_test.rb @@ -4,6 +4,15 @@ require 'helpers/iqdb_test_helper' class PostReplacementTest < ActiveSupport::TestCase include IqdbTestHelper + def upload_file(path, filename, &block) + Tempfile.open do |file| + file.write(File.read(path)) + file.seek(0) + uploaded_file = ActionDispatch::Http::UploadedFile.new(tempfile: file, filename: filename) + yield uploaded_file + end + end + def setup mock_iqdb_service! Delayed::Worker.delay_jobs = true # don't delete the old images right away @@ -168,7 +177,7 @@ class PostReplacementTest < ActiveSupport::TestCase assert_equal("cad1da177ef309bf40a117c17b8eecf5", @post.md5) assert_equal("cad1da177ef309bf40a117c17b8eecf5", Digest::MD5.file(@post.file_path).hexdigest) - assert_equal("https://i1.pixiv.net/img-zip-ugoira/img/2017/04/04/08/57/38/62247364_ugoira1920x1080.zip", @post.source) + assert_equal("https://i.pximg.net/img-zip-ugoira/img/2017/04/04/08/57/38/62247364_ugoira1920x1080.zip", @post.source) assert_equal([{"file"=>"000000.jpg", "delay"=>125}, {"file"=>"000001.jpg", "delay"=>125}], @post.pixiv_ugoira_frame_data.data) end end @@ -195,13 +204,9 @@ class PostReplacementTest < ActiveSupport::TestCase context "a post with an uploaded file" do should "work" do - Tempfile.open do |file| - file.write(File.read("#{Rails.root}/test/files/test.png")) - file.seek(0) - uploaded_file = ActionDispatch::Http::UploadedFile.new(tempfile: file, filename: "test.png") - - @post.replace!(replacement_file: uploaded_file, replacement_url: "") - assert_equal(@post.md5, Digest::MD5.file(file).hexdigest) + upload_file("#{Rails.root}/test/files/test.png", "test.png") do |file| + @post.replace!(replacement_file: file, replacement_url: "") + assert_equal(@post.md5, Digest::MD5.file(file.tempfile).hexdigest) assert_equal("file://test.png", @post.replacements.last.replacement_url) end end @@ -226,5 +231,15 @@ class PostReplacementTest < ActiveSupport::TestCase assert_equal(image_url, @post.replacements.last.replacement_url) end end + + context "a post with the same file" do + should "not raise a duplicate error" do + upload_file("#{Rails.root}/test/files/test.jpg", "test.jpg") do |file| + assert_nothing_raised do + @post.replace!(replacement_file: file, replacement_url: "") + end + end + end + end end end