main.R

logger::log_threshold(Sys.getenv("LOG_LEVEL", "INFO"))
logger::log_formatter(logger::formatter_glue)

# set general i/o paths --------------------------------------------------------

scenario_preparation_inputs_path <- Sys.getenv(
  "SCENARIO_PREPARATION_INPUTS_PATH",
  "./inputs"
)

if (fs::dir_exists(scenario_preparation_inputs_path)) {
  logger::log_info("Setting scenario preparation inputs path: {scenario_preparation_inputs_path}")
} else {
  logger::log_error("Scenario preparation inputs path does not exist: {scenario_preparation_inputs_path}")
  stop()
}


scenario_preparation_outputs_path <- Sys.getenv(
  "SCENARIO_PREPARATION_OUTPUTS_PATH",
  "./outputs"
)

if (fs::dir_exists(scenario_preparation_outputs_path)) {
  logger::log_info("Setting scenario preparation outputs path: {scenario_preparation_outputs_path}")
} else {
  logger::log_error("Scenario preparation outputs path does not exist: {scenario_preparation_outputs_path}")
  stop()
}


# config -----------------------------------------------------------------------

logger::log_info("Loading configuration file.")

config_name <- Sys.getenv("R_CONFIG_ACTIVE")

logger::log_info("Getting config: {config_name}")

config <- config::get(
  file = "config.yml",
  config = config_name,
  use_parent = FALSE
)


# create timestamped output directory ------------------------------------------

system_timestamp <-
  format(
    Sys.time(),
    format = "%Y%m%d_T%H%M%SZ",
    tz = "UTC"
  )

scenario_preparation_outputs_path <-
  file.path(
    scenario_preparation_outputs_path,
    paste0(config_name, "_", system_timestamp)
  )

if (dir.exists(scenario_preparation_outputs_path)) {
  logger::log_warn("POTENTIAL DATA LOSS: Output directory already exists, and files may be overwritten ({scenario_preparation_outputs_path}).")
  warning("Output directory exists. Files may be overwritten.")
} else {
  logger::log_trace("Creating output directory: \"{scenario_preparation_outputs_path}\"")
  dir.create(scenario_preparation_outputs_path, recursive = TRUE)
}

logger::log_info("Files will be saved in directory: \"{scenario_preparation_outputs_path}\"")


# building scenarios -----------------------------------------------------------

logger::log_info("Determining scenarios to include.")

scenarios_to_include <- config$inherits

logger::log_info("Scenarios to be included: {scenarios_to_include}")

logger::log_info("Processing scenarios.")

for (scenario in scenarios_to_include) {
  source(paste0("process_scenario_", scenario, ".R"))
}

logger::log_info("Finished processing scenarios.")


# format and save scenarios for use in P4B -------------------------------------

logger::log_info("Determining scenarios to include.")

scenarios_to_include <- config$inherits

logger::log_info("Scenarios to be included: {scenarios_to_include}")

logger::log_info("Processing scenarios for P4B input.")

source("format_scenarios_for_p4b.R")