R/aws.R

In ropenscilabs/snowball: Spin up a managed cluster and perform parallel calculations

#' Start an AWS instance
#'
#' @param ami
#' @param type
#' @param subNet
#' @param securityGroup
#' @param keyName
#' @param userData
#'
#' @return NULL
#' @export
#'
#' @examples
#'
#' \dontrun{
#' my_instance <- startInstance()
#' instance_status(my_instance$item$instanceId[[1]])
#' }
# startInstance <- function(ami="ami-1170382b",
startInstance <- function(ami="ami-36290455",
                          type="t2.micro",
                          subNet=-1,
                          securityGroup=-1,
                          userData="",
                          keypair=NULL) {
  # library(aws.ec2)

  #describe_images(ami)

  if (subNet==-1) {
    subNet <- describe_subnets()[[1]]
  }

  if (securityGroup==-1) {
    securityGroup <- describe_sgroups()[[1]]
  }

  if (userData!=""){
    userData=base64enc::base64encode(charToRaw(userData))
  }

  i <- run_instances(image    = ami,
                     type     = type,
                     subnet   = subNet,
                     sgroup   = securityGroup,
                     userdata = userData,
                     IAMInstanceProfile = profile,
                     keypair  = keypair)

  message(paste0("Starting a ",i$item$instanceType[[1]]," instance, id ",i$item$instanceId[[1]]))

  return(i)

}

## NOTE: PERMISSIONS NEED TO BE SET UP ON BUCKETS:
# AWS bucket policy `{
#   "Id": "Policy1461293793710",
#   "Version": "2012-10-17",
#   "Statement": [
#     {
#       "Sid": "Stmt1461293785500",
#       "Action": "s3:*",
#       "Effect": "Allow",
#       "Resource": "arn:aws:s3:::auunconfdt2",
#       "Principal": {
#         "AWS": [
#           "arn:aws:iam::830136488105:user/auunconf"
#           ]
#       }
#     }
#     ]
# }`
#


#' List contents of a S3 bucket
#'
#' @param bucket
#'
#' @return vector of data.frame contents
#' @export
#'
#' @examples
#' \dontrun{
#' my_bucket <- get_bucket("bucketName")
#' bucket_contents(my_bucket)
#' }
bucket_contents <- function(bucket) {

  lapply(my_bucket, function(x) x[["Key"]]) %>% unlist %>% unname

}

## save_to_s3

# s3save(x, object="x", bucket=my_bucket)

#' Load an object from S3
#'
#' @param object_name name of object to be retrieved
#' @param bucket a s3_bucket object or character string of such
#'
#' @return NULL
#' @export
#'
load_from_s3 <- function(object_name, bucket) {
  y <- get_object(object_name, bucket=bucket)
  load(rawConnection(y))
  return(NULL)
}