# Post-require hooks for acts_as_solr and sunspot if this 
# gem is loaded and WEBSOLR_URL is defined.

if ENV["WEBSOLR_URL"]
  require "uri"
  
  begin
    require "sunspot/rails/configuration"
    require "sunspot/rails/searchable"
    require "sunspot/rails/request_lifecycle"
    module Sunspot #:nodoc:
      module Rails #:nodoc:
        class Configuration
          def hostname
            URI.parse(ENV["WEBSOLR_URL"]).host
          end
          def port
            URI.parse(ENV["WEBSOLR_URL"]).port
          end
          def path
            URI.parse(ENV["WEBSOLR_URL"]).path
          end
        end
      end
    end
    
    module Sunspot #:nodoc:
      module Rails #:nodoc:
        # 
        # This module adds an after_filter to ActionController::Base that commits
        # the Sunspot session if any documents have been added, changed, or removed
        # in the course of the request.
        #
        module RequestLifecycle
          class <<self
            def included(base) #:nodoc:
              base.after_filter do
                begin
                  if Sunspot::Rails.configuration.auto_commit_after_request?
                    Sunspot::Rails.master_session.commit_if_dirty
                  elsif Sunspot::Rails.configuration.auto_commit_after_delete_request?
                    Sunspot::Rails.master_session.commit_if_delete_dirty
                  end
                rescue Exception => e
                  ActionController::Base.logger.error e.message
                  ActionController::Base.logger.error e.backtrace.join("\n")
                  false
                end
              end
            end
          end
        end
      end
    end
    
    
    module Sunspot
      module Rails 
        module Searchable
          module InstanceMethods
            %w[index index! remove_from_index remove_from_index!].each do |method|
              new_name = method =~ /!/ ? method.gsub("!", "") + "_with_caught_errors!" : "#{method}_with_caught_errors"
              old_name = new_name.sub("_with_", "_without_")
              define_method(new_name) do
                begin
                  send(old_name)
                rescue Exception => e
                  logger.error e.message
                  logger.error e.backtrace.join("\n")
                  false
                end
              end
              alias_method_chain method, :caught_errors
            end
          end
        end
      end
    end
    
  rescue LoadError
    #ignore
  end

  begin
    require "acts_as_solr"
    module ActsAsSolr
      class Post        
        def self.execute(request)
          begin
            connection = Solr::Connection.new(ENV["WEBSOLR_URL"])
            return connection.send(request)
          rescue 
            raise ActsAsSolr::ConnectionError, 
                  "Couldn't connect to the Solr server at #{ENV["WEBSOLR_URL"]}. #{$!}"
            false
          end
        end
      end
    end
  rescue LoadError
    #ignore
  end

end