R/database.R

In shunsambongi/dbiconf: DBI Configuration

methods::setClass("dbiconf_database",
  contains = "DBIConnector",
  slots = c(".name" = "character"),
  prototype = list(".name" = NA_character_)
)

methods::setValidity("dbiconf_database", function(object) {
  if (!rlang::is_string(object@.name)) {
    "@.name must be a string"
  } else {
    TRUE
  }
})

methods::setMethod("show", "dbiconf_database", function(object) {
  cat(sprintf("[database.%s]\n", crayon::bold(format_toml_key(object@.name))))

  drv <- object@.drv
  if (methods::is(drv, "dbiconf_driver")) {
    drv <- format_database_param(drv@driver)
  } else {
    drv <- format(drv)
  }
  cat(crayon::magenta("_driver"), " = ", drv, "\n", sep = "")

  args <- object@.conn_args
  if (rlang::is_empty(args)) {
    return()
  }

  nms <- rlang::names2(args)
  vals <- purrr::map(args, format_database_param)
  cat(paste0(crayon::cyan(nms), " = ", vals, collapse = "\n"), "\n", sep = "")
})

format_database_param <- function(x) {
  out <- format_toml(x)
  template <- attr(x, "template", exact = TRUE)
  if (!is.null(template)) {
    template <- format_toml_key(template)
    out <- paste(out, crayon::silver(sprintf("# [template.%s]", template)))
  }
  out
}

database_args <- function(database) {
  purrr::map_if(database@.conn_args, rlang::is_function, rlang::exec)
}