R/rstudio-projects.R

In csgillespie/rprofile: My Rprofile in a Handy R Package

ip = function(path = getwd()) {
  rstudioapi::initializeProject(path)
  op(path)
}

#' @title RStudio projects
#'
#' @description A command line version of opening RStudio projects.
#' @param path Path to the (proposed) RStudio project.
#' @examples
#' \dontrun{
#' # Open project in current working directory
#' op()
#' # Open project in current working directory
#' op("/path/to/project")
#' }
#' @export
op = function(path = ".") {
  if (file.exists(path)) path = normalizePath(path)
  proj = list.files(path, pattern = "\\.Rproj$")
  if (grepl("\\.Rproj$", path)) {
    setwd(dirname(path))
    rstudioapi::openProject(path)
  } else if (length(proj) == 0L) {
    message("No available project.")
    create_project = readline(prompt = "Create (y/N): ")
    if (tolower(create_project) == "y") {
      ip(path)
    }
  } else {
    setwd(path)
    rstudioapi::openProject(proj)
  }
}

get_shorten_paths = function(paths) {
  path_split = stringr::str_split(paths, pattern = "/")
  path_lens = unlist(lapply(path_split, length))
  path_start = path_lens - 2
  path_start = pmax(path_start, 0)

  shorten_path = vector("character", length(path_split))
  for (i in seq_along(shorten_path)) { #nolint
    p = path_split[[i]]
    p = p[path_start[i]:path_lens[i]]
    shorten_path[i] = paste(p, collapse = .Platform$file.sep)
  }
  return(shorten_path)
}

get_rprojs = function() {
  fname = "~/.local/share/rstudio/monitored/lists/project_mru" #nolint
  if (file.exists(fname)) return(readLines(fname))
  fname = "~/.rstudio-desktop/monitored/lists/project_mru" #nolint
  if (file.exists(fname)) return(readLines(fname))
  stop("Unable to find Rproject list")
}

#' @title Choose an RStudio Project
#'
#' @description Command line version for choosing RStudio projects
#' @param path Default \code{NULL}. If not \code{NULL}, path is passed to \code{op}.
#' @examples
#' \dontrun{
#' cp()
#' }
#' @export
cp = function(path = NULL) {
  if (!is.null(path)) {
    op(path)
    return(invisible(NULL))
  }
  projs_paths = get_rprojs()
  projs_paths = c(getwd(),
                  projs_paths[file.exists(projs_paths)][1:9])
  projs_paths = projs_paths[!is.na(projs_paths)]
  projs_paths = normalizePath(projs_paths)

  ## Get project name
  projs = gsub(basename(projs_paths), pattern = ".Rproj", replacement = "")
  projs[1] = "Current dir"
  projs = paste0(crayon::bold(seq_along(projs) - 1), ". ", crayon::cyan(projs))
  projs[1] = paste0(" ", projs[1])

  ## Get project directory
  projs_dir = dirname(projs_paths)
  tilde = path.expand("~")
  projs_dir = stringr::str_replace(projs_dir, tilde, "~")

  shorten_paths = get_shorten_paths(paths = projs_dir)
  all = paste0(projs, " (", crayon::italic(shorten_paths), ")\n")
  cat(all)

  proj_number = readline("Select Project: ")
  if (nchar(proj_number) == 0) return(invisible(NULL))
  proj_number = as.numeric(proj_number)

  if (is.na(proj_number)) return(invisible(NULL))
  else if (proj_number == 0) op()
  else op(projs_paths[proj_number + 1])
}