Method: LibGems::Commands::MirrorCommand#execute

Defined in:
lib/libgems/commands/mirror_command.rb

#executeObject



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
# File 'lib/libgems/commands/mirror_command.rb', line 28

def execute
  config_file = File.join LibGems.user_home, '.gemmirrorrc'

  raise "Config file #{config_file} not found" unless File.exist? config_file

  mirrors = YAML.load_file config_file

  raise "Invalid config file #{config_file}" unless mirrors.respond_to? :each

  mirrors.each do |mir|
    raise "mirror missing 'from' field" unless mir.has_key? 'from'
    raise "mirror missing 'to' field" unless mir.has_key? 'to'

    get_from = mir['from']
    save_to = File.expand_path mir['to']

    raise "Directory not found: #{save_to}" unless File.exist? save_to
    raise "Not a directory: #{save_to}" unless File.directory? save_to

    gems_dir = File.join save_to, "gems"

    if File.exist? gems_dir then
      raise "Not a directory: #{gems_dir}" unless File.directory? gems_dir
    else
      Dir.mkdir gems_dir
    end

    source_index_data = ''

    say "fetching: #{get_from}/Marshal.#{LibGems.marshal_version}.Z"

    get_from = URI.parse get_from

    if get_from.scheme.nil? then
      get_from = get_from.to_s
    elsif get_from.scheme == 'file' then
      # check if specified URI contains a drive letter (file:/D:/Temp)
      get_from = get_from.to_s
      get_from = if get_from =~ /^file:.*[a-z]:/i then
                   get_from[6..-1]
                 else
                   get_from[5..-1]
                 end
    end

    open File.join(get_from.to_s, "Marshal.#{LibGems.marshal_version}.Z"), "rb" do |y|
      source_index_data = Zlib::Inflate.inflate y.read
      open File.join(save_to, "Marshal.#{LibGems.marshal_version}"), "wb" do |out|
        out.write source_index_data
      end
    end

    source_index = Marshal.load source_index_data

    progress = ui.progress_reporter source_index.size,
                                    "Fetching #{source_index.size} gems"
    source_index.each do |fullname, gem|
      gem_file = gem.file_name
      gem_dest = File.join gems_dir, gem_file

      unless File.exist? gem_dest then
        begin
          open "#{get_from}/gems/#{gem_file}", "rb" do |g|
            contents = g.read
            open gem_dest, "wb" do |out|
              out.write contents
            end
          end
        rescue
          old_gf = gem_file
          gem_file = gem_file.downcase
          retry if old_gf != gem_file
          alert_error $!
        end
      end

      progress.updated gem_file
    end

    progress.done
  end
end