Examples with PropertyModel cz.habarta.typescript.generator.parser.PropertyModel used on opensource projects

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;
        }
    }));
}