mirror of https://github.com/docusealco/docuseal
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
3.4 KiB
98 lines
3.4 KiB
# frozen_string_literal: true
|
|
|
|
module Templates
|
|
module ProcessDocument
|
|
DPI = 200
|
|
FORMAT = '.jpg'
|
|
ATTACHMENT_NAME = 'preview_images'
|
|
SECURED_ATTACHMENT_NAME = 'preview_secured_images'
|
|
|
|
PDF_CONTENT_TYPE = 'application/pdf'
|
|
Q = 35
|
|
MAX_WIDTH = 1400
|
|
|
|
module_function
|
|
|
|
def call(attachment, data)
|
|
if attachment.content_type == PDF_CONTENT_TYPE
|
|
generate_pdf_preview_images(attachment, data)
|
|
elsif attachment.image?
|
|
generate_preview_image(attachment, data)
|
|
end
|
|
|
|
attachment
|
|
end
|
|
|
|
def generate_preview_image(attachment, data)
|
|
ActiveStorage::Attachment.where(name: ATTACHMENT_NAME, record: attachment).destroy_all
|
|
|
|
image = Vips::Image.new_from_buffer(data, '')
|
|
image = image.autorot.resize(MAX_WIDTH / image.width.to_f)
|
|
|
|
io = StringIO.new(image.write_to_buffer(FORMAT, Q: Q, interlace: true))
|
|
|
|
ActiveStorage::Attachment.create!(
|
|
blob: ActiveStorage::Blob.create_and_upload!(
|
|
io:, filename: "0#{FORMAT}",
|
|
metadata: { analyzed: true, identified: true, width: image.width, height: image.height }
|
|
),
|
|
name: ATTACHMENT_NAME,
|
|
record: attachment
|
|
)
|
|
end
|
|
|
|
def generate_pdf_preview_images(attachment, data)
|
|
ActiveStorage::Attachment.where(name: ATTACHMENT_NAME, record: attachment).destroy_all
|
|
number_of_pages = HexaPDF::Document.new(io: StringIO.new(data)).pages.size - 1
|
|
|
|
(0..number_of_pages).each do |page_number|
|
|
page = Vips::Image.new_from_buffer(data, '', dpi: DPI, page: page_number)
|
|
page = page.resize(MAX_WIDTH / page.width.to_f)
|
|
|
|
io = StringIO.new(page.write_to_buffer(FORMAT, Q: Q, interlace: true))
|
|
|
|
ActiveStorage::Attachment.create!(
|
|
blob: ActiveStorage::Blob.create_and_upload!(
|
|
io:, filename: "#{page_number}#{FORMAT}",
|
|
metadata: { analyzed: true, identified: true, width: page.width, height: page.height }
|
|
),
|
|
name: ATTACHMENT_NAME,
|
|
record: attachment
|
|
)
|
|
end
|
|
end
|
|
|
|
def generate_pdf_secured_preview_images(template, attachment, data)
|
|
ActiveStorage::Attachment.where(name: SECURED_ATTACHMENT_NAME, record: attachment).destroy_all
|
|
number_of_pages = PDF::Reader.new(StringIO.new(data)).pages.size - 1
|
|
(0..number_of_pages).each do |page_number|
|
|
pdf = Vips::Image.new_from_buffer(data, '', dpi: DPI, page: page_number)
|
|
pdf = pdf.resize(MAX_WIDTH / pdf.width.to_f)
|
|
redacted_boxes = template.fields.select { |field| field['type'] == 'redact' && field['areas'][0]['page'] == page_number }
|
|
if !redacted_boxes.empty?
|
|
redacted_boxes.each do |box|
|
|
x = (box['areas'][0]['x'] * pdf.width).to_i
|
|
y = (box['areas'][0]['y'] * pdf.height).to_i
|
|
w = (box['areas'][0]['w'] * pdf.width).to_i
|
|
h = (box['areas'][0]['h'] * pdf.height).to_i
|
|
black_rect = Vips::Image.black(w, h)
|
|
pdf = pdf.insert(black_rect, x, y)
|
|
end
|
|
end
|
|
|
|
io = StringIO.new(pdf.write_to_buffer(FORMAT, Q: Q))
|
|
|
|
ActiveStorage::Attachment.create!(
|
|
blob: ActiveStorage::Blob.create_and_upload!(
|
|
io: io, filename: "#{page_number}#{FORMAT}",
|
|
metadata: { analyzed: true, identified: true, width: pdf.width, height: pdf.height }
|
|
),
|
|
name: SECURED_ATTACHMENT_NAME,
|
|
record: attachment
|
|
)
|
|
end
|
|
end
|
|
|
|
end
|
|
end
|