require 'erb'
require 'yaml'

module Sunspot #:nodoc:
  module Rails #:nodoc:
    #
    # Sunspot::Rails is configured via the config/sunspot.yml file, which
    # contains properties keyed by environment name. A sample sunspot.yml file
    # would look like:
    #
    #   development:
    #     solr:
    #       hostname: localhost
    #       port: 8982
    #       memory: 1G
    #       solr_jar: /some/path/solr15/start.jar
    #       bind_address: 0.0.0.0
    #       proxy: false
    #     disabled: false
    #   test:
    #     solr:
    #       hostname: localhost
    #       port: 8983
    #       log_level: OFF
    #       open_timeout: 0.5
    #       read_timeout: 2
    #       proxy: false
    #   production:
    #     solr:
    #       scheme: http
    #       user: username
    #       pass: password
    #       hostname: localhost
    #       port: 8983
    #       path: /solr/myindex
    #       log_level: WARNING
    #       solr_home: /some/path
    #       open_timeout: 0.5
    #       read_timeout: 2
    #       proxy: http://proxy.com:12345
    #     master_solr:
    #       hostname: localhost
    #       port: 8982
    #       path: /solr
    #     auto_index_callback: after_commit
    #     auto_remove_callback: after_commit
    #     auto_commit_after_request: true
    #
    # Sunspot::Rails uses the configuration to set up the Solr connection, as
    # well as for starting Solr with the appropriate port using the
    # <code>rake sunspot:solr:start</code> task.
    #
    # If the <code>master_solr</code> configuration is present, Sunspot will use
    # the Solr instance specified here for all write operations, and the Solr
    # configured under <code>solr</code> for all read operations.
    #
    class Configuration
      # ActiveSupport log levels are integers; this array maps them to the
      # appropriate java.util.logging.Level constant
      LOG_LEVELS = %w(FINE INFO WARNING SEVERE SEVERE INFO)

      attr_writer :user_configuration
      #
      # The host name at which to connect to Solr. Default 'localhost'.
      #
      # ==== Returns
      #
      # String:: host name
      #
      def hostname
        unless defined?(@hostname)
          @hostname   = solr_url.host if solr_url
          @hostname ||= user_configuration_from_key('solr', 'hostname')
          @hostname ||= default_hostname
        end
        @hostname
      end

      #
      # The port at which to connect to Solr.
      # Defaults to 8981 in test, 8982 in development and 8983 in production.
      #
      # ==== Returns
      #
      # Integer:: port
      #
      def port
        unless defined?(@port)
          @port   = solr_url.port if solr_url
          @port ||= user_configuration_from_key('solr', 'port')
          @port ||= default_port
          @port   = @port.to_i
        end
        @port
      end

      #
      # The scheme to use, http or https.
      # Defaults to http
      #
      # ==== Returns
      #
      # String:: scheme
      #
      def scheme
        unless defined?(@scheme)
          @scheme   = solr_url.scheme if solr_url
          @scheme ||= user_configuration_from_key('solr', 'scheme')
          @scheme ||= default_scheme
        end
        @scheme
      end

      #
      # The userinfo used for authentication, a colon-delimited string like "user:pass"
      # Defaults to nil, which means no authentication
      #
      # ==== Returns
      #
      # String:: userinfo
      #
      def userinfo
        unless defined?(@userinfo)
          @userinfo   = solr_url.userinfo if solr_url
          user = user_configuration_from_key('solr', 'user')
          pass = user_configuration_from_key('solr', 'pass')
          @userinfo ||= [ user, pass ].compact.join(":") if user && pass
          @userinfo ||= default_userinfo
        end
        @userinfo
      end

      #
      # The url path to the Solr servlet (useful if you are running multicore).
      # Default '/solr/default'.
      #
      # ==== Returns
      #
      # String:: path
      #
      def path
        unless defined?(@path)
          @path   = solr_url.path if solr_url
          @path ||= user_configuration_from_key('solr', 'path')
          @path ||= default_path
        end
        @path
      end

      #
      # The host name at which to connect to the master Solr instance. Defaults
      # to the 'hostname' configuration option.
      #
      # ==== Returns
      #
      # String:: host name
      #
      def master_hostname
        @master_hostname ||= (user_configuration_from_key('master_solr', 'hostname') || hostname)
      end

      #
      # The port at which to connect to the master Solr instance. Defaults to
      # the 'port' configuration option.
      #
      # ==== Returns
      #
      # Integer:: port
      #
      def master_port
        @master_port ||= (user_configuration_from_key('master_solr', 'port') || port).to_i
      end

      #
      # The path to the master Solr servlet (useful if you are running multicore).
      # Defaults to the value of the 'path' configuration option.
      #
      # ==== Returns
      #
      # String:: path
      #
      def master_path
        @master_path ||= (user_configuration_from_key('master_solr', 'path') || path)
      end

      #
      # True if there is a master Solr instance configured, otherwise false.
      #
      # ==== Returns
      #
      # Boolean:: bool
      #
      def has_master?
        @has_master = !!user_configuration_from_key('master_solr')
      end

      #
      # The default log_level that should be passed to solr. You can
      # change the individual log_levels in the solr admin interface.
      # If no level is specified in the sunspot configuration file,
      # use a level similar to Rails own logging level.
      #
      # ==== Returns
      #
      # String:: log_level
      #
      def log_level
        @log_level ||= (
          user_configuration_from_key('solr', 'log_level') ||
          LOG_LEVELS[::Rails.logger.level]
        )
      end

      #
      # Should the solr index receive a commit after each http-request.
      # Default true
      #
      # ==== Returns
      #
      # Boolean: auto_commit_after_request?
      #
      def auto_commit_after_request?
        @auto_commit_after_request ||=
          user_configuration_from_key('auto_commit_after_request') != false
      end

      #
      # As for #auto_commit_after_request? but only for deletes
      # Default false
      #
      # ==== Returns
      #
      # Boolean: auto_commit_after_delete_request?
      #
      def auto_commit_after_delete_request?
        @auto_commit_after_delete_request ||=
          (user_configuration_from_key('auto_commit_after_delete_request') || false)
      end


      #
      # The log directory for solr logfiles
      #
      # ==== Returns
      #
      # String:: log_dir
      #
      def log_file
        @log_file ||= (user_configuration_from_key('solr', 'log_file') || default_log_file_location )
      end

      def pid_dir
        @pid_dir ||= user_configuration_from_key('solr', 'pid_dir') || File.join(::Rails.root, 'solr', 'pids', ::Rails.env)
      end


      #
      # The solr home directory. Sunspot::Rails expects this directory
      # to contain a config, data and pids directory. See
      # Sunspot::Rails::Server.bootstrap for more information.
      #
      # ==== Returns
      #
      # String:: solr_home
      #
      def solr_home
        @solr_home ||=
          if user_configuration_from_key('solr', 'solr_home')
            user_configuration_from_key('solr', 'solr_home')
          else
            File.join(::Rails.root, 'solr')
          end
      end

      #
      # Solr start jar
      #
      def solr_executable
        @solr_executable ||= user_configuration_from_key('solr', 'solr_executable')
      end

      #
      # java heap size for Solr instance
      #
      def memory
        @memory ||= user_configuration_from_key('solr', 'memory')
      end

      #
      # Interface on which to run Solr
      #
      def bind_address
        @bind_address ||= user_configuration_from_key('solr', 'bind_address')
      end

      def read_timeout
        @read_timeout ||= user_configuration_from_key('solr', 'read_timeout')
      end

      def open_timeout
        @open_timeout ||= user_configuration_from_key('solr', 'open_timeout')
      end

      def proxy
        @proxy ||= user_configuration_from_key('solr', 'proxy')
      end

      #
      # Whether or not to disable Solr.
      # Defaults to false.
      #
      def disabled?
        @disabled ||= (user_configuration_from_key('disabled') || false)
      end

      #
      # The callback to use when automatically indexing records.
      # Defaults to after_save.
      #
      def auto_index_callback
        @auto_index_callback ||=
          (user_configuration_from_key('auto_index_callback') || 'after_save')
      end

      #
      # The callback to use when automatically removing records after deletation.
      # Defaults to after_destroy.
      #
      def auto_remove_callback
        @auto_remove_callback ||=
          (user_configuration_from_key('auto_remove_callback') || 'after_destroy')
      end

      private

      #
      # Logging in rails_root/log as solr_<environment>.log as a
      # default.
      #
      # ===== Returns
      #
      # String:: default_log_file_location
      #
      def default_log_file_location
        File.join(::Rails.root, 'log', "solr_" + ::Rails.env + ".log")
      end

      #
      # return a specific key from the user configuration in config/sunspot.yml
      #
      # ==== Returns
      #
      # Mixed:: requested_key or nil
      #
      def user_configuration_from_key( *keys )
        keys.inject(user_configuration) do |hash, key|
          hash[key] if hash
        end
      end

      #
      # Memoized hash of configuration options for the current Rails environment
      # as specified in config/sunspot.yml
      #
      # ==== Returns
      #
      # Hash:: configuration options for current environment
      #
      def user_configuration
        @user_configuration ||=
          begin
            path = File.join(::Rails.root, 'config', 'sunspot.yml')
            if File.exist?(path)
              File.open(path) do |file|
                processed = ERB.new(file.read).result
                YAML.load(processed)[::Rails.env]
              end
            else
              {}
            end
          end
      end

    protected

      #
      # When a specific hostname, port and path aren't provided in the
      # sunspot.yml file, look for a key named 'url', then check the
      # environment, then fall back to a sensible localhost default.
      #

      def solr_url
        if ENV['SOLR_URL'] || ENV['WEBSOLR_URL']
          URI.parse(ENV['SOLR_URL'] || ENV['WEBSOLR_URL'])
        end
      end

      def default_hostname
        'localhost'
      end

      def default_port
        { 'test'        => 8981,
          'development' => 8982,
          'production'  => 8983
        }[::Rails.env]  || 8983
      end

      def default_scheme
        'http'
      end

      def default_userinfo
        nil
      end

      def default_path
        '/solr/default'
      end

    end
  end
end