R/check_cetesb_code.R

In quishqa/qualR: An R package to download São Paulo and Rio de Janeiro air pollution data

#' Internal function - Check if CETESB AQS code or name is valid
#'
#' @param aqs data frame with cetesb aqs code and names
#' @param aqs_code user's input code or name
#'
#' @noRd
#' @keywords internal

check_cetesb_code <- function(aqs, aqs_code){
  if (is.numeric(aqs_code) & aqs_code %in% aqs$code){
    aqs_code <- aqs_code
    aqs_name <- aqs$ascii[aqs$code == aqs_code]
  } else if (is.character(aqs_code)){
    aqs_code_ascii <- iconv(aqs_code, from = "UTF-8", to = "ASCII//TRANSLIT")
    aqs_name <- aqs_code_ascii
    aqs_code <- aqs$code[aqs$ascii == aqs_code_ascii]
  } else {
    stop("Wrong aqs_code value or aqs name, please check cetesb_aqs",  # nocov
         call. = FALSE)                                                # nocov
  }
  aqs_code_name <- c(aqs_name, aqs_code)
  return(aqs_code_name)
}