R/utils-subscriptions.R

In JakeVestal/InteractiveTradeR: API

subscribe <- function(
  fun_name,
  req_name,
  req_id,
  req       = list(NULL),
  sock
){
  
  new_subscription <- dplyr::bind_cols(
    tibble::tibble(
      req_name  = req_name
    ),
    purrr::map_dfc(
      purrr::compact(req),
      function(x){
        if(length(x) > 1){return(list(x))}
        x
      }
    ),
    tibble::tibble(
      req_id    = req_id,
      conn_row  = attr(sock, "conn_row"),
      client_id = attr(sock, "client_id")
    )
  )
  
  subs <- get0(
    fun_name,
    envir = get("subscriptions")
  )
  
  if(is.null(subs)){
    assign(
      fun_name,
      value = new_subscription,
      envir = get("subscriptions")
    )
  } else {
    
    req_name_match <- which(subs$req_name == req_name)
    
    if(length(req_name_match) == 0){
      assign(
        fun_name,
        value = dplyr::bind_rows(subs, new_subscription),
        envir = get("subscriptions")
      )
    } else {
      subs[req_name_match,] <- new_subscription
      assign(fun_name, value = subs, envir = get("subscriptions"))
    }
    
  }
  
  invisible()
}

unsubscribe <- function(
  subscription_name,
  request,
  msg,
  id_or_name,
  channel = NULL
){
  
  subscriptions <- get("subscriptions")
  
  sock <- select_sock_for_api_fun()
  
  writeBin(object = msg, con = sock, endian = "big")
  
  read_sock_drawer()
  
  if(isTRUE(nrow(subscriptions[[subscription_name]]) == 1)){
    rm(list = subscription_name, envir = subscriptions)
  } else {
    assign(
      subscription_name,
      value = subscriptions[[subscription_name]][
        subscriptions[[subscription_name]][id_or_name] != request,
        ],
      envir = subscriptions
    )
  }
  
}