add documents editor

2026-06-10 15:12:16 +03:00
parent ac66809a05
commit 13874b8830
19 changed files with 4130 additions and 56 deletions
@@ -0,0 +1,69 @@
+# frozen_string_literal: true
+
+module Templates
+  module CreateDocumentCrop
+    MAX_SCAN_SIZE = 1400
+
+    module_function
+
+    def call(template, attachment, params)
+      scan = params[:scan]
+
+      bytes, width, height = Leptonica.crop_document(attachment.download, params[:corners].map(&:to_h),
+                                                     scan:,
+                                                     rotate: params[:rotate]&.to_i,
+                                                     flip_h: params[:flip_h],
+                                                     flip_v: params[:flip_v])
+
+      image = load_image(bytes, width, height)
+      image = pad_scan_image(image, template.account) if scan
+
+      data = scan ? encode_png(image) : encode_jpeg(image)
+
+      create_document!(template, attachment, data, scan)
+    end
+
+    def create_document!(template, attachment, data, scan)
+      blob = ActiveStorage::Blob.create_and_upload!(
+        io: StringIO.new(data),
+        filename: "#{attachment.filename.base}.#{scan ? 'png' : 'jpg'}",
+        metadata: { identified: true, analyzed: true },
+        content_type: scan ? 'image/png' : 'image/jpeg'
+      )
+
+      document = template.documents.create!(blob:)
+
+      Templates::ProcessDocument.call(document, data)
+
+      document
+    end
+
+    def load_image(bytes, width, height)
+      Vips::Image.new_from_memory_copy(bytes, width, height, 4, :uchar)
+                 .extract_band(0, n: 3)
+                 .copy(interpretation: :srgb)
+    end
+
+    def pad_scan_image(image, account)
+      scale = MAX_SCAN_SIZE / [image.width, image.height].max.to_f
+      image = image.resize(scale) if scale < 1
+
+      base_size = Templates::ModifyDocuments.default_page_size(account)
+      page_width, page_height = image.width > image.height ? base_size.reverse : base_size
+
+      target_width = [image.width, (image.height * page_width / page_height.to_f).round].max
+      target_height = [image.height, (image.width * page_height / page_width.to_f).round].max
+
+      image.gravity('centre', target_width, target_height, extend: :background, background: 255)
+    end
+
+    def encode_png(image)
+      image.write_to_buffer(Templates::ProcessDocument::FORMAT,
+                            compression: 6, filter: 0, bitdepth: 4, palette: true, dither: 0, strip: true)
+    end
+
+    def encode_jpeg(image)
+      image.write_to_buffer('.jpg', Q: 90, strip: true)
+    end
+  end
+end