R/messy_string.R

In joshyam-k/messy: What the Package Does (One Line, Title Case)

#' @title messy_string
#'
#' @description Generate a vector of capitalization variants of a string
#'
#' @param string A string for which you would like to generate capitalization variants
#'
#' @return A vector of strings
#' @examples
#' string <- "United States"
#' messy_string(string)
#' @export
#' @importFrom rlang .data

new_strings <- function(string) {


  if (stringr::str_detect(string, "-")) {

    string <- stringr::str_replace_all(string, "-", " - ")

  }

  words <- stringr::str_to_lower(stringr::str_split(string, " ")[[1]])

  n <- length(words)

  binary_tibble <- as.data.frame(binary_strings(n), .name_repair = "unique")

  full_word_cap <- binary_tibble
  first_letter_cap <- binary_tibble

  for (row in 1:nrow(binary_tibble)) {

    for (col in 1:n) {

      if (binary_tibble[row, col] == "0") {

        first_letter_cap[row, col] <- words[col]
        full_word_cap[row, col] <- words[col]

      } else {

        first_letter_cap[row, col] <-
          paste0(stringr::str_to_upper(stringr::str_sub(words[col], 1, 1)), stringr::str_sub(words[col], 2, -1))

        full_word_cap[row, col] <- stringr::str_to_upper(words[col])

      }

    }
  }


  first_letter_cap <- tidyr::unite(first_letter_cap, phrase, 1:ncol(first_letter_cap), sep = " ")
  full_word_cap <- tidyr::unite(full_word_cap, phrase, 1:ncol(full_word_cap), sep = " ")

  final_phrases <- dplyr::distinct(rbind(first_letter_cap, full_word_cap), .keep_all = T)

  if (stringr::str_detect(string, "'")) {

    to_add1 <- apostrophe_rm(final_phrases)

    final_phrases <- rbind(final_phrases, to_add1)

  }


  if (stringr::str_detect(string, "\\.")) {

    to_add2 <- period_rm(final_phrases)

    final_phrases <- rbind(final_phrases, to_add2)

  }

  if (stringr::str_detect(string, ",")) {


    to_add3 <- comma_rm(final_phrases)

    final_phrases <- rbind(final_phrases, to_add3)

  }

  if (stringr::str_detect(string, "!")) {

    to_add4 <- exclamation_rm(final_phrases)

    final_phrases <- rbind(final_phrases, to_add4)

  }

  unlist(final_phrases, use.names = F)

}


messy_string <- function(string) {

  if (stringr::str_detect(string, "-")) {

    locs <- stringr::str_locate_all(string, "-")[[1]][ ,1]
    hyphen_bits <- binary_strings(length(locs))
    hyphen_options <- 2**(length(locs))

    string_hyphen_rm <- stringr::str_replace_all(string, "-", " ")


    item_add <- function(x, loc, item) {

      stringr::str_sub(x, loc, loc) <- item; x

    }

    final_hyphens <- c()
    for (i in 1:hyphen_options) {

      final_phrases_piece <- new_strings(string_hyphen_rm)

      for (col in 1:length(locs)) {

        if (hyphen_bits[i, col] == "0") {

          final_phrases_piece <- purrr::map_chr(final_phrases_piece, ~item_add(.x, locs[col], " "))

        } else {

          final_phrases_piece <- purrr::map_chr(final_phrases_piece, ~item_add(.x, locs[col], "-"))


        }

      }

      final_hyphens <- c(final_hyphens, final_phrases_piece)


    }

    ret <- final_hyphens

  } else {

    ret <- new_strings(string)

  }

  ret

}