R/azure_info.R

In billspat/bayesrun: simplified management of Azure resources to run R code

# azureInfo.R
# get information and status about specific Azure resources (relevant to running batch VMs)


# previous standalone code had this
# library(AzureVM)
# library(AzureRMR)
# library(AzureStor)
# library(stringr)

# vector of names of all the resources in a resource group
allResourceNames <- function(rgName = getOption('azurerg')){
    rg<-get_rg(rgName)
    # get a vector of the names of resources, each vector item named for the type of resource
    resourcenames <- unlist(lapply(rg$list_resources(), function(x){x$name}))
    return(resourcenames)

}

# given a collection of Azure resources, return a vector of their namespull out just the names
resourceNameList <- function(resources){
    all_names <- unlist(lapply(resources, function(x){x$name}))
    return(all_names)
}

# return a list of all storage accounts.  From one you can get a list of containers
storageAccounts <- function(rgName = getOption('azurerg')){
    rg<-get_rg(rgName)
    return(rg$list_resources(filter="resourceType eq 'Microsoft.Storage/storageAccounts'"))
}

# given a storage account name, get vector of its containers
listContainers <- function(storageAccount, rgName = getOption('azurerg')){
    rg<-get_rg(rgName)
    storageObject <- rg$get_storage(storageAccount)
    be <- storageObject$get_blob_endpoint()
    se <- AzureStor::storage_endpoint(be$url, be$key)
    return(AzureStor::list_storage_containers(se))

}

# return vector of all containers in all storage accounts in a resource group
listAllContainers <- function( rgName =getOption('azurerg') ) {
    rg <- get_rg(rgName)
    sa <- storageAccounts(rg)
    containerlist <- lapply(sa, rg=rg, listContainers)
}

# return list of resources that start with a string
getResourcesByName <- function(namePart, rgName = getOption('azurerg') ){
    rg<-get_rg(rgName)
    resources <- rg$list_resources(filter=paste0("substringof('", namePart, "', name)"))
    return(resources)
    # all_names <- rg$list_resources()
    # get a vector of the names of resources, each vector item named for the type of resource
    # all_names <- unlist(lapply(all_resources, function(x){x$name}))
    # these names <- all_resources[grep(namePattern, all_names)])
}




### move to another script
#' delete all resources with names that match a pattern
#'
#' since many cloning scripts create cllections of resources all with similar name
#' prefixes or suffixes, this lets us delete them all at once
#' @return character vector of the names of resources deleted
deleteResourcesByName <- function(namePart, rgName = getOption('azurerg'), confirm=TRUE ) {
    # $filter=substringof('demo', name)
    rg<-get_rg(rgName)

    resources <- getResourcesByName(namePart)
    deleted_resources <- c('')

    if(confirm) {
        print(paste0(resourceNameList(resources)))
        answer <-  readline(prompt="confirm delete these resources? type 'Yes' ")
    } else { answer <- 'Yes'}

    if(answer == 'Yes' || answer == 'yes'){
        for(r in resources){
            print(paste('deleting resource...', r$name))
            #TODO trycatch()
            r$delete(confirm=FALSE)
            append(deleted_resources, r$name)
            #TODO check tag for Sys.info()['user']
        }

    } else {
        message("deletion cancelled")
    }
    # vector of names
    return(deleted_resources)
}