package ru.pyatigil.db.migrate.liquibase;

import liquibase.database.Database;
import liquibase.sql.Sql;
import liquibase.sql.UnparsedSql;
import liquibase.sqlgenerator.SqlGeneratorChain;
import liquibase.sqlgenerator.core.CreateTableGenerator;
import liquibase.statement.core.CreateTableStatement;

/**
 * User: looser
 * Date: 12/03/11
 */
@SuppressWarnings({"UnusedDeclaration"}) // used by liquibase
public class InnoDbCreateTableGenerator extends CreateTableGenerator {
    @Override
    public Sql[] generateSql(CreateTableStatement statement, Database database, SqlGeneratorChain sqlGeneratorChain) {
        Sql[] sqls = super.generateSql(statement, database, sqlGeneratorChain);
        if (database.getTypeName().equals("mysql")) {
            if (sqls.length != 1) {
                throw new IllegalStateException("was expecting exactly 1 sql");
            }
            Sql sql = sqls[0];
            return new Sql[] {
                    new UnparsedSql(sql.toSql() + " engine INNODB")
            };
        } else {
            return sqls;
        }
    }

    @Override
    public int getPriority() {
        return super.getPriority() + 1;
    }
}