* This file is part of JuniperBot.
 * JuniperBot is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.

 * JuniperBot is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * GNU General Public License for more details.

 * You should have received a copy of the GNU General Public License
 * along with JuniperBot. If not, see <http://www.gnu.org/licenses/>.
package ru.juniperbot.common.persistence.types.json;

import org.hibernate.annotations.common.reflection.java.JavaXMember;
import org.hibernate.type.descriptor.WrapperOptions;
import org.hibernate.type.descriptor.java.AbstractTypeDescriptor;
import org.hibernate.type.descriptor.java.MutableMutabilityPlan;
import org.hibernate.usertype.DynamicParameterizedType;
import ru.juniperbot.common.utils.JacksonUtil;

import java.lang.reflect.Field;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.util.List;
import java.util.Map;
import java.util.Properties;

public class JsonTypeDescriptor
        extends AbstractTypeDescriptor<Object>
        implements DynamicParameterizedType {

    private static final long serialVersionUID = -4842954367890483417L;

    private Class<?> jsonObjectClass;

    private Type type;

    public void setParameterValues(Properties parameters) {
        jsonObjectClass = ((ParameterType) parameters.get(PARAMETER_TYPE)).getReturnedClass();
        try {
            Field typeField = JavaXMember.class.getDeclaredField("type");
            if (!typeField.isAccessible()) {
            type = (Type) typeField.get(parameters.get(XPROPERTY));
        } catch (IllegalAccessException | NoSuchFieldException e) {
            throw new RuntimeException(e);

    public JsonTypeDescriptor() {
        super(Object.class, new MutableMutabilityPlan<Object>() {
            private static final long serialVersionUID = 1606718143878984537L;

            protected Object deepCopyNotNull(Object value) {
                return JacksonUtil.clone(value);

    public boolean areEqual(Object one, Object another) {
        if (one == another) {
            return true;
        if (one == null || another == null) {
            return false;
        return JacksonUtil.toJsonNode(JacksonUtil.toString(one)).equals(

    public String toString(Object value) {
        return JacksonUtil.toString(value);

    public Object fromString(String string) {
        if (type instanceof ParameterizedType) {
            ParameterizedType pType = (ParameterizedType) type;
            if (List.class.isAssignableFrom((Class) pType.getRawType())) {
                return JacksonUtil.mapJsonToObjectList(string, (Class) pType.getActualTypeArguments()[0]);

            if (Map.class.isAssignableFrom((Class) pType.getRawType())) {
                return JacksonUtil.mapJsonToMap(string, (Class) pType.getActualTypeArguments()[0],
                        (Class) pType.getActualTypeArguments()[1]);
        return JacksonUtil.fromString(string, jsonObjectClass);

    public <X> X unwrap(Object value, Class<X> type, WrapperOptions options) {
        if (value == null) {
            return null;
        if (String.class.isAssignableFrom(type)) {
            return (X) toString(value);
        if (Object.class.isAssignableFrom(type)) {
            return (X) JacksonUtil.toJsonNode(toString(value));
        throw unknownUnwrap(type);

    public <X> Object wrap(X value, WrapperOptions options) {
        if (value == null) {
            return null;
        return fromString(value.toString());