/*
 * Copyright 2011 James Moger.
 *
 * Licensed 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.
 */

package com.iciql;

import java.sql.SQLException;

import org.postgresql.util.PGobject;

import com.iciql.Iciql.DataTypeAdapter;
import com.iciql.TableDefinition.IndexDefinition;
import com.iciql.util.StatementBuilder;

/**
 * PostgreSQL database dialect.
 */
public class SQLDialectPostgreSQL extends SQLDialectDefault {

	@Override
	public Class<? extends java.util.Date> getDateTimeClass() {
		return java.sql.Timestamp.class;
	}

	@Override
	public String convertSqlType(String sqlType) {
		if ("DOUBLE".equals(sqlType)) {
			return "DOUBLE PRECISION";
		} else if ("TINYINT".equals(sqlType)) {
			// PostgreSQL does not have a byte type
			return "SMALLINT";
		} else if ("CLOB".equals(sqlType)) {
			return "TEXT";
		} else if ("BLOB".equals(sqlType)) {
			return "BYTEA";
		}
		return sqlType;
	}

	@Override
	protected boolean prepareColumnDefinition(StatementBuilder buff, String dataType,
			boolean isAutoIncrement, boolean isPrimaryKey) {
		String convertedType = convertSqlType(dataType);
		if (isIntegerType(dataType)) {
			if (isAutoIncrement) {
				if ("BIGINT".equals(dataType)) {
					buff.append("BIGSERIAL");
				} else {
					buff.append("SERIAL");
				}
			} else {
				buff.append(convertedType);
			}
		} else {
			buff.append(convertedType);
		}
		return false;
	}

	@Override
	public void prepareCreateIndex(SQLStatement stat, String schemaName, String tableName,
			IndexDefinition index) {
		StatementBuilder buff = new StatementBuilder();
		buff.append("CREATE ");
		switch (index.type) {
		case UNIQUE:
			buff.append("UNIQUE ");
			break;
		case UNIQUE_HASH:
			buff.append("UNIQUE ");
			break;
		}
		buff.append("INDEX ");
		buff.append(index.indexName);
		buff.append(" ON ");
		buff.append(tableName);

		switch (index.type) {
		case HASH:
			buff.append(" USING HASH");
			break;
		case UNIQUE_HASH:
			buff.append(" USING HASH");
			break;
		}

		buff.append(" (");
		for (String col : index.columnNames) {
			buff.appendExceptFirst(", ");
			buff.append(prepareColumnName(col));
		}
		buff.append(") ");

		stat.setSQL(buff.toString().trim());
	}

	/**
	 * Handles transforming raw strings to/from the Postgres JSON data type.
	 */
	public class JsonStringAdapter implements DataTypeAdapter<String> {

		@Override
		public String getDataType() {
			return "json";
		}

		@Override
		public Class<String> getJavaType() {
			return String.class;
		}

		@Override
		public Object serialize(String value) {
			PGobject pg = new PGobject();
			pg.setType(getDataType());
			try {
				pg.setValue(value);
			} catch (SQLException e) {
				// not thrown on base PGobject
			}
			return pg;
		}

		@Override
		public String deserialize(Object value) {
			return value.toString();
		}
	}

	/**
	 * Handles transforming raw strings to/from the Postgres XML data type.
	 */
	public class XmlStringAdapter implements DataTypeAdapter<String> {

		@Override
		public String getDataType() {
			return "xml";
		}

		@Override
		public Class<String> getJavaType() {
			return String.class;
		}

		@Override
		public Object serialize(String value) {
			PGobject pg = new PGobject();
			pg.setType(getDataType());
			try {
				pg.setValue(value);
			} catch (SQLException e) {
				// not thrown on base PGobject
			}
			return pg;
		}

		@Override
		public String deserialize(Object value) {
			return value.toString();
		}
	}
}