Implement approve if build success

1 files changed, 11 insertions, 0 deletions

diff --git a/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketRepository.java b/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketRepository.java
index f027919..c084391 100644..100755
--- a/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketRepository.java
+++ b/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketRepository.java
@@ -71,6 +71,9 @@ public class BitbucketRepository {
     public void addFutureBuildTasks(Collection<BitbucketPullRequestResponseValue> pullRequests) {
         for(BitbucketPullRequestResponseValue pullRequest : pullRequests) {
             String commentId = postBuildStartCommentTo(pullRequest);
+            if ( this.trigger.getApproveIfSuccess() ) {
+                deletePullRequestApproval(pullRequest.getId());
+            }
             BitbucketCause cause = new BitbucketCause(
                     pullRequest.getSource().getBranch().getName(),
                     pullRequest.getDestination().getBranch().getName(),
@@ -101,6 +104,14 @@ public class BitbucketRepository {
         this.client.postPullRequestComment(pullRequestId, comment);
     }
 
+    public void deletePullRequestApproval(String pullRequestId) {
+        this.client.deletePullRequestApproval(pullRequestId);
+    }
+
+    public void postPullRequestApproval(String pullRequestId) {
+        this.client.postPullRequestApproval(pullRequestId);
+    }
+
     private boolean isBuildTarget(BitbucketPullRequestResponseValue pullRequest) {
     	
         boolean shouldBuild = true;