github.com/zppinho/prow@v0.0.0-20240510014325-1738badeb017/pkg/plugins/transfer-issue/transfer-issue.go

/*
Copyright 2021 The Kubernetes Authors.

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 transferissue implements the `/transfer-issue` command which allows members of the org
// to transfer issues between repos
package transferissue

import (
	"context"
	"fmt"
	"regexp"
	"strings"

	githubql "github.com/shurcooL/githubv4"
	"github.com/sirupsen/logrus"

	"sigs.k8s.io/prow/pkg/config"
	"sigs.k8s.io/prow/pkg/github"
	"sigs.k8s.io/prow/pkg/pluginhelp"
	"sigs.k8s.io/prow/pkg/plugins"
)

const pluginName = "transfer-issue"

var (
	transferRe = regexp.MustCompile(`(?mi)^/transfer(?:-issue)?(?: +(.*))?$`)
)

type githubClient interface {
	GetRepo(org, name string) (github.FullRepo, error)
	CreateComment(org, repo string, number int, comment string) error
	IsMember(org, user string) (bool, error)
	MutateWithGitHubAppsSupport(context.Context, interface{}, githubql.Input, map[string]interface{}, string) error
}

func init() {
	plugins.RegisterGenericCommentHandler(pluginName, handleGenericComment, helpProvider)
}

func helpProvider(_ *plugins.Configuration, _ []config.OrgRepo) (*pluginhelp.PluginHelp, error) {
	pluginHelp := &pluginhelp.PluginHelp{
		Description: "The transfer-issue plugin transfers a GitHub issue from one repo to another in the same organization.",
	}
	pluginHelp.AddCommand(pluginhelp.Command{
		Usage:       "/transfer[-issue] <destination repo in same org>",
		Description: "Transfers an issue to a different repo in the same org.",
		Featured:    true,
		WhoCanUse:   "Org members.",
		Examples:    []string{"/transfer-issue kubectl", "/transfer test-infra"},
	})
	return pluginHelp, nil
}

func handleGenericComment(pc plugins.Agent, e github.GenericCommentEvent) error {
	return handleTransfer(pc.GitHubClient, pc.Logger, e)
}

func handleTransfer(gc githubClient, log *logrus.Entry, e github.GenericCommentEvent) error {
	org := e.Repo.Owner.Login
	srcRepoName := e.Repo.Name
	srcRepoPair := org + "/" + srcRepoName
	user := e.User.Login

	if e.IsPR || e.Action != github.GenericCommentActionCreated {
		return nil
	}
	matches := transferRe.FindAllStringSubmatch(e.Body, -1)
	if len(matches) == 0 {
		return nil
	}
	if len(matches) != 1 || len(matches[0]) != 2 || len(matches[0][1]) == 0 {
		return gc.CreateComment(
			org, srcRepoName, e.Number,
			plugins.FormatResponseRaw(e.Body, e.HTMLURL, user, "/transfer-issue must only be used once and with a single destination repo."),
		)
	}

	dstRepoName := strings.TrimSpace(matches[0][1])
	dstRepoPair := org + "/" + dstRepoName

	dstRepo, err := gc.GetRepo(org, dstRepoName)
	if err != nil {
		log.WithError(err).WithField("dstRepo", dstRepoPair).Warning("could not fetch destination repo")
		// TODO: Might want to add another GetRepo type call that checks if a repo exists vs a bad request
		return gc.CreateComment(
			org, srcRepoName, e.Number,
			plugins.FormatResponseRaw(e.Body, e.HTMLURL, user, fmt.Sprintf("Something went wrong or the destination repo %s does not exist.", dstRepoPair)),
		)
	}

	isMember, err := gc.IsMember(org, user)
	if err != nil {
		return fmt.Errorf("unable to fetch if %s is an org member of %s: %w", user, org, err)
	}
	if !isMember {
		return gc.CreateComment(
			org, srcRepoName, e.Number,
			plugins.FormatResponseRaw(e.Body, e.HTMLURL, user, "You must be an org member to transfer this issue."),
		)
	}

	m, err := transferIssue(gc, org, dstRepo.NodeID, e.NodeID)
	if err != nil {
		log.WithError(err).WithFields(logrus.Fields{
			"issueNumber": e.Number,
			"srcRepo":     srcRepoPair,
			"dstRepo":     dstRepoPair,
		}).Error("issue could not be transferred")
		return err
	}
	log.WithFields(logrus.Fields{
		"user":        user,
		"org":         org,
		"srcRepo":     srcRepoName,
		"issueNumber": e.Number,
		"dstURL":      m.TransferIssue.Issue.URL,
	}).Infof("successfully transferred issue")
	return nil
}

// TransferIssueMutation is a GraphQL mutation struct compatible with shurcooL/githubql's client
//
// See https://docs.github.com/en/graphql/reference/input-objects#transferissueinput
type transferIssueMutation struct {
	TransferIssue struct {
		Issue struct {
			URL githubql.URI
		}
	} `graphql:"transferIssue(input: $input)"`
}

// TransferIssue will move an issue from one repo to another in the same org.
//
// See https://docs.github.com/en/graphql/reference/mutations#transferissue
//
// In the future we may want to interact with the TransferredEvent on the issue IssueTimeline
// See https://docs.github.com/en/graphql/reference/objects#transferredevent
// https://docs.github.com/en/graphql/reference/unions#issuetimelineitem
func transferIssue(gc githubClient, org, dstRepoNodeID string, issueNodeID string) (*transferIssueMutation, error) {
	m := &transferIssueMutation{}
	input := githubql.TransferIssueInput{
		IssueID:      githubql.ID(issueNodeID),
		RepositoryID: githubql.ID(dstRepoNodeID),
	}
	err := gc.MutateWithGitHubAppsSupport(context.Background(), m, input, nil, org)
	return m, err
}