R语言：如何通过自定义环境$方法捕获内部函数调用参数值？

我需要处理以下场景：

• 使用来自其他包的自定义环境some_env，其中包含some_function()等函数；
• 有多个面向用户的函数，内部会调用该环境中的函数。

目标是自动给所有用户函数的返回结果附加内部函数的调用表达式（带实际参数值）作为属性，避免每个用户函数重复编写捕获表达式、添加属性的代码。之前尝试给环境添加custom_class类并自定义$方法，但只能拿到未求值的调用表达式，无法获取参数的实际值。

解决方案：通过包装函数捕获调用上下文 #

核心思路是在自定义$方法中返回一个包装函数，这个函数会在用户调用时捕获完整的调用信息（包括实际参数值），执行原函数后给结果附加属性。

完整代码如下：

library(rlang)

# 自定义环境（模拟来自其他包）
some_env <- new.env(parent = emptyenv())
class(some_env) <- "custom_class"
some_env$some_function <- function(x) {
  print(x)
}

# 自定义$方法，返回包装后的函数
`$.custom_class` <- function(x, name) {
  # 获取环境中的原函数
  original_fun <- get(name, envir = x, inherits = FALSE)
  
  # 返回包装函数，捕获调用并处理
  function(...) {
    # 获取当前调用的完整表达式（包含显式/默认参数）
    call_details <- rlang::call_match(defaults = TRUE)
    # 构建带环境前缀的完整调用表达式
    env_fun_sym <- rlang::sym(paste0(deparse(substitute(x)), "$", name))
    full_call_expr <- rlang::expr(!!env_fun_sym(!!!call_details[-1]))
    
    # 执行原函数得到结果
    result <- original_fun(...)
    
    # 给结果添加调用表达式属性
    attr(result, "original_query") <- full_call_expr
    
    result
  }
}

# 示例用户函数（无需修改）
user_facing_fn <- function(val = 1) {
  some_env$some_function(val)
}

# 测试：传入自定义参数
output <- user_facing_fn(3)
output
#> [1] 3
#> attr(,"original_query")
#> some_env$some_function(val = 3)

# 测试：使用默认参数
output_default <- user_facing_fn()
output_default
#> [1] 1
#> attr(,"original_query")
#> some_env$some_function(val = 1)

关键细节解释 #

1. 包装函数的作用：
   当用户调用some_env$some_function(val)时，自定义$方法不会直接返回原函数，而是返回一个包装函数。这个包装函数运行在用户函数的上下文里，能直接获取到参数的实际值。

2. 捕获完整调用信息：

   • rlang::call_match(defaults = TRUE)会获取当前调用的所有参数（包括显式传入的和默认值），返回一个包含函数和参数的调用对象。
   • 通过准引用（!!和!!!）构建完整的some_env$some_function(...)表达式，把参数替换为实际值。

3. 兼容多调用场景：
   如果用户函数中多次调用环境中的函数，每个调用都会被独立包装，各自生成对应的属性：

   user_facing_multi <- function(a, b) {
     res1 <- some_env$some_function(a)
     res2 <- some_env$some_function(b)
     list(res1, res2)
   }
   
   multi_output <- user_facing_multi(5, 10)
   multi_output[[1]]
   #> [1] 5
   #> attr(,"original_query")
   #> some_env$some_function(a = 5)
   multi_output[[2]]
   #> [1] 10
   #> attr(,"original_query")
   #> some_env$some_function(b = 10)
   

为什么之前的方法无法获取参数值？ #

之前用rlang::frame_call()拿到的是用户函数中的原始表达式some_env$some_function(val)，此时val还是一个未求值的符号，仅存在于用户函数的上下文中。而包装函数是在参数传递的时刻执行的，此时参数已经被求值（或使用默认值），所以能直接拿到实际值。

内容的提问来源于stack exchange，提问作者bretauv