com.wangjiegulu.rapidooo.library.compiler.OOOProcess Maven / Gradle / Ivy

Go to download
package com.wangjiegulu.rapidooo.library.compiler;

import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;
import com.wangjiegulu.rapidooo.library.compiler.oooentry.OOOEntry;
import com.wangjiegulu.rapidooo.library.compiler.oooentry.OOOGenerator;
import com.wangjiegulu.rapidooo.library.compiler.oooentry.OOOSEntry;
import com.wangjiegulu.rapidooo.library.compiler.part.PartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.CreateMethodPartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.DefaultConstructorMethodPartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.FieldAndGetterSetterPartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.FromMethodPartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.InterfacePartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.PoolPartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.ToMethod1PartBrew;
import com.wangjiegulu.rapidooo.library.compiler.part.impl.ToMethod2PartBrew;
import com.wangjiegulu.rapidooo.library.compiler.util.ElementUtil;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Locale;

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

/**
 * Author: wangjie
 * Email: [email protected]
 * Date: 11/04/2018.
 */
public class OOOProcess {
    private static final SimpleDateFormat DATE_FORMAT = new SimpleDateFormat("yyyy-MM-dd HH:mm:SSS", Locale.getDefault());

    private OOOGenerator oooGenerator;

    private List partBrews = new ArrayList<>();

    public OOOProcess() {
        partBrews.add(new InterfacePartBrew());
        partBrews.add(new DefaultConstructorMethodPartBrew());
        partBrews.add(new FieldAndGetterSetterPartBrew());
        partBrews.add(new FromMethodPartBrew());
        partBrews.add(new CreateMethodPartBrew());
        partBrews.add(new ToMethod1PartBrew());
        partBrews.add(new ToMethod2PartBrew());
        partBrews.add(new PoolPartBrew());
    }

    public void setGeneratorClassEl(Element mGeneratorClassEl) {
        oooGenerator = new OOOGenerator(mGeneratorClassEl);
        oooGenerator.parse();
    }

    public void brewJava(Filer filer) throws Throwable {
        OOOSEntry ooosEntry = oooGenerator.getOoosEntry();
        for(OOOEntry oooEntry : ooosEntry.getOoos()){

            ///////////////////// class /////////////////////
            TypeSpec.Builder result = TypeSpec.classBuilder(oooEntry.getTargetClassSimpleName())
                    .addModifiers(Modifier.PUBLIC)
                    .addJavadoc("From POJO: {@link $T}\nGenerate By: {@link $T}\nGenerate Time: " + DATE_FORMAT.format(new Date(System.currentTimeMillis())) + "\n",
                            oooEntry.getFromTypeName(), oooGenerator.getGeneratorClassType()
                    );

            ///////////////////// super class /////////////////////
            TypeName supperTypeName = oooEntry.getTargetSupperType();
            if (!ElementUtil.isSameType(supperTypeName, TypeName.OBJECT)) {
                result.superclass(supperTypeName);
            }

            for(PartBrew partBrew : partBrews){
                partBrew.brew(oooEntry, result);
            }

            JavaFile.builder(oooEntry.getTargetClassPackage(), result.build())
                    .addFileComment("GENERATED CODE BY RapidOOO. DO NOT MODIFY! https://github.com/wangjiegulu/RapidOOO")
                    .skipJavaLangImports(true)
                    .build()
                    .writeTo(filer);


        }


    }








}