require 'sprockets'
require 'sprockets/engines'
require 'forwardable'
require 'sprockets-vendor_gems'

module Flowerbox::Delivery
  class SprocketsHandler
    extend Forwardable

    attr_reader :files, :options

    def_delegators :environment, :append_path, :register_engine

    def initialize(options)
      @options = options

      @files = UniqueAssetList.new
    end

    def add(asset)
      paths_for(asset).each { |path| add_paths_for_compiled_asset(path) }
    end

    def paths_for(asset)
      environment.find_asset(asset).to_a.collect(&:pathname)
    end

    def environment
      return @environment if @environment

      @environment = Sprockets::EnvironmentWithVendoredGems.new
      @environment.unregister_postprocessor('application/javascript', Sprockets::SafetyColons)
      @environment.register_postprocessor('application/javascript', Flowerbox::Delivery::Tilt::EnsureSavedFile)
      @environment.unregister_bundle_processor('text/css', Sprockets::CharsetNormalizer)
      @environment.register_engine('.js', Flowerbox::Delivery::Tilt::JSTemplate)
      @environment.register_engine('.css', Flowerbox::Delivery::Tilt::CSSTemplate)
      @environment.register_engine('.jst', Flowerbox::Delivery::Tilt::JSTTemplate)

      options[:asset_paths].each { |path| append_path(path) }

      @environment
    end

    def asset_for(*args)
      environment.find_asset(*args)
    end

    def add_paths_for_compiled_asset(path)
      asset_for(path, :bundle => false).to_a.each { |file_path| @files.add(file_path) }
    end
  end
end