1 files changed, 44 insertions, 0 deletions
diff --git a/lib/rubygems/indexer/master_index_builder.rb b/lib/rubygems/indexer/master_index_builder.rb
new file mode 100644
index 0000000000..f435c44e41
--- /dev/null
+++ b/lib/rubygems/indexer/master_index_builder.rb
@@ -0,0 +1,44 @@
+require 'rubygems/indexer'
+
+# Construct the master Gem index file.
+class Gem::Indexer::MasterIndexBuilder < Gem::Indexer::AbstractIndexBuilder
+
+  def start_index
+    super
+    @index = Gem::SourceIndex.new
+  end
+
+  def end_index
+    super
+    @file.puts @index.to_yaml
+  end
+
+  def cleanup
+    super
+
+    index_file_name = File.join @directory, @filename
+
+    compress index_file_name, "Z"
+    compressed_file_name = "#{index_file_name}.Z"
+
+    paranoid index_file_name, compressed_file_name
+
+    @files << compressed_file_name
+  end
+
+  def add(spec)
+    @index.add_spec(spec)
+  end
+
+  private
+
+  def paranoid(fn, compressed_fn)
+    data = File.open(fn, 'rb') do |fp| fp.read end
+    compressed_data = File.open(compressed_fn, 'rb') do |fp| fp.read end
+
+    if data != unzip(compressed_data) then
+      fail "Compressed file #{compressed_fn} does not match uncompressed file #{fn}"
+    end
+  end
+
+end