R/facebook_heartbeat.R

In gonzalezben81/facebookanalysis: Analyses the emotional sentiment of Facebook message data

# facebook_heartbeat
#
# You can learn more about package authoring with RStudio at:
#
#   http://r-pkgs.had.co.nz/
#
# Some useful keyboard shortcuts for package authoring:
#
#   Build and Reload Package:  'Ctrl + Shift + B'
#   Check Package:             'Ctrl + Shift + E'
#   Test Package:   'Ctrl + Shift + T'
#' Facebook: Heartbeat Creator
#'
#' This function pulls the data from your messages folder created by the facebook_message_pull function.
#' The sentiment of the data is then calculated and a "heartbeat" chart is created. The chart spans the entire length of the
#' conversation that is present in your messenger data.
#'
#'
#' @param folder filepath to data
#' @keywords facebook_heartbeat
#' @return Sentiment image showing valence of message.
#' @name facebook_heartbeat
#' @aliases facebook_heartbeat
#' @import tm
#' @import syuzhet
#' @import stringr
#' @usage facebook_heartbeat(folder)
#' @title facebook_heartbeat
#' @examples
#'
#' library(facebookanalysis)
#'
#'  \dontrun{facebook_heartbeat(folder = data)}
#'
#' @export



facebook_heartbeat <- function(folder){
  # require(tm,quietly = TRUE)
  # require(syuzhet,quietly = TRUE)
  # require(stringr,quietly = TRUE)
  # requireNamespace("tm",quietly = TRUE)
  # requireNamespace('stringr',quietly = TRUE)
  # requireNamespace('syuzhet',quietly = TRUE)

  filelist <- list.files(path = paste0("./",folder),pattern = '.txt', full.names = TRUE)
  # print(filelist)
  for(f in 1:length(filelist)){
    ##Find the file path to the folder the facebook messages are in
    text_file_path <- file.path(paste0("./",folder))
    ##Create the corpus from the folder the messages are in
    document <- Corpus(DirSource(text_file_path[1]))
    ##Unlist the documents to allow for sentiment analysis
    docs <- unlist(document)
    ##Look at the ith (f) in the vector list
    docs <- docs[f]
    ##Get the file names
    file_name <- filelist[f]
    # print(paste0(file_name," printed"))
    ##Get NRC Sentiment from the ith (f) text file
    ##If directory does not exist create directory. Else if it exists write .txt data to directory
    if(!dir.exists("./heartbeat")){
      dir.create("./heartbeat")
    }else{
      ##Create the file name from the filelist and name the .csv file this way
      # name<- gsub(pattern = './messages',replacement = "",x = file_name)
      # name <- gsub(pattern = ".txt*",replacement = "",x = name)
      # write.csv(x = values,file = paste0('./nrc_sentences/',name,' sentences.csv'))
    }
    s_v <- get_sentences(docs)
    s_v_sentiment <- get_sentiment(s_v)
    name<- gsub(pattern = './messages',replacement = "",x = file_name)
    name <- gsub(pattern = ".txt*",replacement = "",x = name)
    name <- gsub(pattern = "/",replacement = "",x = name)
    myfile_path<- file.path(".","heartbeat",paste0(name,"timeline heartbeat.pdf"))
    pdf(file = myfile_path)
    plot(
      s_v_sentiment,
      type="l",
      main= paste0(name," Messenger Timeline"),
      xlab = "Messenger Timeline",
      ylab= "Emotional Valence"
    )
    dev.off()

    myfile_path<- file.path(".","heartbeat",paste0(name,"timeline heartbeat.png"))
    png(file = myfile_path)
    plot(
      s_v_sentiment,
      type="l",
      main= paste0(name," Messenger Timeline"),
      xlab = "Messenger Timeline",
      ylab= "Emotional Valence",
      col = 'blue'
    )
    dev.off()
    print(paste0(name," facebook heartbeat charts created."))
  }


}