/*
 * Copyright (c) 2010-2013 Evolveum
 *
 * 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.evolveum.midpoint.repo.sql.util;

import com.evolveum.midpoint.init.RepositoryFactory;
import com.evolveum.midpoint.repo.sql.SqlRepositoryConfiguration;
import com.evolveum.midpoint.repo.sql.testing.TestSqlRepositoryFactory;
import com.evolveum.midpoint.util.logging.Trace;
import com.evolveum.midpoint.util.logging.TraceManager;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;

/**
 * This bean post processor is only used in {@link com.evolveum.midpoint.repo.sql.DataSourceTest} because that
 * test class needs hibernate property hibernate.hbm2ddl.auto set to update. Test configuration can be overridden
 * through system option -Dconfig=<properties file> so this processor guarantees that hbm2ddl is always set to
 * "update".
 *
 * @author lazyman
 */
public class DataSourceTestBeanPostprocessor implements BeanPostProcessor, ApplicationContextAware {

    private static final Trace LOGGER = TraceManager.getTrace(DataSourceTestBeanPostprocessor.class);

    private ApplicationContext context;

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
        if (!(bean instanceof RepositoryFactory)) {
            return bean;
        }

        TestSqlRepositoryFactory factory = context.getBean("testSqlRepositoryFactory", TestSqlRepositoryFactory.class);
        SqlRepositoryConfiguration config = factory.getSqlConfiguration();

        if (!config.isUsingH2()) {
            return bean;
        }

        System.out.println("Changing hibernate.hbm2ddl.auto to update");
        LOGGER.info("Changing hibernate.hbm2ddl.auto to update");

        config.setHibernateHbm2ddl("update");

        return bean;
    }

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.context = applicationContext;
    }
}