R/check_expressions.R

In queryparser: Translate 'SQL' Queries into 'R' Expressions

# Copyright 2023 Cloudera Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

are_aggregate_expressions <- function(exprs) {
  vapply(exprs, is_aggregate_expression, TRUE)
}

is_aggregate_expression <- function(expr) {
  if (is_aggregate_function_call(expr)) {
    return(TRUE)
  }
  if (length(expr) == 1) {
    return(FALSE)
  } else {
    out <- lapply(expr, is_aggregate_expression)
  }
  any(vapply(out, isTRUE, TRUE))
}

are_valid_expressions_in_aggregation <- function(exprs, allowed_names) {
  vapply(exprs, is_valid_expression_in_aggregation, TRUE, allowed_names = allowed_names)
}

is_valid_expression_in_aggregation <- function(expr, allowed_names, var_names = all_cols(expr), agg = FALSE) {
  if (deparse(expr) %in% allowed_names) {
    return(TRUE)
  } else if (length(expr) == 1) {
    if (!agg && deparse(expr) %in% var_names) {
      return(FALSE)
    } else {
      return(TRUE)
    }
  } else {
    if (is_aggregate_function_call(expr)) {
      agg <- TRUE
    }
    out <- lapply(expr, is_valid_expression_in_aggregation, allowed_names, var_names, agg)
  }
  all(vapply(out, isTRUE, TRUE))
}

is_aggregate_function_call <- function(expr) {
  identical(typeof(expr), "language") && is.call(expr) &&
    (deparse(expr[[1]]) %in% r_aggregate_functions ||
       (deparse(expr[[1]]) %in% c("paste","paste0") && "collapse" %in% names(expr) &&
          !is.null(expr[[which(names(expr) == "collapse")]])))
}

are_valid_expressions_in_distinct <- function(exprs, allowed_names) {
  vapply(exprs, is_valid_expression_in_distinct, TRUE, allowed_names = allowed_names)
}

is_valid_expression_in_distinct <- function(expr, allowed_names, var_names = all_cols(expr)) {
  if (deparse(expr) %in% allowed_names) {
    return(TRUE)
  } else if (length(expr) == 1) {
    if (deparse(expr) %in% var_names) {
      return(FALSE)
    } else {
      return(TRUE)
    }
  } else {
    out <- lapply(expr, is_valid_expression_in_distinct, allowed_names, var_names)
  }
  all(vapply(out, isTRUE, TRUE))
}