Land rapid7#5000, tools/dev/add_pr_fetch.rb

Author: wvu

tools/dev/add_pr_fetch.rb

@@ -0,0 +1,33 @@
+#!/usr/bin/env ruby
+
+toplevel = %x{git rev-parse --show-toplevel}.strip
+infile = "#{toplevel}/.git/config"
+outfile = infile
+$stderr.puts "Rewriting #{infile}"
+data = File.open(infile, 'rb') {|f| f.read f.stat.size}
+newdata = ""
+data.each_line do |line|
+  newdata << line
+  case line
+  when /^(\s*)fetch\s*=.*remotes\/([^\/]+)\//
+    ws = $1
+    remote = $2
+    pr_line = "fetch = +refs/pull/*/head:refs/remotes/#{remote}/pr/*"
+    next if line.strip == pr_line.strip
+    if data.include? pr_line
+      $stderr.puts "Skipping #{remote}, already present"
+      next
+    else
+      @new_pr_line ||= true
+      $stderr.puts "Adding pull request fetch for #{remote}"
+      newdata << "#{ws}#{pr_line}\n"
+    end
+  end
+end
+
+if @new_pr_line
+  File.open(outfile, 'wb') {|f| f.write newdata}
+  $stderr.puts "Wrote #{outfile}"
+else
+  $stderr.puts "No changes to #{outfile}"
+end