postdoc 0.3.0 → 0.3.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 529772412289f1fe24ffeedd9644f2f8799ee955c3dea07f256a33e23ca1bc51
-  data.tar.gz: 2b27cdf8389e6c096c1dc64f3005396542c7808e04449f79d81f1f62990b1f0a
+  metadata.gz: 74295695d3dbf550fa41c48f13ed409547fc0e30559d4baa215c5d8875ec291d
+  data.tar.gz: f4e09afaa99866596a382fc7d4dea72c1cbbfe6df972ca336da97949f2c6f637
 SHA512:
-  metadata.gz: 948c6386bc64c6ee7d5d9423a905dfec09dbbc2a5ed80e8032817929f13e5b682ded267a4751aa3fe66cd5d38f83b66abf4ee0989f1b2a4bf2e3fb0bdc4ea9b4
-  data.tar.gz: ce0193c8374d7c6fb44f8196829531156b79b4684ace11d172224bc1bdcf87f089a82757344e00edad9d377d76e037d83215d482247ce30e0b115af496d9d75b
+  metadata.gz: 7cc297db02152057e2bfa61042430e29d67a0d10e4bd109e70791ea5bf065324d0cb25df9c6fd305f8671fbe9583b3c7470c8a843744ed76a972777ce7e331b4
+  data.tar.gz: 19395b879f1a9837e11986eba31f832818f8765f05f13861c09f00141bc3d2da41712845984fef87e476a2d3daf09347124e6e63a11456ae8e9f65ec1bd2432b

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']
-
-    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,34 @@
+# 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 Errno::ECONNREFUSED
+      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,49 @@
+# 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 }
+      raise 'ChromeClient couldn\'t launch' if @client.blank?
+    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,
+        marginLeft: options[:margin_left] || 1,
+        marginRight: options[:margin_right] || 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 Exception
+      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.0
+  version: 0.3.6
 platform: ruby
 authors:
 - Frank Groeneveld
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-12-16 00:00:00.000000000 Z
+date: 2021-03-26 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