EDIT

/**

* Copyright (C) 2007 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.hibernate;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Types;

import org.apache.log4j.Logger;

import org.hibernate.HibernateException;

import org.hibernate.engine.spi.SessionImplementor;

import org.hibernate.type.StandardBasicTypes;

import org.hibernate.usertype.UserType;

import org.jadira.usertype.dateandtime.shared.spi.AbstractUserType;

import org.joda.time.DateTimeFieldType;

import org.joda.time.Partial;

/**

 * Persist {@link org.joda.time.Partial} via hibernate.

 * This is a preliminary implementation that fulfills the needs of CDM but does not fully store a Partial.

 * Only year, month and day is stored. Since 5.0 also hour and minute is supported.

 *

 * @author a.mueller

 * @since 11.11.2008

 */

public class PartialUserType extends AbstractUserType implements UserType /* extends AbstractSingleColumnUserType<Partial, String, ColumnMapper<Partial,String>> implements UserType */ {

	private static final long serialVersionUID = -5323104403077597869L;

	private static final Logger logger = Logger.getLogger(PartialUserType.class);

	//not required

	public final static PartialUserType INSTANCE = new PartialUserType();

	private static final int[] SQL_TYPES = new int[]{

	    Types.VARCHAR,

	};

	@Override

	public Partial nullSafeGet(ResultSet rs, String[] names, SessionImplementor session, Object owner)

			throws HibernateException, SQLException {

		String partial = (String)StandardBasicTypes.STRING.nullSafeGet(rs, names, session, owner);

		Partial result = new Partial();

		if (partial == null || "00000000".equals(partial) || "0000000000000".equals(partial)) {

			return null;

		}else if (partial.length() != 8 &&  partial.length() != 13){

		    throw new HibernateException("Format for Partial not supported. Length mus be 8 or 13: " + partial);

		}

		Integer year = Integer.valueOf(partial.substring(0,4));

		Integer month = Integer.valueOf(partial.substring(4,6));

		Integer day = Integer.valueOf(partial.substring(6,8));

		Integer hour = null;

		Integer minute = null;

		if (partial.length() == 13){

	        hour = Integer.valueOf(partial.substring(9,11));

            minute = Integer.valueOf(partial.substring(11,13));

		}

		if (year != 0){

			result = result.with(DateTimeFieldType.year(), year);

		}

		if (month != 0){

			result = result.with(DateTimeFieldType.monthOfYear(), month);

		}

		if (day != 0){

			result = result.with(DateTimeFieldType.dayOfMonth(), day);

		}

	    if (hour != null){

	        result = result.with(DateTimeFieldType.hourOfDay(), hour);

	    }

        if (minute != null){

            result = result.with(DateTimeFieldType.minuteOfHour(), minute);

        }

        return isEmptyOrNull(result)? null:result;

	}

    private boolean isEmptyOrNull(Partial partial) {

        return partial == null ? true : partial.getValues().length == 0;

    }

    @Override

	public void nullSafeSet(PreparedStatement preparedStatement, Object value, int index,

			SessionImplementor session) throws HibernateException, SQLException {

		if (isEmptyOrNull((Partial)value)){

			StandardBasicTypes.STRING.nullSafeSet(preparedStatement, null, index, session);

		}else {

			Partial p = ((Partial) value);

			StandardBasicTypes.STRING.nullSafeSet(preparedStatement, partialToString(p), index, session);

		}

	}

	/**

	 * @param p

	 * @return an ISO 8601 like time representations of the form yyyyMMdd

	 */

	public static String partialToString(Partial p) {

		//FIXME reduce code by use org.joda.time.format.ISODateTimeFormat.basicDate() instead ?

		//      for a date with unknown day this will produce e.g. 195712??

		//

		String strYear = getNullFilledString(p, DateTimeFieldType.year(),4);

		String strMonth = getNullFilledString(p, DateTimeFieldType.monthOfYear(),2);

		String strDay = getNullFilledString(p, DateTimeFieldType.dayOfMonth(),2);

		String strHour = getNullFilledString(p, DateTimeFieldType.hourOfDay(),2);

		String strMinute = getNullFilledString(p, DateTimeFieldType.minuteOfHour(),2);

		boolean timeExists = timeExists(p);

        String result = strYear + strMonth + strDay;

        if (timeExists) {

            result = result + "_" + strHour + strMinute;

        }

        return result;

	}

    private static boolean timeExists(Partial partial) {

        return partial.isSupported(DateTimeFieldType.hourOfDay()) ||

                partial.isSupported(DateTimeFieldType.minuteOfHour());

    }

    private static String getNullFilledString(Partial partial, DateTimeFieldType type, int count){

		String nul = "0000000000";

		if (! partial.isSupported(type)){

			return nul.substring(0, count);

		}else{

			int value = partial.get(type);

			String result = String.valueOf(value);

			if (result.length() > count){

				logger.error("value to long");

				result = result.substring(0, count);

			}else if (result.length() < count){

				result = nul.substring(0, count - result.length()) +  result;

			}

			return result;

		}

	}

    @Override

    public Object deepCopy(Object value) throws HibernateException {

        if (value == null) {

            return null;

        }

        return value;

    }

	@Override

	public int[] sqlTypes() {

		return SQL_TYPES;

	}

	@Override

	public Class returnedClass() {

		return Partial.class;

	}

}