R/mni_combinar_docs.R

In jjesusfilho/tjsp: Coleta e organização de dados do Tribunal de Justiça de São Paulo

#' Combina docs baixados com a partir de mni_base64_docs
#'
#' @param arquivos Vetor de arquivos
#' @param dir_origem Diretório de origem, se não
#'     informar arquivos
#' @param dir_destino Diretório destino
#' @param nivel Juntar processo inteiro ou por documento?
#'
#' @return único pdf
#' @export
#'
mni_combinar_docs <- function(arquivos = NULL,
                              dir_origem = ".",
                              dir_destino = NULL,
                              nivel = c("processo", "doc")){
  
  if (is.null(dir_destino) || !dir.exists(dir_destino)){
    
    stop("Voc\u00ea deve informar um diret\u00f3rio existente")
    
  }
  
  if (is.null(arquivos)){
    
    arquivos <- list.files(dir_origem, full.names = TRUE, pattern = "pdf$")
    
  }
  
  nivel = nivel[1]
  
  
  lista <- tibble::tibble(arquivos) |>
    dplyr::mutate(processo = stringr::str_extract(arquivos,"\\d{20}"),
                  id_documento = stringr::str_extract(arquivos,'(?<=documento_)\\d+'),
                  digito = stringr::str_extract(arquivos, "(?<=digito_)\\d+")) |> 
    dplyr::mutate(dplyr::across(id_documento:digito, as.integer)) |> 
    dplyr::arrange(processo,id_documento,digito)
  
  if (nivel == "processo"){
    
    lista <- dplyr::group_split(lista, processo)
    
    
    purrr::walk(lista, purrr::possibly(~{
      
      processo <- unique(.x$processo)
      
      qpdf::pdf_combine(.x$arquivos, file.path(dir_destino,paste0(processo,".pdf")))
      
    },NULL))
    
    
  } else {
    
    lista <- dplyr::group_split(lista, processo, id_documento)
    
    purrr::walk(lista, purrr::possibly(~{
      processo <- unique(.x$processo)
      id_documento <- unique(.x$id_documento)
      suppressWarnings(
        qpdf::pdf_combine(.x$arquivos, file.path(dir_destino,
                                                 paste0(processo,"_id_documento_",id_documento, ".pdf")))
      )
    }, NULL))
  }
  

}