stRoke/R/add_padding.R

#' Add padding to string
#'
#' @param d vector of strings or numbers
#' @param length final string length
#' @param after if padding should be added after as opposed to default before
#' @param pad padding string of length 1
#'
#' @return vector or character strings of same length.
#' @export
#'
#' @examples
#' add_padding(sample(1:200,5))
add_padding <- function(d,length=NULL,after=FALSE,pad="0"){
  if (!is.vector(d)) {
    stop("Please supply vector")
  }
  
  if (nchar(pad)!=1) {
    stop("Padding value should be just a single character or digit")
  }
  
  ns <- nchar(d)
  
  if (is.null(length)){
    l <- max(ns)
  } else {
    l <- length
  }
  
  ps <- unlist(lapply(l-ns,function(i){
    paste(rep(pad,i),collapse="")}))
  
  if (after) {
    paste0(d,ps)
  } else {
    paste0(ps,d)
  }
}