class Bundler::Source::Rubygems

def self.from_lock(options)
  s = new(options)
  Array(options["remote"]).each { |r| s.add_remote(r) }
  s
end

def add_remote(source)
  @remotes << normalize_uri(source)
end

def cache(spec)
  cached_path = cached_gem(spec)
  raise GemNotFound, "Missing gem file '#{spec.full_name}.gem'." unless cached_path
  return if File.dirname(cached_path) == Bundler.app_cache.to_s
  Bundler.ui.info "  * #{File.basename(cached_path)}"
  FileUtils.cp(cached_path, Bundler.app_cache)
end

def cached!
  @allow_cached = true
end

def cached_gem(spec)
  possibilities = @caches.map { |p| "#{p}/#{spec.full_name}.gem" }
  possibilities.find { |p| File.exist?(p) }
end

def cached_specs
  @cached_specs ||= begin
    idx = Index.new
    @caches.each do |path|
      Dir["#{path}/*.gem"].each do |gemfile|
        next if name == 'bundler'
        s = Gem::Format.from_file_by_path(gemfile).spec
        s.source = self
        idx << s
      end
    end
    idx
  end
end

def download_gem_from_uri(spec, uri)
  spec.fetch_platform
  download_path = Bundler.requires_sudo? ? Bundler.tmp : Gem.dir
  gem_path = "#{Gem.dir}/cache/#{spec.full_name}.gem"
  FileUtils.mkdir_p("#{download_path}/cache")
  Gem::RemoteFetcher.fetcher.download(spec, uri, download_path)
  if Bundler.requires_sudo?
    sudo "mkdir -p #{Gem.dir}/cache"
    sudo "mv #{Bundler.tmp}/cache/#{spec.full_name}.gem #{gem_path}"
  end
  gem_path
end

def eql?(o)
  Rubygems === o
end

def fetch(spec)
  action = @spec_fetch_map[spec.full_name]
  action.call if action
end

def fetch_all_remote_specs(&blk)
  begin
    # Fetch all specs, minus prerelease specs
    Gem::SpecFetcher.new.list(true, false).each(&blk)
    # Then fetch the prerelease specs
    begin
      Gem::SpecFetcher.new.list(false, true).each(&blk)
    rescue Gem::RemoteFetcher::FetchError
      Bundler.ui.warn "Could not fetch prerelease specs from #{self}"
    end
  rescue Gem::RemoteFetcher::FetchError
    Bundler.ui.warn "Could not reach #{self}"
  end
end

def fetch_specs
  Index.build do |idx|
    idx.use installed_specs
    idx.use cached_specs if @allow_cached
    idx.use remote_specs if @allow_remote
  end
end

def hash
  Rubygems.hash
end

def initialize(options = {})
  @options = options
  @remotes = (options["remotes"] || []).map { |r| normalize_uri(r) }
  @allow_remote = false
  @allow_cached = false
  # Hardcode the paths for now
  @caches = [ Bundler.app_cache ] + Gem.path.map { |p| File.expand_path("#{p}/cache") }
  @spec_fetch_map = {}
end

def install(spec)
  path = cached_gem(spec)
  if installed_specs[spec].any?
    Bundler.ui.info "Using #{spec.name} (#{spec.version}) "
    return
  end
  Bundler.ui.info "Installing #{spec.name} (#{spec.version}) "
  install_path = Bundler.requires_sudo? ? Bundler.tmp : Gem.dir
  installer = Gem::Installer.new path,
    :install_dir         => install_path,
    :ignore_dependencies => true,
    :wrappers            => true,
    :env_shebang         => true,
    :bin_dir             => "#{install_path}/bin"
  installer.install
  # SUDO HAX
  if Bundler.requires_sudo?
    sudo "mkdir -p #{Gem.dir}/gems #{Gem.dir}/specifications"
    sudo "cp -R #{Bundler.tmp}/gems/#{spec.full_name} #{Gem.dir}/gems/"
    sudo "cp -R #{Bundler.tmp}/specifications/#{spec.full_name}.gemspec #{Gem.dir}/specifications/"
  end
  spec.loaded_from = "#{Gem.dir}/specifications/#{spec.full_name}.gemspec"
end

def installed_specs
  @installed_specs ||= begin
    idx = Index.new
    have_bundler = false
    Gem::SourceIndex.from_installed_gems.to_a.reverse.each do |dont_use_this_var, spec|
      next if spec.name == 'bundler' && spec.version.to_s != VERSION
      have_bundler = true if spec.name == 'bundler'
      spec.source = self
      idx << spec
    end
    # Always have bundler locally
    unless have_bundler
     # We're running bundler directly from the source
     # so, let's create a fake gemspec for it (it's a path)
     # gemspec
     bundler = Gem::Specification.new do |s|
       s.name     = 'bundler'
       s.version  = VERSION
       s.platform = Gem::Platform::RUBY
       s.source   = self
       # TODO: Remove this
       s.loaded_from = 'w0t'
     end
     idx << bundler
    end
    idx
  end
end

def name
  ":gems"
end

def normalize_uri(uri)
  uri = uri.to_s
  uri = "#{uri}/" unless uri =~ %r'/$'
  uri = URI(uri)
  raise ArgumentError, "The source must be an absolute URI" unless uri.absolute?
  uri
end

def options
  { "remotes" => @remotes.map { |r| r.to_s } }
end

def remote!
  @allow_remote = true
end

def remote_specs
  @remote_specs ||= begin
    idx     = Index.new
    remotes = self.remotes.map { |uri| uri.to_s }
    old     = Gem.sources
    remotes.each do |uri|
      Bundler.ui.info "Fetching source index for #{uri}"
      Gem.sources = ["#{uri}"]
      fetch_all_remote_specs do |n,v|
        v.each do |name, version, platform|
          next if name == 'bundler'
          spec = RemoteSpecification.new(name, version, platform, uri)
          spec.source = self
          # Temporary hack until this can be figured out better
          @spec_fetch_map[spec.full_name] = lambda do
            path = download_gem_from_uri(spec, uri)
            s = Gem::Format.from_file_by_path(path).spec
            spec.__swap__(s)
          end
          idx << spec
        end
      end
    end
    idx
  ensure
    Gem.sources = old
  end
end

def specs
  @specs ||= fetch_specs
end

def sudo(str)
  Bundler.sudo(str)
end

def to_lock
  out = "GEM\n"
  out << remotes.map {|r| "  remote: #{r}\n" }.join
  out << "  specs:\n"
end

def to_s
  remotes = self.remotes.map { |r| r.to_s }.join(', ')
  "rubygems repository #{remotes}"
end