Fix author-in-team (#154)

Use the right GH API to resolve team membership

Signed-off-by: Galo Navarro <[email protected]>

Author: srvaroa

cmd/action.go

@@ -231,7 +231,7 @@ func newLabeler(gh *github.Client, config *labeler.LabelerConfigV1) *labeler.Lab
 			GetRawDiff: func(owner, repo string, prNumber int) (string, error) {
 				diff, _, err := gh.PullRequests.GetRaw(ctx,
 					owner, repo, prNumber,
-					github.RawOptions{github.Diff})
+					github.RawOptions{Type: github.Diff})
 				return diff, err
 			},
 			ListIssuesByRepo: func(owner, repo string) ([]*github.Issue, error) {
@@ -244,8 +244,8 @@ func newLabeler(gh *github.Client, config *labeler.LabelerConfigV1) *labeler.Lab
 					owner, repo, &github.PullRequestListOptions{})
 				return prs, err
 			},
-			IsUserMemberOfTeam: func(user, team string) (bool, error) {
-				membership, _, err := gh.Organizations.GetOrgMembership(ctx, user, team)
+			IsUserMemberOfTeam: func(org, user, team string) (bool, error) {
+				membership, _, err := gh.Teams.GetTeamMembershipBySlug(ctx, org, team, user)
 				if err != nil {
 					return false, err
 				}

pkg/condition_author_in_team.go

@@ -17,7 +17,11 @@ func AuthorInTeamCondition(l *Labeler) Condition {
 				return false, fmt.Errorf("author-in-team is not set in config")
 			}
 			// check if author is a member of team
-			return l.GitHubFacade.IsUserMemberOfTeam(target.Author, matcher.AuthorInTeam)
+			return l.GitHubFacade.IsUserMemberOfTeam(
+				target.Owner,
+				target.Author,
+				matcher.AuthorInTeam, // this is the team slug
+			)
 		},
 	}
 }

pkg/labeler.go

@@ -69,7 +69,7 @@ type GitHubFacade struct {
 	GetRawDiff         func(owner, repo string, prNumber int) (string, error)
 	ListIssuesByRepo   func(owner, repo string) ([]*gh.Issue, error)
 	ListPRs            func(owner, repo string) ([]*gh.PullRequest, error)
-	IsUserMemberOfTeam func(user, team string) (bool, error)
+	IsUserMemberOfTeam func(org, user, team string) (bool, error)
 }
 
 type Labeler struct {

pkg/labeler_test.go

@@ -1075,7 +1075,7 @@ func NewTestLabeler(t *testing.T, tc TestCase) Labeler {
 				return string(data), nil
 			},
 			// Will return true whenever team contains the given user name
-			IsUserMemberOfTeam: func(user, team string) (bool, error) {
+			IsUserMemberOfTeam: func(org, user, team string) (bool, error) {
 				return strings.Contains(team, user), nil
 			},
 		},