Workbench launcher concept

Author: shikokuchuo

DESCRIPTION

@@ -30,7 +30,8 @@ Imports:
     nanonext (>= 1.6.2.9002)
 Suggests: 
     cli,
-    litedown
+    litedown,
+    rstudioapi
 Enhances: 
     parallel,
     promises

NAMESPACE

@@ -60,6 +60,7 @@ export(stop_cluster)
 export(stop_mirai)
 export(unresolved)
 export(with_daemons)
+export(workbench_config)
 importFrom(nanonext,"opt<-")
 importFrom(nanonext,.advance)
 importFrom(nanonext,.context)

R/launchers.R

@@ -94,6 +94,23 @@ launch_remote <- function(n = 1L, remote = remote_config(), ..., tls = NULL, .co
   dots <- if (...length()) parse_dots(envir, ...) else envir[["dots"]]
   if (is.null(tls)) tls <- envir[["tls"]]
 
+  if (length(remote) == 2L) {
+    requireNamespace("rstudioapi", quietly = TRUE) || stop(._[["rstudio_api"]])
+    rstudioapi::launcherAvailable()
+    cluster <- remote[["name"]]
+    container <- rstudioapi::launcherContainer(remote[["image"]])
+    lapply(
+      seq_len(n),
+      function(x) rstudioapi::launcherSubmitJob(
+        sprintf("mirai_daemon_%d", x),
+        cluster = cluster,
+        command = launch_remote(),
+        container = container
+      )
+    )
+    return(invisible())
+  }
+
   command <- remote[["command"]]
   rscript <- remote[["rscript"]]
   quote <- remote[["quote"]]
@@ -392,6 +409,34 @@ cluster_config <- function(command = "sbatch", options = "", rscript = "Rscript"
   list(command = "/bin/sh", args = args, rscript = rscript, quote = NULL)
 }
 
+#' Workbench Remote Launch Configuration
+#'
+#' Generates a remote configuration for launching daemons using the default
+#' launcher configured in Posit Workbench.
+#'
+#' @inherit remote_config return
+#'
+#' @seealso [ssh_config()], [cluster_config()], and [remote_config()] for other
+#'   remote launch configurations.
+#'
+#' @examples
+#' tryCatch(workbench_config(), error = identity)
+#'
+#' \dontrun{
+#'
+#' # Launch 2 daemons using the Workbench default launcher:
+#' daemons(n = 2, url = host_url(), remote = workbench_config())
+#' }
+#'
+#' @export
+#'
+workbench_config <- function() {
+  requireNamespace("rstudioapi", quietly = TRUE) || stop(._[["rstudio_api"]])
+  rstudioapi::launcherAvailable()
+  cluster <- rstudioapi::launcherGetInfo()[["clusters"]][[1L]]
+  list(name = cluster[["name"]], image = cluster[["defaultImage"]])
+}
+
 #' URL Constructors
 #'
 #' `host_url()` constructs a valid host URL (at which daemons may connect) based

R/mirai-package.R

@@ -80,6 +80,8 @@
     n_zero = "the number of daemons must be zero or greater",
     not_found = "compute profile `%s` not found",
     numeric_n = "`n` must be numeric, did you mean to provide `url`?",
+    rstudio_api = "workbench launcher requires the `rstudioapi` package",
+    rstudio_unavailable = "workbench launcher requires a compatible environment",
     sync_daemons = "mirai: initial sync with daemon(s) [%d secs elapsed]",
     sync_dispatcher = "mirai: initial sync with dispatcher [%d secs elapsed]",
     within_map = "cannot create local daemons from within mirai map"