Return WebURL of newly created pull request

Signed-off-by: Yiannis <[email protected]>

Author: yiannistri

github/client_repository_pullrequest.go

@@ -18,6 +18,7 @@ package github
 
 import (
 	"context"
+
 	"github.com/fluxcd/go-git-providers/gitprovider"
 	"github.com/google/go-github/v32/github"
 )
@@ -32,7 +33,7 @@ type PullRequestClient struct {
 }
 
 // Create creates a pull request with the given specifications.
-func (c *PullRequestClient) Create(ctx context.Context, title, branch, baseBranch, description string) error {
+func (c *PullRequestClient) Create(ctx context.Context, title, branch, baseBranch, description string) (gitprovider.PullRequest, error) {
 
 	prOpts := &github.NewPullRequest{
 		Title: &title,
@@ -41,9 +42,10 @@ func (c *PullRequestClient) Create(ctx context.Context, title, branch, baseBranc
 		Body:  &description,
 	}
 
-	if _, _, err := c.c.Client().PullRequests.Create(ctx, c.ref.GetIdentity(), c.ref.GetRepository(), prOpts); err != nil {
-		return err
+	pr, _, err := c.c.Client().PullRequests.Create(ctx, c.ref.GetIdentity(), c.ref.GetRepository(), prOpts)
+	if err != nil {
+		return nil, err
 	}
 
-	return nil
+	return newPullRequest(c.clientContext, pr), nil
 }

github/integration_test.go

@@ -408,9 +408,9 @@ var _ = Describe("GitHub Provider", func() {
 		_, err = userRepo.Commits().Create(ctx, branchName, "added config file", files)
 		Expect(err).ToNot(HaveOccurred())
 
-		err = userRepo.PullRequests().Create(ctx, "Added config file", branchName, *defaultBranch, "added config file")
+		pr, err := userRepo.PullRequests().Create(ctx, "Added config file", branchName, *defaultBranch, "added config file")
 		Expect(err).ToNot(HaveOccurred())
-
+		Expect(pr.Get().WebURL).ToNot(BeEmpty())
 	})
 
 	AfterSuite(func() {

github/resource_pullrequest.go

@@ -0,0 +1,51 @@
+/*
+Copyright 2021 The Flux CD contributors.
+
+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 github
+
+import (
+	"github.com/fluxcd/go-git-providers/gitprovider"
+	"github.com/google/go-github/v32/github"
+)
+
+func newPullRequest(ctx *clientContext, apiObj *github.PullRequest) *pullrequest {
+	return &pullrequest{
+		clientContext: ctx,
+		pr:            *apiObj,
+	}
+}
+
+var _ gitprovider.PullRequest = &pullrequest{}
+
+type pullrequest struct {
+	*clientContext
+
+	pr github.PullRequest
+}
+
+func (pr *pullrequest) Get() gitprovider.PullRequestInfo {
+	return pullrequestFromAPI(&pr.pr)
+}
+
+func (pr *pullrequest) APIObject() interface{} {
+	return &pr.pr
+}
+
+func pullrequestFromAPI(apiObj *github.PullRequest) gitprovider.PullRequestInfo {
+	return gitprovider.PullRequestInfo{
+		WebURL: apiObj.GetHTMLURL(),
+	}
+}

gitlab/client_repository_pullrequest.go

@@ -18,6 +18,7 @@ package gitlab
 
 import (
 	"context"
+
 	"github.com/fluxcd/go-git-providers/gitprovider"
 	"github.com/xanzy/go-gitlab"
 )
@@ -32,7 +33,7 @@ type PullRequestClient struct {
 }
 
 // Create creates a pull request with the given specifications.
-func (c *PullRequestClient) Create(ctx context.Context, title, branch, baseBranch, description string) error {
+func (c *PullRequestClient) Create(ctx context.Context, title, branch, baseBranch, description string) (gitprovider.PullRequest, error) {
 
 	prOpts := &gitlab.CreateMergeRequestOptions{
 		Title:        &title,
@@ -41,9 +42,10 @@ func (c *PullRequestClient) Create(ctx context.Context, title, branch, baseBranc
 		Description:  &description,
 	}
 
-	if _, _, err := c.c.Client().MergeRequests.CreateMergeRequest(getRepoPath(c.ref), prOpts); err != nil {
-		return err
+	mr, _, err := c.c.Client().MergeRequests.CreateMergeRequest(getRepoPath(c.ref), prOpts)
+	if err != nil {
+		return nil, err
 	}
 
-	return nil
+	return newPullRequest(c.clientContext, mr), nil
 }

gitlab/integration_test.go

@@ -729,9 +729,9 @@ var _ = Describe("GitLab Provider", func() {
 		_, err = userRepo.Commits().Create(ctx, branchName, "added config file", files)
 		Expect(err).ToNot(HaveOccurred())
 
-		err = userRepo.PullRequests().Create(ctx, "Added config file", branchName, defaultBranch, "added config file")
+		pr, err := userRepo.PullRequests().Create(ctx, "Added config file", branchName, defaultBranch, "added config file")
 		Expect(err).ToNot(HaveOccurred())
-
+		Expect(pr.Get().WebURL).ToNot(BeEmpty())
 	})
 
 	AfterSuite(func() {

gitlab/resource_pullrequest.go

@@ -0,0 +1,51 @@
+/*
+Copyright 2021 The Flux CD contributors.
+
+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 gitlab
+
+import (
+	"github.com/fluxcd/go-git-providers/gitprovider"
+	"github.com/xanzy/go-gitlab"
+)
+
+func newPullRequest(ctx *clientContext, apiObj *gitlab.MergeRequest) *pullrequest {
+	return &pullrequest{
+		clientContext: ctx,
+		pr:            *apiObj,
+	}
+}
+
+var _ gitprovider.PullRequest = &pullrequest{}
+
+type pullrequest struct {
+	*clientContext
+
+	pr gitlab.MergeRequest
+}
+
+func (pr *pullrequest) Get() gitprovider.PullRequestInfo {
+	return pullrequestFromAPI(&pr.pr)
+}
+
+func (pr *pullrequest) APIObject() interface{} {
+	return &pr.pr
+}
+
+func pullrequestFromAPI(apiObj *gitlab.MergeRequest) gitprovider.PullRequestInfo {
+	return gitprovider.PullRequestInfo{
+		WebURL: apiObj.WebURL,
+	}
+}

gitprovider/client.go

@@ -228,5 +228,5 @@ type BranchClient interface {
 // This client can be accessed through Repository.PullRequests().
 type PullRequestClient interface {
 	// Create creates a pull request with the given specifications.
-	Create(ctx context.Context, title, branch, baseBranch, description string) error
+	Create(ctx context.Context, title, branch, baseBranch, description string) (PullRequest, error)
 }

gitprovider/resources.go

@@ -138,3 +138,12 @@ type Commit interface {
 	// Get returns high-level information about this commit.
 	Get() CommitInfo
 }
+
+type PullRequest interface {
+	// Object implements the Object interface,
+	// allowing access to the underlying object returned from the API.
+	Object
+
+	// Get returns high-level information about this pull request.
+	Get() PullRequestInfo
+}

gitprovider/types_repository.go

@@ -199,3 +199,9 @@ type CommitFile struct {
 	// +required
 	Content *string `json:"content"`
 }
+
+type PullRequestInfo struct {
+	// WebURL is the URL of the pull request in the git provider web interface.
+	// +required
+	WebURL string `json:"web_url"`
+}