package org.argeo.slc.hibernate.unit;

import java.sql.Connection;
import java.util.List;
import java.util.Properties;

import org.hibernate.cfg.Configuration;
import org.hibernate.cfg.Environment;
import org.hibernate.tool.hbm2ddl.SchemaExport;

/**
 * Creates a relational data model from Hibernate mapping files. The benefit of
 * this class is to be able to use Hibernate to have test data which are
 * independent from the type of database used.
 */
public class DbModel {
	private String dialect;
	private List<String> mappings;

	/** Sets the Hibernate dialect to use. */
	public void setDialect(String dialect) {
		this.dialect = dialect;
	}

	/** Sets the list of mappings to consider. */
	public void setMappings(List<String> mappings) {
		this.mappings = mappings;
	}

	/**
	 * Creates an Hibernate schema export tool, in order to create the
	 * underlying datamodel.
	 */
	public SchemaExport createSchemaExport(Connection connection) {
		Configuration configuration = new Configuration();
		Properties properties = new Properties();
		properties.setProperty(Environment.DIALECT, dialect);
		properties.setProperty(Environment.HBM2DDL_AUTO, "create");
		configuration.setProperties(properties);

		for (String mapping : mappings) {
			configuration.addResource(mapping.trim());
		}

		return new SchemaExport(configuration, connection);
	}
}