R/utils.R

In business-science/riingo: An R Interface to the 'Tiingo' Stock Price API

# ------------------------------------------------------------------------------
# Reexports



# ------------------------------------------------------------------------------
# Pure imports

# Colors for error messages
# green = user entry, yellow = correct entry
#' @importFrom crayon green
#' @importFrom crayon yellow
#' @importFrom crayon italic

# ------------------------------------------------------------------------------

# map_dfr(), map2_dfr(), and flatten_dfr() without a dplyr dependency

riingo_map_dfr <- function(.x, .f, ...) {
  .f <- purrr::as_mapper(.f, ...)
  res <- purrr::map(.x, .f, ...)
  vctrs::vec_rbind(!!!res)
}

riingo_map2_dfr <- function(.x, .y, .f, ...) {
  .f <- purrr::as_mapper(.f, ...)
  res <- purrr::map2(.x, .y, .f, ...)
  vctrs::vec_rbind(!!!res)
}

riingo_flatten_dfr <- function(x) {
  res <- purrr::flatten(x)
  vctrs::vec_rbind(!!!res)
}


# ------------------------------------------------------------------------------
# Error messages

glue_stop <- function(..., .sep = "", .envir = parent.frame()) {
  msg <- glue::glue(..., .sep, .envir = .envir)
  abort(msg)
}

validate_not_all_null <- function(x) {
  all_null <- all(purrr::map_lgl(x, is.null))

  if (all_null) {
    glue_stop("All tickers failed to download any data. Aborting.")
  }

  invisible(x)
}