postdoc 0.2.3 → 0.3.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d21422293d756a0a95e7a697efcf2d35e9d8fc558124d8a665c05c5e3eef2e3d
-  data.tar.gz: bbfda4ff89ce4066f8ea2ea1da1190ce30c21990508db035c05aa733dc1e9014
+  metadata.gz: a6a52e2638a2925d5834959e3a716295aaf5a66e8be797452c507b72f4fa7e0a
+  data.tar.gz: cca67e571e577b9ac1b770a4b9049b66953beab03ab13d6046d90684af0daf91
 SHA512:
-  metadata.gz: 32451b802853cbe1e0383ae26e7c264956cf136c0703c7e3cbdbd3ff4d5d8364db2c5647aa010d196c70b9759dfeff7668d0caff1750ee8e3600a688e13ad56b
-  data.tar.gz: 7ee471438ee718694588e0b1827e601e6a4b68b22685c937e76a97448d9fc64a50f3010d02fc0c0a50d0679da894290b9db77fffab9a39458c021c583d5abb57
+  metadata.gz: e7ccb76057f1cf0112dc9150fba8de4b9aef0261ed0f229a15af3030a4e4b10bb80fb8929dddf40aa16af10bf0133778da3cb7d5ed99f81a93b73267cb4e3b21
+  data.tar.gz: 7e5b5c6eca2c4020b4e8c27078aed985bcfb4e1b4f0e4bd0c3a7f8585e094069a8eb723619045f1fe2971c9cc9d3c4bff92205b6225dacdfc9e1849f10dee3be

data/lib/postdoc.rb

@@ -1,55 +1,22 @@
-require 'chrome_remote'
+# frozen_string_literal: true
+
+require 'action_controller'
+require 'postdoc/chrome_process'
+require 'postdoc/client'
+require 'postdoc/html_document'
 
 module Postdoc
-  ActionController::Renderers.add :pdf do |filename, options|
+  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
-
-    # random port at 1025 or higher
-    random_port = 1024 + Random.rand(65535 - 1024)
-
-    pid = Process.spawn "chrome --remote-debugging-port=#{random_port} --headless"
-
-    # FIXME
-    sleep 1
-
-    begin
-      chrome = ChromeRemote.client port: random_port
-      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
-      Process.kill 'KILL', pid
-      Process.wait pid
-
-      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::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.3
+  version: 0.3.4
 platform: ruby
 authors:
 - Frank Groeneveld
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-12-13 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,11 +68,14 @@ 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
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -73,8 +90,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.1.4
+signing_key:
 specification_version: 4
 summary: Summary of Postdoc.
 test_files: []