Search in sources :

Example 1 with PropertyModel

use of cz.habarta.typescript.generator.parser.PropertyModel in project openremote by openremote.

the class JsonSerializeExtension method getTransformers.

@Override
public List<cz.habarta.typescript.generator.Extension.TransformerDefinition> getTransformers() {
    return Arrays.asList(new cz.habarta.typescript.generator.Extension.TransformerDefinition(ModelCompiler.TransformationPhase.BeforeTsModel, new ModelTransformer() {

        @Override
        public TsModel transformModel(SymbolTable symbolTable, TsModel model) {
            return model;
        }

        @Override
        public Model transformModel(SymbolTable symbolTable, Model model) {
            // Look for @JsonSerialize annotation and modify the property type accordingly
            List<BeanModel> beans = model.getBeans();
            beans.replaceAll(bean -> {
                AtomicBoolean modified = new AtomicBoolean(false);
                List<PropertyModel> properties = bean.getProperties().stream().map(p -> {
                    Member member = p.getOriginalMember();
                    JsonSerialize jsonSerialize = null;
                    if (member instanceof Field) {
                        Field field = (Field) member;
                        jsonSerialize = field.getAnnotation(JsonSerialize.class);
                    } else if (member instanceof Method) {
                        Method method = (Method) member;
                        jsonSerialize = method.getAnnotation(JsonSerialize.class);
                    }
                    if (jsonSerialize != null) {
                        // TODO: Add support for other options
                        if (jsonSerialize.as() != Void.class) {
                            modified.set(true);
                            return new PropertyModel(p.getName(), jsonSerialize.as(), p.isOptional(), p.getAccess(), p.getOriginalMember(), p.getPullProperties(), p.getContext(), p.getComments());
                        }
                        if (jsonSerialize.converter() != Converter.None.class) {
                            // Type info is not accessible with reflection so instantiate the converter
                            Method convertMethod = Arrays.stream(jsonSerialize.converter().getMethods()).filter(m -> m.getName().equals("convert")).findFirst().orElse(null);
                            if (convertMethod != null) {
                                modified.set(true);
                                return new PropertyModel(p.getName(), convertMethod.getGenericReturnType(), p.isOptional(), p.getAccess(), p.getOriginalMember(), p.getPullProperties(), p.getContext(), p.getComments());
                            }
                        }
                    }
                    return p;
                }).collect(Collectors.toList());
                if (modified.get()) {
                    return new BeanModel(bean.getOrigin(), bean.getParent(), bean.getTaggedUnionClasses(), bean.getDiscriminantProperty(), bean.getDiscriminantLiteral(), bean.getInterfaces(), properties, bean.getComments());
                }
                return bean;
            });
            return model;
        }
    }));
}
Also used : Arrays(java.util.Arrays) Converter(com.fasterxml.jackson.databind.util.Converter) PropertyModel(cz.habarta.typescript.generator.parser.PropertyModel) EmitterExtensionFeatures(cz.habarta.typescript.generator.emitter.EmitterExtensionFeatures) Member(java.lang.reflect.Member) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) Field(java.lang.reflect.Field) TsModel(cz.habarta.typescript.generator.emitter.TsModel) Collectors(java.util.stream.Collectors) List(java.util.List) ModelTransformer(cz.habarta.typescript.generator.compiler.ModelTransformer) JsonSerialize(com.fasterxml.jackson.databind.annotation.JsonSerialize) SymbolTable(cz.habarta.typescript.generator.compiler.SymbolTable) BeanModel(cz.habarta.typescript.generator.parser.BeanModel) Model(cz.habarta.typescript.generator.parser.Model) ModelCompiler(cz.habarta.typescript.generator.compiler.ModelCompiler) Method(java.lang.reflect.Method) BeanModel(cz.habarta.typescript.generator.parser.BeanModel) PropertyModel(cz.habarta.typescript.generator.parser.PropertyModel) SymbolTable(cz.habarta.typescript.generator.compiler.SymbolTable) Method(java.lang.reflect.Method) JsonSerialize(com.fasterxml.jackson.databind.annotation.JsonSerialize) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) Field(java.lang.reflect.Field) ModelTransformer(cz.habarta.typescript.generator.compiler.ModelTransformer) TsModel(cz.habarta.typescript.generator.emitter.TsModel) PropertyModel(cz.habarta.typescript.generator.parser.PropertyModel) TsModel(cz.habarta.typescript.generator.emitter.TsModel) BeanModel(cz.habarta.typescript.generator.parser.BeanModel) Model(cz.habarta.typescript.generator.parser.Model) Member(java.lang.reflect.Member)

Aggregations

JsonSerialize (com.fasterxml.jackson.databind.annotation.JsonSerialize)1 Converter (com.fasterxml.jackson.databind.util.Converter)1 ModelCompiler (cz.habarta.typescript.generator.compiler.ModelCompiler)1 ModelTransformer (cz.habarta.typescript.generator.compiler.ModelTransformer)1 SymbolTable (cz.habarta.typescript.generator.compiler.SymbolTable)1 EmitterExtensionFeatures (cz.habarta.typescript.generator.emitter.EmitterExtensionFeatures)1 TsModel (cz.habarta.typescript.generator.emitter.TsModel)1 BeanModel (cz.habarta.typescript.generator.parser.BeanModel)1 Model (cz.habarta.typescript.generator.parser.Model)1 PropertyModel (cz.habarta.typescript.generator.parser.PropertyModel)1 Field (java.lang.reflect.Field)1 Member (java.lang.reflect.Member)1 Method (java.lang.reflect.Method)1 Arrays (java.util.Arrays)1 List (java.util.List)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1 Collectors (java.util.stream.Collectors)1