# -*- coding: utf-8 -*-

from __future__ import division, print_function, unicode_literals


from pynfe.utils.flags import (
    NAMESPACE_MDFE,
    MODELO_MDFE,
    NAMESPACE_MDFE_METODO,
    NAMESPACE_SOAP,
    NAMESPACE_XSI,
    NAMESPACE_XSD,
)
from pynfe.utils.webservices import (
    MDFE_WS_URL,
    MDFE_WS_METODO,
    WS_MDFE_CONSULTA,
    WS_MDFE_STATUS_SERVICO,
    WS_MDFE_CONSULTA_NAO_ENCERRADOS,
    WS_MDFE_RECEPCAO,
    WS_MDFE_RET_RECEPCAO,
    WS_MDFE_RECEPCAO_EVENTO,
)
from .comunicacao import ComunicacaoSefaz

from mdfelib.v3_00.consStatServMDFe import TConsStatServ
from mdfelib.v3_00.consSitMDFe import TConsSitMDFe
from mdfelib.v3_00.consMDFeNaoEnc import TConsMDFeNaoEnc
from mdfelib.v3_00.enviMDFe import TEnviMDFe


class ComunicacaoMDFE(ComunicacaoSefaz):

    _modelo = MODELO_MDFE
    _namespace = NAMESPACE_MDFE
    _versao = '3.00'
    _ws_url = MDFE_WS_URL
    _ws_metodo = MDFE_WS_METODO
    _header = 'mdfeCabecMsg'
    _envio_mensagem = 'mdfeDadosMsg'
    _retorno_mensagem = 'mdfeRecepcaoResult'
    _namespace_metodo = NAMESPACE_MDFE_METODO
    _accept = False
    _soap_action = False
    _namespace_soap = NAMESPACE_SOAP
    _namespace_xsi = NAMESPACE_XSI
    _namespace_xsd = NAMESPACE_XSD
    _soap_version = 'soap12'

    def status_servico(self):

        url, metodo = self._get_url_metodo(WS_MDFE_STATUS_SERVICO)

        raiz = TConsStatServ(
            versao=self._versao,
            tpAmb=str(self._ambiente),
            xServ='STATUS',
        )
        raiz.original_tagname_ = 'consStatServMDFe'

        xml = self._construir_xml_soap(
            metodo, self._construir_etree_ds(raiz)
        )

        return self._post(url, xml)

    def consulta(self, chave):

        url, metodo = self._get_url_metodo(WS_MDFE_CONSULTA)
        raiz = TConsSitMDFe(
            versao=self._versao,
            tpAmb=str(self._ambiente),
            xServ='CONSULTAR',
            chMDFe=chave,
        )
        raiz.original_tagname_ = 'consSitMDFe'
        xml = self._construir_xml_soap(
            metodo,
            self._construir_etree_ds(raiz)
        )
        return self._post(url, xml)

    def consulta_nao_encerrados(self, cnpj):

        url, metodo = self._get_url_metodo(WS_MDFE_CONSULTA_NAO_ENCERRADOS)
        raiz = TConsMDFeNaoEnc(
            versao=self._versao,
            tpAmb=str(self._ambiente),
            xServ='CONSULTAR NÃO ENCERRADOS',
            CNPJ=cnpj,
        )
        raiz.original_tagname_ = 'consMDFeNaoEnc'
        xml = self._construir_xml_soap(
            metodo,
            self._construir_etree_ds(raiz)
        )
        return self._post(url, xml)

    def autorizacao(self, documento, id_lote=1):

        url, metodo = self._get_url_metodo(WS_MDFE_RECEPCAO)

        raiz = TEnviMDFe(
            versao=self._versao,
            idLote=id_lote,
            MDFe=documento,
        )
        raiz.original_tagname_ = 'enviMDFe'

        xml = self._construir_xml_soap(
            metodo,
            self._construir_etree_ds(raiz)
        )
        # Faz request no Servidor da Sefaz
        retorno = self._post(url, xml)

        # TODO: Processar o retorno
        return retorno