R/AddIn_FindAndReplace.R

In mikebesso/two.laws.big.bang:

#' Find and Replace Tokens in a Document
#'
#' Call this as an addin to replace occurrences of a particular sequence of
#' characters in a document with a new sequence of characters.
#'
#' @export
AddIn_FindAndReplace <- function() {

  ui <- miniUI::miniPage(

    includeHighlightJs(),
    miniUI::gadgetTitleBar("Find and Replace"),
    miniUI::miniContentPanel(
      shiny::h4("Replace the text 'from' with the text 'to'."),
      shiny::hr(),
      stableColumnLayout(
        shiny::checkboxInput("boundaries", "Use Word Boundaries?", value = TRUE),
        shiny::uiOutput("changes")
      ),
      stableColumnLayout(
        shiny::textInput("from", "From:"),
        shiny::textInput("to", "To:")
      ),
      shiny::uiOutput("document", container = rCodeContainer)
    )
  )

  server <- function(input, output, session) {

    context <- rstudioapi::getActiveDocumentContext()
    original <- context$contents

    reactiveRefactor <- shiny::reactive({

      from <- input$from
      to <- input$to
      boundaries <- input$boundaries

      valid <- nzchar(from) && nzchar(to)
      if (valid)
        return(performRefactor(original, from, to, boundaries))
      else
        return(list(refactored = original, changes = 0))

    })

    output$changes <- renderUI({
      spec <- reactiveRefactor()
      if (spec$changes == 0)
        return(div("No changes to be made."))

      instances <- if (spec$changes == 1) "instance" else "instances"
      shiny::div(paste(spec$changes, instances, "will be replaced."))
    })

    output$document <- renderCode({
      spec <- reactiveRefactor()
      highlightCode(session, "document")
      paste(spec$refactored, collapse = "\n")
    })

    shiny::observeEvent(input$done, {
      spec <- reactiveRefactor()
      transformed <- paste(spec$refactored, collapse = "\n")
      rstudioapi::setDocumentContents(transformed, id = context$id)
      invisible(stopApp())
    })

  }

  viewer <- dialogViewer("Find and Replace", width = 1000, height = 800)
  runGadget(ui, server, viewer = viewer)
}

performRefactor <- function(contents, from, to, useWordBoundaries = TRUE) {

  reFrom <- if (useWordBoundaries)
    paste("\\b", from, "\\b", sep = "")
  else
    from

  reTo <- to
  matches <- gregexpr(reFrom, contents, perl = TRUE)
  changes <- sum(unlist(lapply(matches, function(x) {
    if (x[[1]] == -1) 0 else length(x)
  })))

  refactored <- unlist(lapply(contents, function(x) {
    gsub(reFrom, reTo, x, perl = TRUE)
  }))

  list(
    refactored = refactored,
    changes = changes
  )
}