package org.apache.torque.engine.platform;

 /*

  * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE

  * file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file

  * to you under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the

  * License. You may obtain a copy of the License at

  * 

  * http://www.apache.org/licenses/LICENSE-2.0

  * 

  * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on

  * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the

  * specific language governing permissions and limitations under the License.

  */

 import java.sql.Connection;

 import java.sql.PreparedStatement;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import org.apache.commons.lang.StringUtils;

 import org.apache.torque.engine.database.model.Domain;

 import org.apache.torque.engine.database.model.SchemaType;

 /**

  * MySql Platform implementation.

  * 

  * @author <a href="mailto:mpoeschl@marmot.at">Martin Poeschl</a>

  * @version $Id: PlatformMysqlImpl.java,v 1.1.6.1 2008-04-18 17:04:37 jkeller Exp $

  */

 public class PlatformMysqlImpl extends PlatformDefaultImpl {

     /**

      * Default constructor.

      */

     public PlatformMysqlImpl() {

         super();

         initialize();

     }

     /**

      * Initializes db specific domain mapping.

      */

     private void initialize() {

         setSchemaDomainMapping(new Domain(SchemaType.NUMERIC, "DECIMAL"));

         setSchemaDomainMapping(new Domain(SchemaType.LONGVARCHAR, "MEDIUMTEXT"));

         setSchemaDomainMapping(new Domain(SchemaType.DATE, "DATETIME"));

         setSchemaDomainMapping(new Domain(SchemaType.BINARY, "BLOB"));

         setSchemaDomainMapping(new Domain(SchemaType.VARBINARY, "MEDIUMBLOB"));

         setSchemaDomainMapping(new Domain(SchemaType.LONGVARBINARY, "LONGBLOB"));

         setSchemaDomainMapping(new Domain(SchemaType.BLOB, "LONGBLOB"));

         setSchemaDomainMapping(new Domain(SchemaType.CLOB, "LONGTEXT"));

     }

     /**

      * @see Platform#getAutoIncrement()

      */

     public String getAutoIncrement() {

         return "AUTO_INCREMENT";

     }

     /**

      * @see Platform#hasSize(String)

      */

     public boolean hasSize(String sqlType) {

         return !("MEDIUMTEXT".equals(sqlType) || "LONGTEXT".equals(sqlType) || "BLOB".equals(sqlType)

                 || "MEDIUMBLOB".equals(sqlType) || "LONGBLOB".equals(sqlType));

     }

     protected String getBaseUrl(String url) {

         if (url == null) {

             return null;

         }

         url = url.trim();

         if (!url.startsWith("jdbc:mysql://")) {

             return null;

         }

         int count = StringUtils.countMatches(url, "/");

         if (count != 3) {

             return null;

         }

         int pos = url.lastIndexOf("/");

         return url.substring(0, pos + 1);

     }

     protected String getOptions(String url) {

         int pos = url.indexOf("?");

         if (pos == -1) {

             return "";

         }

         return url.substring(pos);

     }

     /**

      * jdbc:mysql://[host:port],[host:port]/[database][?property1][=value1][&property2][=value2]

      */

     @Override

     public String getServerUrl(String url) {

         String baseUrl = getBaseUrl(url);

         if (baseUrl == null) {

             return url;

         }

         String options = getOptions(url);

         return baseUrl + options;

     }

     @Override

     public String getSchemaName(String artifactId) {

         String s = super.getSchemaName(artifactId);

         return s.toLowerCase();

     }

     @Override

     public String filterInvalidDefaultValues(String defaultValue) {

         if (defaultValue == null) {

             return null;

         }

         defaultValue = defaultValue.replace("SYS_GUID()", "");

         defaultValue = defaultValue.replace("SYSDATE", "");

         defaultValue = defaultValue.replace("USERENV(\'SESSIONID\')", "");

         return defaultValue.trim();

     }

     @Override

     public Long getSequenceNextVal(Connection con, String schema, String sequenceName) {

         try {

             PreparedStatement ps = con

                     .prepareStatement("SELECT auto_increment FROM information_schema.tables WHERE table_schema = ? AND table_name = ?");

             Long nextVal = 0L;

             ps.setString(1, schema);

             ps.setString(2, sequenceName);

             ResultSet rs = ps.executeQuery();

             if (rs.next()) {

                 nextVal = rs.getLong(1);

             }

             rs.close();

             ps.close();

             System.out.println("Next Val for " + schema + "." + sequenceName + "=" + nextVal);

             return nextVal;

         } catch (SQLException ex) {

             System.err.println("Unable to extract sequence definition: " + schema + "." + sequenceName);

             ex.printStackTrace();

             return 0L;

         }

     }

     @Override

     public String getViewDefinition(Connection con, String schema, String viewName) {

         try {

             PreparedStatement ps = con

                     .prepareStatement("SELECT view_definition FROM information_schema.views WHERE table_schema = ? AND table_name = ?");

             String definition = "";

             ps.setString(1, schema);

             ps.setString(2, viewName);

             ResultSet rs = ps.executeQuery();

             if (rs.next()) {

                 definition = rs.getString(1);

             }

             rs.close();

             ps.close();

             return definition;

         } catch (SQLException ex) {

             System.err.println("Unable to extract view definition: " + schema + "." + viewName);

             ex.printStackTrace();

             return "";

         }

     }

 }