How to configure UnitSourceGenerator not to add imports for "java.lang" classes

[fjakop]

Good morning,
we're using UnitSourceGenerator to generate Java files from other Java files like that

void createPojo(Class<?> clazz) {
        final var sourceGenerator = UnitSourceGenerator.create("my.cool.package");
        final var clazzSimpleName = clazz.getSimpleName();

        final var classSourceGenerator = ClassSourceGenerator.create(TypeDeclarationSourceGenerator
                        .create(clazzSimpleName));

        Arrays.stream(clazz.getDeclaredFields())
                .forEach(field -> classSourceGenerator
                        .addField(VariableSourceGenerator.create(field.getType(), field.getName())
                                .addModifier(Modifier.PRIVATE))
                );

        sourceGenerator.addClass(classSourceGenerator);

        log.debug(sourceGenerator.make());
        sourceGenerator.storeToClassPath(outputDirectory);
    }

This works really well except if the source clazz has member fields belonging to "java.lang" package. Normally usage of these types would not result in an import directive, since "java.lang" is always available.
However, the UnitSourceGenerator generates these import directives.

Does anyone know how to switch this off?

Kind regards
Frank

[Roberto-Gentili]

Try this:

	static void createPojo(Class<?> clazz) {
		UnitSourceGenerator sourceGenerator = UnitSourceGenerator.create("my.cool.package");
		String clazzSimpleName = clazz.getSimpleName();

		ClassSourceGenerator classSourceGenerator = ClassSourceGenerator.create(TypeDeclarationSourceGenerator
                        .create(clazzSimpleName));

        Arrays.stream(clazz.getDeclaredFields())
                .forEach(field -> {
                	if (field.getType().getName().startsWith("java.lang")) {
                		classSourceGenerator
                        	.addField(VariableSourceGenerator.create(field.getType().getSimpleName() + " " +  field.getName())
                                .addModifier(Modifier.PRIVATE));
                	} else {
                		classSourceGenerator
                        .addField(VariableSourceGenerator.create(field.getType(), field.getName())
                                .addModifier(Modifier.PRIVATE));
                	}
                
                });

        sourceGenerator.addClass(classSourceGenerator);

        System.out.println(sourceGenerator.make());
    }

[fjakop]

thanks a lot, this works well.