Add Zeitwerk and a helper to stream PDFs via send_pdf

Gemfile.lock

@@ -4,6 +4,7 @@ PATH
     phlex-pdf (0.1.2)
       matrix (~> 0.4)
       prawn (~> 2.0)
+      zeitwerk (~> 2.0)
 
 GEM
   remote: https://rubygems.org/
@@ -41,6 +42,7 @@ GEM
     rspec-support (3.13.0)
     ruby-rc4 (0.1.5)
     ttfunk (1.7.0)
+    zeitwerk (2.6.13)
 
 PLATFORMS
   arm64-darwin-23

lib/phlex/pdf.rb

@@ -3,9 +3,16 @@
 require_relative "pdf/version"
 require "prawn"
 require "matrix"
+require "zeitwerk"
 
 module Phlex
   class PDF
+    Loader = Zeitwerk::Loader.for_gem_extension(Phlex).tap do |loader|
+      loader.ignore "#{__dir__}/generators"
+      loader.inflector.inflect("pdf" => "PDF")
+      loader.setup
+    end
+
     include Prawn::View
 
     def document = @document

lib/phlex/pdf/rails.rb

@@ -0,0 +1,13 @@
+module Phlex::PDF::Rails
+  module Helpers
+    # Sends a PDF to the browser via Rails controllers.
+    def send_pdf(pdf, disposition: "inline", type: "application/pdf", **kwargs)
+      send_data(
+        pdf.to_pdf,
+        disposition: disposition,
+        type: type,
+        **kwargs
+      )
+    end
+  end
+end

phlex-pdf.gemspec

@@ -35,9 +35,7 @@ Gem::Specification.new do |spec|
   # Uncomment to register a new dependency of your gem
   spec.add_dependency "prawn", "~> 2.0"
   spec.add_dependency "matrix", "~> 0.4"
+  spec.add_dependency "zeitwerk", "~> 2.0"
 
-  spec.add_development_dependency("pdf-inspector", "~> 1.0")
-
-  # For more information and examples about making a new gem, check out our
-  # guide at: https://bundler.io/guides/creating_gem.html
+  spec.add_development_dependency "pdf-inspector", "~> 1.0"
 end

spec/phlex/pdf_spec.rb

@@ -102,3 +102,9 @@ def view_template
     end
   end
 end
+
+RSpec.describe Phlex::PDF::Loader do
+  it "resolves modules" do
+    expect { Phlex::PDF::Rails::Helpers }.not_to raise_error
+  end
+end