lib/jekyll/gzip.rb in jekyll-gzip-1.1.0 vs lib/jekyll/gzip.rb in jekyll-gzip-2.0.0

@@ -1,13 +1,34 @@
-require "jekyll/gzip/version"
-require "jekyll/gzip/compressor"
+# frozen_string_literal: true
 
+require 'jekyll/gzip/version'
+require 'jekyll/gzip/config'
+require 'jekyll/gzip/compressor'
+require 'pathname'
+
 module Jekyll
   module Gzip
   end
 end
 
+Jekyll::Hooks.register :site, :after_init do |site|
+  config = site.config['gzip'] || {}
+  site.config['gzip'] = Jekyll::Gzip::DEFAULT_CONFIG.merge(config) || {}
+end
+
 Jekyll::Hooks.register :site, :post_write do |site|
-  if ENV["JEKYLL_ENV"] == "production"
-    Jekyll::Gzip::Compressor.new(site).compress
+  Jekyll::Gzip::Compressor.compress_site(site) if Jekyll.env == 'production'
+end
+
+begin
+  require 'jekyll-assets'
+
+  Jekyll::Assets::Hook.register :env, :after_write do |env|
+    if Jekyll.env == 'production'
+      path = Pathname.new("#{env.jekyll.config['destination']}#{env.prefix_url}")
+      Jekyll::Gzip::Compressor.compress_directory(path, env.jekyll)
+    end
   end
-end
\ No newline at end of file
+rescue LoadError
+  # The Jekyll site doesn't use Jekyll::Assets, so no need to compress those
+  # files.
+end