com.wangjiegulu.rapidooo.library.compiler.part.statement.parcelable.ParcelableArrayStatementBrew Maven / Gradle / Ivy

Go to download
package com.wangjiegulu.rapidooo.library.compiler.part.statement.parcelable;

import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.TypeName;
import com.wangjiegulu.rapidooo.library.compiler.oooentry.type.OOOArrayTypeEntry;
import com.wangjiegulu.rapidooo.library.compiler.oooentry.type.OOOTypeEntry;
import com.wangjiegulu.rapidooo.library.compiler.part.statement.contact.IParcelableStatementBrew;
import com.wangjiegulu.rapidooo.library.compiler.util.ElementUtil;
import com.wangjiegulu.rapidooo.library.compiler.util.TextUtil;

/**
 * Author: wangjie Email: [email protected] Date: 2019-06-18.
 */
public class ParcelableArrayStatementBrew implements IParcelableStatementBrew {
    @Override
    public boolean match(OOOTypeEntry typeEntry) {
        return typeEntry.isArray();
    }

    @Override
    public void read(MethodSpec.Builder methodBuilder, String statementPrefix, Object[] statementPrefixTypes, String fieldCode, OOOTypeEntry oooTypeEntry, String fieldName) {
        TypeName arrayItemType = ((OOOArrayTypeEntry) oooTypeEntry).getArrayItemTypeName();
        // TODO: 2019-06-21 wangjie
        // #id__ChatBO[]
        if (ElementUtil.isParcelableClassType(arrayItemType)) {
            methodBuilder.addStatement(statementPrefix + fieldCode + " = parcel.createTypedArray($T.CREATOR)", TextUtil.addAll(statementPrefixTypes, arrayItemType));
        } else { // java.lang.String[]
            if (arrayItemType.isPrimitive()) {
                methodBuilder.addStatement(statementPrefix + fieldCode + " = parcel.create" + TextUtil.firstCharUpper(arrayItemType.toString()) + "Array()", statementPrefixTypes);
            } else {
                methodBuilder.addStatement(statementPrefix + fieldCode + " = ($T) parcel.readArray($T.class.getClassLoader())",
                        TextUtil.addAll(statementPrefixTypes, oooTypeEntry.getTypeName(), oooTypeEntry.getTypeName()));
            }
        }
    }

    @Override
    public void write(MethodSpec.Builder methodBuilder, String statementPrefix, Object[] statementPrefixTypes, String fieldCode, OOOTypeEntry oooTypeEntry, String fieldName) {
        TypeName arrayItemType = ((OOOArrayTypeEntry) oooTypeEntry).getArrayItemTypeName();
        // java.util.List<#id__ChatBO>
        if (ElementUtil.isParcelableClassType(arrayItemType)) {
            methodBuilder.addStatement(statementPrefix + "dest.writeTypedArray(" + fieldCode + ", flags)", TextUtil.addAll(statementPrefixTypes, arrayItemType));
        } else { // java.util.List
            if (arrayItemType.isPrimitive()) {
                methodBuilder.addStatement(statementPrefix + "dest.write" + TextUtil.firstCharUpper(arrayItemType.toString()) + "Array(" + fieldCode + ")", statementPrefixTypes);
            } else {
                methodBuilder.addStatement(statementPrefix + "dest.writeArray(" + fieldCode + ")", statementPrefixTypes);
            }
        }
    }
}