96 lines
3.2 KiB
Go
96 lines
3.2 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) {
|
|
workflowInitiator := os.Getenv(envVarWorkflowInitiator)
|
|
shouldExcludeWorkflowInitiatorRaw := os.Getenv(envVarExcludeWorkflowInitiatorAsApprover)
|
|
shouldExcludeWorkflowInitiator, parseBoolErr := strconv.ParseBool(shouldExcludeWorkflowInitiatorRaw)
|
|
if parseBoolErr != nil {
|
|
return nil, fmt.Errorf("error parsing exclude-workflow-initiator-as-approver flag: %w", parseBoolErr)
|
|
}
|
|
|
|
approvers := []string{}
|
|
requiredApproversRaw := os.Getenv(envVarApprovers)
|
|
requiredApprovers := strings.Split(requiredApproversRaw, ",")
|
|
|
|
for i := range requiredApprovers {
|
|
requiredApprovers[i] = strings.TrimSpace(requiredApprovers[i])
|
|
}
|
|
|
|
for _, approverUser := range requiredApprovers {
|
|
expandedUsers := expandGroupFromUser(client, repoOwner, approverUser, workflowInitiator, shouldExcludeWorkflowInitiator)
|
|
if expandedUsers != nil {
|
|
approvers = append(approvers, expandedUsers...)
|
|
} else if strings.EqualFold(workflowInitiator, approverUser) && shouldExcludeWorkflowInitiator {
|
|
fmt.Printf("Not adding user '%s' as an approver as they are the workflow initiator\n", approverUser)
|
|
} 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, workflowInitiator string, shouldExcludeWorkflowInitiator bool) []string {
|
|
fmt.Printf("Attempting to expand user %s/%s as a group (may not succeed)\n", org, userOrTeam)
|
|
|
|
// GitHub replaces periods in the team name with hyphens. If a period is
|
|
// passed to the request it would result in a 404. So we need to replace
|
|
// and occurrences with a hyphen.
|
|
formattedUserOrteam := strings.ReplaceAll(userOrTeam, ".", "-")
|
|
|
|
users, _, err := client.Teams.ListTeamMembersBySlug(context.Background(), org, formattedUserOrteam, &github.TeamListTeamMembersOptions{})
|
|
if err != nil {
|
|
fmt.Printf("%v\n", err)
|
|
return nil
|
|
}
|
|
|
|
userNames := make([]string, 0, len(users))
|
|
for _, user := range users {
|
|
userName := user.GetLogin()
|
|
if strings.EqualFold(userName, workflowInitiator) && shouldExcludeWorkflowInitiator {
|
|
fmt.Printf("Not adding user '%s' from group '%s' as an approver as they are the workflow initiator\n", userName, userOrTeam)
|
|
} else {
|
|
userNames = append(userNames, userName)
|
|
}
|
|
}
|
|
|
|
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
|
|
}
|