Implement notes-github-pr post-receive hook

Author: k0kubun

bin/notes-github-pr.rb

@@ -0,0 +1,93 @@
+#!/usr/bin/env ruby
+# Add GitHub pull request reference to git notes.
+
+require 'net/http'
+require 'uri'
+require 'tmpdir'
+require 'json'
+
+class GitHub
+  ENDPOINT = URI.parse('https://api.github.com')
+
+  def initialize(access_token)
+    @access_token = access_token
+  end
+
+  # https://developer.github.com/changes/2019-04-11-pulls-branches-for-commit/
+  def pulls(owner:, repo:, commit_sha:)
+    resp = get("/repos/#{owner}/#{repo}/commits/#{commit_sha}/pulls", accept: 'application/vnd.github.groot-preview+json')
+    JSON.parse(resp.body)
+  end
+
+  private
+
+  def get(path, accept: 'application/vnd.github.v3+json')
+    Net::HTTP.start(ENDPOINT.host, ENDPOINT.port, use_ssl: ENDPOINT.scheme == 'https') do |http|
+      headers = { 'Accept': accept, 'Authorization': "bearer #{@access_token}" }
+      http.get(path, headers).tap(&:value)
+    end
+  end
+end
+
+module Git
+  class << self
+    def abbrev_ref(refname)
+      git('rev-parse', '--symbolic', '--abbrev-ref', refname).strip
+    end
+
+    def rev_parse(arg, first_parent: false)
+      git('rev-parse', *[('--first-parent' if first_parent)].compact, arg).lines.map(&:chomp)
+    end
+
+    def commit_message(sha)
+      git('log', '-1', '--pretty=format:%B', sha)
+    end
+
+    def notes_message(sha)
+      git('log', '-1', '--pretty=format:%N', sha)
+    end
+
+    private
+
+    def git(*cmd)
+      out = IO.popen(['git', *cmd], &:read)
+      unless $?.success?
+        abort "Failed to execute: git #{cmd.join(' ')}\n#{out}"
+      end
+      out
+    end
+  end
+end
+
+github = GitHub.new(File.read(File.expand_path('~git/config/github-access-token')).chomp)
+
+repo_path, *rest = ARGV
+rest.each_slice(3).map do |oldrev, newrev, refname|
+  branch = Git.abbrev_ref(refname)
+  next if branch != 'master' # we use pull requests only for master branches
+
+  Dir.mktmpdir do |workdir|
+    depth = Git.rev_parse("#{oldrev}..#{newrev}").size + 1
+    system('git', 'clone', "--depth=#{depth}", "--branch=#{branch}", "file:///#{repo_path}", workdir)
+    Dir.chdir(workdir)
+
+    updated = false
+    Git.rev_parse("#{oldrev}..#{newrev}", first_parent: true).each do |sha|
+      github.pulls(owner: 'ruby', repo: 'ruby', commit_sha: sha).each do |pull|
+        number = pull.fetch('number')
+        url = pull.fetch('url')
+
+        message = Git.commit_message(sha)
+        notes = Git.notes_message(sha)
+        if !message.include?(url) && !message.include?("(##{number})") && !notes.include?(url)
+          system('git', 'notes', 'append', '-m', "Merged: #{url}", sha)
+          updated = true
+        end
+      end
+    end
+
+    if updated
+      system('git', 'push', 'origin', 'refs/notes/commits')
+    end
+  end
+end

hooks/post-receive-pre.sh

@@ -36,7 +36,7 @@ curl -s "https://bugs.ruby-lang.org/sys/fetch_changesets?key=`cat ~git/config/re
 log "==> auto-style"
 SVN_ACCOUNT_NAME=git "${ruby_commit_hook}/bin/auto-style.rb" "$ruby_git" $*
 
-log "==> update-version.h.rb"
+log "==> update-version.h"
 SVN_ACCOUNT_NAME=git "${ruby_commit_hook}/bin/update-version.h.rb" git "$ruby_git" $*
 
 log "### end ###\n"

hooks/post-receive.sh

@@ -35,7 +35,10 @@ curl -s "https://bugs.ruby-lang.org/sys/fetch_changesets?key=`cat ~git/config/re
 log "==> auto-style"
 SVN_ACCOUNT_NAME=git "${ruby_commit_hook}/bin/auto-style.rb" "$ruby_git" $*
 
-log "==> update-version.h.rb"
+log "==> update-version.h"
 SVN_ACCOUNT_NAME=git "${ruby_commit_hook}/bin/update-version.h.rb" git "$ruby_git" $*
 
+log "==> notes-github-pr"
+SVN_ACCOUNT_NAME=git "${ruby_commit_hook}/bin/notes-github-pr.rb" git "$ruby_git" $*
+
 log "### end ###\n"