R/stata_collectcode.r

In Hemken/Statamarkdown: 'Stata' Markdown

stata_collectcode <- function() {
  # Message when Statamarkdown loads
  if (file.exists(file.path(dirname(find_stata(message=FALSE)), "sysprofile.do"))) {
    packageStartupMessage("Found a 'sysprofile.do'")
  }
  if (file.exists(file.path(dirname(find_stata(message=FALSE)), "profile.do"))) {
    packageStartupMessage("Found a 'profile.do' in the STATA executable directory.")
    packageStartupMessage("  This prevents 'collectcode' from working.")
    packageStartupMessage("  Please rename this 'sysprofile.do'.")
  }
  if (file.exists("profile.do")){
#    print(sys.frames())
#    print(sys.calls())
#    print(sys.nframe())
    assign("oprofile", readLines("profile.do"), pos=2)
#    oprofile <- readLines("profile.do")
    packageStartupMessage("Found an existing 'profile.do'")
    packageStartupMessage(paste("  ", oprofile, "\n"))
  }
  else {
    oprofile <- NULL
  }
  # Hook for code processing
  knitr::knit_hooks$set(collectcode = function(before, options, envir) {
    if (!before) {
        if (options$engine == "stata") {
# print(options$engine.path$stata)
          if (file.exists(file.path(dirname(options$engine.path$stata), "profile.do"))) {
            packageStartupMessage("Found a 'profile.do' in the STATA executable directory.")
            packageStartupMessage("  This prevents 'collectcode' from working properly.")
            packageStartupMessage("  Please rename this 'sysprofile.do'.")
          }
            autoexec <- file("profile.do", open="at")
            writeLines(options$code, autoexec)
            close(autoexec)
# print(sys.frames())
# print(sys.calls())
          if (utils::packageVersion('knitr') < '1.45') {
              do.call("on.exit",
                      list(quote(unlink("profile.do")), add=TRUE),
                      envir = sys.frame(-9))
            } else if (utils::packageVersion('knitr') >= '1.45') {
              do.call("on.exit",
                      list(quote(unlink("profile.do")), add=TRUE),
                      envir = sys.frame(-10))
            }

        if (!is.null(oprofile)) { # replace the original "profile.do"
          if (utils::packageVersion('knitr') < '1.45') {
            do.call("on.exit",
                    list(quote(writeLines(oprofile, "profile.do")), add=TRUE),
                    envir = sys.frame(-9))
          } else if (utils::packageVersion('knitr') >= '1.45') {
            do.call("on.exit",
                    list(quote(writeLines(oprofile, "profile.do")), add=TRUE),
                    envir = sys.frame(-10))
          }
# sys.frame(1) or sys.frame(-10) is rmarkdown::render()
# sys.frame(-9) is knitr::knit()
        }
        }
    }
})
}