R/proust_random.R

In ColinFay/proustr: Tools for Natural Language Processing in French

#' Create a Random Proust extract
#' 
#' Create your own flavor of Proust with this random extractor.
#'
#' @param count the number of line you want to randomly extract and paste. 
#' @param collapse if FALSE, the output will be a tibble. Default is TRUE, a character vector. 
#' 
#' @importFrom attempt stop_if_not
#' @importFrom tokenizers tokenize_sentences
#' 
#' @return a character vector
#' @export
#'
#' @examples
#' proust_random(4)

proust_random <- function(count = 1, collapse = TRUE){
  stop_if_not(inherits(count, "numeric"), msg = "count should be a number")
  stop_if_not(inherits(collapse, "logical"), msg = "collapse should be a logical")
  temp <- sample(unlist(lapply(proust_books()$text, tokenize_sentences)), count) 
  if(collapse){
    return(paste(temp, sep = " ", collapse = " "))
  } else {
    structure(data.frame(text = temp) , class = c("tbl_df", "tbl", "data.frame"))
  }
}