feat: add wrapper function

Author: averissimo

R/utils.R

@@ -280,3 +280,35 @@ assert_single_selection <- function(x,
   }
   invisible(TRUE)
 }
+
+#' Wrappers around `srv_transform_teal_data` that allows to decorate the data
+#' @inheritParams teal::srv_transform_teal_data
+#' @param expr (`expression`) to evaluate on the output of the decoration.
+#' Must be inline code. See [within()]
+#' Default is `NULL` which won't append any expression.
+#' @details
+#' `srv_decorate_teal_data` is a wrapper around `srv_transform_teal_data` that
+#' allows to decorate the data with additional reactive expressions.
+#' When original `teal_data` object is in error state, it will show that error
+#' first.
+#'
+#' @keywords internal
+srv_decorate_teal_data <- function(id, data, decorators, expr = NULL) {
+  expr_quosure <- rlang::enexpr(expr)
+  decorated_output <- srv_transform_teal_data(id, data = data, transformators = decorators)
+
+  reactive({
+    req(data(), decorated_output()) # ensure original errors are displayed
+    if (is.null(expr_quosure)) {
+      decorated_output()
+    } else {
+      eval_code(decorated_output(), expr_quosure)
+    }
+  })
+}
+
+#' @rdname srv_decorate_teal_data
+#' @details
+#' `ui_decorate_teal_data` is a wrapper around `ui_transform_teal_data`.
+#' @keywords internal
+ui_decorate_teal_data <- teal::ui_transform_teal_data