grover 0.1.2 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 3a06e4797cf17f2c9d7e1e4d85188aa12c7a2eef
-  data.tar.gz: 9ce156b3e3d80b6e14d98878264f124c4877b816
+  metadata.gz: cf861057bca3ab2b9c59bf0c52b329bb70cd1e5c
+  data.tar.gz: 910e24710cfd955ca779b600680ce7b50bc8458b
 SHA512:
-  metadata.gz: e0f96d8f404409c1c6980bdc692ee97955ab9dd8bbd685ae945cc319b945231a6539b6bd07f48deb2b10f42d4ffed0eac4e5817b3d37cf81bdfe06e9fcb7103c
-  data.tar.gz: e002838a7f7b42256545a4590fb4624a75b3425af3764eff7a78b5cae8f973e62930ea1db9db47977fcc6bf527df86489389d055957aea80b0ec24435f20b9ab
+  metadata.gz: 6c101f3d8bf4b9cdefa35ed26994e24498c9eede363ed798450b1cc849b288547b34e28ded856387bda3e3d20335b76aed0547527baaa5632a6e3164f5bbc1a8
+  data.tar.gz: 3fbc34457ebb9a151c85e01217ddc884baaac14c36532c0eec77856557188b7351389f6ec73cdc400ad2379589fccf158fd0b20c5f2626f27de5e850ffa7ff31

data/lib/grover.rb

@@ -2,4 +2,52 @@ require 'grover/version'
 
 require 'grover/utils'
 require 'grover/processor'
-require 'grover/grover'
+require 'grover/middleware'
+
+#
+# Grover interface for converting HTML to PDF
+#
+class Grover
+  #
+  # @param [String] url URL of the page to convert
+  # @param [Hash] options Optional parameters to pass to PDF processor
+  #   see https://github.com/GoogleChrome/puppeteer/blob/master/docs/api.md#pagepdfoptions
+  #
+  def initialize(url, options = {})
+    @url = url
+    @root_path = options.delete :root_path
+    @options = options
+  end
+
+  #
+  # Request URL with provided options and create PDF
+  #
+  # @param [String] path Optional path to write the PDF to
+  # @return [Array<Integer>] Byte array of the resulting PDF
+  #
+  def to_pdf(path = nil)
+    options = @options.dup
+    options[:path] = path if path
+    result = processor.convert_pdf @url, options
+    result['data'].pack('c*')
+  end
+
+  def inspect
+    format(
+      '#<%<class_name>s:0x%<object_id>p @url="%<url>s">',
+      class_name: self.class.name,
+      object_id: object_id,
+      url: url
+    )
+  end
+
+  private
+
+  def root_path
+    @root_path ||= Dir.pwd
+  end
+
+  def processor
+    Grover::Processor.new(root_path)
+  end
+end

data/lib/grover/middleware.rb

@@ -0,0 +1,55 @@
+class Grover
+  #
+  # Rack middleware for catching PDF requests and returning the upstream HTML as a PDF
+  #
+  class Middleware
+    def initialize(app)
+      @app = app
+      @render_pdf = false
+    end
+
+    def call(env)
+      @request = Rack::Request.new(env)
+      @render_pdf = false
+
+      set_request_to_render_as_pdf(env) if render_as_pdf?
+      status, headers, response = @app.call(env)
+
+      if rendering_pdf? && headers['Content-Type'] =~ %r{text/html|application/xhtml\+xml}
+        body = response.respond_to?(:body) ? response.body : response.join
+        body = body.join if body.is_a?(Array)
+
+        body = Grover.new(body).to_pdf
+        response = [body]
+
+        # Do not cache PDFs
+        headers.delete 'ETag'
+        headers.delete 'Cache-Control'
+
+        headers['Content-Length'] = (body.respond_to?(:bytesize) ? body.bytesize : body.size).to_s
+        headers['Content-Type'] = 'application/pdf'
+      end
+
+      [status, headers, response]
+    end
+
+    private
+
+    def rendering_pdf?
+      @render_pdf
+    end
+
+    def render_as_pdf?
+      @request.path.end_with?('.pdf')
+    end
+
+    def set_request_to_render_as_pdf(env)
+      @render_pdf = true
+      env['HTTP_ACCEPT'] = concat(env['HTTP_ACCEPT'], Rack::Mime.mime_type('.html'))
+    end
+
+    def concat(accepts, type)
+      (accepts || '').split(',').unshift(type).compact.join(',')
+    end
+  end
+end

data/lib/grover/processor.rb

@@ -18,7 +18,11 @@ class Grover
         try {
           browser = await puppeteer.launch(#{launch_params});
           const page = await browser.newPage();
-          await page.goto(url, { waitUntil: 'networkidle2' });
+          if (url.match(/^http/i)) {
+            await page.goto(url, { waitUntil: 'networkidle2' });
+          } else {
+            await page.setContent(url);
+          }
           return await page.pdf(options);
         } finally {
           if (browser) {

data/lib/grover/version.rb

@@ -1,3 +1,3 @@
 class Grover
-  VERSION = '0.1.2'.freeze
+  VERSION = '0.2.0'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: grover
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.2.0
 platform: ruby
 authors:
 - Andrew Bromwich
@@ -97,14 +97,12 @@ dependencies:
 description: A Ruby gem to transform HTML into PDFs using Google Puppeteer/Chromium
 email:
 - abromwich@studiosity.com
-executables:
-- console
+executables: []
 extensions: []
 extra_rdoc_files: []
 files:
-- bin/console
 - lib/grover.rb
-- lib/grover/grover.rb
+- lib/grover/middleware.rb
 - lib/grover/processor.rb
 - lib/grover/utils.rb
 - lib/grover/version.rb
@@ -132,5 +130,5 @@ rubygems_version: 2.6.14.1
 signing_key: 
 specification_version: 4
 summary: A Ruby gem to transform HTML into PDFs wrapper the NodeJS Google Puppeteer
-  project using Chromium
+  driver for Chromium
 test_files: []

data/bin/console

@@ -1,7 +0,0 @@
-#!/usr/bin/env ruby
-
-require 'bundler/setup'
-require 'grover'
-
-require 'irb'
-IRB.start

data/lib/grover/grover.rb

@@ -1,43 +0,0 @@
-#
-# Grover interface for converting HTML to PDF
-#
-class Grover
-  #
-  # @param [String] url URL of the page to convert
-  # @param [Hash] options Optional parameters to pass to PDF processor
-  #   see https://github.com/GoogleChrome/puppeteer/blob/master/docs/api.md#pagepdfoptions
-  #
-  def initialize(url, options = {})
-    @url = url
-    @root_path = options.delete :root_path
-    @options = options
-  end
-
-  #
-  # Request URL with provided options and create PDF
-  #
-  # @param [String] path Optional path to write the PDF to
-  # @return [Array<Integer>] Byte array of the resulting PDF
-  #
-  def to_pdf(path = nil)
-    options = @options.dup
-    options[:path] = path if path
-    result = Grover::Processor.new(root_path).convert_pdf(@url, options)
-    result['data'].pack('c*')
-  end
-
-  def inspect
-    format(
-      '#<%<class_name>s:0x%<object_id>p @url="%<url>s">',
-      class_name: self.class.name,
-      object_id: object_id,
-      url: url
-    )
-  end
-
-  private
-
-  def root_path
-    @root_path ||= File.expand_path(__dir__)
-  end
-end