com.sap.cloud.mt.subscription.DbCredentialsBuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of multi-tenant-subscription Show documentation
Spring Boot Enablement Parent
There is a newer version: 3.3.1
Show newest version
/*******************************************************************************
 *   © 2019-2024 SAP SE or an SAP affiliate company. All rights reserved.
 ******************************************************************************/

package com.sap.cloud.mt.subscription;

import com.sap.cloud.mt.subscription.exceptions.InternalError;
import org.apache.commons.lang3.StringUtils;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

public class DbCredentialsBuilder {
	private static final String DBNAME = "dbname";
	private static final String DATABASE = "database";
	private static final String URI = "uri";
	private static final String JDBC = "jdbc:";
	private static final String POSTGRES = "postgres:";
	private static final String POSTGRESQL = "postgresql:";
	private static final String H2 = "h2";
	private static final String USERNAME = "username";
	private static final String PASSWORD = "password";
	private static final String HOSTNAME = "hostname";
	private static final String PORT = "port";
	private Map credentials = new HashMap<>();
	DbIdentifiers.DB db = DbIdentifiers.DB.NONE;

	private DbCredentialsBuilder() {
	}

	public DbCredentials build() throws InternalError {
		switch (db) {
			case POSTGRESQL:
				return createPostgresCredentials();
			case H2:
				return createH2Credentials();
			default:
				throw new InternalError("Don't support this DB");
		}

	}

	public static DbCredentialsBuilder create() {
		return new DbCredentialsBuilder();
	}

	public DbCredentialsBuilder db(DbIdentifiers.DB db) {
		//make sure not to overwrite a db type auto determined via credentials
		if ((this.db == null || this.db == DbIdentifiers.DB.NONE)
				&& (db != null && db != DbIdentifiers.DB.NONE)) {
			this.db = db;
		}
		return this;
	}

	public DbCredentialsBuilder credentials(Map credentials) {
		this.credentials = new HashMap<>(credentials);
		setDbType(credentials);
		return this;
	}

	private void setDbType(Map credentials) {
		if (credentials.containsKey(URI)) {
			String uri = (String) credentials.get(URI);
			if (StringUtils.isNotEmpty(uri)) {
				if (isDbType(uri, POSTGRES, POSTGRESQL)) {
					db = DbIdentifiers.DB.POSTGRESQL;
				} else if (isDbType(uri, H2)) {
					db = DbIdentifiers.DB.H2;
				}
			}
		}
	}

	private boolean isDbType(String uri, String... dbStartString) {
		return Arrays.stream(dbStartString).anyMatch(start -> uri.startsWith(JDBC + start) || uri.startsWith(start));
	}

	private DbCredentials createPostgresCredentials() throws InternalError {
		String username = (String) credentials.get(USERNAME);
		String password = (String) credentials.get(PASSWORD);
		String hostname = (String) credentials.get(HOSTNAME);
		String port = (String) credentials.get(PORT);
		String uri = (String) credentials.get(URI);
		String database = null;
		if (credentials.containsKey(DBNAME)) {
			database = (String) credentials.get(DBNAME);
		}
		if (credentials.containsKey(DATABASE)) {
			database = (String) credentials.get(DATABASE);
		}
		DbCredentialsPostgreSQL credentialsPostgreSQL = new DbCredentialsPostgreSQL(username, password, hostname, port, database, uri);
		if (StringUtils.isEmpty(credentialsPostgreSQL.getUser())) {
			throw new InternalError("No DB user specified");
		}
		if (StringUtils.isEmpty(credentialsPostgreSQL.getPassword())) {
			throw new InternalError("No DB password specified");
		}
		if (StringUtils.isEmpty(credentialsPostgreSQL.getUrl()) &&
				StringUtils.isEmpty(credentialsPostgreSQL.getHost())) {
			throw new InternalError("No DB uri and host specified");
		}
		return credentialsPostgreSQL;
	}

	private DbCredentials createH2Credentials() throws InternalError {
		String username = (String) credentials.get(USERNAME);
		String password = (String) credentials.get(PASSWORD);
		String database = null;
		if (credentials.containsKey(DBNAME)) {
			database = (String) credentials.get(DBNAME);
		}
		if (credentials.containsKey(DATABASE)) {
			database = (String) credentials.get(DATABASE);
		}
		DbCredentialsH2 credentialsH2 = new DbCredentialsH2(username, password, database);
		if (StringUtils.isEmpty(credentialsH2.getUser())) {
			throw new InternalError("No DB user specified");
		}
		return credentialsH2;
	}
}