Added API methods for creating and updating extractor labels

Author: bodom0015

app/api/Extractions.scala

@@ -674,4 +674,26 @@ class Extractions @Inject()(
     Ok(toJson("added new event"))
   }
 
+  def createExtractorsLabel() = ServerAdminAction(parse.json) { implicit request =>
+    val name = (request.body \ "name").as[String]
+    val category = (request.body \ "category").asOpt[String]
+    val assignedExtractors = (request.body \ "extractors").as[List[String]]
+
+    extractors.createExtractorsLabel(name, category, assignedExtractors) match {
+      case Some(lbl) => { Ok(Json.toJson(lbl)) }
+      case None => { BadRequest("Failed to create new label") }
+    }
+  }
+
+  def updateExtractorsLabel(id: UUID) = ServerAdminAction(parse.json) { implicit request =>
+    val name = (request.body \ "name").as[String]
+    val category = (request.body \ "category").asOpt[String]
+    val assignedExtractors = (request.body \ "extractors").as[List[String]]
+
+    val label = ExtractorsLabel(id, name, category, assignedExtractors)
+    extractors.updateExtractorsLabel(label) match {
+      case Some(lbl) => { Ok(Json.toJson(lbl)) }
+      case None => { BadRequest("Failed to update existing label") }
+    }
+  }
 }

app/models/Extraction.scala

@@ -185,3 +185,21 @@ case class ExtractionGroup(
                           latestMsg: String,
                           allMsgs: Map[UUID, List[Extraction]]
                           )
+
+case class ExtractorsLabel(
+                            id: UUID,
+                            name: String,
+                            category: Option[String],
+                            extractors: List[String]
+                         )
+
+object ExtractorsLabel {
+  implicit val extractorsLabelWrites = Json.writes[ExtractorsLabel]
+
+  /*implicit val extractorsLabelReads: Reads[ExtractorsLabel] = (
+    (JsPath \ "id").read[UUID] and
+      (JsPath \ "name").read[String] and
+      (JsPath \ "category").read[String] and
+      (JsPath \ "extractors").read[List[String]].orElse(Reads.pure(List.empty))
+    )(ExtractorsLabel.apply _)*/
+}

app/services/ExtractorService.scala

@@ -42,4 +42,10 @@ trait ExtractorService {
   def updateExtractorInfo(e: ExtractorInfo): Option[ExtractorInfo]
 
   def deleteExtractor(extractorName: String)
+
+  def listExtractorsLabels(): List[ExtractorsLabel]
+
+  def createExtractorsLabel(name: String, category: Option[String], extractors: List[String]): Option[ExtractorsLabel]
+
+  def updateExtractorsLabel(label: ExtractorsLabel): Option[ExtractorsLabel]
 }

app/services/mongodb/MongoDBExtractorService.scala

@@ -216,6 +216,28 @@ class MongoDBExtractorService extends ExtractorService {
 
     }
   }
+
+  def listExtractorsLabels(): List[ExtractorsLabel] = {
+    ExtractorsLabelDAO.findAll().toList
+  }
+
+  def createExtractorsLabel(name: String, category: Option[String], assignedExtractors: List[String]): Option[ExtractorsLabel] = {
+    val id = UUID.generate()
+    val label = ExtractorsLabel(id, name, category, assignedExtractors)
+    ExtractorsLabelDAO.save(label)
+    Some(label)
+  }
+
+  def updateExtractorsLabel(updated: ExtractorsLabel): Option[ExtractorsLabel] = {
+    val query = MongoDBObject("id" -> updated.id)
+    ExtractorsLabelDAO.findOne(query) match {
+      case Some(old) => {
+        ExtractorsLabelDAO.update(query, updated, false, false, WriteConcern.Safe)
+        Some(updated)
+      }
+      case None => None
+    }
+  }
 }
 
 object ExtractorServer extends ModelCompanion[ExtractorServer, ObjectId] {
@@ -259,6 +281,13 @@ object ExtractorInfoDAO extends ModelCompanion[ExtractorInfo, ObjectId] {
   }
 }
 
+object ExtractorsLabelDAO extends ModelCompanion[ExtractorsLabel, ObjectId] {
+  val dao = current.plugin[MongoSalatPlugin] match {
+    case None => throw new RuntimeException("No MongoSalatPlugin");
+    case Some(x) => new SalatDAO[ExtractorsLabel, ObjectId](collection = x.collection("extractors.labels")) {}
+  }
+}
+
 object ExtractorsForInstanceDAO extends ModelCompanion[ExtractorsForInstance, ObjectId] {
   val dao = current.plugin[MongoSalatPlugin] match {
     case None => throw new RuntimeException("No MongoSalatPlugin");