docgen-adapter 1.1.0 → 1.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 436def6a7f3f7cf576ee57d20432b0f81598d4e8cf203bd399a55016bbebfdb9
-  data.tar.gz: 2d306870e9a52bda078216596d734b0bb1b8736d647fcf739c8792b06054ba15
+  metadata.gz: afe6bdfcb9c5d9773d369ba750d43c797614ff24f9e9bd4c10eeb645b12e122e
+  data.tar.gz: cfdb37ee53d4d4ead59f99a7f5e1a767211c848c8109ed1f2b71ff84ae8e47d1
 SHA512:
-  metadata.gz: a1d79bff9ac3c1d4c846d042a9fb061ea9128a93061346f50a69e4155d4270152830ea77aea7197bd460089e2eb69f99290c11964d2879ab908a037ff5cbd88d
-  data.tar.gz: eeaf82f5f161738451e49cb11c2e5c7c6d0009a12ac3b9f3f282cd7237e1633afa865eb97cc03fec16168f7184a770c021d894bdbc58805479c5afda6f9600cf
+  metadata.gz: 0c9a445451d9637e8c601d9ea4f86fb2748144f4cafb82a9d2ebc08c4092c0b88fec3e0d0b37aaf909ac95ac5fa472d2448a0c8c9c0cb900b95157b0aa949825
+  data.tar.gz: eb0baf8397ce28db7ed2755931458a00168d1ea909b003af73b1de32197d5b95aed75df267050bbc4232eaa4ff3c828916d2f67dee33cfebe796bdbc7475293c

data/README.md

@@ -65,6 +65,14 @@ def json(params = {})
 end
 ```
 
+Добавьте метод `files` для обработки и передачи дополнительных файлов в задание генерации. В ответе метода `files` должен отдаваться массив ссылок (URL) на файлы либо массив локальный путей файлов.
+
+```ruby
+def files(params)
+  []
+end
+```
+
 Задайте название генерируемого файла в `metadata`. Например:
 
 ```ruby

data/lib/docgen_adapter.rb

@@ -4,20 +4,22 @@ require 'docgen_subject'
 
 class DocgenAdapter
 
-  def initialize(type:, doc_id:, debug: false)
+  def initialize(type:, doc_id:, debug: false, open: false)
     # аргументы сервиса
     @type = type
     @doc_id = doc_id
     @debug = Rails.env.development? && debug
+    @open = Rails.env.development? && open
     @task_id = SecureRandom.uuid
   end
 
   # выполнение генерации документа
   def generate(params = {})
-    @service_params = build_params(params)
+    @files = handle_files(params)
+    @task_params = build_task_params(params)
 
     # отображение параметров для проверки
-    puts JSON.pretty_generate(JSON.parse(@service_params)) if @debug
+    puts JSON.pretty_generate(JSON.parse(@task_params)) if @debug
 
     # подписка ка канал и ожидание сообщений в другом потоке
     Thread.new { subscribe_and_listen }
@@ -26,7 +28,14 @@ class DocgenAdapter
     sleep(0.5)
 
     # отправка сообщения с заданием
-    redis.publish('TASKS', @service_params)
+    redis.publish('TASKS', @task_params)
+
+    sleep(0.5)
+
+    # отправка файлов при наличии
+    @files.each do |file|
+      redis.publish("files-#{@task_id}", file[:content])
+    end
 
     # путь до сгенерированного файла в ответе
     filename
@@ -44,6 +53,10 @@ class DocgenAdapter
       end
   end
 
+  def document_instance
+    @document_instance ||= document_class.new
+  end
+
   # формирование названия генерируемого файла
   def filename
     @filename ||= "#{Dir.mktmpdir}/#{document_class.metadata_opts[:filename]}.#{@type}"
@@ -76,15 +89,55 @@ class DocgenAdapter
     end
   end
 
-  # формирование объекта с параметрами
-  def build_params(params)
-    {
+  # обработка доп. файлов для документа
+  def handle_files(params)
+    return [] unless document_class.method_defined?(:files)
+
+    document_instance.files(params).map do |file|
+      # обработка ссылки или локального пути для получения содержимого файла и расширения
+      if file =~ URI::regexp
+        opened = URI.open(file)
+        content = opened.read
+        extension = url_file_extname(opened.meta['content-type'])
+      else
+        content = File.read(file)
+        extension = File.extname(file).gsub('.', '')
+      end
+
+      # MD5 содержимого файла
+      md5 = Digest::MD5.base64digest(content)
+
+      { content:, extension:, md5: }
+    end
+  end
+
+  def url_file_extname(content_type)
+    case content_type
+    when 'image/png'
+      'png'
+    end
+  end
+
+  # формирование объекта с параметрами задания
+  def build_task_params(params)
+    # базовый объект параметров
+    object = {
       ИдентификаторЗадачи: @task_id,
       ИдентификаторДокумента: @doc_id,
       ТипДокумента: @type,
-      Md5ДополнительныхФайлов: [],
-      ПараметрыДокумента: document_class.new.json(params)
-    }.to_json
+      ПараметрыДокумента: document_instance.json(params)
+    }
+
+    # параметры доп. файлов
+    object[:ДополнительныеФайлы] =
+      @files.map do |file|
+        {
+          Md5: file[:md5],
+          Формат: file[:extension]
+        }
+      end
+
+    object.to_json
   end
 
   # обработка сообщения пришедшего в канал Redis
@@ -122,6 +175,9 @@ class DocgenAdapter
 
     # сообщение с файлом приходит последним - отписка от канала Redis
     redis.unsubscribe
+
+    # открывать файл для проверки
+    %x(open '#{filename}') if @open
   end
 
   def handle_exception(object)
@@ -130,7 +186,7 @@ class DocgenAdapter
     if Rails.env.development?
       puts text
     else
-      Rollbar.error(text, params: @service_params)
+      Rollbar.error(text, params: @task_params)
     end
   end
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: docgen-adapter
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.2.0
 platform: ruby
 authors:
 - Павел Бабин
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-10-23 00:00:00.000000000 Z
+date: 2024-11-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis