postdoc 0.2.2 → 0.3.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a85e6b50205b46d1127b53214c2bad5e53aef5765081e17a1998eafb7c1bae4b
-  data.tar.gz: 6d00e103be5c33ce38e20e0bfa3cf8939ee2a8824a9a13e32f15662fea31c0b6
+  metadata.gz: 866cf4c6e0cfe521087c96c5aea85b003a139d55467ccd77e757c6c45084cb17
+  data.tar.gz: c2789db75c33ede4a402e06c815d7e9a33689960d53773940fca96f350ec1f2c
 SHA512:
-  metadata.gz: 416f6e0522cbd000347afb73c1e66fc7ad1b054db46d0bb4ae2a3b21dfc24720b8bbdd1eca7fbe7c13860815a3916aa30e1d7d009e1f2675aca5f8c599755f7d
-  data.tar.gz: a5b397c41b09ddf29125eb3ddd4e967971abcabefcf3a7b57222888c827b783f6978fb14316d51170baf8c7c04b0e701c5054fdaac151438452b943552ebe5fc
+  metadata.gz: d8fedb4b38ebc3fbb45577423c478211c7d801bb917742b2493c1f57869b51aaba1877efac94383e1fbdfb77bd07d3c0ec5eb9edb2a8e6f732453045e865a797
+  data.tar.gz: af3070ea0e915b3f44349f926e5fe08d1c2270f4bc7143a1352145a679e0882392b5f1e02800e6a7240612c16fa182fa1282424262498e171d89a6b7726d656b

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|
-    render_from_string_to_pdf render_to_string(options), 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,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,15 +1,29 @@
 --- !ruby/object:Gem::Specification
 name: postdoc
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.3.3
 platform: ruby
 authors:
 - Frank Groeneveld
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-12-11 00:00:00.000000000 Z
+date: 2021-03-03 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: []