require 'rubygems/user_interaction'

require 'forwardable'

module Gem

  # The SourceIndex object indexes all the gems available from a
  # particular source (e.g. a list of gem directories, or a remote
  # source).  A SourceIndex maps a gem full name to a gem
  # specification.
  #
  # NOTE:: The class used to be named Cache, but that became
  #        confusing when cached source fetchers where introduced.
  #        The constant Gem::Cache is an alias for this class to allow
  #        old YAMLized source index objects to load properly.
  #
  class SourceIndex
    extend Forwardable
    include Enumerable

    class << self
      include Gem::UserInteraction
    end
    
    # Constructs a source index instance from the provided
    # specifications
    #
    # specifications::
    #   [Hash] hash of [Gem name, Gem::Specification] pairs
    #
    def initialize(specifications)
      @gems = specifications
    end
    
    # Factory method to construct a source index instance for a given
    # path.
    # 
    # source_dirs::
    #   List of gem directories to search for specifications.  The
    #   default is the "specification" directories under each
    #   directory in Gem.path.
    #
    # return::
    #   SourceIndex instance
    #
    def self.from_installed_gems(*spec_dirs)
      gems = {}
      if spec_dirs.empty?
        spec_dirs = Gem.path.collect { |dir| File.join(dir, "specifications") }
      end
      Dir.glob("{#{spec_dirs.join(',')}}/*.gemspec").each do |file_name|
        gemspec = load_specification(file_name)
        gems[gemspec.full_name] = gemspec if gemspec
      end
      self.new(gems)
    end

    # Load a specification from a file (eval'd Ruby code)
    # 
    # file_name:: [String] The .gemspec file
    # return:: Specification instance or nil if an error occurs
    #
    def self.load_specification(file_name)
      begin
        spec_code = File.read(file_name)
        gemspec = eval(spec_code)
        if gemspec.is_a?(Gem::Specification)
          gemspec.loaded_from = file_name
          return gemspec
        end
        alert_warning "File '#{file_name}' does not evaluate to a gem specification"
      rescue SyntaxError => e
        alert_warning e
        alert_warning spec_code
      rescue Exception => e
        alert_warning(e.inspect.to_s + "\n" + spec_code)
        alert_warning "Invalid .gemspec format in '#{file_name}'"
      end
      return nil
    end
    
    # Iterate over the specifications in the source index.
    #
    # &block:: [yields gem.long_name, Gem::Specification]
    #
    def each(&block)
      @gems.each(&block)
    end

    def_delegators :@gems, :size, :length

    # Find a gem by an exact match on the short name.
    def find_name(gem_name, version_requirement=Version::Requirement.new(">= 0"))
      search(/^#{gem_name}$/, version_requirement)
    end

    # Search for a gem by short name pattern and optional version
    #
    # gem_name::
    #   [String] a partial for the (short) name of the gem, or
    #   [Regex] a pattern to match against the short name
    # version_requirement::
    #   [String | default=Version::Requirement.new(">= 0")] version to
    #   find
    # return::
    #   [Array] list of Gem::Specification objects in sorted (version)
    #   order.  Empty if not found.
    #
    def search(gem_pattern, version_requirement=Version::Requirement.new(">= 0"))
      #FIXME - remove duplication between this and RemoteInstaller.search
      gem_pattern = /#{ gem_pattern }/i if String === gem_pattern
      version_requirement = Gem::Version::Requirement.create(version_requirement)
      result = []
      @gems.each do |full_spec_name, spec|
        next unless spec.name =~ gem_pattern
        result << spec if version_requirement.satisfied_by?(spec.version)
      end
      result = result.sort
      result
    end

    # Refresh the source index from the local file system.
    #
    # return:: Returns a pointer to itself.
    #
    def refresh!
      spec_dirs = Gem.path.collect {|dir| File.join(dir, "specifications")}
      files = Dir.glob("{#{spec_dirs.join(',')}}/*.gemspec")
      current_loaded_files = @gems.values.collect {|spec| spec.loaded_from}
      (files - current_loaded_files).each do |spec_file|
        gemspec = Gem::SourceIndex.load_specification(spec_file)

        @gems[gemspec.full_name] = gemspec if gemspec
      end
      self
    end
    
  end

  # Cache is an alias for SourceIndex to allow older YAMLized source
  # index objects to load properly.
  Cache = SourceIndex

end