R/parse_trains.R

In ntyndall/translink.bot: Translink Slack Bot

#' @title Parse Trains
#' 
#' @export


parse_trains <- function(dbr, event) {
  
  # Get the actual message
  myMessage <- event$text

  # Make sure to statement exists
  if ("to" %in% myMessage) {
  
    # Get to index
    myInd <- "to" %>% 
      `==`(myMessage) %>%
      which
    
    # Parse out stations
    startSt <- myMessage[1:(myInd %>% `-`(1))]
    stopSt <- myMessage[(myInd %>% `+`(1)):(myMessage %>% length)]

    # Get station list
    station.list <- translink.bot::get_stations()

    # Upper-case first character
    startSt %<>% translink.bot::format_name()
    stopSt %<>% translink.bot::format_name()

    # Get the start code
    startCode <- station.list$code %>% 
      `[`(startSt %>% 
            stringdist::stringdist(station.list$name, method = 'jw') %>%
            which.min
      )
    
    # Make sure stopStation has a match!
    stopSt <- station.list$name %>% 
      `[`(stopSt %>% stringdist::stringdist(station.list$name, method = 'jw') %>% which.min)
    
    # Get calling information
    allresults <- startCode %>% 
      translink.bot::query_live()
    
    # Make sure trains are still running
    if (allresults$myresults %>% is.null) {
      slackTxt <- data.frame(
        fallback = "Trains are not running right now - check back in the morning!",
        pretext = "Trains are not running right now - check back in the morning!"
      )
    } else {
      correctWay <- lapply(
        X = allresults$callingpoints,
        FUN = function(x) if (stopSt %in% (x %>% `[[`("Name") %>% as.character)) T else F 
      ) %>% 
        purrr::flatten_lgl()
      
      # Subset the right way details
      allresults$callingpoints %<>% `[`(correctWay)
      allresults$myresults %<>% subset(correctWay)
      
      slackTxt <- allresults %>% 
        translink.bot::create_text(
          dbr = dbr,
          startStation = startSt,
          stopStation = stopSt
        ) 
    }
  } else {
    slackTxt <- data.frame(
      fallback = "Could not find `to` tag...",
      pretext = "Could not find `to` tag..."
    )
  }
  
  # Return text for slack
  return(list(attachments = slackTxt) %>% jsonlite::toJSON())
}