Skip to content
This repository
Fetching contributors…

Octocat-spinner-32-eaf2f5

Cannot retrieve contributors at this time

executable file 120 lines (105 sloc) 3.585 kb
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
#--
# Copyright 2006 by Chad Fowler, Rich Kilmer, Jim Weirich and others.
# All rights reserved.
# See LICENSE.txt for permissions.
#++

require 'rubygems/source_index'

module Kernel
  alias gem_original_require require # :nodoc:

  #
  # We replace Ruby's require with our own, which is capable of
  # loading gems on demand.
  #
  # When you call <tt>require 'x'</tt>, this is what happens:
  # * If the file can be loaded from the existing Ruby loadpath, it
  # is.
  # * Otherwise, installed gems are searched for a file that matches.
  # If it's found in gem 'y', that gem is activated (added to the
  # loadpath).
  #
  # The normal <tt>require</tt> functionality of returning false if
  # that file has already been loaded is preserved.
  #
  def require(path) # :nodoc:
    gem_original_require path
  rescue LoadError => load_error
    begin
      if spec = Gem.searcher.find(path)
        Gem.activate(spec.name, false, "= #{spec.version}")
        gem_original_require path
      else
        raise load_error
      end
    end
  end
end # module Kernel

module Gem

  #
  # GemPathSearcher has the capability to find loadable files inside
  # gems. It generates data up front to speed up searches later.
  #
  class GemPathSearcher
    
    #
    # Initialise the data we need to make searches later.
    #
    def initialize
      # We want a record of all the installed gemspecs, in the order
      # we wish to examine them.
      @gemspecs = init_gemspecs
      # Map gem spec to glob of full require_path directories.
      # Preparing this information may speed up searches later.
      @lib_dirs = {}
      @gemspecs.each do |spec|
        @lib_dirs[spec.object_id] = lib_dirs(spec)
      end
    end

    #
    # Look in all the installed gems until a matching _path_ is found.
    # Return the _gemspec_ of the gem where it was found. If no match
    # is found, return nil.
    #
    # The gems are searched in alphabetical order, and in reverse
    # version order.
    #
    # For example:
    #
    # find('log4r') # -> (log4r-1.1 spec)
    # find('log4r.rb') # -> (log4r-1.1 spec)
    # find('rake/rdoctask') # -> (rake-0.4.12 spec)
    # find('foobarbaz') # -> nil
    #
    # Matching paths can have various suffixes ('.rb', '.so', and
    # others), which may or may not already be attached to _file_.
    # This method doesn't care about the full filename that matches;
    # only that there is a match.
    #
    def find(path)
      @gemspecs.each do |spec|
        return spec if matching_file(spec, path)
      end
      nil
    end

    private

    # Attempts to find a matching path using the require_paths of the
    # given _spec_.
    #
    # Some of the intermediate results are cached in @lib_dirs for
    # speed.
    def matching_file(spec, path) # :doc:
      glob = File.join @lib_dirs[spec.object_id], "#{path}#{Gem.suffix_pattern}"
      return true unless Dir[glob].select { |f| File.file?(f.untaint) }.empty?
    end

    # Return a list of all installed gemspecs, sorted by alphabetical
    # order and in reverse version order.
    def init_gemspecs
      Gem.source_index.map { |_, spec| spec }.sort { |a,b|
        (a.name <=> b.name).nonzero? || (b.version <=> a.version)
      }
    end

    # Returns library directories glob for a gemspec. For example,
    # '/usr/local/lib/ruby/gems/1.8/gems/foobar-1.0/{lib,ext}'
    def lib_dirs(spec)
      "#{spec.full_gem_path}/{#{spec.require_paths.join(',')}}"
    end

  end # class Gem::GemPathLoader

end # module Gem

Something went wrong with that request. Please try again.