#15 Add method to transform an attribute to optional attribute

TheHive-Project · Jun 9, 2017 · b42e888 · b42e888
1 parent 0000324
commit b42e888
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/app/org/elastic4play/models/Attributes.scala b/app/org/elastic4play/models/Attributes.scala
@@ -316,7 +316,8 @@ object AttachmentAttributeFormat extends AttributeFormat[Attachment]("attachment
       }
   }
 
-  override def fromInputValue(subNames: Seq[String], value: InputValue): Attachment Or Every[AttributeError] = sys.error("not implemented")
+  override def fromInputValue(subNames: Seq[String], value: InputValue): Attachment Or Every[AttributeError] =
+    Bad(One(InvalidFormatAttributeError("", name, value)))
 
   override def elasticToJson(values: Seq[Any]): Option[JsValue] = values match {
     case Seq(m: util.Map[_, _]) ⇒
@@ -576,6 +577,15 @@ case class Attribute[T](
     description: String) {
   def defaultValueJson: Option[JsValue] = defaultValue.map(d ⇒ format.jsFormat.writes(d()))
 
+  def toOptional: Attribute[Option[T]] =
+    Attribute[Option[T]](
+      modelName,
+      name,
+      OptionalAttributeFormat(format),
+      options,
+      defaultValue.map(a ⇒ () ⇒ Some(a())),
+      description)
+
   lazy val isMulti: Boolean = format match {
     case _: MultiAttributeFormat[_] ⇒ true
     case _                          ⇒ false