Delete the first comment when build is finished for a cleaner appearance.

Implements #8

1 files changed, 2 insertions, 5 deletions

diff --git a/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketBuilds.java b/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketBuilds.java
index 8052cc6..9600e26 100644
--- a/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketBuilds.java
+++ b/src/main/java/bitbucketpullrequestbuilder/bitbucketpullrequestbuilder/BitbucketBuilds.java
@@ -56,10 +56,7 @@ public class BitbucketBuilds {
         else {
             buildUrl = rootUrl + build.getUrl();
         }
-        if(result == Result.SUCCESS) {
-            repository.postFinishedComment(cause.getPullRequestId(), cause.getCommitHash(), true, buildUrl);
-        } else if (result == Result.FAILURE || result == Result.UNSTABLE) {
-            repository.postFinishedComment(cause.getPullRequestId(), cause.getCommitHash(), false, buildUrl);
-        }
+        repository.deletePullRequestComment(cause.getPullRequestId(), cause.getBuildStartCommentId());
+        repository.postFinishedComment(cause.getPullRequestId(), cause.getCommitHash(), result == Result.SUCCESS, buildUrl);
     }
 }