R/firefox.R

In duncantl/RBrowserCookies: Get cookies from browsers

getLoginHandle =
function(host, cookie = getLoginCookie(host), ..., curl = getCurlHandle())
{
   curlSetOpt(..., cookie = cookie, curl = curl)
   curl
}

getLoginCookie = getLoginCookieFirefox =
function(host, cookiesDB = getFirefoxCookiesFile(), con = getCookiesDBCon(cookiesDB))
{
    if(missing(con))
        on.exit(dbDisconnect(con))
    
   ck = dbGetQuery(con, sprintf("SELECT * FROM moz_cookies WHERE host = '%s'", host))
   paste(ck$name, ck$value, sep = "=", collapse = ";")   
}


getFirefoxCookiesFile =
function()
{
   dir = "~/Library/Application Support/Firefox/profiles"
   ver = R.Version()
   if(grepl("cygwin", ver$os)) 
       dir = sprintf("C:/Users/%s/AppData/Roaming/Mozilla/Firefox/Profiles", Sys.getenv("USERNAME"))
    
   ff = list.files(dir, pattern = "^cookies.sqlite$", recursive = TRUE, full.names = TRUE)
   if(length(ff) == 0)
       stop("cannot determine Firefox profile")
   if(length(ff) > 1)
      ff[ which.max(file.info(ff)$mtime) ]
   else
      ff
}

getCookiesDBCon =
function(file, copyIfLocked = TRUE)
{
    drv = SQLite()

    tryCatch({ con = dbConnect(drv, file); dbListTables(con); con},
             error = function(e) {
                 if(grepl("locked", e$message)) {
                     tmp = tempfile()
                     file.copy(file, tmp)
                     con = dbConnect(drv, tmp)
                     reg.finalizer(con@ptr, function(obj) file.remove(tmp) )
                     con
                 } else
                     stop(e)
             })
}


mkCookie =
function(name, value)    
{
    if(is.data.frame(name) && missing(value)) {
        value = name$value
        name = name$name
    }

    paste(name, value, sep = "=", collapse = "; ")
}