/**
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this file,
 * You can obtain one at http://mozilla.org/MPL/2.0/.
 */
package org.mifosplatform.infrastructure.core.boot.db;

import javax.sql.DataSource;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.jdbc.core.JdbcTemplate;
import org.springframework.stereotype.Service;

/**
 * Service which "fixes up" the schemaServerPort in the tenants table of the
 * mifosplatform-tenants schema. It sets it to the actual current port of
 * MariaDB4j, instead of the default 3306 which is hard-coded in the initial
 * set-up SQL script (V1__mifos-platform-shared-tenants.sql).
 *
 * This service is called by the TenantDatabaseUpgradeService
 * "just at the right time" during start-up, that is AFTER the initialization of
 * the mifosplatform-tenants but BEFORE Flyway runs on all tenant schemas.
 *
 * It's difficult to achieve the same goal elsewhere, because we cannot do this
 * e.g. in MariaDB4jSetupService, as that's too early
 * (TenantDatabaseUpgradeService has not yet run), nor e.g. in
 * TomcatJdbcDataSourcePerTenantService because in there we do not have access
 * to and should not depend on MariaDB4j configuration.
 */
@Service
public class TenantDataSourcePortFixService {
	private static final Logger logger = LoggerFactory.getLogger(TenantDataSourcePortFixService.class);

	/**
	 * While what this class does is convenient in 96% you may would like to
	 * completely disable it if you've hand tuned your tenants table for an
	 * advanced configuration and use e.g. different databases maybe on
	 * different hosts for different tenants.
	 */
    public final static String ENABLED = "mifos.tenantdb.fixup";
    @Value("${" + ENABLED + ":true}")
    private boolean enabled;

	// required=false is important here, because in
	// WebApplicationInitializerConfiguration for classic WAR there
	// is (intentionally) no MariaDB4j nor a DataSourceProperties
	// bean (because in the WAR we're using a DS from JNDI)
	private @Autowired(required = false) DataSourceProperties dsp;

	private JdbcTemplate jdbcTemplate;

    @Autowired
    public TenantDataSourcePortFixService(@Qualifier("tenantDataSourceJndi") final DataSource dataSource) {
        this.jdbcTemplate = new JdbcTemplate(dataSource);
    }

    public void fixUpTenantsSchemaServerPort() {
	if (!enabled)  {
		logger.info("No schema_server_port UPDATE made to tenant_server_connections table of the mifosplatform-tenants schema, because " + ENABLED + " = false");
		return;
	}
	if (dsp == null) {
		// we don't have any generic mechanism to know the DB port, given just a tenant DataSource
		logger.debug("No schema_server_port UPDATE made to tenant_server_connections table of the mifosplatform-tenants schema (because neither MariaDB4j nor our own Spring Boot DataSourceConfiguration is used in a traditional WAR)");
		return;
	}
		int r = jdbcTemplate
				.update("UPDATE tenant_server_connections SET schema_server = ?, schema_server_port = ?, schema_username = ?, schema_password = ?",
						dsp.getHost(), dsp.getPort(), dsp.getUsername(), dsp.getPassword());
	if ( r == 0 )
		logger.warn("UPDATE tenant_server_connections SET ... did not update ANY rows - something is probably wrong");
	else
			logger.info("Upated "
					+ r
					+ " rows in the tenant_server_connections table of the mifosplatform-tenants schema to the real current host: "
					+ dsp.getHost() + ", port: " + dsp.getPort());
    }

}