com.virgilsecurity.keyknox.utils.Serializer.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2015-2020, Virgil Security, Inc.
 *
 * Lead Maintainer: Virgil Security Inc. 
 *
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *     (1) Redistributions of source code must retain the above copyright notice, this
 *     list of conditions and the following disclaimer.
 *
 *     (2) Redistributions in binary form must reproduce the above copyright notice,
 *     this list of conditions and the following disclaimer in the documentation
 *     and/or other materials provided with the distribution.
 *
 *     (3) Neither the name of virgil nor the names of its
 *     contributors may be used to endorse or promote products derived from
 *     this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

package com.virgilsecurity.keyknox.utils

import com.google.gson.*
import com.virgilsecurity.keyknox.model.CloudEntries
import com.virgilsecurity.keyknox.model.CloudEntry
import java.lang.reflect.Type
import java.util.*

interface Serializer {

    companion object {

        val gson: Gson by lazy {
            val dateConverter = object : JsonSerializer, JsonDeserializer {

                override fun serialize(date: Date?, typeOfSrc: Type?, context: JsonSerializationContext?): JsonElement? {
                    return if (date == null) {
                        JsonNull.INSTANCE
                    } else {
                        JsonPrimitive(date.time)
                    }
                }

                override fun deserialize(json: JsonElement?, typeOfT: Type?, context: JsonDeserializationContext?): Date? {
                    return if (json != null) {
                        Date(json.asLong)
                    } else {
                        Date()
                    }
                }
            }

            val byteArrayConverter = object : JsonSerializer, JsonDeserializer {
                override fun serialize(array: ByteArray?, typeOfSrc: Type?, context: JsonSerializationContext?): JsonElement {
                    return if (array == null) {
                        JsonNull.INSTANCE
                    } else {
                        JsonPrimitive(base64Encode(array))
                    }
                }

                override fun deserialize(json: JsonElement?, typeOfT: Type?, context: JsonDeserializationContext?): ByteArray {
                    return if (json != null) {
                        base64Decode(json.asString)
                    } else {
                        byteArrayOf()
                    }
                }
            }

            val cloudEntriesConverter = object : JsonSerializer, JsonDeserializer {
                override fun serialize(src: CloudEntries?, typeOfSrc: Type?, context: JsonSerializationContext?): JsonElement {
                    val json = JsonObject()
                    src as Map<*, *>
                    src.entries.forEach { entry ->
                        json.add(entry.key, gson.toJsonTree(entry.value))
                    }
                    return json
                }

                override fun deserialize(json: JsonElement?, typeOfT: Type?, context: JsonDeserializationContext?): CloudEntries {
                    val map = mutableMapOf()
                    json?.asJsonObject?.entrySet()?.forEach { entry ->
                        val cloudEntry = gson.fromJson(entry.value, CloudEntry::class.java)
                        if (cloudEntry.meta == null) {
                            cloudEntry.meta = mutableMapOf()
                        }
                        map[entry.key] = cloudEntry
                    }
                    return CloudEntries(map)
                }
            }

            val gson = GsonBuilder().registerTypeAdapter(Date::class.java, dateConverter)
                    .registerTypeAdapter(ByteArray::class.java, byteArrayConverter)
                    .registerTypeAdapter(CloudEntries::class.java, cloudEntriesConverter).disableHtmlEscaping().create()
            gson
        }

    }
}

@Target(AnnotationTarget.FIELD)
annotation class Base64EncodedArray

@Target(AnnotationTarget.FIELD)
annotation class DateAsTimestamp