githubmunger: label PRs based on certain files touched

github/github.go

@@ -161,9 +161,19 @@ func GetLabelsWithPrefix(labels []github.Label, prefix string) []string {
 	return ret
 }
 
+func GetLabelSet(labels []github.Label) util.StringSet {
+	set := util.NewStringSet()
+	for _, label := range labels {
+		if label.Name != nil {
+			set.Insert(*label.Name)
+		}
+	}
+	return set
+}
+
 func (config *GithubConfig) AddLabels(prNum int, labels []string) error {
 	if config.DryRun {
-		glog.Infof("Would have added labels %v to PR %d --dry-run is set", labels, prNum)
+		glog.Infof("Would have added labels %v to PR %d but --dry-run is set", labels, prNum)
 		return nil
 	}
 	if _, _, err := config.client.Issues.AddLabelsToIssue(config.Org, config.Project, prNum, labels); err != nil {
@@ -175,7 +185,7 @@ func (config *GithubConfig) AddLabels(prNum int, labels []string) error {
 
 func (config *GithubConfig) RemoveLabel(prNum int, label string) error {
 	if config.DryRun {
-		glog.Infof("Would have removed label %q to PR %d --dry-run is set", label, prNum)
+		glog.Infof("Would have removed label %q to PR %d but --dry-run is set", label, prNum)
 		return nil
 	}
 	if _, err := config.client.Issues.RemoveLabelForIssue(config.Org, config.Project, prNum, label); err != nil {

mungegithub/pulls/label_based_on_path.go

@@ -0,0 +1,73 @@
+/*
+Copyright 2015 The Kubernetes Authors All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package pulls
+
+import (
+	"fmt"
+	"strings"
+
+	github_util "k8s.io/contrib/github"
+	"k8s.io/contrib/mungegithub/config"
+	"k8s.io/kubernetes/pkg/util"
+
+	"github.com/golang/glog"
+	"github.com/google/go-github/github"
+	"github.com/spf13/cobra"
+)
+
+var (
+	_        = fmt.Print
+	labelMap = map[string]string{
+		"docs/proposals":         "kind/design",
+		"pkg/api/register.go":    "kind/new-api",
+		"pkg/expapi/register.go": "kind/new-api",
+		"pkg/api/types.go":       "kind/api-change",
+		"pkg/expapi/types.go":    "kind/api-change",
+	}
+)
+
+type PathLabelMunger struct{}
+
+func init() {
+	RegisterMungerOrDie(PathLabelMunger{})
+}
+
+func (PathLabelMunger) Name() string { return "path-label" }
+
+func (PathLabelMunger) AddFlags(cmd *cobra.Command) {}
+
+func (PathLabelMunger) MungePullRequest(config *config.MungeConfig, pr *github.PullRequest, issue *github.Issue, commits []github.RepositoryCommit, events []github.IssueEvent) {
+	glog.V(8).Infof("Checking out PR %d\n", *pr.Number)
+
+	currentLabels := github_util.GetLabelSet(issue.Labels)
+
+	mungerLabels := util.StringSet{}
+	for _, c := range commits {
+		for _, f := range c.Files {
+			for prefix, label := range labelMap {
+				if strings.HasPrefix(*f.Filename, prefix) {
+					mungerLabels.Insert(label)
+				}
+			}
+		}
+	}
+
+	needsLabels := mungerLabels.Difference(currentLabels)
+	if len(needsLabels) != 0 {
+		config.AddLabels(*pr.Number, needsLabels.List())
+	}
+}

mungegithub/run.sh

@@ -16,7 +16,7 @@
 
 while true; do
     /mungegithub \
-      --pr-mungers=blunderbuss,needs-rebase,ok-to-test,ping-ci,size \
+      --pr-mungers=blunderbuss,needs-rebase,ok-to-test,ping-ci,size,path-label \
       --blunderbuss-config=/blunderbuss.yml \
       --token-file=/etc/secret-volume/token
     sleep 600