/*
 * Hibernate OGM, Domain model persistence for NoSQL datastores
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.ogm.jpa.impl;

import java.util.List;
import java.util.Map;

import javax.persistence.Cache;
import javax.persistence.EntityGraph;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceUnitUtil;
import javax.persistence.Query;
import javax.persistence.SynchronizationType;
import javax.persistence.criteria.CriteriaBuilder;
import javax.persistence.metamodel.EntityType;
import javax.persistence.metamodel.Metamodel;

import org.hibernate.SessionFactory;
import org.hibernate.engine.spi.SessionFactoryImplementor;
import org.hibernate.jpa.HibernateEntityManagerFactory;
import org.hibernate.jpa.internal.EntityManagerImpl;
import org.hibernate.ogm.hibernatecore.impl.OgmSessionFactoryImpl;

/**
 * An OGM {@code EntityManagerFactory} which delegates most work to the underlying ORM {@code EntityManagerFactory}.
 *
 * Note: Returns an {@code OgmEntityManager} to cope with query operations.
 *
 * @author Emmanuel Bernard <emmanuel@hibernate.org>
 */
public class OgmEntityManagerFactory implements HibernateEntityManagerFactory {

	private final HibernateEntityManagerFactory hibernateEmf;

	public OgmEntityManagerFactory(HibernateEntityManagerFactory hibernateEmf) {
		this.hibernateEmf = hibernateEmf;
	}

	@Override
	public EntityManager createEntityManager() {
		return new OgmEntityManager( this, (EntityManagerImpl) hibernateEmf.createEntityManager() );
	}

	@Override
	public EntityManager createEntityManager(Map map) {
		return new OgmEntityManager( this, (EntityManagerImpl) hibernateEmf.createEntityManager( map ) );
	}

	@Override
	public EntityManager createEntityManager(SynchronizationType synchronizationType) {
		return new OgmEntityManager(
				this, (EntityManagerImpl) hibernateEmf.createEntityManager( synchronizationType )
		);
	}

	@Override
	public EntityManager createEntityManager(SynchronizationType synchronizationType, Map map) {
		return new OgmEntityManager(
				this, (EntityManagerImpl) hibernateEmf.createEntityManager( synchronizationType, map )
		);
	}

	@Override
	public SessionFactoryImplementor getSessionFactory() {
		final SessionFactory sessionFactory = hibernateEmf.getSessionFactory();
		return new OgmSessionFactoryImpl( (SessionFactoryImplementor) sessionFactory );
	}

	@Override
	public void addNamedQuery(String name, Query query) {
		hibernateEmf.addNamedQuery( name, query );
	}

	@Override
	public <T> void addNamedEntityGraph(String graphName, EntityGraph<T> entityGraph) {
		throw new IllegalStateException( "Hibernate OGM does not support entity graphs" );
	}

	//Delegation

	@Override
	public CriteriaBuilder getCriteriaBuilder() {
		return hibernateEmf.getCriteriaBuilder();
	}

	@Override
	public Metamodel getMetamodel() {
		return hibernateEmf.getMetamodel();
	}

	@Override
	public boolean isOpen() {
		return hibernateEmf.isOpen();
	}

	@Override
	public void close() {
		hibernateEmf.close();
	}

	@Override
	public Map<String, Object> getProperties() {
		return hibernateEmf.getProperties();
	}

	@Override
	public Cache getCache() {
		return hibernateEmf.getCache();
	}

	@Override
	public PersistenceUnitUtil getPersistenceUnitUtil() {
		return hibernateEmf.getPersistenceUnitUtil();
	}

	@Override
	public EntityType<?> getEntityTypeByName(String entityName) {
		return hibernateEmf.getEntityTypeByName( entityName );
	}

	@Override
	public <T> T unwrap(Class<T> cls) {
		if ( cls != null && cls.isAssignableFrom( getClass() ) ) {
			@SuppressWarnings("unchecked")
			T result = (T) this;
			return result;
		}

		return hibernateEmf.unwrap( cls );
	}

	@Override
	public <T> List<EntityGraph<? super T>> findEntityGraphsByType(Class<T> entityClass) {
		return hibernateEmf.findEntityGraphsByType( entityClass );
	}

	@Override
	public String getEntityManagerFactoryName() {
		return hibernateEmf.getEntityManagerFactoryName();
	}
}