add tag support

Author: shawjef3

src/main/scala/me/jeffshaw/digitalocean/Droplet.scala

@@ -164,7 +164,7 @@ case class Droplet(
   }
 
   def refresh()(implicit client: DigitalOceanClient, ec: ExecutionContext): Future[Droplet] = {
-   Droplet(id)
+    Droplet(id)
   }
 
   override def equals(obj: scala.Any): Boolean = {

src/main/scala/me/jeffshaw/digitalocean/Tag.scala

@@ -0,0 +1,150 @@
+package me.jeffshaw.digitalocean
+
+import org.json4s.JsonDSL.WithBigDecimal._
+import scala.concurrent.{ExecutionContext, Future}
+
+case class Tag(
+  name: String,
+  resources: Tag.Resources
+) {
+  def delete()(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    Tag.delete(name)
+  }
+
+  def tagDropletIds(
+    droplets: Seq[BigInt]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    Tag.tagDropletIds(name, droplets)
+  }
+
+  def tagDroplets(
+    droplets: Seq[Droplet]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    Tag.tagDroplets(name, droplets)
+  }
+
+  def untagDropletIds(
+    droplets: Seq[BigInt]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    Tag.untagDropletIds(name, droplets)
+  }
+
+  def untagDroplets(
+    droplets: Seq[Droplet]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    Tag.untagDroplets(name, droplets)
+  }
+}
+
+object Tag extends Path {
+
+  override protected val path: Seq[String] = Seq("tags")
+
+  def apply(
+    name: String
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Tag] = {
+    client.get[responses.Tag](path :+ name).map(_.tag)
+  }
+
+  def list()(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Seq[Tag]] = {
+    client.get[responses.Tags](path).map(_.tags)
+  }
+
+  def create(
+    name: String
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Tag] = {
+    client.post[responses.Tag](path, "name" -> name).map(_.tag)
+  }
+
+  def delete(
+    name: String
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    client.delete(path :+ name)
+  }
+
+  def tagDroplets(
+    name: String,
+    droplets: Seq[Droplet]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    tagDropletIds(name, droplets.map(_.id))
+  }
+
+  def tagDropletIds(
+    name: String,
+    dropletIds: Seq[BigInt]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    if (dropletIds.isEmpty) {
+      Future.successful(())
+    } else {
+      val postBody =
+        "resources" -> (
+          for {
+            id <- dropletIds
+          } yield ("resource_id" -> id) ~ ("resource_type" -> "droplet")
+        )
+      client.postWithEmptyResponse(path :+ name, postBody)
+    }
+  }
+
+  def untagDroplets(
+    name: String,
+    droplets: Seq[Droplet]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    untagDropletIds(name, droplets.map(_.id))
+  }
+
+  def untagDropletIds(
+    name: String,
+    dropletIds: Seq[BigInt]
+  )(implicit client: DigitalOceanClient,
+    ec: ExecutionContext
+  ): Future[Unit] = {
+    if (dropletIds.isEmpty) {
+      Future.successful(())
+    } else {
+      val postBody =
+        "resources" -> (
+          for {
+            id <- dropletIds
+          } yield ("resource_id" -> id) ~ ("resource_type" -> "droplet")
+        )
+      client.delete(path :+ name, maybeMessage = Some(postBody))
+    }
+  }
+
+  case class Resources(
+    droplets: Resources.Droplets
+  )
+
+  object Resources {
+    case class Droplets(
+      count: BigInt,
+      lastTagged: Option[Droplet]
+    )
+  }
+
+}

src/main/scala/me/jeffshaw/digitalocean/responses/package.scala

@@ -376,4 +376,12 @@ package object responses {
     firewall: Firewall
   )
 
+  private[digitalocean] case class Tag(
+    tag: digitalocean.Tag
+  )
+
+  private [digitalocean] case class Tags(
+    tags: Seq[digitalocean.Tag]
+  )
+
 }

src/test/scala/me/jeffshaw/digitalocean/FirewallSpec.scala

@@ -7,8 +7,10 @@ import scala.concurrent.duration._
 
 class FirewallSpec extends Suite {
 
+  val randomSuffix = Random.nextInt().toString
+
   test("create, update") {
-    val name = firewallNamePrefix + Random.nextInt()
+    val name = firewallNamePrefix + randomSuffix
     val inboundRule = Firewall.InboundRule(
       protocol = Firewall.Protocol.Tcp(Firewall.Port.Range(1, 2)),
       sources = Firewall.Source(addresses = Seq(Firewall.Source.Address(InetAddress.getByName("1.1.1.1"), cidr = Some(8))))
@@ -22,15 +24,16 @@ class FirewallSpec extends Suite {
     assert(firewall.tags.isEmpty)
     assert(firewall.dropletIds.isEmpty)
 
-
     val outboundRule =
       Firewall.OutboundRule(
         protocol = Firewall.Protocol.Icmp,
         destinations = Firewall.Destination(
-          tags = Seq("hi")
+          tags = Seq(name)
         )
       )
 
+    val tag = Await.result(Tag.create(name), 10 seconds)
+
     val updated =
       Await.result(
         firewall.update(
@@ -42,6 +45,7 @@ class FirewallSpec extends Suite {
 
     assert(updated.inboundRules.isEmpty)
     assertResult(Seq(outboundRule))(updated.outboundRules)
+    Await.result(tag.delete(), 10 seconds)
   }
 
   override protected def afterAll(): Unit = {