package net.mainseek.wlw.config;

import org.apache.commons.dbcp.BasicDataSource;
import org.apache.log4j.Logger;
import org.hibernate.SessionFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.orm.hibernate3.HibernateTransactionManager;
import org.springframework.orm.hibernate3.annotation.AnnotationSessionFactoryBean;

import javax.sql.DataSource;
import java.util.Properties;

@Configuration
public class HibernateConfig {

    private static final Logger LOG = Logger.getLogger(HibernateConfig.class);

    private static final String[] PACKAGES_TO_SCAN = {
            "net.mainseek.wlw.user.entity",
            "net.mainseek.wlw.location.entity",
            "net.mainseek.wlw.company.entity"
    };

    @Bean
    public SessionFactory sessionFactory(DataSource dataSource,
                                         @Value("${hibernate.dialect}") String dialect,
                                         @Value("${hibernate.hbm2ddl.auto}") String hbm2ddl) throws Exception {

        AnnotationSessionFactoryBean factory = new AnnotationSessionFactoryBean();
        factory.setDataSource(dataSource);
        factory.setPackagesToScan(PACKAGES_TO_SCAN);
        factory.setHibernateProperties(createHibernateProperties(dialect, hbm2ddl));
        factory.afterPropertiesSet();
        return factory.getObject();
    }

    private Properties createHibernateProperties(String dialect, String hbm2ddl) {
        Properties properties = new Properties();
        properties.setProperty("hibernate.dialect", dialect);
        properties.setProperty("hibernate.hbm2ddl.auto", hbm2ddl);
        return properties;
    }

    @Bean
    public DataSource dataSource(@Value("${jdbc.driver}") String driverClassName,
                                 @Value("${jdbc.url}") String url,
                                 @Value("${jdbc.username}") String username,
                                 @Value("${jdbc.password}") String password) {
        BasicDataSource dataSource = new BasicDataSource();
        LOG.debug("driverClassName = " + driverClassName);
        LOG.debug("url = " + url);
        LOG.debug("username = " + username);
        LOG.debug("password = " + password);
        dataSource.setDriverClassName(driverClassName);
        dataSource.setUrl(url);
        dataSource.setUsername(username);
        dataSource.setPassword(password);
        return dataSource;
    }

    @Bean
    public HibernateTransactionManager transactionManager(SessionFactory sessionFactory) {
        return new HibernateTransactionManager(sessionFactory);
    }
}