PRDONE-94: updated to use interpreter for getting environment credentials, updated interpreter to load Dotenv files as environment overrides

This commit is contained in:
Tim Chamberlain
2022-08-31 12:05:35 -05:00
parent 4bf1fe8638
commit f08ffe691f
5 changed files with 95 additions and 37 deletions

View File

@ -23,9 +23,12 @@ package com.kingsrook.qqq.backend.core.instances;
import java.lang.reflect.Method; import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Locale; import java.util.Locale;
import java.util.Map; import java.util.Map;
import com.kingsrook.qqq.backend.core.exceptions.QException; import com.kingsrook.qqq.backend.core.exceptions.QException;
import io.github.cdimascio.dotenv.Dotenv;
import io.github.cdimascio.dotenv.DotenvEntry;
import org.apache.logging.log4j.LogManager; import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger; import org.apache.logging.log4j.Logger;
@ -45,7 +48,22 @@ public class QMetaDataVariableInterpreter
{ {
private static final Logger LOG = LogManager.getLogger(QMetaDataVariableInterpreter.class); private static final Logger LOG = LogManager.getLogger(QMetaDataVariableInterpreter.class);
private Map<String, String> customEnvironment; private Map<String, String> environmentOverrides;
/*******************************************************************************
**
*******************************************************************************/
public QMetaDataVariableInterpreter()
{
environmentOverrides = new HashMap<>();
Dotenv dotenv = Dotenv.configure().ignoreIfMissing().load();
for(DotenvEntry e : dotenv.entries())
{
environmentOverrides.put(e.getKey(), e.getValue());
}
}
@ -124,7 +142,7 @@ public class QMetaDataVariableInterpreter
if(value.startsWith(envPrefix) && value.endsWith("}")) if(value.startsWith(envPrefix) && value.endsWith("}"))
{ {
String envVarName = value.substring(envPrefix.length()).replaceFirst("}$", ""); String envVarName = value.substring(envPrefix.length()).replaceFirst("}$", "");
String envValue = getEnvironment().get(envVarName); String envValue = getEnvironmentVariable(envVarName);
return (envValue); return (envValue);
} }
@ -149,13 +167,13 @@ public class QMetaDataVariableInterpreter
/******************************************************************************* /*******************************************************************************
** Setter for customEnvironment - protected - meant to be called (at least at this ** Setter for environmentOverrides - protected - meant to be called (at least at this
** time), only in unit test ** time), only in unit test
** **
*******************************************************************************/ *******************************************************************************/
protected void setCustomEnvironment(Map<String, String> customEnvironment) protected void setEnvironmentOverrides(Map<String, String> environmentOverrides)
{ {
this.customEnvironment = customEnvironment; this.environmentOverrides = environmentOverrides;
} }
@ -163,13 +181,13 @@ public class QMetaDataVariableInterpreter
/******************************************************************************* /*******************************************************************************
** **
*******************************************************************************/ *******************************************************************************/
private Map<String, String> getEnvironment() private String getEnvironmentVariable(String key)
{ {
if(this.customEnvironment != null) if(this.environmentOverrides.containsKey(key))
{ {
return (this.customEnvironment); return (this.environmentOverrides.get(key));
} }
return System.getenv(); return System.getenv(key);
} }
} }

View File

@ -102,7 +102,7 @@ class QMetaDataVariableInterpreterTest
QMetaDataVariableInterpreter secretReader = new QMetaDataVariableInterpreter(); QMetaDataVariableInterpreter secretReader = new QMetaDataVariableInterpreter();
String key = "CUSTOM_PROPERTY"; String key = "CUSTOM_PROPERTY";
String value = "ABCD-9876"; String value = "ABCD-9876";
secretReader.setCustomEnvironment(Map.of(key, value)); secretReader.setEnvironmentOverrides(Map.of(key, value));
assertNull(secretReader.interpret(null)); assertNull(secretReader.interpret(null));
assertEquals("foo", secretReader.interpret("foo")); assertEquals("foo", secretReader.interpret("foo"));

View File

@ -50,6 +50,48 @@ public class S3BackendMetaData extends AbstractFilesystemBackendMetaData
/*******************************************************************************
** Fluent setter for basePath
**
*******************************************************************************/
@SuppressWarnings("unchecked")
@Override
public S3BackendMetaData withBasePath(String basePath)
{
setBasePath(basePath);
return this;
}
/*******************************************************************************
** Fluent setter for name
**
*******************************************************************************/
@SuppressWarnings("unchecked")
@Override
public S3BackendMetaData withName(String name)
{
setName(name);
return this;
}
/*******************************************************************************
** Fluent setter for backendType
**
*******************************************************************************/
@SuppressWarnings("unchecked")
@Override
public S3BackendMetaData withBackendType(String backendType)
{
setBackendType(backendType);
return this;
}
/******************************************************************************* /*******************************************************************************
** Getter for bucketName ** Getter for bucketName
** **

View File

@ -25,9 +25,8 @@ package com.kingsrook.qqq.backend.module.rdbms.jdbc;
import java.sql.Connection; import java.sql.Connection;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.Collections; import java.util.Collections;
import java.util.Objects; import com.kingsrook.qqq.backend.core.instances.QMetaDataVariableInterpreter;
import com.kingsrook.qqq.backend.module.rdbms.model.metadata.RDBMSBackendMetaData; import com.kingsrook.qqq.backend.module.rdbms.model.metadata.RDBMSBackendMetaData;
import io.github.cdimascio.dotenv.Dotenv;
import org.junit.jupiter.api.Disabled; import org.junit.jupiter.api.Disabled;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertNotNull; import static org.junit.jupiter.api.Assertions.assertNotNull;
@ -103,13 +102,13 @@ class ConnectionManagerTest
private RDBMSBackendMetaData getAuroraBacked() private RDBMSBackendMetaData getAuroraBacked()
{ {
Dotenv dotenv = Dotenv.configure().ignoreIfMissing().load(); QMetaDataVariableInterpreter interpreter = new QMetaDataVariableInterpreter();
String vendor = (dotenv.get("RDBMS_VENDOR") != null) ? dotenv.get("RDBMS_VENDOR") : System.getenv("RDBMS_VENDOR"); String vendor = interpreter.interpret("${env.RDBMS_VENDOR}");
String hostname = (dotenv.get("RDBMS_HOSTNAME") != null) ? dotenv.get("RDBMS_HOSTNAME") : System.getenv("RDBMS_HOSTNAME"); String hostname = interpreter.interpret("${env.RDBMS_HOSTNAME}");
Integer port = (dotenv.get("RDBMS_PORT") != null) ? Integer.valueOf(Objects.requireNonNull(dotenv.get("RDBMS_PORT"))) : Integer.valueOf(System.getenv("RDBMS_PORT")); Integer port = Integer.valueOf(interpreter.interpret("${env.RDBMS_PORT}"));
String databaseName = (dotenv.get("RDBMS_DATABASE_NAME") != null) ? dotenv.get("RDBMS_DATABASE_NAME") : System.getenv("RDBMS_DATABASE_NAME"); String databaseName = interpreter.interpret("${env.RDBMS_DATABASE_NAME}");
String userName = (dotenv.get("RDBMS_USERNAME") != null) ? dotenv.get("RDBMS_USERNAME") : System.getenv("RDBMS_USERNAME"); String username = interpreter.interpret("${env.RDBMS_USERNAME}");
String password = (dotenv.get("RDBMS_PASSWORD") != null) ? dotenv.get("RDBMS_PASSWORD") : System.getenv("RDBMS_PASSWORD"); String password= interpreter.interpret("${env.RDBMS_PASSWORD}");
return new RDBMSBackendMetaData() return new RDBMSBackendMetaData()
.withName("aurora-test") .withName("aurora-test")
@ -117,7 +116,7 @@ class ConnectionManagerTest
.withHostName(hostname) .withHostName(hostname)
.withPort(port) .withPort(port)
.withDatabaseName(databaseName) .withDatabaseName(databaseName)
.withUsername(userName) .withUsername(username)
.withPassword(password); .withPassword(password);
} }
} }

View File

@ -23,13 +23,13 @@ package com.kingsrook.sampleapp;
import java.util.List; import java.util.List;
import java.util.Objects;
import com.amazonaws.regions.Regions; import com.amazonaws.regions.Regions;
import com.kingsrook.qqq.backend.core.actions.dashboard.QuickSightChartRenderer; import com.kingsrook.qqq.backend.core.actions.dashboard.QuickSightChartRenderer;
import com.kingsrook.qqq.backend.core.actions.processes.BackendStep; import com.kingsrook.qqq.backend.core.actions.processes.BackendStep;
import com.kingsrook.qqq.backend.core.exceptions.QException; import com.kingsrook.qqq.backend.core.exceptions.QException;
import com.kingsrook.qqq.backend.core.exceptions.QValueException; import com.kingsrook.qqq.backend.core.exceptions.QValueException;
import com.kingsrook.qqq.backend.core.instances.QInstanceEnricher; import com.kingsrook.qqq.backend.core.instances.QInstanceEnricher;
import com.kingsrook.qqq.backend.core.instances.QMetaDataVariableInterpreter;
import com.kingsrook.qqq.backend.core.model.actions.processes.RunBackendStepInput; import com.kingsrook.qqq.backend.core.model.actions.processes.RunBackendStepInput;
import com.kingsrook.qqq.backend.core.model.actions.processes.RunBackendStepOutput; import com.kingsrook.qqq.backend.core.model.actions.processes.RunBackendStepOutput;
import com.kingsrook.qqq.backend.core.model.metadata.QAuthenticationType; import com.kingsrook.qqq.backend.core.model.metadata.QAuthenticationType;
@ -63,7 +63,6 @@ import com.kingsrook.qqq.backend.module.filesystem.local.model.metadata.Filesyst
import com.kingsrook.qqq.backend.module.filesystem.local.model.metadata.FilesystemTableBackendDetails; import com.kingsrook.qqq.backend.module.filesystem.local.model.metadata.FilesystemTableBackendDetails;
import com.kingsrook.qqq.backend.module.rdbms.model.metadata.RDBMSBackendMetaData; import com.kingsrook.qqq.backend.module.rdbms.model.metadata.RDBMSBackendMetaData;
import com.kingsrook.sampleapp.dashboard.widgets.PersonsByCreateDateBarChart; import com.kingsrook.sampleapp.dashboard.widgets.PersonsByCreateDateBarChart;
import io.github.cdimascio.dotenv.Dotenv;
/******************************************************************************* /*******************************************************************************
@ -135,11 +134,11 @@ public class SampleMetaDataProvider
.withName(PersonsByCreateDateBarChart.class.getSimpleName()) .withName(PersonsByCreateDateBarChart.class.getSimpleName())
.withCodeReference(new QCodeReference(PersonsByCreateDateBarChart.class, null))); .withCodeReference(new QCodeReference(PersonsByCreateDateBarChart.class, null)));
Dotenv dotenv = Dotenv.configure().ignoreIfMissing().load(); QMetaDataVariableInterpreter interpreter = new QMetaDataVariableInterpreter();
String accountId = (dotenv.get("QUICKSIGHT_ACCOUNT_ID") != null) ? dotenv.get("QUICKSIGHT_ACCOUNT_ID") : System.getenv("QUICKSIGHT_ACCOUNT_ID"); String accountId = interpreter.interpret("${env.QUICKSIGHT_ACCOUNT_ID}");
String accessKey = (dotenv.get("QUICKSIGHT_ACCESS_KEY") != null) ? dotenv.get("QUICKSIGHT_ACCESS_KEY") : System.getenv("QUICKSIGHT_ACCESS_KEY"); String accessKey = interpreter.interpret("${env.QUICKSIGHT_ACCESS_KEY}");
String secretKey = (dotenv.get("QUICKSIGHT_SECRET_KEY") != null) ? dotenv.get("QUICKSIGHT_SECRET_KEY") : System.getenv("QUICKSIGHT_SECRET_KEY"); String secretKey = interpreter.interpret("${env.QUICKSIGHT_SECRET_KEY}");
String userArn = (dotenv.get("QUICKSIGHT_USER_ARN") != null) ? dotenv.get("QUICKSIGHT_USER_ARN") : System.getenv("QUICKSIGHT_USER_ARN"); String userArn = interpreter.interpret("${env.QUICKSIGHT_USER_ARN}");
QWidgetMetaDataInterface quickSightChartMetaData = new QuickSightChartMetaData() QWidgetMetaDataInterface quickSightChartMetaData = new QuickSightChartMetaData()
.withAccountId(accountId) .withAccountId(accountId)
@ -211,13 +210,13 @@ public class SampleMetaDataProvider
{ {
if(USE_MYSQL) if(USE_MYSQL)
{ {
Dotenv dotenv = Dotenv.configure().ignoreIfMissing().load(); QMetaDataVariableInterpreter interpreter = new QMetaDataVariableInterpreter();
String vendor = (dotenv.get("RDBMS_VENDOR") != null) ? dotenv.get("RDBMS_VENDOR") : System.getenv("RDBMS_VENDOR"); String vendor = interpreter.interpret("${env.RDBMS_VENDOR}");
String hostname = (dotenv.get("RDBMS_HOSTNAME") != null) ? dotenv.get("RDBMS_HOSTNAME") : System.getenv("RDBMS_HOSTNAME"); String hostname = interpreter.interpret("${env.RDBMS_HOSTNAME}");
Integer port = (dotenv.get("RDBMS_PORT") != null) ? Integer.valueOf(Objects.requireNonNull(dotenv.get("RDBMS_PORT"))) : Integer.valueOf(System.getenv("RDBMS_PORT")); Integer port = Integer.valueOf(interpreter.interpret("${env.RDBMS_PORT}"));
String databaseName = (dotenv.get("RDBMS_DATABASE_NAME") != null) ? dotenv.get("RDBMS_DATABASE_NAME") : System.getenv("RDBMS_DATABASE_NAME"); String databaseName = interpreter.interpret("${env.RDBMS_DATABASE_NAME}");
String userName = (dotenv.get("RDBMS_USERNAME") != null) ? dotenv.get("RDBMS_USERNAME") : System.getenv("RDBMS_USERNAME"); String username = interpreter.interpret("${env.RDBMS_USERNAME}");
String password = (dotenv.get("RDBMS_PASSWORD") != null) ? dotenv.get("RDBMS_PASSWORD") : System.getenv("RDBMS_PASSWORD"); String password= interpreter.interpret("${env.RDBMS_PASSWORD}");
return new RDBMSBackendMetaData() return new RDBMSBackendMetaData()
.withName(RDBMS_BACKEND_NAME) .withName(RDBMS_BACKEND_NAME)
@ -225,7 +224,7 @@ public class SampleMetaDataProvider
.withHostName(hostname) .withHostName(hostname)
.withPort(port) .withPort(port)
.withDatabaseName(databaseName) .withDatabaseName(databaseName)
.withUsername(userName) .withUsername(username)
.withPassword(password); .withPassword(password);
} }
else else