chore: move uninterlace to image processor job
This commit is contained in:
parent
00c905a615
commit
0511a84abf
3 changed files with 101 additions and 13 deletions
|
@ -22,6 +22,7 @@ class ImageProcessorJob < ApplicationJob
|
||||||
return if ActiveStorage::Attachment.find_by(blob_id: blob.id)&.record_type == "ActiveStorage::VariantRecord"
|
return if ActiveStorage::Attachment.find_by(blob_id: blob.id)&.record_type == "ActiveStorage::VariantRecord"
|
||||||
|
|
||||||
auto_rotate(blob) if ["image/jpeg", "image/jpg"].include?(blob.content_type)
|
auto_rotate(blob) if ["image/jpeg", "image/jpg"].include?(blob.content_type)
|
||||||
|
uninterlace(blob) if blob.content_type == "image/png"
|
||||||
create_representations(blob) if blob.representation_required?
|
create_representations(blob) if blob.representation_required?
|
||||||
add_watermark(blob) if blob.watermark_pending?
|
add_watermark(blob) if blob.watermark_pending?
|
||||||
end
|
end
|
||||||
|
@ -40,6 +41,16 @@ class ImageProcessorJob < ApplicationJob
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def uninterlace(blob)
|
||||||
|
blob.open do |file|
|
||||||
|
processed = UninterlaceService.new.process(file)
|
||||||
|
return if processed.blank?
|
||||||
|
|
||||||
|
blob.upload(processed)
|
||||||
|
blob.save!
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
def create_representations(blob)
|
def create_representations(blob)
|
||||||
blob.attachments.each do |attachment|
|
blob.attachments.each do |attachment|
|
||||||
next unless attachment&.representable?
|
next unless attachment&.representable?
|
||||||
|
|
23
app/services/uninterlace_service.rb
Normal file
23
app/services/uninterlace_service.rb
Normal file
|
@ -0,0 +1,23 @@
|
||||||
|
# frozen_string_literal: true
|
||||||
|
|
||||||
|
class UninterlaceService
|
||||||
|
def process(file)
|
||||||
|
uninterlace_png(file)
|
||||||
|
end
|
||||||
|
|
||||||
|
private
|
||||||
|
|
||||||
|
def uninterlace_png(uploaded_file)
|
||||||
|
if interlaced?(uploaded_file.to_path)
|
||||||
|
chunky_img = ChunkyPNG::Image.from_io(uploaded_file.to_io)
|
||||||
|
chunky_img.save(uploaded_file.to_path, interlace: false)
|
||||||
|
uploaded_file.reopen(uploaded_file.to_path, 'rb')
|
||||||
|
end
|
||||||
|
uploaded_file
|
||||||
|
end
|
||||||
|
|
||||||
|
def interlaced?(png_path)
|
||||||
|
png = MiniMagick::Image.open(png_path)
|
||||||
|
png.data["interlace"] != "None"
|
||||||
|
end
|
||||||
|
end
|
|
@ -13,6 +13,7 @@ describe ImageProcessorJob, type: :job do
|
||||||
let(:antivirus_pending) { false }
|
let(:antivirus_pending) { false }
|
||||||
let(:watermark_service) { instance_double("WatermarkService") }
|
let(:watermark_service) { instance_double("WatermarkService") }
|
||||||
let(:auto_rotate_service) { instance_double("AutoRotateService") }
|
let(:auto_rotate_service) { instance_double("AutoRotateService") }
|
||||||
|
let(:uninterlace_service) { instance_double("UninterlaceService") }
|
||||||
|
|
||||||
before do
|
before do
|
||||||
virus_scanner_mock = instance_double("ActiveStorage::VirusScanner", pending?: antivirus_pending)
|
virus_scanner_mock = instance_double("ActiveStorage::VirusScanner", pending?: antivirus_pending)
|
||||||
|
@ -23,9 +24,6 @@ describe ImageProcessorJob, type: :job do
|
||||||
|
|
||||||
allow(WatermarkService).to receive(:new).and_return(watermark_service)
|
allow(WatermarkService).to receive(:new).and_return(watermark_service)
|
||||||
allow(watermark_service).to receive(:process).and_return(true)
|
allow(watermark_service).to receive(:process).and_return(true)
|
||||||
|
|
||||||
allow(AutoRotateService).to receive(:new).and_return(auto_rotate_service)
|
|
||||||
allow(auto_rotate_service).to receive(:process).and_return(true)
|
|
||||||
end
|
end
|
||||||
|
|
||||||
context "when the blob is not scanned yet" do
|
context "when the blob is not scanned yet" do
|
||||||
|
@ -37,11 +35,33 @@ describe ImageProcessorJob, type: :job do
|
||||||
end
|
end
|
||||||
|
|
||||||
describe 'autorotate' do
|
describe 'autorotate' do
|
||||||
|
let(:blob_info) do
|
||||||
|
{
|
||||||
|
filename: file.original_filename,
|
||||||
|
byte_size: file.size,
|
||||||
|
checksum: Digest::SHA256.file(file.path),
|
||||||
|
content_type: file.content_type,
|
||||||
|
# we don't want to run virus scanner on this file
|
||||||
|
metadata: { virus_scan_result: ActiveStorage::VirusScanner::SAFE }
|
||||||
|
}
|
||||||
|
end
|
||||||
|
|
||||||
|
let(:blob) do
|
||||||
|
blob = ActiveStorage::Blob.create_before_direct_upload!(**blob_info)
|
||||||
|
blob.upload(file)
|
||||||
|
blob
|
||||||
|
end
|
||||||
|
|
||||||
|
before do
|
||||||
|
allow(AutoRotateService).to receive(:new).and_return(auto_rotate_service)
|
||||||
|
allow(auto_rotate_service).to receive(:process).and_return(true)
|
||||||
|
end
|
||||||
|
|
||||||
context "when image is not a jpg" do
|
context "when image is not a jpg" do
|
||||||
let(:rotated_file) { Tempfile.new("rotated.png") }
|
let(:file) { fixture_file_upload('spec/fixtures/files/uninterlaced-black.png', 'image/png') }
|
||||||
|
|
||||||
before do
|
before do
|
||||||
allow(rotated_file).to receive(:size).and_return(100)
|
allow(file).to receive(:size).and_return(100)
|
||||||
end
|
end
|
||||||
|
|
||||||
it "it does not process autorotate" do
|
it "it does not process autorotate" do
|
||||||
|
@ -52,6 +72,7 @@ describe ImageProcessorJob, type: :job do
|
||||||
|
|
||||||
context "when image is a jpg " do
|
context "when image is a jpg " do
|
||||||
let(:rotated_file) { Tempfile.new("rotated.jpg") }
|
let(:rotated_file) { Tempfile.new("rotated.jpg") }
|
||||||
|
let(:file) { fixture_file_upload('spec/fixtures/files/image-rotated.jpg', 'image/jpeg') }
|
||||||
|
|
||||||
before do
|
before do
|
||||||
allow(rotated_file).to receive(:size).and_return(100)
|
allow(rotated_file).to receive(:size).and_return(100)
|
||||||
|
@ -118,34 +139,67 @@ describe ImageProcessorJob, type: :job do
|
||||||
describe 'watermark' do
|
describe 'watermark' do
|
||||||
context "when watermark is already done" do
|
context "when watermark is already done" do
|
||||||
before do
|
before do
|
||||||
allow(blob).to receive(:watermark_done?).and_return(true)
|
allow(blob_jpg).to receive(:watermark_done?).and_return(true)
|
||||||
end
|
end
|
||||||
|
|
||||||
it "does not process the watermark" do
|
it "does not process the watermark" do
|
||||||
expect(watermark_service).not_to receive(:process)
|
expect(watermark_service).not_to receive(:process)
|
||||||
described_class.perform_now(blob)
|
described_class.perform_now(blob_jpg)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
context "when the blob is ready to be watermarked" do
|
context "when the blob is ready to be watermarked" do
|
||||||
let(:watermarked_file) { Tempfile.new("watermarked.png") }
|
let(:watermarked_file) { Tempfile.new("watermarked.jpg") }
|
||||||
|
|
||||||
before do
|
before do
|
||||||
allow(watermarked_file).to receive(:size).and_return(100)
|
allow(watermarked_file).to receive(:size).and_return(100)
|
||||||
allow(blob).to receive(:watermark_pending?).and_return(true)
|
allow(blob_jpg).to receive(:watermark_pending?).and_return(true)
|
||||||
end
|
end
|
||||||
|
|
||||||
it "processes the blob with watermark" do
|
it "processes the blob with watermark" do
|
||||||
expect(watermark_service).to receive(:process).and_return(watermarked_file)
|
expect(watermark_service).to receive(:process).and_return(watermarked_file)
|
||||||
|
|
||||||
expect {
|
expect {
|
||||||
described_class.perform_now(blob)
|
described_class.perform_now(blob_jpg)
|
||||||
}.to change {
|
}.to change {
|
||||||
blob.reload.checksum
|
blob_jpg.reload.checksum
|
||||||
}
|
}
|
||||||
|
|
||||||
expect(blob.byte_size).to eq(100)
|
expect(blob_jpg.byte_size).to eq(100)
|
||||||
expect(blob.watermarked_at).to be_present
|
expect(blob_jpg.watermarked_at).to be_present
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
describe 'uninterlace' do
|
||||||
|
let(:blob_info) do
|
||||||
|
{
|
||||||
|
filename: file.original_filename,
|
||||||
|
byte_size: file.size,
|
||||||
|
checksum: Digest::SHA256.file(file.path),
|
||||||
|
content_type: file.content_type,
|
||||||
|
# we don't want to run virus scanner on this file
|
||||||
|
metadata: { virus_scan_result: ActiveStorage::VirusScanner::SAFE }
|
||||||
|
}
|
||||||
|
end
|
||||||
|
|
||||||
|
let(:blob) do
|
||||||
|
blob = ActiveStorage::Blob.create_before_direct_upload!(**blob_info)
|
||||||
|
blob.upload(file)
|
||||||
|
blob
|
||||||
|
end
|
||||||
|
|
||||||
|
before do
|
||||||
|
allow(UninterlaceService).to receive(:new).and_return(uninterlace_service)
|
||||||
|
end
|
||||||
|
|
||||||
|
context "when file is interlaced" do
|
||||||
|
let(:file) { fixture_file_upload('spec/fixtures/files/interlaced-black.png', 'image/png') }
|
||||||
|
let(:uninterlaced_file) { fixture_file_upload('spec/fixtures/files/uninterlaced-black.png', 'image/png') }
|
||||||
|
|
||||||
|
it "it process uninterlace" do
|
||||||
|
expect(uninterlace_service).to receive(:process).and_return(uninterlaced_file)
|
||||||
|
described_class.perform_now(blob)
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue