如何解析knitr文档并调用all_labels等内置函数处理指定输入文件

knitr本身已经提供了足够的底层能力，不需要从零开发解析逻辑，两种实现方案供你选择：

方案1：静态解析（推荐用于缓存管理场景） #

无需运行文档就能拿到所有块的完整信息，直接用rmarkdown包的parse_rmd()函数即可，它会将Rmd文件解析为结构化列表，每个代码块的标签、选项、代码都可以直接提取：

some_utility <- function(input) {
  # 解析Rmd文档
  parsed_doc <- rmarkdown::parse_rmd(input)
  # 提取所有R代码块的标签
  all_chunk_labels <- lapply(parsed_doc, function(node) {
    if (inherits(node, "rmd_chunk") && node$engine == "r") {
      return(node$params$label)
    }
  }) |> unlist() |> na.omit() |> as.character()
  
  return(all_chunk_labels)
}

你还可以直接从解析结果中提取每个块的缓存开关、依赖设置等选项，完全满足缓存管理工具的开发需求。

方案2：兼容knitr原生接口 #

如果你需要严格匹配你预期的调用形式，直接用all_labels()、knit_code$get()这类knitr内置函数操作目标文档，可以通过临时加载文档块到knitr的运行时环境实现：

open_knitr <- function(input) {
  # 备份当前会话已有的代码块，避免污染现有环境
  old_chunks <- knitr::knit_code$get()
  on.exit(knitr::knit_code$restore(old_chunks), add = TRUE)

  # 拆分并解析Rmd文档的所有块
  file_lines <- readLines(input, warn = FALSE)
  chunks <- knitr:::split_file(file_lines)

  # 将目标文档的R代码块加载到knitr的代码存储环境
  for (chunk in chunks) {
    if (chunk$engine != "R" || is.null(chunk$label)) next
    knitr::knit_code$set(
      name = chunk$label,
      value = chunk$code,
      chunk_opts = chunk$params
    )
  }

  return(invisible(knitr::knit_code))
}

# 调用方式和你预期的完全一致
some_utility <- function(input) {
  knit_doc <- open_knitr(input)
  knitr::all_labels()
}

如果担心非导出函数split_file的版本兼容性，可以把这部分替换为rmarkdown::parse_rmd()的结果做适配，逻辑完全一致。

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