actions.go

package main

import (
	"fmt"
	"regexp"
	"strings"

	"github.com/opensourceways/robot-framework-lib/client"
	"k8s.io/apimachinery/pkg/util/sets"
)

var regCheckPr = regexp.MustCompile(`(?mi)^/check-pr\s*$`)

func (bot *robot) clearLabel(evt *client.GenericEvent, org, repo, number string) error {

	labels := bot.getPRLabelSet(org, repo, number)
	v := getLGTMLabelsOnPR(labels)

	if labels.Has(approvedLabel) {
		v = append(v, approvedLabel)
	}

	if len(v) > 0 {

		if ok := bot.cli.RemovePRLabels(org, repo, number, v); !ok {
			return nil
		}

		var noteComment string
		if bot.cli.CheckIfPRSourceCodeUpdateEvent(evt) {
			noteComment = commentClearLabelCaseByReopenPR
		}

		if bot.cli.CheckIfPRSourceCodeUpdateEvent(evt) {
			noteComment = commentClearLabelCaseByPRUpdate
		}

		bot.cli.CreatePRComment(
			org, repo, number,
			fmt.Sprintf(noteComment, strings.Join(v, ", ")),
		)
	}

	return nil
}
func (bot *robot) checkCommenterPermission(org, repo, author, commenter string, fn func()) (pass bool) {
	if author == commenter {
		return true
	}
	pass, success := bot.cli.CheckPermission(org, repo, commenter)
	bot.log.Infof("request success: %t, the %s has permission to the repo[%s/%s]: %t", success, commenter, org, repo, pass)

	if success && !pass {
		fn()
	}
	return pass && success
}

func (bot *robot) getPRLabelSet(org, repo, number string) sets.Set[string] {
	res := sets.New[string]()

	labels, ok := bot.cli.GetPullRequestLabels(org, repo, number)
	if !ok {
		return res
	}

	for _, v := range labels {
		res.Insert(v)
	}

	if res.Has("") {
		res.Delete("")
	}

	return res
}

func (bot *robot) genMergeMethod(org, repo, number string) string {
	mergeMethod := "merge"

	prLabels := bot.getPRLabelSet(org, repo, number)

	for p := range prLabels {
		if strings.HasPrefix(p, "merge/") {
			if strings.Split(p, "/")[1] == "squash" {
				return "squash"
			}

			return strings.Split(p, "/")[1]
		}
	}

	return mergeMethod
}

func (bot *robot) handleCheckPR(configmap *repoConfig, comment, commenter, org, repo, number string) error {
	if !regCheckPr.MatchString(comment) {
		return nil
	}
	if err := bot.handleMerge(configmap, org, repo, number); err != nil {
		claYesLabel := ""
		for _, labelForMerge := range configmap.LabelsForMerge {
			if strings.Contains(labelForMerge, "-cla/yes") {
				claYesLabel = labelForMerge
				break
			}
		}
		comment := fmt.Sprintf("@%s, this pr is not mergeable and the reasons are below:\n%s\n\n***lgtm***: "+
			"A label mandatory for merging a pull request. The repository collaborators can comment '/lgtm' to "+
			"add the label. The creator of a pull request can comment '/lgtm cancel' to remove the label, but "+
			"cannot run the '/lgtm' command to add the label.\n***approved***: A label mandatory for merging a "+
			"pull request. The repository collaborators can comment '/approve' to add the label and comment "+
			"'/approve cancel' to remove the label.\n***%s***:  A label mandatory for merging a pull request. "+
			"The author of each commit of a pull request must sign the Contributor License Agreement (CLA). "+
			"Otherwise, the pull request will fail to be merged. After signing the CLA, the author can comment "+
			"'/check-cla' to check the CLA status again.\n***wait_confirm***: A label for confirming pull request "+
			"merging. A pull request with this label cannot be automatically merged. This label is added because "+
			"members (including maintainers, committers, and repository administrators) are to be added to "+
			"sig-info.yaml in the pull request. To remove the label, all members to be added must comment "+
			"'/lgtm' in the pull request.",
			commenter, err.Error(), claYesLabel)
		bot.cli.CreatePRComment(org, repo, number, comment)
		return err
	}
	return nil
}