package com.saas.shared.config;

import com.saas.shared.datasource.DynamicHikariDataSourceManager;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
import org.springframework.orm.jpa.JpaTransactionManager;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import javax.sql.DataSource;
import jakarta.persistence.EntityManagerFactory;
import java.util.HashMap;
import java.util.Map;

/**
 * Admin JPA Configuration
 * 
 * Configures a separate EntityManagerFactory for admin entities only.
 * This ensures admin tables (User, Tenant, Role, Permission, etc.)
 * are created only in saas_db, not in tenant databases.
 * 
 * Package scope: com.saas.admin.entity + com.saas.shared.audit
 */
@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(basePackages = { "com.saas.admin.repository", "com.saas.shared.audit",
        "com.saas.subscription.repository" }, entityManagerFactoryRef = "adminEntityManagerFactory", transactionManagerRef = "adminTransactionManager")
public class AdminJpaConfig {

    @Autowired
    private DynamicHikariDataSourceManager dataSourceManager;

    /**
     * Admin EntityManagerFactory - scans ONLY admin entities
     */
    @Primary
    @Bean(name = "adminEntityManagerFactory")
    public LocalContainerEntityManagerFactoryBean adminEntityManagerFactory() {

        DataSource adminDataSource = dataSourceManager.getAdminDataSource();

        LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
        em.setDataSource(adminDataSource);
        em.setPackagesToScan("com.saas.admin.entity", "com.saas.shared.audit", "com.saas.subscription.entity");
        em.setPersistenceUnitName("adminPU");

        HibernateJpaVendorAdapter vendorAdapter = new HibernateJpaVendorAdapter();
        vendorAdapter.setShowSql(false);
        vendorAdapter.setGenerateDdl(true);
        em.setJpaVendorAdapter(vendorAdapter);

        Map<String, Object> properties = new HashMap<>();
        properties.put("hibernate.hbm2ddl.auto", "update");
        properties.put("hibernate.dialect", "org.hibernate.dialect.MariaDBDialect");
        properties.put("hibernate.format_sql", true);
        properties.put("hibernate.physical_naming_strategy",
                "org.hibernate.boot.model.naming.CamelCaseToUnderscoresNamingStrategy");
        em.setJpaPropertyMap(properties);

        return em;
    }

    @Primary
    @Bean(name = "adminTransactionManager")
    public PlatformTransactionManager adminTransactionManager(
            @Qualifier("adminEntityManagerFactory") EntityManagerFactory adminEntityManagerFactory) {
        return new JpaTransactionManager(adminEntityManagerFactory);
    }
}