handle when message is a function

NAMESPACE

@@ -1,6 +1,5 @@
 # Generated by roxygen2: do not edit by hand
 
 export("%!%")
-export(slap)
 import(cli)
 import(rlang)

R/slap.R

@@ -24,13 +24,14 @@
 #' @name slap
 #' @export
 `%!%` <- function(expr, message) {
-  message <- enexpr(message)
-
   quo <- quo(
     withCallingHandlers(
       {{ expr }},
-      error = function(.__slap__.) {
-        message <- !!message
+      error = function(err) {
+        message <- {{ message }}
+        if (is.function(message)) {
+          message <- message(err)
+        }
 
         error_call <- env$.__error_call__.
         if (is.null(error_call)) {
@@ -40,16 +41,11 @@
           error_call <- NULL
         }
 
-        cli::cli_abort(message, parent = .__slap__., call = error_call)
+        cli::cli_abort(message, parent = err, call = error_call)
       }
     )
   )
 
   env <- caller_env()
   eval_tidy(quo, env = env, data = list(env = env))
 }
-
-#' @export
-slap <- function(error_call = current_env()) {
-  get(".__slap__.", caller_env()) %||% cli_abort("{.fn slap()} can only be used on the rhs of a `%!%`.", call = error_call)
-}