com.regnosys.rosetta.common.serialisation.mixin.RosettaEnumBuilderIntrospector Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rosetta-common Show documentation

Rune Common is a java library that is utilised by Rosetta Code Generators and models expressed in the Rosetta DSL.

There is a newer version: 11.27.1

package com.regnosys.rosetta.common.serialisation.mixin;

/*-
 * ==============
 * Rune Common
 * ==============
 * Copyright (C) 2018 - 2024 REGnosys
 * ==============
 * Licensed 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.
 * ==============
 */

import com.fasterxml.jackson.databind.introspect.AnnotatedClass;
import com.fasterxml.jackson.databind.introspect.AnnotatedField;
import com.rosetta.model.lib.annotations.RosettaEnum;
import com.rosetta.model.lib.annotations.RosettaEnumValue;

import java.lang.reflect.Field;
import java.util.function.BiFunction;
import java.util.stream.Stream;

public class RosettaEnumBuilderIntrospector {

    private final EnumNameFunc enumNameFunc;
    private final EnumAliasFunc enumAliasFunc;

    public RosettaEnumBuilderIntrospector(boolean supportRosettaEnumValue) {
        if (supportRosettaEnumValue) {
            this.enumNameFunc = (annotation, javaEnumName) -> !annotation.displayName().isEmpty() ? annotation.displayName() : annotation.value();
        } else {
            this.enumNameFunc = (annotation, javaEnumName) -> !annotation.displayName().isEmpty() ? annotation.displayName() : javaEnumName;
        }
        this.enumAliasFunc = (annotation, javaEnumName) -> !annotation.displayName().isEmpty() ?
                new String[]{javaEnumName, annotation.displayName(), annotation.value()}:
                new String[]{javaEnumName,  annotation.value()};
    }

    public boolean isApplicable(AnnotatedClass enumType) {
        return enumType.getAnnotation(RosettaEnum.class) != null;
    }

    public void findEnumValues(AnnotatedClass enumType, Enum[] enumValues, String[] names) {
        for (AnnotatedField f : enumType.fields()) {
            if (f.hasAnnotation(RosettaEnumValue.class)) {
                RosettaEnumValue annotation = f.getAnnotation(RosettaEnumValue.class);
                final String name = f.getName();
                for (int i = 0, end = enumValues.length; i < end; ++i) {
                    if (name.equals(enumValues[i].name())) {
                        names[i] = enumNameFunc.apply(annotation, name);
                        break;
                    }
                }
            }
        }
    }

    public void findEnumAliases(AnnotatedClass enumType, Enum[] enumValues, String[][] aliasList) {
        for (AnnotatedField f : enumType.fields()) {
            if (f.hasAnnotation(RosettaEnumValue.class)) {
                RosettaEnumValue annotation = f.getAnnotation(RosettaEnumValue.class);
                final String name = f.getName();
                for (int i = 0, end = enumValues.length; i < end; ++i) {
                    if (name.equals(enumValues[i].name())) {
                        aliasList[i] = enumAliasFunc.apply(annotation, name);
                        break;
                    }
                }
            }
        }
    }


    interface EnumNameFunc extends BiFunction {

    }

    interface EnumAliasFunc extends BiFunction {

    }
}