R/modify_note.R

In pxmake: Make PX-Files in R

handle_notes <- function(x, value, keyword, validate) {
  colname <- tolower(keyword)
  error_msg <- stringr::str_glue("Argument 'value' has wrong format. See ?{colname}.")

  if (missing(value)) {
    value_table2 <- get_table2_value(x, keyword)
    value_variables2 <- get_variables2_value(x, colname)

    if (is.null(value_table2)) {
      return(value_variables2)
    } else if (is.null(value_variables2)) {
      return(value_table2)
    } else {
      return(list(value_table2, value_variables2))
    }
  }

  if (is.null(value)) {
    x <- remove_keyword_table2(x, keyword)
    x <- remove_keyword_variables2(x, keyword)
  } else if (is.character(value)) {
    x <- modify_table2(x, keyword, value)
  } else if (is.data.frame(value)) {
    if (all(c("language", "value") %in% colnames(value))) {
      x <- modify_table2(x, keyword, value)
    } else if (all(c("variable-code", colname) %in% colnames(value))) {
      x <- modify_variables2(x, colname, value)
    } else {
      error(error_msg)
    }
  } else if (is.list(value)) {
    for (i in seq_along(value)) {
      x <- get(paste0("px_", colname, ".px"))(x, value[[i]])
    }
  } else {
    error(error_msg)
  }

  return_px(x, validate)
}


#' @rdname px_note.px
#' @export
px_note <- function(x, value, validate) {
  UseMethod("px_note")
}

#' @eval add_documentation_note("NOTE")
px_note.px <- function(x, value, validate = TRUE) {
  handle_notes(x, value, "NOTE", validate)
}


#' @rdname px_notex.px
#' @export
px_notex <- function(x, value, validate) {
  UseMethod("px_notex")
}

#' @eval add_documentation_note("NOTEX")
px_notex.px <- function(x, value, validate = TRUE) {
  handle_notes(x, value, "NOTEX", validate)
}