22 lines
737 B
Java
22 lines
737 B
Java
package it.cavallium.data.generator.plugin.classgen;
|
|
|
|
import com.squareup.javapoet.TypeSpec;
|
|
import it.cavallium.data.generator.plugin.ClassGenerator;
|
|
import java.util.stream.Stream;
|
|
import javax.lang.model.element.Modifier;
|
|
|
|
public class GenSuperType extends ClassGenerator {
|
|
|
|
public GenSuperType(ClassGeneratorParams params) {
|
|
super(params);
|
|
}
|
|
|
|
@Override
|
|
protected Stream<GeneratedClass> generateClasses() {
|
|
var superTypeClass = TypeSpec.enumBuilder("SuperType");
|
|
superTypeClass.addModifiers(Modifier.PUBLIC);
|
|
dataModel.getSuperTypesComputed().forEach(superType -> superTypeClass.addEnumConstant(superType.getName()));
|
|
return Stream.of(new GeneratedClass(dataModel.getRootPackage(basePackageName), superTypeClass));
|
|
}
|
|
}
|