Made some changes to ojo_parse_county and added ojo_list_counties

NAMESPACE

@@ -2,6 +2,8 @@
 
 export(limit)
 export(ojo_counties)
+export(ojo_list_counties)
+export(ojo_oscn_counties)
 export(ojo_parse_county)
 importFrom(dplyr,if_else)
 importFrom(stringr,str_remove_all)

R/county.R

@@ -16,14 +16,27 @@ ojo_counties <- c(
   "tillman", "tulsa", "wagoner", "washington", "washita", "woods", "woodward"
 )
 
+#' @title Oklahoma OSCN reporting counties
+#'
+#' @export ojo_oscn_counties
+#'
+ojo_oscn_counties <- c(
+  "adair", "canadian", "cleveland", "comanche",
+  "ellis", "garfield", "logan", "oklahoma", "payne",
+  "pushmataha", "rogermills", "rogers", "tulsa"
+)
+
 #' @title Parse Oklahoma Counties
 #'
 #' @importFrom stringr str_remove_all str_to_lower
 #' @importFrom dplyr if_else
 #'
 #' @export ojo_parse_county
 #'
-ojo_parse_county <- function(county, ..., case = "lower", squish = NULL, suffix = NULL, counties = ojo_counties, .silent = FALSE) {
+ojo_parse_county <- function(county, ..., case = "lower", squish = FALSE, suffix = NULL,
+                             counties = ojo_counties, .silent = FALSE) {
+
+  # Check case arg
   case <- rlang::arg_match0(
     arg = stringr::str_to_lower(case),
     values = c("lower", "upper", "title")
@@ -42,21 +55,32 @@ ojo_parse_county <- function(county, ..., case = "lower", squish = NULL, suffix
 
   # Apply Casing
   county <- dplyr::case_when(
+    # Adjusting for unusual district names
+    tolower(county) == "rogermills" & case == "lower" ~ "roger mills",
+    tolower(county) == "rogermills" & case == "title" ~ "Roger Mills",
+    tolower(county) == "rogermills" & case == "upper" ~ "ROGER MILLS",
+    tolower(county) == "leflore" & case == "title" ~ "LeFlore",
+    tolower(county) == "mcclain" & case == "title" ~ "McClain",
+    tolower(county) == "mccurtain" & case == "title" ~ "McCurtain",
+    # Normal behavior
     case == "lower" ~ stringr::str_to_lower(county),
     case == "upper" ~ stringr::str_to_upper(county),
-    case == "title" ~ stringr::str_to_title(county)
+    case == "title" ~ stringr::str_to_title(county),
   )
 
-  if (case == "title") {
-    county <- dplyr::case_when(
-      county == "Leflore" ~ "LeFlore",
-      county == "Mcclain" ~ "McClain",
-      county == "Mccurtain" ~ "McCurtain",
-      county == "Rogermills" ~ "Roger Mills",
-      TRUE ~ county
+  # Add suffix if requested
+  if (!is.null(suffix)) {
+    # Change case...
+    suffix <- dplyr::case_when(
+      case == "lower" ~ stringr::str_to_lower(suffix),
+      case == "upper" ~ stringr::str_to_upper(suffix),
+      case == "title" ~ stringr::str_to_title(suffix),
     )
+    # ...then add
+    county <- stringr::str_c(county, suffix, sep = " ")
   }
 
+  # Remove all spaces if requested
   if (isTRUE(squish)) {
     county <- stringr::str_remove_all(county, "\\s")
   }
@@ -66,3 +90,33 @@ ojo_parse_county <- function(county, ..., case = "lower", squish = NULL, suffix
 
   return(county)
 }
+
+#' @title List Oklahoma Counties
+#'
+#' @export ojo_list_counties
+#'
+ojo_list_counties <- function(case = "lower", squish = FALSE, suffix = NULL,
+                              oscn_only = FALSE){
+
+  # Clean the list of counties using ojo_parse_counties() and then return
+  county <- ojo_counties |>
+    dplyr::as_tibble() |>
+    dplyr::mutate(
+      value = ojoutils::ojo_parse_county(county = value,
+                                         case = case,
+                                         squish = squish,
+                                         suffix = suffix,
+                                         .silent = TRUE)
+    )
+
+  # Only return OSCN counties?
+  if (oscn_only) {
+    county <- county |>
+      dplyr::filter(
+        ojo_parse_county(value, case = "lower", squish = TRUE) %in% ojo_oscn_counties
+      )
+  }
+
+  return(county)
+
+}