JsonNode, ArrayNode and ObjectNode extensions

Author: sokomishalov

src/main/kotlin/com/fasterxml/jackson/module/kotlin/Extensions.kt

@@ -3,17 +3,16 @@ package com.fasterxml.jackson.module.kotlin
 import com.fasterxml.jackson.core.JsonParser
 import com.fasterxml.jackson.core.TreeNode
 import com.fasterxml.jackson.core.type.TypeReference
-import com.fasterxml.jackson.databind.JsonMappingException
-import com.fasterxml.jackson.databind.MappingIterator
-import com.fasterxml.jackson.databind.ObjectMapper
-import com.fasterxml.jackson.databind.ObjectReader
-import com.fasterxml.jackson.databind.module.SimpleModule
-import com.fasterxml.jackson.databind.JsonDeserializer
-import com.fasterxml.jackson.databind.JsonSerializer
+import com.fasterxml.jackson.databind.*
 import com.fasterxml.jackson.databind.json.JsonMapper
+import com.fasterxml.jackson.databind.module.SimpleModule
+import com.fasterxml.jackson.databind.node.ArrayNode
+import com.fasterxml.jackson.databind.node.ObjectNode
 import java.io.File
 import java.io.InputStream
 import java.io.Reader
+import java.math.BigDecimal
+import java.math.BigInteger
 import java.net.URL
 import java.util.*
 import kotlin.reflect.KClass
@@ -54,6 +53,41 @@ inline fun <reified T> ObjectReader.readValueTyped(jp: JsonParser): T = readValu
 inline fun <reified T> ObjectReader.readValuesTyped(jp: JsonParser): Iterator<T> = readValues(jp, jacksonTypeRef<T>())
 inline fun <reified T> ObjectReader.treeToValue(n: TreeNode): T? = treeToValue(n, T::class.java)
 
+operator fun ArrayNode.plus(element: Boolean) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: Short) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: Int) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: Long) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: Float) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: Double) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: BigDecimal) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: BigInteger) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: String) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: ByteArray) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(element: JsonNode) = Unit.apply { add(element) }
+operator fun ArrayNode.plus(elements: ArrayNode) = Unit.apply { addAll(elements) }
+operator fun ArrayNode.plusAssign(element: Boolean) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: Short) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: Int) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: Long) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: Float) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: Double) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: BigDecimal) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: BigInteger) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: String) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: ByteArray) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(element: JsonNode) = Unit.apply { add(element) }
+operator fun ArrayNode.plusAssign(elements: ArrayNode) = Unit.apply { addAll(elements) }
+operator fun ArrayNode.minus(index: Int) = Unit.apply { remove(index) }
+operator fun ArrayNode.minusAssign(index: Int) = Unit.apply { remove(index) }
+
+operator fun ObjectNode.minus(field: String) = Unit.apply { remove(field) }
+operator fun ObjectNode.minus(fields: Collection<String>) = Unit.apply { remove(fields) }
+operator fun ObjectNode.minusAssign(field: String) = Unit.apply { remove(field) }
+operator fun ObjectNode.minusAssign(fields: Collection<String>) = Unit.apply { remove(fields) }
+
+operator fun JsonNode.contains(field: String) = has(field)
+operator fun JsonNode.contains(index: Int) = has(index)
+
 internal fun JsonMappingException.wrapWithPath(refFrom: Any?, refFieldName: String) = JsonMappingException.wrapWithPath(this, refFrom, refFieldName)
 internal fun JsonMappingException.wrapWithPath(refFrom: Any?, index: Int) = JsonMappingException.wrapWithPath(this, refFrom, index)