com.digitolio.jdbi.codegen.experimental.H2Generator Maven / Gradle / Ivy

Go to download
package com.digitolio.jdbi.codegen.experimental;

import com.digitolio.jdbi.annotations.PK;
import com.digitolio.jdbi.codegen.Scanner;
import com.digitolio.jdbi.strategy.SnakeCaseTranslatingStrategy;
import com.digitolio.jdbi.table.Column;
import com.digitolio.jdbi.table.Table;
import com.digitolio.jdbi.table.TableResolver;
import com.google.common.base.Joiner;
import com.google.common.collect.Lists;
import com.google.common.collect.Ordering;
import com.google.common.io.Files;

import java.io.BufferedWriter;
import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.nio.charset.Charset;
import java.util.Comparator;
import java.util.Date;
import java.util.List;
import java.util.Set;

/**
 * @author C.Koc
 */
public class H2Generator {

    private final Table table;
    private final Class clazz;
    private final File targetDir;

    public H2Generator(Table table, Class clazz, File targetDir) {
        this.table = table;
        this.clazz = clazz;
        this.targetDir = targetDir;
    }

    public static void main(String[] args) throws IOException {
        if (args[0].contains("OVERWRITE_THIS_VALUE")) {
            return;
        }
        SnakeCaseTranslatingStrategy strategy = new SnakeCaseTranslatingStrategy();
        TableResolver tableResolver = new TableResolver();

//        Set> inputClasses = new Scanner().scanPackage("com.digitolio.jdbi.codegen.test");
        Set> inputClasses = scanPackage(args[0]);
        List> classes = orderClasses(inputClasses);
        cleanDirectoryForNewDll(new File(args[1].concat("/db/h2/")));
//        File targetDir = new File("D:\\PersonalProjects\\digitolio\\jdbi-codegen\\src\\main\\java\\cemo");
        File targetDir = new File(args[1]);
        StringBuilder allDdl = new StringBuilder();

        for (Class aClass : classes) {
            Table resolve = tableResolver.resolve(aClass, strategy);
            H2Generator h2Generator = new H2Generator(resolve, aClass, targetDir);
            Object generate = h2Generator.generate();
            String content = generate.toString();
            allDdl.append(content).append("\n\n");
            File file = new File(args[1].concat("/db/h2/").concat(resolve.getTableName().toLowerCase().concat(".ddl")));
            Files.createParentDirs(file);
            BufferedWriter bufferedWriter = Files.newWriter(file, Charset.defaultCharset());
            bufferedWriter.write(content);
            bufferedWriter.close();
        }

        File file = new File(args[1].concat("/db/h2/all-").concat(args[2]).concat(".ddl"));
        Files.createParentDirs(file);
        BufferedWriter bufferedWriter = Files.newWriter(file, Charset.defaultCharset());
        bufferedWriter.write(allDdl.toString());
        bufferedWriter.close();

    }

    @SuppressWarnings({"ConstantConditions", "ResultOfMethodCallIgnored"})
    private static void cleanDirectoryForNewDll(File targetDir) {
        if (targetDir.exists()) {
            for (File file : targetDir.listFiles()) {
                boolean delete = file.delete();
            }
        }
    }

    private static Set> scanPackage(String arg) {
        return new Scanner().scanPackage(arg);
    }

    private static List> orderClasses(Set> inputClasses) {
        return Ordering.from(new Comparator() {
            @Override
            public int compare(Class o1, Class o2) {
                return o1.getSimpleName().compareTo(o2.getSimpleName());
            }
        }).sortedCopy(inputClasses);
    }


    private Object generate() {
        StringBuilder builder = new StringBuilder();
        builder.append("drop table ").append(table.getTableName()).append(" if exists;\n");
        builder.append("create table ").append(table.getTableName()).append("(\n");
        int a = 0;

        for (Column column : table.getAllColumns()) {
            builder.append("\t");
            if (a++ > 0) {
                builder.append(",");
            }
            builder.append(column.getDatabaseName())
                   .append(" ")
                   .append(getDbType(column))
                   .append(getNullInfo(column))
                   .append(getDefaultInfo(column))
                   .append(getAutoGeneratedInfo(column))
                   .append("\n");
        }

        builder.append("\t,primary key(");
        List l = Lists.newArrayList();
        for (Column column : table.getPrimaryKeyColumns()) {
            l.add(column.getDatabaseName());
        }
        builder.append(Joiner.on(", ").join(l));
        builder.append(")");

        builder.append("\n);");
//        System.out.println(builder.toString());
        return builder.toString();
    }

    private String getDefaultInfo(Column column) {
        com.digitolio.jdbi.annotations.Column annotation = column.getField().getAnnotation(
                com.digitolio.jdbi.annotations.Column.class);
        String defaultValue = "";
        if (annotation != null) {
            if (annotation.defaultValue().equals(com.digitolio.jdbi.annotations.Column.defaultValueConstant)) {
                defaultValue = "";
            } else {
                defaultValue = "default '" +annotation.defaultValue() + "'";
            }
        }
        return defaultValue;
    }

    private String getAutoGeneratedInfo(Column column) {
        Field field = column.getField();
        PK annotation = field.getAnnotation(PK.class);
        if (annotation != null && annotation.autoIncrement()) {
            return " auto_increment";
        } else {
            return "";
        }
    }

    private String getNullInfo(Column column) {
        com.digitolio.jdbi.annotations.Column annotation = column.getField().getAnnotation(
                com.digitolio.jdbi.annotations.Column.class);
        boolean nullable = false;
        if (annotation != null) {
            nullable = annotation.nullable();
        }
        if (nullable) {
            return " null ";
        } else {
            return " not null ";
        }
    }

    private String getDbType(Column column) {
        Class clazz = column.getField().getType();
        if (Integer.class.isAssignableFrom(clazz)) {
            return "int";
        } else if (Long.class.isAssignableFrom(clazz)) {
            return "bigint";
        } else if (String.class.isAssignableFrom(clazz)) {
            return "varchar(128)";
        } else if (Date.class.isAssignableFrom(clazz)) {
            return "datetime";
        } else if (Enum.class.isAssignableFrom(clazz)) {
            return "varchar(128)";
        }else if (Boolean.class.isAssignableFrom(clazz)) {
            return "boolean";
        }

        return "doktor bu ne";

    }

/*
    drop table shopping_cart_item if exists;
    create table shopping_cart_item (
       order_id       int        not null
       ,user_id   int not null
       ,item_count   int not null
       ,item_price   int not null
       ,primary key(user_id, order_id)
    );

    drop table product_order if exists;
    create table product_order (
       order_id    int        not null  primary key auto_increment
       ,user_id   int not null
       ,order_type   varchar(20) not null
       ,order_time datetime not null
       ,process_time datetime not null
       ,activate_time datetime not null
       ,refund_time datetime not null
    );
*/

}