postdoc 0.3.1 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 347fc91d568be3659d11cbd1c350ab6777a480d38efafbda841cdcf18a897a40
-  data.tar.gz: 7ef1235281009ae11450ca7b7dab86ad4245d5428a1d605891de21c3ef006843
+  metadata.gz: 866cf4c6e0cfe521087c96c5aea85b003a139d55467ccd77e757c6c45084cb17
+  data.tar.gz: c2789db75c33ede4a402e06c815d7e9a33689960d53773940fca96f350ec1f2c
 SHA512:
-  metadata.gz: c9e5086f120958a770f3662579b3a9f67ea51ab44625e7d49067af8748762f184dfc56aaff652b9f2d907e839e6336ca4e0b520a2b0d581a227c77b2a64ed2e3
-  data.tar.gz: 053424d30b3ae859235c24f784f20a194ea40a33e306ae4e7f8129f14199afbd47b59d94a1a90542e6777ba73287c7e738d45a9e5355eb070aa4ff772daaf93e
+  metadata.gz: d8fedb4b38ebc3fbb45577423c478211c7d801bb917742b2493c1f57869b51aaba1877efac94383e1fbdfb77bd07d3c0ec5eb9edb2a8e6f732453045e865a797
+  data.tar.gz: af3070ea0e915b3f44349f926e5fe08d1c2270f4bc7143a1352145a679e0882392b5f1e02800e6a7240612c16fa182fa1282424262498e171d89a6b7726d656b

data/lib/postdoc.rb

@@ -1,72 +1,22 @@
 # frozen_string_literal: true
 
-require 'chrome_remote'
+require 'action_controller'
+require 'postdoc/chrome_process'
+require 'postdoc/client'
+require 'postdoc/html_document'
 
 module Postdoc
   ActionController::Renderers.add :pdf do |_filename, options|
     Postdoc.render_from_string render_to_string(options), options
   end
 
-  def self.render_from_string(string, options)
-    htmlfile = Tempfile.new ['input', '.html'], Rails.root.join('tmp')
-
-    htmlfile.write string
-    htmlfile.flush
-
-    if options[:client].nil?
-      # random port at 1025 or higher
-      random_port = 1024 + Random.rand(65_535 - 1024)
-      pid = Process.spawn "chrome --remote-debugging-port=#{random_port} --headless"
-    end
-    
-    success = false
-    10.times do
-      begin
-        TCPSocket.new('localhost', random_port)
-        success = true 
-        break
-      rescue
-      end
-      sleep 1
-    end
-    
-    return unless success
-
-    begin
-      chrome = options[:client].nil? ? ChromeRemote.client(port: random_port) : options[:client]
-
-      chrome.send_cmd 'Page.enable'
-      chrome.send_cmd 'Page.navigate', url: "file://#{htmlfile.path}"
-      chrome.wait_for 'Page.loadEventFired'
-
-      if options[:header_template].present? || options[:footer_template].present?
-        displayHeaderFooter = true
-      else
-        displayHeaderFooter = false
-      end
-
-      response = chrome.send_cmd 'Page.printToPDF', {
-        landscape: options[:landscape] || false,
-        printBackground: true,
-        marginTop: options[:margin_top] || 1,
-        marginBottom: options[:margin_bottom] || 1,
-        displayHeaderFooter: displayHeaderFooter,
-        headerTemplate: options[:header_template] || '',
-        footerTemplate: options[:footer_template] || ''
-      }
-      result = Base64.decode64 response['data']
-    ensure
-      if options[:client].nil?
-        Process.kill 'KILL', pid
-        Process.wait pid
-      else
-        chrome.send_cmd 'Page.close'
-      end
-
-      htmlfile.close
-      htmlfile.unlink
-    end
-
-    result
+  def self.render_from_string(content, **options)
+    server = ChromeProcess.new(**options)
+    html_file = HTMLDocument.new(content, **options)
+    server.client
+        .print_pdf_from_html(html_file.path, **options)
+  ensure
+    server.kill
+    html_file.cleanup
   end
 end

data/lib/postdoc/chrome_process.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+module Postdoc
+  class ChromeProcess
+    attr_reader :pid, :port
+
+    def initialize(port: Random.rand(65_535 - 1024), **_options)
+      @port = port
+      @pid = Process.spawn "chrome --remote-debugging-port=#{port} --headless",
+          out: File::NULL, err: File::NULL
+    end
+
+    def alive?
+      @alive ||= test_socket!
+    rescue
+      false
+    end
+
+    def kill
+      Process.kill 'KILL', pid
+      Process.wait pid
+    end
+
+    def client
+      Client.new port
+    end
+
+    private
+
+    def test_socket!
+      TCPSocket.new('localhost', port)
+    end
+  end
+end

data/lib/postdoc/client.rb

@@ -0,0 +1,46 @@
+# frozen_string_literal: true
+
+require 'chrome_remote'
+
+module Postdoc
+  class Client
+    attr_accessor :client
+
+    def initialize(port)
+      @port = port
+      100.times { setup_connection_or_wait && break }
+    end
+
+    def print_pdf_from_html(file_path,
+        header_template: false,
+        footer_template: false,
+        **options)
+
+      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,
+        displayHeaderFooter: !!(header_template || footer_template),
+        headerTemplate: header_template || '',
+        footerTemplate: footer_template || ''
+      }
+
+      Base64.decode64 response['data']
+    end
+
+    private
+
+    def setup_connection_or_wait
+      @client = ChromeRemote.client(port: @port)
+      true
+    rescue
+      sleep(0.1)
+      false
+    end
+  end
+end

data/lib/postdoc/html_document.rb

@@ -0,0 +1,24 @@
+# frozen_string_literal: true
+
+require 'rails'
+
+module Postdoc
+  class HTMLDocument
+    attr_accessor :file
+
+    delegate :path, to: :file
+
+    def initialize(content, path: nil, **_options)
+      path ||= Rails.root&.join('tmp') || '/tmp'
+      @file = Tempfile.new ['input', '.html'], path
+
+      file.write content
+      file.flush
+    end
+
+    def cleanup
+      file.close
+      file.unlink
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: postdoc
 version: !ruby/object:Gem::Version
-  version: 0.3.1
+  version: 0.3.3
 platform: ruby
 authors:
 - Frank Groeneveld
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-01-13 00:00:00.000000000 Z
+date: 2021-03-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: chrome_remote
@@ -44,6 +44,20 @@ dependencies:
     - - "<"
       - !ruby/object:Gem::Version
         version: '6.1'
+- !ruby/object:Gem::Dependency
+  name: mocha
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Description of Postdoc.
 email:
 - frank@ivaldi.nl
@@ -54,6 +68,9 @@ files:
 - LICENSE
 - Rakefile
 - lib/postdoc.rb
+- lib/postdoc/chrome_process.rb
+- lib/postdoc/client.rb
+- lib/postdoc/html_document.rb
 homepage: https://github.com/ivaldi/postdoc
 licenses:
 - BSD-2-Clause