From bc91aa701257c729813594184a35e75d8894d239 Mon Sep 17 00:00:00 2001
From: Loic Nageleisen <loic.nageleisen@gmail.com>
Date: Fri, 21 Feb 2014 16:14:18 +0100
Subject: [PATCH] extended auxiliary files support

- support for css in header/footer
- support for js/coffee/whatever
- main integrated into render/inject/tmp flow

=> all html files get every css/js injected
---
 Gemfile.lock     |   6 +++
 lib/tilt/pdf.rb  | 100 ++++++++++++++++++++++++++++++++++++++---------
 tilt-pdf.gemspec |   1 +
 3 files changed, 88 insertions(+), 19 deletions(-)

diff --git a/Gemfile.lock b/Gemfile.lock
index ed7a15a..05cac46 100644
--- a/Gemfile.lock
+++ b/Gemfile.lock
@@ -11,8 +11,13 @@ GEM
     celluloid (0.15.2)
       timers (~> 1.1.0)
     coderay (1.0.9)
+    coffee-script (2.2.0)
+      coffee-script-source
+      execjs
+    coffee-script-source (1.7.0)
     commonjs (0.2.7)
     diff-lcs (1.2.4)
+    execjs (2.0.2)
     ffi (1.9.0)
     formatador (0.2.4)
     guard (2.1.1)
@@ -67,6 +72,7 @@ PLATFORMS
   ruby
 
 DEPENDENCIES
+  coffee-script
   guard-rspec
   less
   pry
diff --git a/lib/tilt/pdf.rb b/lib/tilt/pdf.rb
index 2ea47c9..98afd5d 100644
--- a/lib/tilt/pdf.rb
+++ b/lib/tilt/pdf.rb
@@ -11,23 +11,23 @@ module Tilt
     def prepare; end
 
     def evaluate(scope, locals, &block)
-      main = render_html(main_html_file, scope, locals, &block)
-
-      render_to_tmp(*aux_files, scope, locals, block) do |tmp|
+      files = aux_files
+      files << main_html_file
+      render_to_tmp(*files, scope, locals, block) do |tmp|
         opts = pdfkit_options
 
         if header
-          htmp = tmp.select { |_, f| f =~ /#{File.basename header}/ }.first[1]
+          htmp = tmp.select { |_, f, _| f == header }.first[2]
           opts.merge!('header-html' => htmp) if header
         end
 
         if footer
-          ftmp = tmp.select { |_, f| f =~ /#{File.basename footer}/ }.first[1]
+          ftmp = tmp.select { |_, f, _| f == footer }.first[2]
           opts.merge!('footer-html' => ftmp) if footer
         end
 
-        kit = PDFKit.new(main, opts)
-        tmp.each { |t, f| kit.stylesheets << f if t == 'text/css' }
+        main = tmp.select { |_, f, _| f == main_html_file }.first[2]
+        kit = PDFKit.new(File.read(main), opts)
 
         @output = kit.to_pdf
       end
@@ -46,7 +46,9 @@ module Tilt
     end
 
     def aux_files
-      files = css_files
+      files = []
+      files.concat css_files
+      files.concat js_files
       files << header if header
       files << footer if footer
 
@@ -85,6 +87,16 @@ module Tilt
       config.fetch('stylesheets', []).map { |f| absolutize(f) }
     end
 
+    def js_files
+      js_from_config || find_js
+    end
+
+    def js_from_config
+      return unless config.key?('javascripts')
+
+      config.fetch('javascripts', []).map { |f| absolutize(f) }
+    end
+
     def pdfkit_options
       config.fetch('pdfkit', {})
     end
@@ -101,33 +113,83 @@ module Tilt
       Dir.glob(File.join(dirname, name + '.css*'))
     end
 
+    def find_js
+      Dir.glob(File.join(dirname, name + '.js*'))
+    end
+
     def render_html(file, scope, locals, &block)
       Tilt.new(file).render(scope, locals, &block)
     end
 
+    def inject_css!(document, stylesheet)
+      append_to_head!(document, "<style>#{stylesheet}</style>")
+    end
+
+    def inject_js!(document, script)
+      append_to_head!(document, "<script>#{script}</script>")
+    end
+
+    def append_to_head!(document, tag)
+      if document.match(/<\/head>/)
+        document.gsub!(/(<\/head>)/) { |s| tag + s }
+      else
+        document.insert(0, tag)
+      end
+    end
+
     def render_to_tmp(*files, scope, locals, block)
       tmps = []
-      noop = %w[html css]
+      no_tilt = %w[html css js]
 
-      css = files.map do |file|
+      result = files.map do |file|
         ext = File.extname(file).sub(/^\./, '')
-        if noop.include?(ext)
-          ["text/#{ext}", file]
+        if no_tilt.include?(ext)
+          mime = case ext
+                 when 'js', 'javascript' then 'application/javascript'
+                 else "text/#{ext}"
+                 end
+          rendered = File.read(file)
         else
           template = Tilt.new(file)
           mime = template.class.default_mime_type
           ext = mime.split('/').last
-          tmp = Tempfile.new([File.basename(file), '.' + ext])
-          tmps << tmp
           rendered = template.render(scope, locals, &block)
-          tmp.write(rendered)
-          tmp.close
-
-          [mime, tmp.path]
         end
+
+        [ext, mime, file, rendered]
       end
 
-      yield css
+      result.sort! do |a, b|
+        a[1] <=> b[1]  # css < html
+      end
+
+      styles = result.select { |_, mime, _, _| mime == 'text/css' }
+      scripts = result.select { |_, mime, _, _| mime == 'application/javascript' }
+      result.map! do |ext, mime, file, rendered|
+        if mime == 'text/html'
+          styles.each do |_, _, _, style|
+            inject_css!(rendered, style)
+          end
+          scripts.each do |_, _, _, script|
+            inject_js!(rendered, script)
+          end
+        end
+
+        [ext, mime, file, rendered]
+      end
+
+      result.map! do |ext, mime, file, rendered|
+        tmp = Tempfile.new([File.basename(file), '.' + ext])
+        tmps << tmp
+        tmp.write(rendered)
+        tmp.close
+
+        path = tmp.path
+
+        [mime, file, path]
+      end
+
+      yield result
     ensure
       tmps.each { |tmp| tmp.close! }
     end
diff --git a/tilt-pdf.gemspec b/tilt-pdf.gemspec
index 54f970b..be402c3 100644
--- a/tilt-pdf.gemspec
+++ b/tilt-pdf.gemspec
@@ -22,6 +22,7 @@ Gem::Specification.new do |s|
 
   s.add_development_dependency 'therubyracer'
   s.add_development_dependency 'less'
+  s.add_development_dependency 'coffee-script'
   s.add_development_dependency 'slim'
   s.add_development_dependency 'rspec', '~> 2.14'
   s.add_development_dependency 'rake'