Examples with TsModel cz.habarta.typescript.generator.emitter.TsModel used on opensource projects

Example 6 with TsModel

use of cz.habarta.typescript.generator.emitter.TsModel in project typescript-generator by vojtechhabarta.

the class BeanPropertyPathExtensionTest method basicTest.

@Test
public void basicTest() throws Exception {
    final StringBuilder data = new StringBuilder();
    final EmitterExtension.Writer writer = new EmitterExtension.Writer() {

        @Override
        public void writeIndentedLine(String line) {
            data.append(line + "\n");
        }
    };
    final Settings settings = new Settings();
    settings.sortDeclarations = true;
    final TypeProcessor typeProcessor = new DefaultTypeProcessor();
    final Model model = new Jackson2Parser(settings, typeProcessor).parseModel(ClassA.class);
    final TsModel tsModel = new ModelCompiler(settings, typeProcessor).javaToTypeScript(model);
    new BeanPropertyPathExtension().emitElements(writer, settings, false, tsModel);
    String dataStr = data.toString();
    final String expected = Utils.readString(getClass().getResourceAsStream("/ext/expected.ts"), "\n");
    Assert.assertEquals(expected.trim(), dataStr.trim());
}

Example 7 with TsModel

use of cz.habarta.typescript.generator.emitter.TsModel in project typescript-generator by vojtechhabarta.

the class ClassEnumExtension method getTransformers.

@Override
public List<TransformerDefinition> getTransformers() {
    return Arrays.asList(new TransformerDefinition(ModelCompiler.TransformationPhase.BeforeEnums, new ModelTransformer() {

        @Override
        public TsModel transformModel(SymbolTable symbolTable, TsModel model) {
            List<TsBeanModel> beans = model.getBeans();
            List<TsBeanModel> classEnums = new ArrayList<>();
            for (TsBeanModel bean : beans) {
                if (bean.getName().getSimpleName().contains(classEnumPattern))
                    classEnums.add(bean);
            }
            List<TsEnumModel> stringEnums = new ArrayList<>();
            for (TsBeanModel tsBeanModel : classEnums) {
                List<EnumMemberModel> members = new ArrayList<>();
                for (Field declaredField : tsBeanModel.getOrigin().getDeclaredFields()) {
                    if (declaredField.getType().getName().equals(tsBeanModel.getOrigin().getName())) {
                        members.add(new EnumMemberModel(declaredField.getName(), declaredField.getName(), null));
                    }
                }
                TsEnumModel temp = new TsEnumModel(tsBeanModel.getOrigin(), tsBeanModel.getName(), EnumKind.StringBased, members, null);
                stringEnums.add(temp);
            }
            return model.withEnums(stringEnums).withoutBeans(classEnums);
        }
    }));
}