R/remove_company_suffixes.R

In julianbarg/DataAnalysisTools: Basic Data Analysis Tools

#' Remove company suffixes from a string.
#'
#' @param string Input vector. Either a character vector, or something coercible to one.
#'
#' @return A character vector.
#' 
#' @export
#'
#' @examples
#' remove_company_suffixes(c("Enterprise Products Operating LLC", "Mercedes-Benz AG"))
remove_company_suffixes <- function(string) {
  company_suffixes <- c("LLC", "AG", "A.G.", "L.P.", "LP", "CORP", "CO", "INC.", "CO.", 
                        "INC", "L.L.C.", "THE", "L.P", "LLP", "LLC.", "II", "L. P.", "LC", 
                        "Limited Partnership")
  
  company_suffixes <- c(stringr::str_to_upper(company_suffixes), 
                        stringr::str_to_lower(company_suffixes), 
                        stringr::str_to_title(company_suffixes))
  # Remove either if space before and after, or if space before and end of string ($) after.
  in_end <- sapply(company_suffixes, function(x) paste0(" ", x, "$"), USE.NAMES = FALSE)
  after_comma <- sapply(company_suffixes, function(x) paste0(",", x, "$"), USE.NAMES = FALSE)
  pattern <- paste0(c(in_end, after_comma), collapse = "|")
  
  wo_suffix <- stringr::str_remove_all(string, pattern = pattern)
  wo_suffix <- stringr::str_remove(wo_suffix, ",$")
  # Run a second time, for combinations of company endings.
  wo_suffix <- stringr::str_remove_all(wo_suffix, pattern = pattern)
  wo_suffix <- stringr::str_remove(wo_suffix, ",$")
  return(wo_suffix)
}