32 lines
931 B
Java
32 lines
931 B
Java
package it.cavallium.datagen.plugin.classgen;
|
|
|
|
import com.squareup.javapoet.ClassName;
|
|
import com.squareup.javapoet.TypeSpec;
|
|
import it.cavallium.datagen.plugin.ClassGenerator;
|
|
import it.cavallium.datagen.plugin.ComputedVersion;
|
|
import java.io.Serializable;
|
|
import java.util.stream.Stream;
|
|
import javax.lang.model.element.Modifier;
|
|
|
|
public class GenIType extends ClassGenerator {
|
|
|
|
public GenIType(ClassGeneratorParams params) {
|
|
super(params);
|
|
}
|
|
|
|
@Override
|
|
protected Stream<GeneratedClass> generateClasses() {
|
|
return dataModel.getVersionsSet().stream().map(this::generateClass);
|
|
}
|
|
|
|
private GeneratedClass generateClass(ComputedVersion version) {
|
|
var interfaceBuilder = TypeSpec.interfaceBuilder("IType");
|
|
|
|
interfaceBuilder.addModifiers(Modifier.PUBLIC);
|
|
|
|
interfaceBuilder.addSuperinterface(ClassName.get(Serializable.class));
|
|
|
|
return new GeneratedClass(version.getPackage(basePackageName), interfaceBuilder);
|
|
}
|
|
}
|