R语言：如何在foo<-替换函数中获取子集索引（不可添加类）？

背景与问题现象 #

我们先定义如下foo函数和对应的替换函数foo<-：

foo <- function(x) {
  x
}

`foo<-` <- function(x, value) {
  value
}

正常的子集赋值操作能得到预期结果：

x <- 1:3
foo(x)[2] <- 40
x
# [1]  1 40  3

但当子集索引超出范围时，会产生意外的结果：

x <- 1:3
foo(x)[-5] <- 40
x
# [1] 40 40 40

核心疑问 #

根据《R语言定义》3.4.4节“子集赋值”的说明，上述异常代码大致等价于：

x <- 1:3
`*tmp*` <- x
x <- `foo<-`(`*tmp*`, value=`[<-`(foo(`*tmp*`), -5, value=40))
rm(`*tmp*`)

这意味着子集索引在调用foo<-之前就已经在[<-的调用中处理了。那么是否仍能在foo<-内部获取到该调用中的索引i = -5？

注意：我无法/不想为x添加类属性并创建[<-.foo方法，比如以下方案不可行：

# 无法添加类属性，因此此方法不可行：
foo <- function(x) {
  class(x) <- c("foo", class(x))
  x
}

`[<-.foo` <- function(x, i, value) {
  cat("index: ", i, "\n")
  # <<< 对i进行操作 >>>
  NextMethod("[<-")
}

x <- 1:3
foo(x)[1] <- 40
x

很遗憾，在当前的实现逻辑下，无法在foo<-内部直接获取到子集索引i。

原因很明确：根据R的子集赋值机制，[<-会先完成索引处理和值替换的逻辑，然后才把最终生成的完整向量传递给foo<-的value参数。此时foo<-拿到的只是一个已经被修改后的结果向量，完全看不到原始的索引信息。

如果不能通过添加自定义类来拦截[<-调用，还有一种变通思路：不直接用foo(x)[i] <- value的语法，而是封装一个专门的函数来显式传递索引，比如：

foo_subset_assign <- function(x, i, value) {
  # 在这里校验索引i的合法性
  if (any(abs(i) > length(x))) {
    stop("索引超出范围！")
  }
  # 执行赋值逻辑
  tmp <- foo(x)
  tmp[i] <- value
  `foo<-`(x, tmp)
}

# 使用示例
x <- 1:3
foo_subset_assign(x, 2, 40)
x
# [1]  1 40  3

# 测试非法索引
foo_subset_assign(x, -5, 40)
# Error in foo_subset_assign(x, -5, 40) : 索引超出范围！

这种方式绕开了R原生的子集赋值语法，通过显式传递索引来实现校验逻辑，完全不需要修改原对象的类属性。

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