From c56e7a962407cd7a80fe8568267a91258f535586 Mon Sep 17 00:00:00 2001
From: Dylan Hall <dehall@mitre.org>
Date: Mon, 8 Jul 2024 07:33:45 -0400
Subject: [PATCH] Fix issue with setting config via CLI

---
 src/main/java/App.java                        | 26 ++++++++++++++++---
 .../org/mitre/synthea/export/Exporter.java    |  2 +-
 2 files changed, 23 insertions(+), 5 deletions(-)

diff --git a/src/main/java/App.java b/src/main/java/App.java
index 423d9e2b90..10cf7171ed 100644
--- a/src/main/java/App.java
+++ b/src/main/java/App.java
@@ -66,6 +66,7 @@ public static void main(String[] args) throws Exception {
 
     boolean validArgs = true;
     boolean overrideFutureDateError = false;
+    boolean reloadConfig = false;
     if (args != null && args.length > 0) {
       try {
         Queue<String> argsQ = new LinkedList<String>(Arrays.asList(args));
@@ -104,6 +105,7 @@ public static void main(String[] args) throws Exception {
           } else if (currArg.equalsIgnoreCase("-p")) {
             String value = argsQ.poll();
             options.population = Integer.parseInt(value);
+            Config.set("generate.default_population", value);
           } else if (currArg.equalsIgnoreCase("-o")) {
             String value = argsQ.poll();
             options.overflow = Boolean.parseBoolean(value);
@@ -132,10 +134,7 @@ public static void main(String[] args) throws Exception {
             String value = argsQ.poll();
             File configFile = new File(value);
             Config.load(configFile);
-            // Any options that are automatically set by reading the configuration
-            // file during options initialization need to be reset here.
-            options.population = Config.getAsInteger("generate.default_population", 1);
-            options.threadPoolSize = Config.getAsInteger("generate.thread_pool_size", -1);
+            reloadConfig = true;
           } else if (currArg.equalsIgnoreCase("-d")) {
             String value = argsQ.poll();
             File localModuleDir = new File(value);
@@ -252,6 +251,7 @@ public static void main(String[] args) throws Exception {
             }
 
             Config.set(configSetting, value);
+            reloadConfig = true;
           } else if (options.state == null) {
             options.state = currArg;
           } else {
@@ -266,12 +266,30 @@ public static void main(String[] args) throws Exception {
       }
     }
 
+    if (reloadConfig) {
+      resetOptionsFromConfig(options, exportOptions);
+    }
+
     if (validArgs && validateConfig(options, overrideFutureDateError)) {
       Generator generator = new Generator(options, exportOptions);
       generator.run();
     }
   }
 
+  /**
+   * Reset the fields of the provided options to the current values in the Config.
+   */
+  private static void resetOptionsFromConfig(Generator.GeneratorOptions options,
+		  Exporter.ExporterRuntimeOptions exportOptions) {
+      // Any options that are automatically set by reading the configuration
+      // file during options initialization need to be reset here.
+      options.population = Config.getAsInteger("generate.default_population", 1);
+      options.threadPoolSize = Config.getAsInteger("generate.thread_pool_size", -1);
+
+      exportOptions.yearsOfHistory = Config.getAsInteger("exporter.years_of_history", 10);
+      exportOptions.terminologyService = !Config.get("generate.terminology_service_url", "").isEmpty();
+  }
+
   private static boolean validateConfig(Generator.GeneratorOptions options,
           boolean overrideFutureDateError) {
     boolean valid = true;
diff --git a/src/main/java/org/mitre/synthea/export/Exporter.java b/src/main/java/org/mitre/synthea/export/Exporter.java
index 26faa5b44d..f6c3e0e641 100644
--- a/src/main/java/org/mitre/synthea/export/Exporter.java
+++ b/src/main/java/org/mitre/synthea/export/Exporter.java
@@ -109,7 +109,7 @@ public static class ExporterRuntimeOptions {
     private List<Mapping> flexporterMappings;
 
     public ExporterRuntimeOptions() {
-      yearsOfHistory = Integer.parseInt(Config.get("exporter.years_of_history"));
+      yearsOfHistory = Config.getAsInteger("exporter.years_of_history", 10);
     }
 
     /**