#1670 Remove tag value in entity when a tag is deleted

TheHive-Project · Mar 6, 2021 · 0f61fb5 · 0f61fb5
1 parent 748004f
commit 0f61fb5
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/thehive/app/org/thp/thehive/services/TagSrv.scala b/thehive/app/org/thp/thehive/services/TagSrv.scala
@@ -92,6 +92,17 @@ class TagSrv @Inject() (
         .when(tag.colour != input.colour)(_.update(_.colour, input.colour))
         .getOrFail("Tag")
     } yield updatedTag
+
+  override def delete(tag: Tag with Entity)(implicit graph: Graph): Try[Unit] = {
+    val tagName = tag.toString
+    Try {
+      get(tag)
+        .sideEffect(
+          _.unionFlat(_.`case`.removeValue(_.tags, tagName), _.alert.removeValue(_.tags, tagName), _.observable.removeValue(_.tags, tagName))
+        )
+        .remove()
+    }
+  }
 }
 
 object TagOps {