ruby_danfe 1.13.2 → 1.15.0

data/lib/ruby_danfe/danfe_nfce_generator.rb

@@ -1,10 +1,11 @@
 # encoding:utf-8
+require 'digest/sha1'
 
 module RubyDanfe
   class DanfeNfceGenerator
     def initialize(xml)
       @xml = xml
-      @pdf = Document.new
+      @pdf = Document.new(page_size: [390, 845])
     end
 
     def generatePDF
@@ -34,7 +35,7 @@ module RubyDanfe
       @pdf.ibox 2.85, 3, 1.5, 2, "", "NFC-e", {:size => 12, :align => :center, :border => 0, :style => :bold}
 
       @pdf.ibox 4.92, 7, 4, 0, '',
-        @xml['emit/xNome'] + "\n" + 
+        @xml['emit/xNome'] + "\n" +
         "CNPJ: " + @xml['emit/CNPJ'] + "\n" +
         @xml['enderEmit/xLgr'] + ", " + @xml['enderEmit/nro'] + "\n" +
         @xml['enderEmit/xBairro']+ " " + "-" + " " + @xml['enderEmit/xMun'] + "/" + @xml['enderEmit/UF'] + "\n" +
@@ -42,22 +43,22 @@ module RubyDanfe
         "CEP: " + @xml['enderEmit/CEP'] + " - " + "IE: " + @xml['emit/IE'], {:align => :center, :valign => :center, size: 8, border: 0}
     end
 
-    def render_info_fixas   
+    def render_info_fixas
       @pdf.ibox 2, 12, 0.9, 4, '',
-        "DANFE NFC-e - Documento Auxiliar da Nota Fiscal Eletrônica" + "\n" + "para Consumidor Final" + "\n \n" + 
+        "DANFE NFC-e - Documento Auxiliar da Nota Fiscal Eletrônica" + "\n" + "para Consumidor Final" + "\n \n" +
         "Não permite aproveitamento de crédito de ICMS", {align: :center, :valign => :center, size: 8}
     end
 
-    def render_detalhes_venda      
+    def render_detalhes_venda
       render_cabecalho_dos_produtos
-      
+
       totais =  @xml.css('total')
-      
+
       @pdf.inumeric 0.70, 4, 0.9, 15.8, "Subtotal", totais.css('vProd').text, {:size => 6}
       @pdf.inumeric 0.70, 4, 4.9, 15.8, "Desconto", totais.css('vDesc').text, {:size => 6}
       @pdf.inumeric 0.70, 4, 8.9, 15.8, "Troco", totais.css('vTroco').text, {:size => 6}
-    end   
-    
+    end
+
     def render_cabecalho_dos_produtos
       @pdf.ibox 9.55, 1.3, 0.9, 6.1, "CÓDIGO", "",{:size => 8, :align => :center, :style => :bold}
       @pdf.ibox 9.55, 1.3, 2.2, 6.1, "UNIDADE"
@@ -108,7 +109,7 @@ module RubyDanfe
       return row
     end
 
-    def render_totais  
+    def render_totais
       qtde_produtos =   @xml.css('det').count
       forma_pgto    =   get_forma_pgto
       valor_total   =   Helper.numerify(@xml.css('total').css('vProd').text)
@@ -120,7 +121,7 @@ module RubyDanfe
 
     def get_forma_pgto
       cod_forma_pgto = @xml.css('tPag').text
-      case cod_forma_pgto 
+      case cod_forma_pgto
       when "01" then "Dinheiro"
         when "02" then "Cheque"
         when "03" then "Cartão de Crédito"
@@ -143,7 +144,7 @@ module RubyDanfe
         "Informação dos Tributos Totais Incidentes (Lei Federal 12.741 /2012):   #{soma_tributos}", {align: :center, :valign => :center, size: 7}
     end
 
-    def render_mensagem_fiscal    
+    def render_mensagem_fiscal
       numero_nota = @xml.css('nNF').text
       serie = @xml.css('serie').text
       data = @xml.css('dhEmi').text
@@ -158,30 +159,38 @@ module RubyDanfe
       end
 
       @pdf.ibox 3, 12, 0.9, 18.9, 'Mensagem Fiscal',
-        "#{cabecalho}" + 
-        "Número #{numero_nota} | Série #{serie} | Emissão #{data_emissao} - Via Consumidor" + 
-        "\n \n" + "Consulte pela Chave de Acesso em sistemas.sefaz.am.gov.br/nfceweb/formConsulta.do" + 
+        "#{cabecalho}" +
+        "Número #{numero_nota} | Série #{serie} | Emissão #{data_emissao} - Via Consumidor" +
+        "\n \n" + "Consulte pela Chave de Acesso em #{get_url_consulta}" +
         "\n \n" + "CHAVE DE ACESSO:" +
-        "\n \n" + chave_acesso + "\n", {align: :center, :valign => :center, size: 7} 
+        "\n \n" + chave_acesso + "\n", {align: :center, :valign => :center, size: 7}
     end
 
     def get_chave
-      chave = @xml.css('infNFe').attr("Id").value.gsub(/^(NFe|CTe)/, "")      
+      chave = @xml.css('infNFe').attr("Id").value.gsub(/^(NFe|CTe)/, "")
       return "%s %s %s %s %s %s %s %s %s %s %s" %[chave[0..3], chave[4..7], chave[8..11], chave[12..15], chave[16..19], chave[20..23], chave[24..27], chave[28..31], chave[32..35], chave[36..39], chave[40..43]]
     end
 
-    def render_consumidor   
+    def render_consumidor
       consumidor = @xml.xpath("//dest")
-      cpf = @xml.regex_string(consumidor.to_s.downcase, "//cpf").text
       nome = @xml.regex_string(consumidor.to_s.downcase, "//xnome").text.split.map(&:capitalize).join(' ')
-      
+
       endereco = @xml.regex_string(consumidor.to_s.downcase, "//xlgr").text + " - " + @xml.regex_string(consumidor.to_s.downcase, "//xbairro").text
-      endereco += " - " + @xml.regex_string(consumidor.to_s.downcase, "//xmun").text + " - " + @xml.regex_string(consumidor.to_s.downcase, "//uf").text
+      endereco += " - " + @xml.regex_string(consumidor.to_s.downcase, "//xmun").text + " - " + @xml.regex_string(consumidor.to_s.downcase, "//uf").text.downcase
       endereco = endereco.split.map(&:capitalize).join(' ')
+      endereco = endereco.upcase
 
       if consumidor.any?
-        consumidor_msg = cpf.empty? ? "" : "CNPJ/CPF/ID Estrangeiro - #{cpf} - "
-        consumidor_msg += nome
+        documento = @xml.css('dest idEstrangeiro, dest CPF, dest CNPJ').text
+        tipo_documento, documento =
+          case documento.length
+            when 11 then ["CPF", "%s.%s.%s-%s" % documento.match(/(...)(...)(...)(..)/).captures]
+            when 14 then ["CNPJ", "%s.%s.%s/%s-%s" % documento.match(/(..)(...)(...)(....)(..)/).captures]
+            else ["ID Estrangeiro", documento]
+          end
+
+        consumidor_msg = nome
+        consumidor_msg += " - #{tipo_documento}: #{documento}"
         consumidor_msg += "\n" + endereco
       else
         consumidor_msg = "CONSUMIDOR NÃO IDENTIFICADO"
@@ -191,15 +200,16 @@ module RubyDanfe
         consumidor_msg, {align: :center, :valign => :top, size: 7}
     end
 
-    def render_qrcode      
-      @pdf.ibox 0.7, 12, 0.9, 23.3, '', ''
-      #@pdf.iqrcode 1.7, 8.2, 6.5, 25, get_qrcode_url
+    def render_qrcode
+      @pdf.iqrcode 5.18, 28, get_qrcode_url, 20
 
       data = @xml.css('dhRecbto').text
       data_receb = Date.parse(data).strftime('%d/%m/%Y %I:%M:%S')
 
       protocolo = @xml.css('nProt').text
-      @pdf.draw_text "\n" + "Protocolo de Autorização: #{protocolo} - Data: #{data_receb}", at: [100, 140], :size => 6
+      msg_protocolo = "Protocolo de Autorização: #{protocolo} - Data: #{data_receb}"
+
+      @pdf.ibox 0.7, 12, 0.9, 23.3, '', msg_protocolo, size: 6, align: :center, valign: :center
     end
 
     def get_qrcode_url
@@ -207,30 +217,59 @@ module RubyDanfe
     end
 
     def get_ender_consulta
-      tipo_ambiente = @xml.css('tpAmb')
-      uf = @xml.css('emit').css('UF').text.upcase
-      url = ""
-
-      if tipo_ambiente == "2" # HOMOLOGAÇÃO
-        case uf 
-          when "AC" then return "http://www.sefaznet.ac.gov.br/nfe/NFe.jsp?opc=3" 
-          when "AM" then return "http://sistemas.sefaz.am.gov.br/nfceweb/consultarNFCe.jsp?"
-          when "MA" then return "http://www.nfce.sefaz.ma.gov.br/portal/consultarNFCe.jsp"
-          when "MT" then return "http://www.sefaz.mt.gov.br/nfe/portal/consultanfce"
-          when "RN" then return "http://www.nfe.rn.gov.br/portal/consultarNFCe.jsp?"
-          when "RS" then return "https://www.sefaz.rs.gov.br/NFCE/NFCE-COM.aspx"
-          when "SE" then return "http://www.nfe.se.gov.br/portal/consultarNFCe.jsp?"
-        end
-      else # PRODUÇÃO
-        case uf 
-          when "AC" then return "http://hml.sefaznet.ac.gov.br" 
-          when "AM" then return "http://homnfe.sefaz.am.gov.br/nfceweb/consultarNFCe.jsp?"
-          when "MA" then return "http://www.hom.nfce.sefaz.ma.gov.br/portal/consultarNFCe.jsp"
-          when "MT" then return "http://www.hom.nfe.sefaz.mt.gov.br/portal/consultarNFCe.jsp"
-          when "RN" then return "http://www.hom.nfe.rn.gov.br/portal/consultarNFCe.jsp?"
-          when "RS" then return "https://www.sefaz.rs.gov.br/NFCE/NFCE-COM.aspx"
-          when "SE" then return "http://www.hom.nfe.se.gov.br/portal/consultarNFCe.jsp?"
-        end
+      url = get_url_qr_code
+
+      params = {
+        "chNFe" => get_chave.gsub(/ /, ""),
+        "nVersao" => @xml.css('nfeProc').attr("versao"),
+        "tpAmb" => @xml['ide/tpAmb'],
+        "dhEmi" => str_to_hex(@xml['ide/dhEmi']),
+        "vNF" => @xml['ICMSTot/vNF'],
+        "vICMS" => @xml['ICMSTot/vICMS'],
+        "digVal" => str_to_hex(@xml['nfe/DigestValue']),
+        "cIdToken" => '000001',
+      }
+
+      unless @xml.css('dest').empty?
+        params["cDest"] = @xml.css('dest idEstrangeiro, dest CPF, dest CNPJ').text
+      end
+
+      params_string = params.to_a.map{|p| p.join("=")}.join("&")
+
+      hash_token = Digest::SHA1.base64digest params_string
+
+      url + params_string + "&cHashQRCode=" + hash_token
+    end
+
+    def get_url_qr_code
+      case @xml.css('emit').css('UF').text.upcase
+        when "AC" then "http://www.sefaznet.ac.gov.br/nfe/NFe.jsp?opc=3"
+        when "AM" then "http://sistemas.sefaz.am.gov.br/nfceweb/consultarNFCe.jsp?"
+        when "MA" then "http://www.nfce.sefaz.ma.gov.br/portal/consultarNFCe.jsp?"
+        when "MT" then "http://www.sefaz.mt.gov.br/nfe/portal/consultanfce?"
+        when "RN" then "http://www.nfe.rn.gov.br/portal/consultarNFCe.jsp?"
+        when "RO" then "http://www.nfce.sefin.ro.gov.br/consultanfce/consulta.jsp?"
+        when "RS" then "https://www.sefaz.rs.gov.br/NFCE/NFCE-COM.aspx?"
+        when "SE" then "http://www.nfe.se.gov.br/portal/consultarNFCe.jsp?"
+      end
+    end
+
+    def get_url_consulta
+      case @xml.css('emit').css('UF').text.upcase
+        when "AC" then "http://www.sefaznet.ac.gov.br/nfe/NFe.jsp?opc=3"
+        when "AM" then "http://sistemas.sefaz.am.gov.br/nfceweb/consultarNFCe.jsp"
+        when "MA" then "http://www.nfce.sefaz.ma.gov.br/portal/consultarNFCe.jsp"
+        when "MT" then "http://www.sefaz.mt.gov.br/nfe/portal/consultanfce"
+        when "RN" then "http://www.nfe.rn.gov.br/portal/consultarNFCe.jsp"
+        when "RO" then "http://www.nfce.sefin.ro.gov.br"
+        when "RS" then "https://www.sefaz.rs.gov.br/NFCE/NFCE-COM.aspx"
+        when "SE" then "http://www.nfe.se.gov.br/portal/consultarNFCe.jsp"
+      end
+    end
+
+    def str_to_hex str
+      str.split("").reduce("") do |str, char|
+        str + char.ord.to_s(16)
       end
     end
   end

data/lib/ruby_danfe/danfse_generator.rb

@@ -0,0 +1,102 @@
+# coding: utf-8
+module RubyDanfe
+  class DanfseGenerator
+    def initialize(xml)
+      @xml = xml
+      @pdf = Document.new
+      @vol = 0
+    end
+
+    attr_reader :municipios
+
+    def municipios
+      lib_path = File.expand_path("../../", __FILE__)
+      @municipios ||= JSON.parse(File.read(File.join(lib_path, 'municipios.json')))
+    end
+
+    def generatePDF
+      render_titulo
+      render_prestador
+      render_tomador
+      render_intermediario
+      render_discriminacao
+      render_valor_total
+      render_outras
+      @pdf
+    end
+
+    private
+    def render_titulo
+      @pdf.ibox 2.55, 16.10, 0.25, 0.42, '',
+        "PREFEITURA DO MUNICÍPIO DE #{municipios[@xml['InfNfse/PrestadorServico/Endereco/CodigoMunicipio']].upcase} \n" +
+        "Secretaria Municipal da Fazenda \n" +
+        "NOTA FISCAL ELETRÔNICA DE SERVIÇOS - NFS-e \n" +
+        "RPS n° #{@xml['IdentificacaoRps/Numero']}, emitido em #{@xml['DataEmissaoRps']}", {:align => :center, :valign => :center}
+
+      @pdf.ibox 0.85, 4.47, 16.35, 0.42, "NÚMERO DA NOTA", "#{@xml['InfNfse/Numero'].rjust(8,'0')}"
+      @pdf.ibox 0.85, 4.47, 16.35, 1.27, "DATA E HORA DE EMISSÃO", "#{@xml['InfNfse/DataEmissao'].gsub('T', ' ')}"
+      @pdf.ibox 0.85, 4.47, 16.35, 2.12, "CÓDIGO DE VERIFICAÇÃO", "#{@xml['CodigoVerificacao']}"
+    end
+
+    def render_prestador
+      @pdf.ibox 4.25, 20.57, 0.25, 2.97
+      @pdf.ibox 0.85, 20.57, 0.25, 2.97, '', 'PRESTADOR DE SERVIÇOS', {border: 0, style: :bold,:align => :center, :valign => :center}
+      @pdf.ibox 0.85, 20.57, 0.25, 3.82, "Nome/Razão Social", "#{@xml['PrestadorServico/RazaoSocial']}", {border: 0}
+      @pdf.ibox 0.85, 12,    0.25, 4.67, "CPF/CNPJ", "#{@xml['PrestadorServico/IdentificacaoPrestador/Cnpj'] || @xml['PrestadorServico/IdentificacaoPrestador/Cpf']}", {border: 0}
+      @pdf.ibox 0.85, 4.47,  12,   4.67, "Inscrição Municipal", "#{@xml['IdentificacaoPrestador/InscricaoMunicipal']}", {border: 0}
+      @pdf.ibox 0.85, 20.57, 0.25, 5.52, "Endereço", "#{@xml['PrestadorServico/Endereco/Endereco']}", {border: 0}
+      @pdf.ibox 0.85, 10,    0.25, 6.37, "Município", "#{municipios[@xml['PrestadorServico/Endereco/CodigoMunicipio']]}", {border: 0}
+      @pdf.ibox 0.85, 4.47,  10,   6.37, "UF", "#{@xml['PrestadorServico/Endereco/Uf']}", {border: 0}
+      @pdf.ibox 0.85, 4.47,  15,   6.37, "E-mail", "#{@xml['PrestadorServico/Contato/Email']}", {border: 0}
+    end
+
+    def render_tomador
+      @pdf.ibox 4.25, 20.57, 0.25, 7.22
+      @pdf.ibox 0.85, 20.57, 0.25, 7.22, '', 'TOMADOR DE SERVIÇOS', {border: 0, style: :bold,:align => :center, :valign => :center}
+      @pdf.ibox 0.85, 20.57, 0.25, 8.07, "Nome/Razão Social", "#{@xml['TomadorServico/RazaoSocial']}", {border: 0}
+      @pdf.ibox 0.85, 12,    0.25, 8.92, "CPF/CNPJ", "#{@xml['TomadorServico/IdentificacaoTomador/CpfCnpj/Cnpj'] || @xml['TomadorServico/IdentificacaoTomador/CpfCnpj/Cpf']}", {border: 0}
+      @pdf.ibox 0.85, 4.47,  12,   8.92, "Inscrição Municipal", "#{@xml['IdentificacaoTomador/InscricaoMunicipal']}", {border: 0}
+      @pdf.ibox 0.85, 20.57, 0.25, 9.77, "Endereço", "#{@xml['TomadorServico/Endereco/Endereco']}", {border: 0}
+      @pdf.ibox 0.85, 10,    0.25, 10.62, "Município", "#{municipios[@xml['TomadorServico/Endereco/CodigoMunicipio']]}", {border: 0}
+      @pdf.ibox 0.85, 4.47,  10,   10.62, "UF", "#{@xml['TomadorServico/Endereco/Uf']}", {border: 0}
+      @pdf.ibox 0.85, 4.47,  15,   10.62, "E-mail", "#{@xml['TomadorServico/Contato/Email']}", {border: 0}
+    end
+
+    def render_intermediario
+      @pdf.ibox 1.70, 20.57, 0.25,  11.47
+      @pdf.ibox 0.85, 20.57, 0.25,  11.47, '', 'INTERMEDIÁRIO DE SERVIÇOS', {border: 0, style: :bold,:align => :center, :valign => :center}
+      @pdf.ibox 0.85, 12,    0.25,  12.32, "Nome/Razão Social", "#{@xml['IdentificacaoIntermediarioServico/RazaoSocial']}", {border: 0}
+      @pdf.ibox 0.85, 8,     12.25, 12.32, "CPF/CNPJ", "#{@xml['IdentificacaoIntermediarioServico/CpfCnpj/Cnpj'] || @xml['IdentificacaoIntermediarioServico/CpfCnpj/Cpf']}", {border: 0}
+    end
+
+    def render_discriminacao
+      @pdf.ibox 9.35, 20.57, 0.25,  13.17
+      @pdf.ibox 0.85, 20.57, 0.25,  13.17, '', 'DISCRIMINAÇÃO DOS SERVIÇOS', {border: 0, style: :bold,:align => :center, :valign => :center}
+      @pdf.ibox 8, 19.57, 0.75,  14.02, "", "#{@xml['Servico/Discriminacao']}", {border: 0}
+    end
+
+    def render_valor_total
+      @pdf.ibox 1.70, 20.57, 0.25,  22.52
+      @pdf.ibox 0.85, 20.57, 0.25,  22.52, '', "VALOR TOTAL DO SERVIÇO = R$#{Helper.numerify(@xml['Servico/Valores/ValorServicos'])}", {border: 0, style: :bold,:align => :center, :valign => :center}
+      @pdf.inumeric 0.85, 4.06,  0.25,  23.37, "INSS", @xml['Servico/Valores/ValorInss']
+      @pdf.inumeric 0.85, 4.06,  4.31,  23.37, "IRRF", @xml['Servico/Valores/ValorIr']
+      @pdf.inumeric 0.85, 4.06,  8.37,  23.37, "CSLL", @xml['Servico/Valores/ValorCsll']
+      @pdf.inumeric 0.85, 4.06,  12.43, 23.37, "COFINS", @xml['Servico/Valores/ValorCofins']
+      @pdf.inumeric 0.85, 4.32,  16.49, 23.37, "PIS/PASEP", @xml['Servico/Valores/ValorPis']
+      @pdf.ibox 0.85, 20.57, 0.25,  24.22, "Código do Serviço", @xml['Servico/CodigoTributacaoMunicipio']
+      @pdf.inumeric 0.85, 3.46,  0.25,  25.07, "Valor Total das Deduções", @xml['Servico/Valores/ValorDeducoes']
+      @pdf.inumeric 0.85, 3.46,  3.71,  25.07 , "Base de Cálculo", @xml['Servico/Valores/BaseCalculo']
+      @pdf.ibox 0.85, 3.46,  7.17,  25.07, "Alíquota", @xml['Servico/Valores/Aliquota']
+      @pdf.inumeric 0.85, 3.46,  10.63, 25.07, "Valor do ISS", @xml['Servico/Valores/ValorIss']
+      @pdf.inumeric 0.85, 6.73,  14.09, 25.07, "Crédito", @xml['InfNfse/ValorCredito']
+      @pdf.ibox 0.85, 10.38, 0.25,  25.92, "Muncípio da Prestação do Serviço", municipios[@xml['OrgaoGerador/CodigoMunicipio']], :style => :bold
+      @pdf.ibox 0.85, 10.19, 10.63, 25.92, "Número Inscrição da Obra", @xml['DadosConstrucaoCivil/CodigoObra'], :style => :bold
+    end
+
+    def render_outras
+      @pdf.ibox 2.55, 20.57, 0.25,  26.77
+      @pdf.ibox 0.85, 20.57, 0.25,  26.77, '', 'OUTRAS INFORMAÇÕES', {border: 0, style: :bold,:align => :center, :valign => :center}
+      @pdf.ibox 1.70, 19.57, 0.75,  27.62, "", "#{@xml['InfNfse/OutrasInformacoes']}", {border: 0}
+    end
+  end
+end

data/lib/ruby_danfe/descricao.rb

@@ -40,5 +40,6 @@ module RubyDanfe
     def self.need_st(det)
       det.css('ICMS/*/vBCST').text.to_i > 0
     end
+
   end
 end

data/lib/ruby_danfe/document.rb

@@ -1,14 +1,16 @@
 module RubyDanfe
   class Document
-    def initialize
-      @document = Prawn::Document.new(
+    def initialize(opts = {})
+      default_opts = {
         :page_size => 'A4',
         :page_layout => :portrait,
         :left_margin => 0,
         :right_margin => 0,
         :top_margin => 0,
         :botton_margin => 0
-      )
+      }
+
+      @document = Prawn::Document.new(default_opts.merge(opts))
 
       @document.font "Times-Roman"
     end
@@ -26,11 +28,12 @@ module RubyDanfe
     end
 
     def ibarcode(h, w, x, y, info)
+      info = info.gsub(/\D/, '')
       Barby::Code128C.new(info).annotate_pdf(self, :x => x.cm, :y => Helper.invert(y.cm), :width => w.cm, :height => h.cm) if info != ''
     end
 
-    def iqrcode(h, w, x, y, info)
-      Barby::QrCode.new(info, :level => :q).annotate_pdf(self, :x => x.cm, :y => Helper.invert(y.cm), :width => w.cm, :height => h.cm) if info != ''
+    def iqrcode(x, y, info, size = nil)
+      Barby::QrCode.new(info, :level => :q, :size => size).annotate_pdf(self, :x => x.cm, :y => Helper.invert(y.cm)) if info != ''
     end
 
     def irectangle(h, w, x, y)

data/lib/ruby_danfe/helper.rb

@@ -10,6 +10,16 @@ module RubyDanfe
         number
     end
 
+    def self.numerify_default_zero(number, decimals = 2)
+      number = number.tr("\n","").delete(" ")
+      return "0,00" if !number || number == ""
+      int, frac = ("%.#{decimals}f" % number).split(".")
+      int.gsub!(/(\d)(?=(\d\d\d)+(?!\d))/, "\\1\.")
+      int + "," + frac
+      rescue
+        number
+    end
+
     def self.invert(y)
       28.7.cm - y
     end

data/lib/ruby_danfe/railtie.rb

@@ -1,11 +1,16 @@
 require "prawn"
+require "prawn/table"
 require "prawn/measurement_extensions"
 require "barby"
 require "barby/barcode/code_128"
+require 'barby/barcode/qr_code'
 require "barby/outputter/prawn_outputter"
 require "nokogiri"
 require 'ostruct'
 require 'yaml'
+require 'date'
+require 'time'
+require 'json'
 
 require_relative '../ruby_danfe.rb'
 require_relative 'options.rb'
@@ -14,7 +19,10 @@ require_relative 'cst.rb'
 require_relative 'document.rb'
 require_relative 'version.rb'
 require_relative 'dacte_generator.rb'
+require_relative 'dacteos_generator.rb'
 require_relative 'danfe_generator.rb'
+require_relative 'danfse_generator.rb'
+require_relative 'danfe_nfce_generator.rb'
 require_relative 'xml.rb'
 require_relative 'descricao.rb'
 require_relative 'ruby_danfe.rb'

data/lib/ruby_danfe/ruby_danfe.rb

@@ -35,15 +35,15 @@ module RubyDanfe
 
       xml = XML.new(xml_string)
 
-      if type == :danfe
-        generator = DanfeGenerator.new(xml)
-      elsif type == :danfe_nfce
-        generator = DanfeNfceGenerator.new(xml)
-      else
-        generator = DacteGenerator.new(xml)
-      end
-
-      pdf = generator.generatePDF
-      pdf
+      generator =
+        case type
+          when :danfe then DanfeGenerator.new(xml)
+          when :danfe_nfce then DanfeNfceGenerator.new(xml)
+          when :dacte then DacteGenerator.new(xml)
+          when :danfse then DanfseGenerator.new(xml)
+          when :dacteos then DacteosGenerator.new(xml)
+          else raise "unknown type #{type}"
+        end
+      generator.generatePDF
     end
 end