postdoc 0.3.7 → 0.3.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4a8f25b2dcb78fb58ec199d021985eef4e9892c55cba5ad9b33b8a1aea14baba
-  data.tar.gz: f2ef8f9083917b3efbd212cdbc7a25c7f574db1a1a963bacf0a194151f2f8fbe
+  metadata.gz: 54aee40fdabd27f1565648b9f80a992a1799214472a7639019ca4385f5d2bd38
+  data.tar.gz: 8d4870650fd4c27a6ebad35af0995a345c70b8edb77fef36c4a1a8d1cc41a82e
 SHA512:
-  metadata.gz: 166ca4c618a7740e02edda59711e1b23a5555e2c4d479ac00b71d6d9dda9c613fdbbd125fe0053f75405881c21e7b31f9494ef5df997cd3afe32b1486b403997
-  data.tar.gz: 963c118d472a7d244fc4841bae66151f211804e796a1a1d8be3f3d1dae25a9f5274122c867abe99cd565a3bfc457a4516b94ce8065866fb929918226331816ce
+  metadata.gz: 8edd792288b47e25ee4c1b76bc7ee80c5533aca499751f0d119acf1b978b0cbf80e7dc10cb2f4eb51db0d86241cb88440c78ff8a07e7604d0ddf67f688870281
+  data.tar.gz: 2f83ad865f53d224044d1a9dd6a1abe758bbfe778ec4fccd9cb59904a2653e565b26e1650c05e29eb0242233f01de130e96e6a36a6d478acb27dea825986ac7c

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,36 @@ 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
     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
+      server.kill
+      batch.cleanup
+    end
+  end
 end

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,26 +13,27 @@ 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 || ''
-      }
+      response = client.send_cmd 'Page.printToPDF', settings.to_cmd
+      
+      client.send_cmd 'Browser.close'
+
+      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'
+      response = client.send_cmd 'Page.printToPDF', settings.to_cmd
 
       client.send_cmd 'Browser.close'
 

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,43 @@
+module Postdoc
+  # Different prints require different settings. This class is used instead of
+  # passing options arguments throughout all of the code.
+  class PrintSettings
+    def initialize(
+        header_template: '',
+        footer_template: '',
+        landscape: false,
+        print_background: true,
+        margin_top: 1,
+        margin_bottom: 1,
+        margin_left: 1,
+        margin_right: 1
+    )
+      @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
+    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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: postdoc
 version: !ruby/object:Gem::Version
-  version: 0.3.7
+  version: 0.3.8
 platform: ruby
 authors:
 - Frank Groeneveld
@@ -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