package liquibase.database.core;

 import liquibase.database.DatabaseConnection;

 import liquibase.database.AbstractDatabase;

 import liquibase.exception.DatabaseException;

 import liquibase.exception.DateParseException;

 import liquibase.statement.DatabaseFunction;

 import liquibase.util.ISODateFormat;

 import java.text.ParseException;

 import java.text.SimpleDateFormat;

 import java.text.DateFormat;

 import java.util.Date;

 import java.util.Arrays;

 import java.util.List;

 public class H2Database extends AbstractDatabase {

     private static String START_CONCAT = "CONCAT(";

     private static String END_CONCAT = ")";

     private static String SEP_CONCAT = ", ";

     public H2Database() {

         this.databaseFunctions.add(new DatabaseFunction("CURRENT_TIMESTAMP()"));

     }

     public String getTypeName() {

         return "h2";

     }

     public String getDefaultDriver(String url) {

         if (url.startsWith("jdbc:h2")) {

             return "org.h2.Driver";

         }

         return null;

     }

     public int getPriority() {

         return PRIORITY_DATABASE;

     }

     public boolean isCorrectDatabaseImplementation(DatabaseConnection conn) throws DatabaseException {

         return "H2".equals(conn.getDatabaseProductName());

     }

     // public void dropDatabaseObjects(String schema) throws DatabaseException {

     // DatabaseConnection conn = getConnection();

     // Statement dropStatement = null;

     // try {

     // dropStatement = conn.createStatement();

     // dropStatement.executeUpdate("DROP ALL OBJECTS");

     // changeLogTableExists = false;

     // changeLogLockTableExists = false;

     // changeLogCreateAttempted = false;

     // changeLogLockCreateAttempted = false;

     // } catch (SQLException e) {

     // throw new DatabaseException(e);

     // } finally {

     // try {

     // if (dropStatement != null) {

     // dropStatement.close();

     // }

     // conn.commit();

     // } catch (SQLException e) {

     // ;

     // }

     // }

     //

     // }

     public boolean supportsTablespaces() {

         return false;

     }

     @Override

     public String getViewDefinition(String schemaName, String name) throws DatabaseException {

         String definition = super.getViewDefinition(schemaName, name);

         if (!definition.startsWith("SELECT")) {

             definition = definition.replaceFirst(".*?\n", ""); // some h2 versions return "create view....as\nselect

         }

         definition = definition.replaceFirst("/\\*.*", ""); // sometimes includes comments at the end

         return definition;

     }

     @Override

     public Date parseDate(String dateAsString) throws DateParseException {

         try {

             if (dateAsString.indexOf(' ') > 0) {

                 return new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSSSSSSSS").parse(dateAsString);

             } else {

                 if (dateAsString.indexOf(':') > 0) {

                     return new SimpleDateFormat("HH:mm:ss").parse(dateAsString);

                 } else {

                     return new SimpleDateFormat("yyyy-MM-dd").parse(dateAsString);

                 }

             }

         } catch (ParseException e) {

             throw new DateParseException(dateAsString);

         }

     }

     @Override

     public boolean isLocalDatabase() throws DatabaseException {

         String url = getConnection().getURL();

         boolean isLocalURL = (super.isLocalDatabase() || url.startsWith("jdbc:h2:file:")

                 || url.startsWith("jdbc:h2:mem:") || url.startsWith("jdbc:h2:zip:") || url.startsWith("jdbc:h2:~"));

         return isLocalURL;

     }

     // @Override

     // public String convertRequestedSchemaToSchema(String requestedSchema) throws DatabaseException {

     // return super.convertRequestedSchemaToSchema(requestedSchema).toLowerCase();

     // }

     @Override

     public boolean supportsSequences() {

         return true;

     }

     @Override

     protected String getDefaultDatabaseSchemaName() throws DatabaseException {

         return "PUBLIC";

     }

     @Override

     public String getAutoIncrementClause() {

         return "GENERATED BY DEFAULT AS IDENTITY IDENTITY";

     }

     @Override

     public String getConcatSql(String... values) {

         if (values == null) {

             return null;

         }

         return getConcatSql(Arrays.asList(values));

     }

     /**

      * Recursive way of building CONCAT instruction

      * 

      * @param values

      *            a non null List of String

      * @return a String containing the CONCAT instruction with all elements, or only a value if there is only one

      *         element in the list

      */

     private String getConcatSql(List<String> values) {

         if (values.size() == 1) {

             return values.get(0);

         } else {

             return START_CONCAT + values.get(0) + SEP_CONCAT + getConcatSql(values.subList(1, values.size()))

                     + END_CONCAT;

         }

     }

     @Override

     public String getDateLiteral(String isoDate) {

         String returnString = isoDate;

         try {

             if (isDateTime(isoDate)) {

                 ISODateFormat isoTimestampFormat = new ISODateFormat();

                 DateFormat dbTimestampFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.S");

                 returnString = dbTimestampFormat.format(isoTimestampFormat.parse(isoDate));

             }

         } catch (ParseException e) {

             throw new RuntimeException("Unexpected date format: " + isoDate, e);

         }

         return "'" + returnString + "'";

     }

     @Override

     public String convertRequestedSchemaToSchema(String requestedSchema) throws DatabaseException {

         return super.convertRequestedSchemaToSchema(requestedSchema).toUpperCase();

     }

     @Override

     public String escapeDatabaseObject(String objectName) {

         if (objectName != null) {

             if (isReservedWord(objectName)) {

                 return "\"" + objectName + "\"";

             }

         }

         return objectName;

     }

     @Override

     public boolean isReservedWord(String objectName) {

         return keywords.contains(objectName.toUpperCase());

     }

     private static List keywords = Arrays.asList(

     // "ADD",

     // "ALL",

     // "ALLOCATE",

     // "ALTER",

     // "AND",

     // "ANY",

     // "ARE",

     // "ARRAY",

     // "AS",

     // "ASENSITIVE",

     // "ASYMMETRIC",

     // "AT",

     // "ATOMIC",

     // "AUTHORIZATION",

     // "BEGIN",

     // "BETWEEN",

     // "BIGINT",

     // "BINARY",

     // "BLOB",

     // "BOOLEAN",

     // "BOTH",

     // "BY",

     // "CALL",

     // "CALLED",

     // "CASCADED",

     // "CASE",

     // "CAST",

     // "CHAR",

     // "CHARACTER",

     // "CHECK",

     // "CLOB",

     // "CLOSE",

     // "COLLATE",

     // "COLUMN",

     // "COMMIT",

     // "CONDITION",

     // "CONNECT",

     // "CONSTRAINT",

     // "CONTINUE",

     // "CORRESPONDING",

     // "CREATE",

     // "CROSS",

     // "CUBE",

     // "CURRENT",

     // "CURRENT_DATE",

     // "CURRENT_DEFAULT_TRANSFORM_GRO",

     // "CURRENT_PATH",

     // "CURRENT_ROLE",

     // "CURRENT_TIME",

     // "CURRENT_TIMESTAMP",

     // "CURRENT_TRANSFORM_GROUP_FOR_T",

     // "CURRENT_USER",

     // "CURSOR",

     // "CYCLE",

     // "DATE",

     // "DAY",

     // "DEALLOCATE",

     // "DEC",

     // "DECIMAL",

     // "DECLARE",

     // "DEFAULT",

     // "DELETE",

     // "DEREF",

     // "DESCRIBE",

     // "DETERMINISTIC",

     // "DISCONNECT",

     // "DISTINCT",

     // "DO",

     // "DOUBLE",

     // "DROP",

     // "DYNAMIC",

     // "EACH",

     // "ELEMENT",

     // "ELSE",

     // "ELSEIF",

     // "END",

     // "ESCAPE",

     // "EXCEPT",

     // "EXEC",

     // "EXECUTE",

     // "EXISTS",

     // "EXIT",

     // "EXTERNAL",

     // "FALSE",

     // "FETCH",

     // "FILTER",

     // "FLOAT",

     // "FOR",

     // "FOREIGN",

     // "FREE",

     // "FROM",

     // "FULL",

     // "FUNCTION",

     // "GET",

     // "GLOBAL",

     // "GRANT",

     // "GROUP",

     // "GROUPING",

     // "HANDLER",

     // "HAVING",

     // "HOLD",

     // "HOUR",

     // "IDENTITY",

     // "IF",

     // "IMMEDIATE",

     // "IN",

     // "INDICATOR",

     // "INNER",

     // "INOUT",

     // "INPUT",

     // "INSENSITIVE",

     // "INSERT",

     // "INT",

     // "INTEGER",

     // "INTERSECT",

     // "INTERVAL",

     // "INTO",

     // "IS",

     // "ITERATE",

     // "JOIN",

     // "LANGUAGE",

     // "LARGE",

     // "LATERAL",

     // "LEADING",

     // "LEAVE",

     // "LEFT",

     // "LIKE",

     // "LOCAL",

     // "LOCALTIME",

     // "LOCALTIMESTAMP",

     // "LOOP",

     // "MATCH",

     // "MEMBER",

     // "MERGE",

     // "METHOD",

     // "MINUTE",

     // "MODIFIES",

     // "MODULE",

     // "MONTH",

     // "MULTISET",

     // "NATIONAL",

     // "NATURAL",

     // "NCHAR",

     // "NCLOB",

     // "NEW",

     // "NO",

     // "NONE",

     // "NOT",

     // "NULL",

     // "NUMERIC",

     // "OF",

     // "OLD",

     // "ON",

     // "ONLY",

     // "OPEN",

     // "OR",

     // "ORDER",

     // "OUT",

     // "OUTER",

     // "OUTPUT",

     // "OVER",

     // "OVERLAPS",

     // "PARAMETER",

     // "PARTITION",

     // "PRECISION",

     // "PREPARE",

     // "PRIMARY",

     // "PROCEDURE",

     // "RANGE",

     // "READS",

     // "REAL",

     // "RECURSIVE",

     // "REF",

     // "REFERENCES",

     // "REFERENCING",

     // "RELEASE",

     // "REPEAT",

     // "RESIGNAL",

     // "RESULT",

     // "RETURN",

     // "RETURNS",

     // "REVOKE",

     // "RIGHT",

     // "ROLLBACK",

     // "ROLLUP",

     // "ROW",

     // "ROWS",

     // "SAVEPOINT",

     // "SCOPE",

     // "SCROLL",

     // "SEARCH",

     // "SECOND",

     // "SELECT",

     // "SENSITIVE",

     // "SESSION_USER",

     // "SET",

     // "SIGNAL",

     // "SIMILAR",

     // "SMALLINT",

     // "SOME",

     // "SPECIFIC",

     // "SPECIFICTYPE",

     // "SQL",

     // "SQLEXCEPTION",

     // "SQLSTATE",

     // "SQLWARNING",

     // "START",

     // "STATIC",

     // "SUBMULTISET",

     // "SYMMETRIC",

     // "SYSTEM",

     // "SYSTEM_USER",

     // "TABLE",

     // "TABLESAMPLE",

     // "THEN",

     // "TIME",

     // "TIMESTAMP",

     // "TIMEZONE_HOUR",

     // "TIMEZONE_MINUTE",

     // "TO",

     // "TRAILING",

     // "TRANSLATION",

     // "TREAT",

     // "TRIGGER",

     // "TRUE",

     // "UNDO",

     // "UNION",

     // "UNIQUE",

     // "UNKNOWN",

     // "UNNEST",

     // "UNTIL",

     // "UPDATE",

             "USER",

             // "USING",

             // "VALUE",

             // "VALUES",

             // "VARCHAR",

             // "VARYING",

             // "WHEN",

             // "WHENEVER",

             // "WHERE",

             // "WHILE",

             // "WINDOW",

             // "WITH",

             // "WITHIN",

             // "WITHOUT",

             // "YEAR",

             // "ALIAS",

             // "AUTOCOMMIT",

             // "CACHED",

             // "CHECKPOINT",

             // "EXPLAIN",

             // "IGNORECASE",

             // "INDEX",

             // "LOGSIZE",

             // "MATCHED",

             // "MAXROWS",

             // "MEMORY",

             // "MINUS",

             // "NEXT",

             // "OPENBRACKET",

             "PASSWORD"

     // "PLAN",

     // "PROPERTY",

     // "READONLY",

     // "REFERENTIAL_INTEGRITY",

     // "RENAME",

     // "RESTART",

     // "SCRIPT",

     // "SCRIPTFORMAT",

     // "SEMICOLON",

     // "SEQUENCE",

     // "SHUTDOWN",

     // "SOURCE",

     // "TEMP",

     // "TEXT",

     // "VIEW",

     // "WRITE_DELAY",

     // "VAR_POP",

     // "VAR_SAMP",

     // "STDDEV_POP",

     // "STDDEV_SAMP",

     // "DEFRAG",

     // "INCREMENT",

     // "TOCHAR",

     // "DATABASE",

     // "SCHEMA",

     // "ROLE",

     // "DOW",

     // "INITIAL"

             );

     public boolean supportsInitiallyDeferrableColumns() {

         return false;

     }

     public String getCurrentDateTimeFunction() {

         if (currentDateTimeFunction != null) {

             return currentDateTimeFunction;

         }

         return "NOW()";

     }

 }