Merge pull request #11 from codacy/build-status-api

Add BuildStatusServices

Author: machadoit

src/main/scala/com/codacy/client/bitbucket/BuildStatus.scala

@@ -0,0 +1,9 @@
+package com.codacy.client.bitbucket
+
+import play.api.libs.json._
+
+case class BuildStatus(state: String, key: String, name: String, url: String, description: String)
+
+object BuildStatus {
+  implicit val fmt = Json.format[BuildStatus]
+}

src/main/scala/com/codacy/client/bitbucket/client/BitbucketClient.scala

@@ -1,12 +1,14 @@
 package com.codacy.client.bitbucket.client
 
+import java.util.concurrent.{SynchronousQueue, ThreadPoolExecutor, TimeUnit}
+
 import com.codacy.client.bitbucket.util.HTTPStatusCodes
 import com.ning.http.client.AsyncHttpClientConfig
-import java.util.concurrent.{SynchronousQueue, ThreadPoolExecutor, TimeUnit}
 import play.api.http.{ContentTypeOf, Writeable}
 import play.api.libs.json.{JsValue, Json, Reads}
 import play.api.libs.oauth._
 import play.api.libs.ws.ning.{NingAsyncHttpClientConfigBuilder, NingWSClient}
+
 import scala.concurrent.Await
 import scala.concurrent.duration.{Duration, SECONDS}
 import scala.util.{Failure, Properties, Success, Try}
@@ -50,14 +52,13 @@ class BitbucketClient(key: String, secretKey: String, token: String, secretToken
   }
 
   /*
-   * Does an API post
+   * Does an API request
    */
-  private def post[D,T](request: Request[T], values: D)(implicit reader: Reads[T], writer: Writeable[D], contentType: ContentTypeOf[D]): RequestResponse[T] = withClientRequest { client =>
-
+  private def performRequest[D, T](method: String, request: Request[T], values: D)(implicit reader: Reads[T], writer: Writeable[D], contentType: ContentTypeOf[D]): RequestResponse[T] = withClientRequest { client =>
     val jpromise = client.url(request.url)
       .sign(requestSigner)
       .withFollowRedirects(follow = true)
-      .post(values)
+      .withMethod(method).withBody(values).execute()
     val result = Await.result(jpromise, requestTimeout)
 
     val value = if (Seq(HTTPStatusCodes.OK, HTTPStatusCodes.CREATED).contains(result.status)) {
@@ -78,11 +79,19 @@ class BitbucketClient(key: String, secretKey: String, token: String, secretToken
   }
 
   def postForm[T](request: Request[T], values: Map[String, Seq[String]])(implicit reader: Reads[T]): RequestResponse[T] = {
-    post(request, values)
+    performRequest("POST", request, values)
   }
 
   def postJson[T](request: Request[T], values: JsValue)(implicit reader: Reads[T]): RequestResponse[T] = {
-    post(request, values)
+    performRequest("POST", request, values)
+  }
+
+  def putForm[T](request: Request[T], values: Map[String, Seq[String]])(implicit reader: Reads[T]): RequestResponse[T] = {
+    performRequest("PUT", request, values)
+  }
+
+  def putJson[T](request: Request[T], values: JsValue)(implicit reader: Reads[T]): RequestResponse[T] = {
+    performRequest("PUT", request, values)
   }
 
   /* copy paste from post ... */

src/main/scala/com/codacy/client/bitbucket/service/BuildStatusServices.scala

@@ -0,0 +1,50 @@
+package com.codacy.client.bitbucket.service
+
+import com.codacy.client.bitbucket.BuildStatus
+import com.codacy.client.bitbucket.client.{BitbucketClient, Request, RequestResponse}
+import play.api.libs.json.Json
+
+class BuildStatusServices(client: BitbucketClient) {
+
+  /*
+   * Gets a commit build status
+   *
+   */
+  def getBuildStatus(owner: String, repository: String, commit: String, key: String): RequestResponse[BuildStatus] = {
+    val url = s"https://bitbucket.org/!api/2.0/repositories/$owner/$repository/commit/$commit/statuses/build/$key"
+
+    client.execute(Request(url, classOf[BuildStatus]))
+  }
+
+
+  /*
+   * Create a build status for a commit
+   *
+   */
+  def createBuildStatus(owner: String, repository: String, commit: String, buildStatus: BuildStatus): RequestResponse[BuildStatus] = {
+    val url = s"https://bitbucket.org/!api/2.0/repositories/$owner/$repository/commit/$commit/statuses/build"
+
+    val values = Map("state" -> Seq(buildStatus.state), "key" -> Seq(buildStatus.key),
+      "name" -> Seq(buildStatus.name), "url" -> Seq(buildStatus.url),
+      "description" -> Seq(buildStatus.description))
+
+    client.postForm(Request(url, classOf[BuildStatus]), values)
+  }
+
+  /*
+ * Update a build status for a commit
+ *
+ */
+  def updateBuildStatus(owner: String, repository: String, commit: String, buildStatus: BuildStatus): RequestResponse[BuildStatus] = {
+    val url = s"https://bitbucket.org/!api/2.0/repositories/$owner/$repository/commit/$commit/statuses/build/${buildStatus.key}"
+
+    val payload = Json.obj(
+      "state" -> buildStatus.state,
+      "name" -> buildStatus.name, "url" -> buildStatus.url,
+      "description" -> buildStatus.description
+    )
+
+    client.putJson(Request(url, classOf[BuildStatus]), payload)
+  }
+}
+