remoter.compiler.builder.FieldBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of remoter Show documentation
Remoter makes developing android remote services intuitive without messing with aidl.
The newest version!
package remoter.compiler.builder;

import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;

import java.util.Map;
import java.util.WeakHashMap;

import javax.annotation.processing.Messager;
import javax.lang.model.element.Element;
import javax.lang.model.element.Modifier;

/**
 * A {@link RemoteBuilder} that knows how to generate the fields for stub and proxy
 */
class FieldBuilder extends RemoteBuilder {


    protected FieldBuilder(Messager messager, Element element) {
        super(messager, element);
    }


    public void addProxyFields(TypeSpec.Builder classBuilder) {
        //add IBinder
        classBuilder.addField(FieldSpec.builder(ClassName.get("android.os", "IBinder"), "mRemote")
                .addModifiers(Modifier.PRIVATE).build());

        classBuilder.addField(FieldSpec.builder(ClassName.bestGuess("DeathRecipient"), "proxyListener")
                .addModifiers(Modifier.PRIVATE).build());


        addCommonFields(classBuilder);
        final int[] lastMethodIndex = {0};
        processRemoterElements(classBuilder, new ElementVisitor() {
            @Override
            public void visitElement(TypeSpec.Builder classBuilder, Element member, int methodIndex, MethodSpec.Builder methodBuilder) {
                addCommonFields(classBuilder, member, methodIndex);
                lastMethodIndex[0] = methodIndex;
            }
        }, null);

        lastMethodIndex[0] ++;
        classBuilder.addField(FieldSpec.builder(TypeName.INT, "TRANSACTION__getStubID")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("android.os.IBinder.FIRST_CALL_TRANSACTION + " + lastMethodIndex[0]).build());

        lastMethodIndex[0] ++;
        classBuilder.addField(FieldSpec.builder(TypeName.INT, "TRANSACTION__getStubProcessID")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("android.os.IBinder.FIRST_CALL_TRANSACTION + " + lastMethodIndex[0]).build());


        classBuilder.addField(FieldSpec.builder(TypeName.INT, "_binderID")
                .addModifiers(Modifier.PRIVATE, Modifier.FINAL).build());

        classBuilder.addField(FieldSpec.builder(TypeName.INT, "_stubProcess")
                .addModifiers(Modifier.PRIVATE, Modifier.FINAL).build());

        classBuilder.addField(FieldSpec.builder(
                ParameterizedTypeName.get(ClassName.get(Map.class),
                        ClassName.get(Object.class),
                        ClassName.get("android.os", "IBinder")), "stubMap")
                .addModifiers(Modifier.PRIVATE)
                .initializer("new $T()", WeakHashMap.class)
                .build());

        classBuilder.addField(FieldSpec.builder(
                ParameterizedTypeName.get(ClassName.get(Map.class),
                        ClassName.get(String.class),
                        ClassName.get(Object.class)), "__global_properties")
                .addModifiers(Modifier.PRIVATE)
                .build());

    }

    public void addStubFields(TypeSpec.Builder classBuilder) {
        classBuilder.addField(FieldSpec.builder(TypeName.get(getRemoterInterfaceElement().asType()), "serviceImpl")
                .addModifiers(Modifier.PRIVATE).build());
        addCommonFields(classBuilder);

        final int[] lastMethodIndex = {0};

        processRemoterElements(classBuilder, new ElementVisitor() {
            @Override
            public void visitElement(TypeSpec.Builder classBuilder, Element member, int methodIndex, MethodSpec.Builder methodBuilder) {
                addCommonFields(classBuilder, member, methodIndex);
                lastMethodIndex[0] = methodIndex;
            }
        }, null);

        classBuilder.addField(FieldSpec.builder(TypeName.INT, "__lastMethodIndex")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("android.os.IBinder.FIRST_CALL_TRANSACTION + " + lastMethodIndex[0]).build());

        classBuilder.addField(FieldSpec.builder(TypeName.INT, "__lastMethodIndexOfProxy")
                .addModifiers(Modifier.PRIVATE)
                .initializer("-1").build());

        classBuilder.addField(FieldSpec.builder(TypeName.BOOLEAN, "__checkStubProxyMatch")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC)
                .initializer("true").build());


        lastMethodIndex[0] ++;

        classBuilder.addField(FieldSpec.builder(TypeName.INT, "TRANSACTION__getStubID")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("android.os.IBinder.FIRST_CALL_TRANSACTION + " + lastMethodIndex[0]).build());

        lastMethodIndex[0] ++;
        classBuilder.addField(FieldSpec.builder(TypeName.INT, "TRANSACTION__getStubProcessID")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("android.os.IBinder.FIRST_CALL_TRANSACTION + " + lastMethodIndex[0]).build());


        classBuilder.addField(FieldSpec.builder(ClassName.bestGuess("BinderWrapper"), "binderWrapper")
                .addModifiers(Modifier.PRIVATE).build());

    }

    private void addCommonFields(TypeSpec.Builder classBuilder) {
        //Add descriptor
        classBuilder.addField(FieldSpec.builder(ClassName.get(String.class), "DESCRIPTOR")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("\"" + getRemoterInterfacePackageName() + "." + getRemoterInterfaceClassName() + "\"")
                .build());
        classBuilder.addField(FieldSpec.builder(TypeName.INT, "REMOTER_EXCEPTION_CODE")
                .addModifiers(Modifier.PRIVATE, Modifier.STATIC, Modifier.FINAL)
                .initializer("-99999")
                .build());

    }

    private void addCommonFields(TypeSpec.Builder classBuilder, Element member, int methodIndex) {
        String methodName = member.getSimpleName().toString();
        classBuilder.addField(FieldSpec.builder(TypeName.INT, "TRANSACTION_" + methodName + "_" + methodIndex)
                .addModifiers(Modifier.PROTECTED, Modifier.STATIC, Modifier.FINAL)
                .initializer("android.os.IBinder.FIRST_CALL_TRANSACTION + " + methodIndex).build());
    }
}