# frozen_string_literal: true

require 'jekyll/zopfli/version'
require 'jekyll/zopfli/config'
require 'jekyll/zopfli/compressor'
require 'pathname'

module Jekyll
  module Zopfli
  end
end

Jekyll::Hooks.register :site, :after_init do |site|
  config = site.config['zopfli'] || {}
  site.config['zopfli'] = Jekyll::Zopfli::DEFAULT_CONFIG.merge(config) || {}
end

Jekyll::Hooks.register :site, :post_write do |site|
  Jekyll::Zopfli::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::Zopfli::Compressor.compress_directory(path, env.jekyll)
    end
  end
rescue LoadError
  # The Jekyll site doesn't use Jekyll::Assets, so no need to compress those
  # files.
end