java source code of DatabaseHelperCreator

/*
 * The MIT License (MIT)
 *
 * Copyright (c) 2015 baoyongzhang <[email protected]>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */
package com.baoyz.airdata.creator;


import com.baoyz.airdata.AbstractDatabase;
import com.baoyz.airdata.AirDatabaseHelper;
import com.baoyz.airdata.ContentValuesWrapper;
import com.baoyz.airdata.TableInfo;
import com.baoyz.airdata.utils.LogUtils;
import com.squareup.javapoet.ArrayTypeName;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.TypeName;
import com.squareup.javapoet.TypeSpec;

import java.util.ArrayList;
import java.util.List;

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

/**
 * AirData
 * Created by baoyz on 15/7/7.
 */
public class DatabaseHelperCreator {

    private ArrayList<TableInfo> tables;
    private Filer filer;
    private String packageName;
    private String className;
    private ClassName airDbType;

    public DatabaseHelperCreator(ArrayList<TableInfo> tables, Filer filer, String packageName, String className, ClassName airDbType) {
        this.tables = tables;
        this.filer = filer;
        this.packageName = packageName;
        this.className = className;
        this.airDbType = airDbType;
    }

    public void create() {

        // dao fields
        List<String> daoFields = new ArrayList<>();

        MethodSpec.Builder constructorBuilder = MethodSpec.constructorBuilder()
                .addModifiers(Modifier.PUBLIC)
                .addParameter(ClassName.get("android.content", "Context"), "context")
                .addParameter(ClassName.get(AbstractDatabase.class), "airDb")
                .addStatement("this.helper = new $T(context, ($T)airDb)", ClassName.get(packageName, className), airDbType)
                .addStatement("this.database = this.helper.getWritableDatabase()");


        for (TableInfo table : tables) {
            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();
            constructorBuilder.addStatement("this.$L = new $L(this.database)", daoField, table.getDaoClassName());
        }

        TypeSpec.Builder typeSpecBuilder = TypeSpec.classBuilder("AirDatabaseHelperImpl")
                .addModifiers(Modifier.PUBLIC)
                .addSuperinterface(ClassName.get(AirDatabaseHelper.class))
                .addField(ClassName.get("android.database.sqlite", "SQLiteDatabase"), "database", Modifier.PRIVATE)
                .addField(ClassName.get(packageName, className), "helper", Modifier.PRIVATE);

        // generate save method
        typeSpecBuilder.addMethod(generateSaveMethod(typeSpecBuilder));

        // generate query method
        typeSpecBuilder.addMethod(generateQueryMethod(typeSpecBuilder));

        // generate queryAll method
        typeSpecBuilder.addMethod(generateQueryAllMethod(typeSpecBuilder));

        // generate rawQuery method
        typeSpecBuilder.addMethod(generateRawQueryMethod());

        // generate update method
        typeSpecBuilder.addMethod(generateUpdateMethod());

        // generate updateById method
        typeSpecBuilder.addMethod(generateUpdateByIdMethod(typeSpecBuilder));

        // generate delete method
        typeSpecBuilder.addMethod(generateDeleteMethod());

        // generate deleteById method
        typeSpecBuilder.addMethod(generateDeleteByIdMethod(typeSpecBuilder));

        // destory method
        MethodSpec destoryMethod = MethodSpec.methodBuilder("destory")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .addStatement("this.database.close()")
                .build();
        typeSpecBuilder.addMethod(destoryMethod);

        // getDatabase method
        MethodSpec getDatabaseMethod = MethodSpec.methodBuilder("getDatabase")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(ClassName.get("android.database.sqlite", "SQLiteDatabase"))
                .addStatement("return this.database")
                .build();
        typeSpecBuilder.addMethod(getDatabaseMethod);

        typeSpecBuilder.addMethod(constructorBuilder.build());

        JavaFile javaFile = JavaFile.builder("com.baoyz.airdata", typeSpecBuilder.build())
                .build();

        try {
            javaFile.writeTo(filer);
        } catch (Exception e) {
            e.printStackTrace();
        }

        LogUtils.debug(javaFile.toString());

    }

    private MethodSpec generateSaveMethod(TypeSpec.Builder typeSpecBuilder) {
        MethodSpec.Builder saveMethodBuidler = MethodSpec.methodBuilder("save")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(TypeName.LONG)
                .addParameter(ClassName.get(Object.class), "bean");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();

            typeSpecBuilder.addField(ClassName.get(table.getPackageName(), table.getDaoClassName()), daoField, Modifier.PRIVATE);
            MethodSpec methodSpec = MethodSpec.methodBuilder("save")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(TypeName.LONG)
                    .addParameter(ClassName.get(table.getPackageName(), table.getClassName()), "bean")
                    .addStatement("return $L.insert(bean)", daoField)
                    .build();
            typeSpecBuilder.addMethod(methodSpec);

            saveMethodBuidler.addStatement("if(bean instanceof $L) return save(($L)bean)", table.getClassName(), table.getClassName());
        }

        saveMethodBuidler.addStatement("return 0");
        return saveMethodBuidler.build();
    }

    private MethodSpec generateUpdateMethod() {
        MethodSpec.Builder saveMethodBuidler = MethodSpec.methodBuilder("update")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(TypeName.INT)
                .addParameter(ClassName.get(Class.class), "clazz")
                .addParameter(ClassName.get(ContentValuesWrapper.class), "valuesWrapper")
                .addParameter(ClassName.get(String.class), "where")
                .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "whereArgs");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();
            saveMethodBuidler.addStatement("if($L.class.equals(clazz)) return $L.update(valuesWrapper, where, whereArgs)", table.getClassName(), daoField);
        }

        saveMethodBuidler.addStatement("return 0");
        return saveMethodBuidler.build();
    }

    private MethodSpec generateUpdateByIdMethod(TypeSpec.Builder typeSpecBuilder) {
        MethodSpec.Builder saveMethodBuidler = MethodSpec.methodBuilder("update")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(TypeName.INT)
                .addParameter(ClassName.get(Object.class), "bean");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();

            MethodSpec methodSpec = MethodSpec.methodBuilder("update")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(TypeName.INT)
                    .addParameter(ClassName.get(table.getPackageName(), table.getClassName()), "bean")
                    .addStatement("return $L.update(bean)", daoField)
                    .build();
            typeSpecBuilder.addMethod(methodSpec);

            saveMethodBuidler.addStatement("if(bean instanceof $L) return update(($L)bean)", table.getClassName(), table.getClassName());
        }

        saveMethodBuidler.addStatement("return 0");
        return saveMethodBuidler.build();
    }

    private MethodSpec generateDeleteMethod() {
        MethodSpec.Builder saveMethodBuidler = MethodSpec.methodBuilder("delete")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(TypeName.INT)
                .addParameter(ClassName.get(Class.class), "clazz")
                .addParameter(ClassName.get(String.class), "where")
                .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "whereArgs");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();

            saveMethodBuidler.addStatement("if($L.class.equals(clazz)) return $L.delete(where, whereArgs)", table.getClassName(), daoField);
        }

        saveMethodBuidler.addStatement("return 0");
        return saveMethodBuidler.build();
    }

    private MethodSpec generateDeleteByIdMethod(TypeSpec.Builder typeSpecBuilder) {
        MethodSpec.Builder saveMethodBuidler = MethodSpec.methodBuilder("delete")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(TypeName.INT)
                .addParameter(ClassName.get(Object.class), "bean");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();

            MethodSpec methodSpec = MethodSpec.methodBuilder("delete")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(TypeName.INT)
                    .addParameter(ClassName.get(table.getPackageName(), table.getClassName()), "bean")
                    .addStatement("return $L.delete(bean)", daoField)
                    .build();
            typeSpecBuilder.addMethod(methodSpec);

            saveMethodBuidler.addStatement("if(bean instanceof $L) return delete(($L)bean)", table.getClassName(), table.getClassName());
        }

        saveMethodBuidler.addStatement("return 0");
        return saveMethodBuidler.build();
    }

    private MethodSpec generateQueryAllMethod(TypeSpec.Builder typeSpecBuilder) {
        MethodSpec.Builder queryMethodBuidler = MethodSpec.methodBuilder("queryAll")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(ClassName.get(List.class))
                .addParameter(ClassName.get(Class.class), "clazz");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();
            MethodSpec methodSpec = MethodSpec.methodBuilder("queryAll")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(ClassName.get(List.class))
                    .addParameter(ClassName.get(table.getPackageName(), table.getClassName()), "bean")
                    .addStatement("return $L.queryAll()", daoField)
                    .build();
            typeSpecBuilder.addMethod(methodSpec);

            queryMethodBuidler.addStatement("if($L.class.equals(clazz)) return $L.queryAll()", table.getClassName(), daoField);
        }

        queryMethodBuidler.addStatement("return null");
        return queryMethodBuidler.build();
    }

    private MethodSpec generateQueryMethod(TypeSpec.Builder typeSpecBuilder) {
        MethodSpec.Builder queryMethodBuidler = MethodSpec.methodBuilder("query")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(ClassName.get(List.class))
                .addParameter(ClassName.get(Class.class), "clazz")
                .addParameter(TypeName.BOOLEAN, "distinct")
                .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "columns")
                .addParameter(ClassName.get(String.class), "selection")
                .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "selectionArgs")
                .addParameter(ClassName.get(String.class), "groupBy")
                .addParameter(ClassName.get(String.class), "having")
                .addParameter(ClassName.get(String.class), "orderBy")
                .addParameter(ClassName.get(String.class), "limit");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();
            MethodSpec methodSpec = MethodSpec.methodBuilder("query")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(ClassName.get(List.class))
                    .addParameter(ClassName.get(table.getPackageName(), table.getClassName()), "bean")
                    .addParameter(TypeName.BOOLEAN, "distinct")
                    .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "columns")
                    .addParameter(ClassName.get(String.class), "selection")
                    .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "selectionArgs")
                    .addParameter(ClassName.get(String.class), "groupBy")
                    .addParameter(ClassName.get(String.class), "having")
                    .addParameter(ClassName.get(String.class), "orderBy")
                    .addParameter(ClassName.get(String.class), "limit")
                    .addStatement("return $L.query(distinct, columns, selection, selectionArgs, groupBy, having, orderBy, limit)", daoField)
                    .build();
            typeSpecBuilder.addMethod(methodSpec);

            queryMethodBuidler.addStatement("if($L.class.equals(clazz)) return $L.query(distinct, columns, selection, selectionArgs, groupBy, having, orderBy, limit)", table.getClassName(), daoField);
        }

        queryMethodBuidler.addStatement("return null");
        return queryMethodBuidler.build();
    }

    private MethodSpec generateRawQueryMethod() {
        MethodSpec.Builder queryMethodBuidler = MethodSpec.methodBuilder("rawQuery")
                .addAnnotation(Override.class)
                .addModifiers(Modifier.PUBLIC)
                .returns(ClassName.get("android.database", "Cursor"))
                .addParameter(ClassName.get(Class.class), "clazz")
                .addParameter(TypeName.BOOLEAN, "distinct")
                .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "columns")
                .addParameter(ClassName.get(String.class), "selection")
                .addParameter(ArrayTypeName.of(ClassName.get(String.class)), "selectionArgs")
                .addParameter(ClassName.get(String.class), "groupBy")
                .addParameter(ClassName.get(String.class), "having")
                .addParameter(ClassName.get(String.class), "orderBy")
                .addParameter(ClassName.get(String.class), "limit");

        for (TableInfo table : tables) {

            String daoField = table.getDaoClassName().replace("$$", "").toLowerCase();

            queryMethodBuidler.addStatement("if($L.class.equals(clazz)) return $L.rawQuery(distinct, columns, selection, selectionArgs, groupBy, having, orderBy, limit)", table.getClassName(), daoField);
        }

        queryMethodBuidler.addStatement("return null");
        return queryMethodBuidler.build();
    }
}