879aad8338
Prior to this change, the approvers could only be explicit users. With this change, you can now specify an org team and this will be expanded out with a user list for approvers. Closes #14.
71 lines
1.8 KiB
Go
71 lines
1.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/google/go-github/v43/github"
|
|
)
|
|
|
|
func retrieveApprovers(client *github.Client, repoOwner string) ([]string, error) {
|
|
approvers := []string{}
|
|
|
|
requiredApproversRaw := os.Getenv(envVarApprovers)
|
|
requiredApprovers := strings.Split(requiredApproversRaw, ",")
|
|
|
|
for _, approverUser := range requiredApprovers {
|
|
expandedUsers := expandGroupFromUser(client, repoOwner, approverUser)
|
|
if expandedUsers != nil {
|
|
approvers = append(approvers, expandedUsers...)
|
|
} else {
|
|
approvers = append(approvers, approverUser)
|
|
}
|
|
}
|
|
|
|
approvers = deduplicateUsers(approvers)
|
|
|
|
minimumApprovalsRaw := os.Getenv(envVarMinimumApprovals)
|
|
minimumApprovals := len(approvers)
|
|
var err error
|
|
if minimumApprovalsRaw != "" {
|
|
minimumApprovals, err = strconv.Atoi(minimumApprovalsRaw)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error parsing minimum number of approvals: %w", err)
|
|
}
|
|
}
|
|
|
|
if minimumApprovals > len(approvers) {
|
|
return nil, fmt.Errorf("error: minimum required approvals (%d) is greater than the total number of approvers (%d)", minimumApprovals, len(approvers))
|
|
}
|
|
|
|
return approvers, nil
|
|
}
|
|
|
|
func expandGroupFromUser(client *github.Client, org, userOrTeam string) []string {
|
|
users, _, err := client.Teams.ListTeamMembersBySlug(context.Background(), org, userOrTeam, &github.TeamListTeamMembersOptions{})
|
|
if err != nil || len(users) == 0 {
|
|
return nil
|
|
}
|
|
|
|
userNames := make([]string, 0, len(users))
|
|
for _, user := range users {
|
|
userNames = append(userNames, user.GetLogin())
|
|
}
|
|
|
|
return userNames
|
|
}
|
|
|
|
func deduplicateUsers(users []string) []string {
|
|
uniqValuesByKey := make(map[string]bool)
|
|
uniqUsers := []string{}
|
|
for _, user := range users {
|
|
if _, ok := uniqValuesByKey[user]; !ok {
|
|
uniqValuesByKey[user] = true
|
|
uniqUsers = append(uniqUsers, user)
|
|
}
|
|
}
|
|
return uniqUsers
|
|
}
|