Mindustry/annotations/src/main/java/mindustry/annotations/remote/SerializerResolver.java

package mindustry.annotations.remote;

import arc.struct.*;

import javax.lang.model.element.*;
import javax.lang.model.type.*;

public class SerializerResolver{

    public static String locate(ExecutableElement elem, TypeMirror mirror, boolean write){
        //generic type
        if((mirror.toString().equals("T") && Array.with(elem.getTypeParameters().get(0).getBounds()).contains(SerializerResolver::isEntity)) ||
            isEntity(mirror)){
            return write ? "mindustry.io.TypeIO.writeEntity" : "mindustry.io.TypeIO.readEntity";
        }
        return null;
    }

    private static boolean isEntity(TypeMirror mirror){
        return !mirror.toString().contains(".") && mirror.toString().endsWith("c");
    }
}