postdoc 0.2.4 → 0.3.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4fd07fd7a356f5e41dd00199937c8824010562d255d6539e59398e3fb55c0f79
-  data.tar.gz: e89f06d78b8bd657ccfd52a2b079eea7035e1b77fda5380e6bc6f7a1f315e254
+  metadata.gz: 374246c4708c1c0492a3c8117f6a891723ae66b3c57476d6370b63f414876eb9
+  data.tar.gz: 30cac5f1aeee47ea6fc52e8985edafbd72ad2b81082aa3e7887f21e787b198ee
 SHA512:
-  metadata.gz: '088597cc28cad883de3f539aa2e6e373285fa55b17cad6a17e619febce77d747c9174e0f2c6b1cca1d3583140b42f32e8b87c86bfdd3ae15361a9e36c6d3d945'
-  data.tar.gz: 84d58a0ea2eeacbef730c0ac47d356125f934fe1e5cfe013cffcf87b5224e5af517b229e60f305f3dbb42e9d5ce753978ff615995a871de91cceebf2d056af4f
+  metadata.gz: 67093df126d95b92849611db7f08d1a6cbd065647398dfcdc8b31aaa68ab9f382b86de6edafb44cbca82a6ed95255e2817ce2bd842d3f9627380eed72d78f48a
+  data.tar.gz: 5b0ace904a20107d6d436c086bd20b183fe702c31b38e8bd4adab00c4f08bf34d28e07571129181fec6c28c98ea3f2cddf0312b841b75319692dd92db02831b4

data/lib/postdoc.rb

@@ -1,62 +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
-
-    # FIXME
-    sleep 1
-
-    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 Errno::EHOSTUNREACH, Errno::ECONNREFUSED
+      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,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: postdoc
 version: !ruby/object:Gem::Version
-  version: 0.2.4
+  version: 0.3.5
 platform: ruby
 authors:
 - Frank Groeneveld
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-10-21 00:00:00.000000000 Z
+date: 2021-03-16 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: chrome_remote
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.2.0
 - !ruby/object:Gem::Dependency
   name: rails
   requirement: !ruby/object:Gem::Requirement
@@ -31,19 +45,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '6.1'
 - !ruby/object:Gem::Dependency
-  name: chrome_remote
+  name: mocha
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.0
-  type: :runtime
+        version: '0'
+  type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.2.0
+        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