postdoc 0.3.7 → 0.4.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4a8f25b2dcb78fb58ec199d021985eef4e9892c55cba5ad9b33b8a1aea14baba
-  data.tar.gz: f2ef8f9083917b3efbd212cdbc7a25c7f574db1a1a963bacf0a194151f2f8fbe
+  metadata.gz: 2562e31f953afcf27300148e00217e1599424efb39bbb1242e0d29f264260c28
+  data.tar.gz: 6b9271a82c70cbe57efd534b1b2d80b8f6be5e952ebdfa167f3cfabb944bbd77
 SHA512:
-  metadata.gz: 166ca4c618a7740e02edda59711e1b23a5555e2c4d479ac00b71d6d9dda9c613fdbbd125fe0053f75405881c21e7b31f9494ef5df997cd3afe32b1486b403997
-  data.tar.gz: 963c118d472a7d244fc4841bae66151f211804e796a1a1d8be3f3d1dae25a9f5274122c867abe99cd565a3bfc457a4516b94ce8065866fb929918226331816ce
+  metadata.gz: 1f71a3cf459f62e022b0f13853084ca2796fb642a6a9c681173b341f961dd257486a141459ec42fe7b3f4caf2c6c6c86aafec2dc7f14e23b7c8c775c8c356ebd
+  data.tar.gz: dec210aedf9353c0e96bf25e5b503088508fbe7a79661b75702ce5c372039f762be6d79f417b55313a83e5857ebbb39bd310b5a4304157b4dcbc072ec5853e01

data/lib/postdoc/batch.rb

@@ -0,0 +1,34 @@
+require 'postdoc/print_settings'
+require 'postdoc/job'
+
+module Postdoc
+  # This module represents a batch. At times we want to render multiple PDF's in
+  # series and batching saves a lot of overhead by not restarting chrome over
+  # and over. Practical applications are  "stitching" PDF's together to get
+  # around markup constraints in chrome or rendering a batch of reports all at
+  # once.
+  class Batch
+    def initialize(jobs: [])
+      @jobs = jobs
+    end
+
+    # Creates a job from a document and add to the jobs.
+    def add_document(document, settings: {})
+      settings = PrintSettings.new(**settings)
+      @jobs << Job.new(document, settings: settings)
+    end
+
+    def add_job(job)
+      @jobs << job
+    end
+
+    # returns the output of the config. Requires a {Postdoc::Client Client}
+    def result(client)
+      @jobs.map { |job| job.result(client) }
+    end
+
+    def cleanup
+      @jobs.each(&:cleanup)
+    end
+  end
+end

data/lib/postdoc/chrome_process.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 module Postdoc
+  # Spins up a Chrome process.
   class ChromeProcess
     attr_reader :pid, :port
 

data/lib/postdoc/client.rb

@@ -3,6 +3,7 @@
 require 'chrome_remote'
 
 module Postdoc
+
   class Client
     attr_accessor :client
 
@@ -12,28 +13,32 @@ module Postdoc
       raise 'ChromeClient couldn\'t launch' if @client.blank?
     end
 
+    # We should move away from passing options like this and collect them in
+    # the prinbt settings.
     def print_pdf_from_html(file_path,
-        header_template: false,
-        footer_template: false,
-        **options)
+        settings: PrintSettings.new)
 
       client.send_cmd 'Page.enable'
       client.send_cmd 'Page.navigate', url: "file://#{file_path}"
       client.wait_for 'Page.loadEventFired'
 
-      response = client.send_cmd 'Page.printToPDF', {
-        landscape: options[:landscape] || false,
-        printBackground: true,
-        marginTop: options[:margin_top] || 1,
-        marginBottom: options[:margin_bottom] || 1,
-        marginLeft: options[:margin_left] || 1,
-        marginRight: options[:margin_right] || 1,
-        displayHeaderFooter: !!(header_template || footer_template),
-        headerTemplate: header_template || '',
-        footerTemplate: footer_template || ''
-      }
-
-      client.send_cmd 'Browser.close'
+      # prevent race condition
+      sleep 0.1 if settings.slow_pc
+
+      response = client.send_cmd 'Page.printToPDF', settings.to_cmd
+
+      Base64.decode64 response['data']
+    end
+
+    def print_document(file_path, settings: PrintSettings.new)
+      client.send_cmd 'Page.enable'
+      client.send_cmd 'Page.navigate', url: "file://#{file_path}"
+      client.wait_for 'Page.loadEventFired'
+
+      # prevent race condition
+      sleep 0.1 if settings.slow_pc
+
+      response = client.send_cmd 'Page.printToPDF', settings.to_cmd
 
       Base64.decode64 response['data']
     end

data/lib/postdoc/html_document.rb

@@ -8,7 +8,7 @@ module Postdoc
 
     delegate :path, to: :file
 
-    def initialize(content, path: nil, **_options)
+    def initialize(content, path: nil)
       path ||= Rails.root&.join('tmp') || '/tmp'
       @file = Tempfile.new ['input', '.html'], path
 

data/lib/postdoc/job.rb

@@ -0,0 +1,18 @@
+module Postdoc
+  # A job is a single "print" in a batch and binds a document to a setting.
+  class Job
+    def initialize(document, settings: PrintSettings.new)
+      @settings = settings
+      @document = HTMLDocument.new(document)
+    end
+
+    # Return the result of the job.
+    def result(client)
+      client.print_document(@document.path, settings: @settings)
+    end
+
+    def cleanup
+      @document.cleanup
+    end
+  end
+end

data/lib/postdoc/print_settings.rb

@@ -0,0 +1,48 @@
+module Postdoc
+  # Different prints require different settings. This class is used instead of
+  # passing options arguments throughout all of the code.
+  class PrintSettings
+
+    attr_accessor :slow_pc
+
+    def initialize(
+        header_template: '',
+        footer_template: '',
+        landscape: false,
+        print_background: true,
+        margin_top: 1,
+        margin_bottom: 1,
+        margin_left: 1,
+        margin_right: 1,
+        slow_pc: false
+    )
+      @header_template = header_template
+      @footer_template = footer_template
+      @landscape = landscape
+      @print_background = print_background
+      @margin_top = margin_top
+      @margin_bottom = margin_bottom
+      @margin_left = margin_left
+      @margin_right = margin_right
+      @slow_pc = slow_pc
+    end
+
+    def to_cmd
+      { landscape: @landscape,
+        printBackground: true,
+        marginTop: @margin_top,
+        marginBottom: @margin_bottom,
+        marginLeft: @margin_left,
+        marginRight: @margin_right,
+        displayHeaderFooter: display_header_and_footer?,
+        headerTemplate: @header_template,
+        footerTemplate: @footer_template }
+    end
+
+    private
+
+    def display_header_and_footer?
+      @header_template.present? || @footer_template.present?
+    end
+  end
+end

data/lib/postdoc.rb

@@ -4,6 +4,8 @@ require 'action_controller'
 require 'postdoc/chrome_process'
 require 'postdoc/client'
 require 'postdoc/html_document'
+require 'postdoc/print_settings'
+require 'postdoc/batch'
 
 module Postdoc
   ActionController::Renderers.add :pdf do |_filename, options|
@@ -11,12 +13,42 @@ module Postdoc
   end
 
   def self.render_from_string(content, **options)
+    print_settings = PrintSettings.new(**options)
     server = ChromeProcess.new(**options)
-    html_file = HTMLDocument.new(content, **options)
+    html_file = HTMLDocument.new(content)
     server.client
-        .print_pdf_from_html(html_file.path, **options)
+        .print_pdf_from_html(html_file.path,
+            settings: print_settings)
   ensure
+    if server.client.client.present?
+      server.client.client.send_cmd 'Browser.close'
+    end
     server.kill
     html_file.cleanup
   end
+
+  # A clean and easy way to render a batch:
+  # ```
+  # html_string_1
+  # html_string_2
+  # with_footer = Postdoc::PrintSettings.new(footer_template: footer_template)
+
+  # result = Postdoc.batch do |batch|
+  #   batch.add_document doc1
+  #   batch.add_document doc2, settings: with_footer
+  # end
+  # ```
+  def self.batch(batch: Batch.new)
+    yield(batch)
+    begin
+      server = ChromeProcess.new
+      batch.result(server.client)
+    ensure
+      if server.client.client.present?
+        server.client.client.send_cmd 'Browser.close'
+      end
+      server.kill
+      batch.cleanup
+    end
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: postdoc
 version: !ruby/object:Gem::Version
-  version: 0.3.7
+  version: 0.4.1
 platform: ruby
 authors:
 - Frank Groeneveld
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-07-19 00:00:00.000000000 Z
+date: 2021-10-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: chrome_remote
@@ -33,7 +33,7 @@ dependencies:
         version: 4.0.0
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '7.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -43,7 +43,7 @@ dependencies:
         version: 4.0.0
     - - "<"
       - !ruby/object:Gem::Version
-        version: '6.1'
+        version: '7.0'
 - !ruby/object:Gem::Dependency
   name: mocha
   requirement: !ruby/object:Gem::Requirement
@@ -68,9 +68,12 @@ files:
 - LICENSE
 - Rakefile
 - lib/postdoc.rb
+- lib/postdoc/batch.rb
 - lib/postdoc/chrome_process.rb
 - lib/postdoc/client.rb
 - lib/postdoc/html_document.rb
+- lib/postdoc/job.rb
+- lib/postdoc/print_settings.rb
 homepage: https://github.com/ivaldi/postdoc
 licenses:
 - BSD-2-Clause