R/zzz.R

In data-science-competitions/IEEE-CIS-Fraud-Detection: R Project Template for Analytic Applications such as Machine Learning Competitions

#nocov start
# .onAttach --------------------------------------------------------------------
.onAttach <- function(...) {
    source_path <- .find_CONFIGURATION_path()
    .load_config(file = source_path, config = c("default", "package"))
    return(invisible())
}

# High-level Functions ---------------------------------------------------------
.load_config <- function(value = NULL, config = Sys.getenv("R_CONFIG_ACTIVE", "default"),
                         file = Sys.getenv("R_CONFIG_FILE", "config.yml"), use_parent = TRUE){

    config <- unique(c("default", config))
    yaml_content <- .read_yaml_without_expression_evaluation(file)
    yaml_content <- yaml_content[config]
    yaml_content <- .remove_special_apostrophe_from_yaml(yaml_content)

    yaml_path <- tempfile()
    base::get("writeLines")(text = yaml_content, con = yaml_path)

    suppressWarnings(
        config::get(value = value, config = config, file = yaml_path, use_parent = use_parent)
    )
}

# Low-level Functions ----------------------------------------------------------
.remove_special_apostrophe_from_yaml <- function(yaml_content){
    stopifnot(is.list(yaml_content))
    target_path <- tempfile()
    yaml::write_yaml(yaml_content, target_path)
    config <- readLines(target_path)
    config <- gsub("'!expr", "!expr", config)
    config <- gsub("('$)", "", config)
    return(config)
}

.read_yaml_without_expression_evaluation <- function(file){
    yaml::read_yaml(
        file,
        eval.expr = FALSE,
        handlers = list(expr = function(x) {paste("!expr", x)})
    )
}

.find_CONFIGURATION_path <- function(){
    try_build_dir <- function(){
        list.files(path = ".",
                   pattern = "CONFIGURATION",
                   all.files = TRUE,
                   full.names = TRUE,
                   recursive = TRUE)[1]
    }

    try_package_dir <- function(){
        system.file("CONFIGURATION", package = utils::packageName())[1]
    }

    if(!is.na(try_build_dir())) return(try_build_dir())
    if(!is.na(try_package_dir())) return(try_package_dir())
    stop("Couldn't find CONFIGURATION")
}

.copy_CONFIGURATION_from_root_to_inst <- function(){
    source <- "CONFIGURATION"
    target <- file.path("inst", "CONFIGURATION")
    dir.create(dirname(target), showWarnings = FALSE, recursive = TRUE)
    file.copy(from = source, to = target, overwrite = TRUE)
}

.is_package_installed <- function(package){
    package %in% rownames(utils::installed.packages())
}
#nocov end