require 'pathname'
require "sprockets"

module Middleman::CoreExtensions::Sprockets
  class << self
    def registered(app)
      app.set :js_compressor, false
      app.set :css_compressor, false
      
      # Cut off every extension after .js (which sprockets eats up)
      app.build_reroute do |destination, request_path|
        if !request_path.match(/\.js\./i)
          false
        else
          [
            destination.gsub(/\.js(\..*)$/, ".js"),
            request_path.gsub(/\.js(\..*)$/, ".js")
          ]
        end
      end
      
      app.ready do
        js_env = Middleman::CoreExtensions::Sprockets::JavascriptEnvironment.new(self)

        vendor_dir = File.join("vendor", "assets", "javascripts")
        gems_with_js = ::Middleman.rubygems_latest_specs.select do |spec|
          ::Middleman.spec_has_file?(spec, vendor_dir)
        end.each do |spec|
          js_env.append_path File.join(spec.full_gem_path, vendor_dir)
        end

        app_dir = File.join("app", "assets", "javascripts")
        gems_with_js = ::Middleman.rubygems_latest_specs.select do |spec|
          ::Middleman.spec_has_file?(spec, app_dir)
        end.each do |spec|
          js_env.append_path File.join(spec.full_gem_path, app_dir)
        end

        # add paths to js_env (vendor/assets/javascripts)
        map "/#{js_dir}" do
          run js_env
        end

        sass.each { |k, v| ::Sprockets::Sass.options[k] = v }
        css_env = Middleman::CoreExtensions::Sprockets::StylesheetEnvironment.new(self)
         
        vendor_dir = File.join("vendor", "assets", "stylesheets")
        gems_with_css = ::Middleman.rubygems_latest_specs.select do |spec|
          ::Middleman.spec_has_file?(spec, vendor_dir)
        end.each do |spec|
          css_env.append_path File.join(spec.full_gem_path, vendor_dir)
        end
  
        app_dir = File.join("app", "assets", "stylesheets")
        gems_with_css = ::Middleman.rubygems_latest_specs.select do |spec|
          ::Middleman.spec_has_file?(spec, app_dir)
        end.each do |spec|
          css_env.append_path File.join(spec.full_gem_path, app_dir)
        end
        
        map("/#{css_dir}") do
          run css_env
        end
      end
    end
    alias :included :registered
  end

  class MiddlemanEnvironment < ::Sprockets::Environment
    def initialize(app)
      super app.source_dir
      
      # Make the app context available to Sprockets
      context_class.send(:define_method, :app) { app }
      context_class.class_eval do
        def method_missing(name)
          if app.respond_to?(name)
            app.send(name)
          else
            super
          end
        end
      end
    end
  end
    
  class JavascriptEnvironment < MiddlemanEnvironment
    def initialize(app)
      super
      
      self.js_compressor = app.js_compressor
      
      # configure search paths
      append_path app.js_dir
    end
    
    def javascript_exception_response(exception)
      expire_index!
      super(exception)
    end
  end
  
  class StylesheetEnvironment < MiddlemanEnvironment
    def initialize(app)
      super
      
      self.css_compressor = app.css_compressor
  
      # configure search paths
      append_path app.css_dir
    end
  
    def css_exception_response(exception)
      expire_index!
      super(exception)
    end
  end
end