package it.cavallium.data.generator.plugin.classgen;

import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterSpec;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;
import com.squareup.javapoet.TypeVariableName;
import com.squareup.javapoet.WildcardTypeName;
import it.cavallium.data.generator.DataSerializer;
import it.cavallium.data.generator.plugin.ClassGenerator;
import it.cavallium.data.generator.plugin.ClassGenerator.ClassGeneratorParams;
import java.io.IOException;
import java.util.stream.Stream;
import javax.lang.model.element.Modifier;

public class GenIVersion extends ClassGenerator {

	public GenIVersion(ClassGeneratorParams params) {
		super(params);
	}

	@Override
	protected Stream<GeneratedClass> generateClasses() {
		var iVersionClass = TypeSpec.interfaceBuilder("IVersion");
		iVersionClass.addModifiers(Modifier.PUBLIC);
		iVersionClass.addTypeVariable(TypeVariableName.get("B"));

		// Add getClass method
		{
			var getClassMethodBuilder = MethodSpec
					.methodBuilder("getClass")
					.addModifiers(Modifier.PUBLIC)
					.addModifiers(Modifier.ABSTRACT)
					.returns(ParameterizedTypeName.get(ClassName.get(Class.class),
							WildcardTypeName.subtypeOf(TypeVariableName.get("B"))
					))
					.addParameter(ParameterSpec
							.builder(ClassName.get(dataModel.getRootPackage(basePackageName), "BaseType"), "type")
							.build());
			iVersionClass.addMethod(getClassMethodBuilder.build());
		}

		// Add getSerializer method
		{
			var getSerializerMethodBuilder = MethodSpec
					.methodBuilder("getSerializer")
					.addModifiers(Modifier.PUBLIC)
					.addModifiers(Modifier.ABSTRACT)
					.addTypeVariable(TypeVariableName.get("T",
							TypeVariableName.get("B")
					))
					.returns(ParameterizedTypeName.get(ClassName.get(DataSerializer.class), TypeVariableName.get("T")))
					.addException(IOException.class)
					.addParameter(ParameterSpec
							.builder(ClassName.get(dataModel.getRootPackage(basePackageName), "BaseType"), "type")
							.build());
			iVersionClass.addMethod(getSerializerMethodBuilder.build());
		}

		// Add getVersion method
		{
			var getVersionMethod = MethodSpec
					.methodBuilder("getVersion")
					.addModifiers(Modifier.PUBLIC)
					.addModifiers(Modifier.ABSTRACT)
					.returns(TypeName.INT)
					.build();
			iVersionClass.addMethod(getVersionMethod);
		}

		return Stream.of(new GeneratedClass(dataModel.getRootPackage(basePackageName), iVersionClass));
	}
}