R/S3--aaa-setup.R

In schloerke/gqlr: 'GraphQL' Server in R

# https://github.com/hadley/adv-r/blob/master/S3.Rmd
# If you're implementing more complicated print() methods, it's a better idea to implement format() methods that return a string, and then implement print.class <- function(x, ...) cat(format(x, ...), "\n". This makes for methods that are much easier to compose, because the side-effects are isolated to a single place.

#' @export
format.AST <- function(x, ..., prompt = NULL, header = NULL, all_fields = FALSE) {
  collapse(
    if (!is.null(header)) collapse(header, "\n", prompt),
    gsub("\n", collapse("\n", prompt), x$.format(..., all_fields = all_fields), fixed = TRUE)
  )
}
#' @export
print.AST <- function(
  x, ...,
  prompt = "  | ",
  header = "<graphql definition>",
  all_fields = FALSE
) {
  cat(format(x, ..., prompt = prompt, header = header, all_fields = all_fields), "\n", sep = "")
}

#' As R6
#'
#' Debug method that strips all gqlr classes and assigns the class as \code{'R6'}
#'
#' @param x any object. If it inherits \code{'R6'}, then the class of \code{x} is set to \code{'R6'}
#' @export
#' @examples
#' Int <- getFromNamespace("Int", "gqlr")$clone()
#' print(Int)
#' print(as_R6(Int))
as_R6 <- function(x) {
  if (inherits(x, "R6")) {
    class(x) <- "R6"
  }
  x
}