com.ebiznext.comet.schema.model.PrivacyLevel.scala Maven / Gradle / Ivy
/*
*
* * Licensed to the Apache Software Foundation (ASF) under one or more
* * contributor license agreements. See the NOTICE file distributed with
* * this work for additional information regarding copyright ownership.
* * The ASF licenses this file to You under the Apache License, Version 2.0
* * (the "License"); you may not use this file except in compliance with
* * the License. You may obtain a copy of the License at
* *
* * http://www.apache.org/licenses/LICENSE-2.0
* *
* * Unless required by applicable law or agreed to in writing, software
* * distributed under the License is distributed on an "AS IS" BASIS,
* * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* * See the License for the specific language governing permissions and
* * limitations under the License.
*
*
*/
package com.ebiznext.comet.schema.model
import java.util.Locale
import com.ebiznext.comet.config.Settings
import com.ebiznext.comet.privacy.PrivacyEngine
import com.fasterxml.jackson.core.JsonParser
import com.fasterxml.jackson.databind.annotation.{JsonDeserialize, JsonSerialize}
import com.fasterxml.jackson.databind.ser.std.ToStringSerializer
import com.fasterxml.jackson.databind.{DeserializationContext, JsonDeserializer}
import scala.collection.JavaConverters._
import scala.reflect.runtime.universe
/**
* How (the attribute should be transformed at ingestion time ?
*
* @param value algorithm to use : NONE, HIDE, MD5, SHA1, SHA256, SHA512, AES
*/
@JsonSerialize(using = classOf[ToStringSerializer])
@JsonDeserialize(using = classOf[PrivacyLevelDeserializer])
sealed case class PrivacyLevel(value: String) {
override def toString: String = value
def crypt(s: String, colMap: Map[String, String])(implicit settings: Settings): String = {
val (privacyAlgo, privacyParams) = PrivacyLevel.ForSettings(settings).all(value)._1
privacyAlgo.crypt(s, colMap, privacyParams)
}
}
object PrivacyLevel {
case class ForSettings(settings: Settings) {
private def make(schemeName: String, encryptionAlgo: String): (PrivacyEngine, List[Any]) = {
val (privacyObject, typedParams) = PrivacyEngine.parse(encryptionAlgo)
val runtimeMirror = universe.runtimeMirror(getClass.getClassLoader)
val module = runtimeMirror.staticModule(privacyObject)
val obj: universe.ModuleMirror = runtimeMirror.reflectModule(module)
val encryption = obj.instance.asInstanceOf[PrivacyEngine]
(encryption, typedParams)
}
def get(schemeName: String): (PrivacyEngine, List[Any]) = {
val encryptionObject = settings.comet.privacy.options.asScala(schemeName)
make(schemeName, encryptionObject)
}
lazy val all = settings.comet.privacy.options.asScala.map {
case (k, objName) =>
val encryption = make(k, objName)
val key = k.toUpperCase(Locale.ROOT)
(key, (encryption, new PrivacyLevel(key)))
}
// Improve Scan performance
def fromString(value: String): PrivacyLevel = all(value.toUpperCase())._2
}
val None: PrivacyLevel = PrivacyLevel("NONE")
}
class PrivacyLevelDeserializer extends JsonDeserializer[PrivacyLevel] {
override def deserialize(jp: JsonParser, ctx: DeserializationContext): PrivacyLevel = {
val value = jp.readValueAs[String](classOf[String])
PrivacyLevel(
value.toUpperCase(Locale.ROOT)
) /* TODO: prior to 2020-02-25 we enforced a valid PrivacyLevel at deser time */
}
}