R/is_holiday.R

In MattCowgill/strayr: Convert Australian State Names to Abbreviations and Vice Versa

#' @title is_holiday
#' @description Identifies if a given date is a public holiday in Australia
#' @param date Date, POSIXct object or a string that can be parsed with `parsedate::parse_date`
#' @param jurisdictions Vector of state jurisdictions you wish to filter on, defaults to all of Australia
#' @return Returns a TRUE or FALSE value
#' @examples
#' \dontrun{
#' if (interactive()) {
#'   # EXAMPLE1
#' }
#' }
#' @rdname is_holiday
#' @export
#' @importFrom purrr map_lgl
is_holiday <- function(date, jurisdictions = c()) {
  ret <- purrr::map_lgl(date, do_is_holiday, jurisdictions = jurisdictions)

  return(ret)
}

do_is_holiday <- function(date, jurisdictions = c()) {
  if (is.na(date)) {
    stop("`date` argument cannot be NA")
  }

  if (!lubridate::is.Date(date) & !lubridate::is.POSIXct(date)) {
    # Attempt to coerce to date
    new_date <- parsedate::parse_date(date)

    if (is.na(new_date)) {
      stop("`date` must be a Date, POSIXct object or a string that can be parsed as a date")
    }

    # Remove any time or time zone element
    new_date <- lubridate::as_date(new_date)
  } else {
    new_date <- lubridate::as_date(date)
  }

  if (length(jurisdictions) == 0) {
    ret <- strayr::auholidays[strayr::auholidays$Date == new_date, ]
  } else {
    ret <- strayr::auholidays[strayr::auholidays$Date == new_date & (strayr::auholidays$Jurisdiction %in% jurisdictions | strayr::auholidays$Jurisdiction == "NAT"), ]
  }

  return(nrow(ret) > 0)
}