R/extract.R

In Wytamma/monitauR: Remotely Monitor Script Progress

extract_steps_as_futures <-
  function(lines,
           token,
           job_id,
           name,
           comment_syntax,
           API_URL,
           STOP_ON_ERROR = FALSE) {
    futures <- list()
    steps <- list()
    for (i in 1:length(lines)) {
      line <- lines[i]
      if (!startsWith(line, comment_syntax)) {
        next
      }
      steps[[sprintf("%02d", i)]] <- line
    }
    if (length(steps) == 0) {
      stop(sprintf(
        'Could not find monituaR style comments (%s) in the file.',
        comment_syntax
      ))

    }
    for (i in 1:length(steps)) {
      step <- steps[i]
      msg <-
        trimws(substr(step, nchar(comment_syntax) + 1, nchar(step)))
      MONITAUR_EXPRESSION <- FALSE
      if (length(grep("^\\{.*\\}$", msg)) > 0) {
        MONITAUR_EXPRESSION <- TRUE
      }
      # create step future
      f <- future::future({
        tryCatch({
          if (MONITAUR_EXPRESSION) {
            STOP_ON_ERROR <- TRUE
            msg <- toString(eval(str2expression(msg)))
            STOP_ON_ERROR <- FALSE
          }
          # do smarter checking for response status
          message(sprintf('%s: %s (%s/%s)', name, msg, i, length(steps)))
          monitauR:::step(token = token,
                          API_URL = API_URL,
                         job_id = job_id,
                         msg = msg)
        },
        error = function(e) {
          if (STOP_ON_ERROR) {
            stop(e)
          }
          warning(e$message)
        })
      }, lazy = TRUE)
      futures[[names(steps)[i]]] <- f
    }
    return(futures)
  }


extract_expressions <- function(lines) {
  exprs <- base::parse(text =  lines, keep.source = TRUE)
  expressions <- list()
  for (i in 1:length(exprs)) {
    e <- exprs[i]
    if (any(startsWith(deparse(e), "expression(monitauR::monitor"))) {
      next
    }
    srcref <- attr(e, "srcref")[[1]][1]
    expressions[sprintf("%02d", srcref)] <- e
  }
  return(expressions)
}