R/pxweb_c.R

In pxweb: R Interface to PXWEB APIs

#' Combine pxweb objects
#'
#' @param x a list with \code{pxweb} objects.
#' 
#' @keywords internal
pxweb_c <- function(x){
  checkmate::assert_class(x, "list")
  if(length(x) == 1) {
    return(x[[1]])
  }
  
  if(inherits(x[[1]], "pxweb_data")) {
    return(pxweb_data_c(x))
  }
  
  if(inherits(x[[1]], "json")) {
    return(x)
  }
  
  if(inherits(x[[1]], "character")) {
    fp <- unlist(x)
    fe <- file.exists(fp)
    if(all(fe)){
      message("PXWEB API did not return JSON. Files has been stored locally (tempdir) and paths has been returned.")
      return(x)
    } else {
      stop("Files doesn't exist:\n", paste(fp[!fe], collapse = "\n"), call. = FALSE)
    }
  }
  
  stop("pxweb_c() not implemented for class '", class(x[[1]])[1], "'.", call. = FALSE)
}


#' Combine pxweb objects
#'
#' @param x a list with \code{pxweb} objects.
#' 
#' @keywords internal
pxweb_data_c <- function(x){
  for (i in seq_along(x)) {
    checkmate::assert_class(x[[i]], "pxweb_data")
  }
  full_pxd <- x[[1]]

  # Assert equal columns
  for(i in 2:length(x)){
    for(j in seq_along(full_pxd$columns)){
      checkmate::assert_set_equal(full_pxd$columns[[j]]$code, x[[i]]$columns[[j]]$code)
      checkmate::assert_set_equal(full_pxd$columns[[j]]$text, x[[i]]$columns[[j]]$text)
      checkmate::assert_set_equal(full_pxd$columns[[j]]$type, x[[i]]$columns[[j]]$type)      
    }
  }
  
  # Add comments if not duplicated
  json_comments <- character(length(full_pxd$comments)) 
  for(i in seq_along(full_pxd$comments)){
    json_comments[i] <- jsonlite::toJSON(full_pxd$comments[[i]])
  }
  for(i in 2:length(x)){
    for(j in seq_along(x[[i]]$comments)){
      json_new_comment <- jsonlite::toJSON(x[[i]]$comments[[j]])
      if(!(json_new_comment %in% json_comments)) json_comments <- c(json_comments, json_new_comment)
    }
  }
  full_pxd$comments <- list()
  for(i in seq_along(json_comments)){
    full_pxd$comments[[i]] <- jsonlite::fromJSON(json_comments[i])
  }
  
  # Combine data
  for(i in 2:length(x)){
    full_pxd$data <- c(full_pxd$data, x[[i]]$data)
  }
  
  checkmate::assert_class(full_pxd, "pxweb_data")
  full_pxd
}