如何判断R包中R脚本函数的调用源是.R还是.Rmd文件

实现思路 #

要区分函数的调用环境，核心是识别当前是否处于Rmarkdown的渲染流程中。下面提供两种可靠的实现方式，按需选择：

方案1：依赖rmarkdown包的内置检测函数 #

这是最简洁且可靠的方法，利用rmarkdown包提供的in_rmarkdown()直接判断：

my_function <- function() {
  # 先检查rmarkdown包是否可用
  if (requireNamespace("rmarkdown", quietly = TRUE)) {
    is_rmd_env <- rmarkdown::in_rmarkdown()
  } else {
    # 未安装rmarkdown时默认判定为普通R脚本环境
    is_rmd_env <- FALSE
  }
  
  if (is_rmd_env) {
    # 被.Rmd调用时执行的逻辑
    cat("当前为Rmarkdown渲染环境，执行对应操作\n")
    # 示例：输出Markdown格式的内容
    cat("**这是Markdown加粗文本**\n")
  } else {
    # 被.R脚本调用时执行的逻辑
    cat("当前为普通R脚本环境，执行对应操作\n")
    # 示例：输出控制台友好的纯文本
    cat("这是普通控制台文本\n")
  }
}

如果你的R包需要依赖rmarkdown，记得在包的DESCRIPTION文件中添加Imports: rmarkdown，确保函数能正常运行。

方案2：无依赖的环境/调用栈检测 #

若不想引入额外依赖，可以通过检测环境变量或调用栈来判断：

my_function <- function() {
  # 检测Rmarkdown渲染时的特征环境变量
  has_rmd_env <- Sys.getenv("RMARKDOWN_INPUT_FILE") != "" || Sys.getenv("RSTUDIO_PANDOC") != ""
  
  # 检测调用栈中是否存在Rmarkdown/Knitr相关函数
  call_stack <- sapply(sys.calls(), deparse)
  has_rmd_call <- any(grepl("rmarkdown::render|knitr::", call_stack))
  
  is_rmd_env <- has_rmd_env || has_rmd_call
  
  if (is_rmd_env) {
    cat("来自Rmarkdown文档的调用\n")
    # 执行Rmd专属操作
  } else {
    cat("来自普通R脚本的调用\n")
    # 执行普通脚本专属操作
  }
}

这种方法兼容性稍弱（受渲染工具版本影响），但无需依赖任何额外包。

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