cli/pkg/cmd/issue/transfer/transfer.go
2021-03-30 18:42:34 +02:00

114 lines
2.7 KiB
Go

package transfer
import (
"context"
"fmt"
"net/http"
"github.com/cli/cli/api"
"github.com/cli/cli/internal/config"
"github.com/cli/cli/internal/ghinstance"
"github.com/cli/cli/internal/ghrepo"
"github.com/cli/cli/pkg/cmd/issue/shared"
"github.com/cli/cli/pkg/cmdutil"
"github.com/cli/cli/pkg/iostreams"
"github.com/shurcooL/githubv4"
"github.com/shurcooL/graphql"
"github.com/spf13/cobra"
)
type TransferOptions struct {
HttpClient func() (*http.Client, error)
Config func() (config.Config, error)
IO *iostreams.IOStreams
BaseRepo func() (ghrepo.Interface, error)
IssueSelector string
DestRepoSelector string
}
func NewCmdTransfer(f *cmdutil.Factory, runF func(*TransferOptions) error) *cobra.Command {
opts := TransferOptions{
IO: f.IOStreams,
HttpClient: f.HttpClient,
Config: f.Config,
}
cmd := &cobra.Command{
Use: "transfer {<number> | <url>} <destination-repo>",
Short: "Transfer issue to another repository",
Args: cmdutil.ExactArgs(2, "issue and destination repository are required"),
RunE: func(cmd *cobra.Command, args []string) error {
opts.BaseRepo = f.BaseRepo
opts.IssueSelector = args[0]
opts.DestRepoSelector = args[1]
if runF != nil {
return runF(&opts)
}
return transferRun(&opts)
},
}
return cmd
}
func transferRun(opts *TransferOptions) error {
httpClient, err := opts.HttpClient()
if err != nil {
return err
}
apiClient := api.NewClientFromHTTP(httpClient)
issue, _, err := shared.IssueFromArg(apiClient, opts.BaseRepo, opts.IssueSelector)
if err != nil {
return err
}
destRepo, err := ghrepo.FromFullName(opts.DestRepoSelector)
if err != nil {
return err
}
url, err := issueTransfer(httpClient, issue.ID, destRepo)
if err != nil {
return err
}
_, err = fmt.Fprintln(opts.IO.Out, url)
return err
}
func issueTransfer(httpClient *http.Client, issueID string, destRepo ghrepo.Interface) (string, error) {
var destinationRepoID string
if r, ok := destRepo.(*api.Repository); ok {
destinationRepoID = r.ID
} else {
apiClient := api.NewClientFromHTTP(httpClient)
r, err := api.GitHubRepo(apiClient, destRepo)
if err != nil {
return "", err
}
destinationRepoID = r.ID
}
var mutation struct {
TransferIssue struct {
Issue struct {
URL string
}
} `graphql:"transferIssue(input: $input)"`
}
variables := map[string]interface{}{
"input": githubv4.TransferIssueInput{
IssueID: issueID,
RepositoryID: destinationRepoID,
},
}
gql := graphql.NewClient(ghinstance.GraphQLEndpoint(destRepo.RepoHost()), httpClient)
err := gql.MutateNamed(context.Background(), "IssueTransfer", &mutation, variables)
return mutation.TransferIssue.Issue.URL, err
}