# -*- coding: utf-8 -*-
# © 2017 Danimar Ribeiro, Trustcode
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

import os
import hashlib
import base64
import requests
from pytrustnfe.xml import render_xml, sanitize_response
from pytrustnfe.certificado import extract_cert_and_key_from_pfx, save_cert_key
from pytrustnfe.nfse.assinatura import Assinatura


def _render(certificado, method, **kwargs):
    path = os.path.join(os.path.dirname(__file__), 'templates')
    xml_send = render_xml(path, '%s.xml' % method, False, **kwargs)

    cert, key = extract_cert_and_key_from_pfx(
        certificado.pfx, certificado.password)
    cert, key = save_cert_key(cert, key)
    signer = Assinatura(cert, key, certificado.password)
    xml_send = signer.assina_xml(xml_send, '')
    return xml_send


def _get_oauth_token(**kwargs):
    if kwargs['ambiente'] == 'producao':
        url = 'https://nfps-e.pmf.sc.gov.br/api/v1/autenticacao/oauth/token'
    else:
        url = 'https://nfps-e-hml.pmf.sc.gov.br/api/v1/autenticacao/oauth/token'

    m = hashlib.md5()
    secret = "%s:%s" % (kwargs["client_id"], kwargs["secret_id"])
    auth = base64.b64encode(secret.encode('utf-8'))
    headers = {
        "Content-Type": "application/x-www-form-urlencoded",
        "Authorization": "Basic %s" % auth.decode('utf-8').replace('\n', '')
    }
    m.update(kwargs["password"].encode('utf-8'))
    password = m.hexdigest().upper()

    dados = "grant_type=password&username=%s&password=%s&client_id=%s&client_secret=%s" % (
        kwargs["username"], password, kwargs["client_id"], kwargs["secret_id"])
    r = requests.post(url, data=dados, headers=headers)
    if r.status_code == 200:
        return r.json()
    else:
        return r.text


def _send(certificado, method, **kwargs):
    if kwargs['ambiente'] == 'producao':
        url = 'https://nfps-e.pmf.sc.gov.br/api/v1/processamento/notas/processa'
    else:
        url = 'https://nfps-e-hml.pmf.sc.gov.br/api/v1/processamento/notas/processa'

    xml_send = '<?xml version="1.0" encoding="UTF-8"?>' + kwargs['xml']

    base = dict(
        ambiente='homologacao', client_id="trustcode-tecnologia-client",
        secret_id="", username="",
        password=""
    )
    token = _get_oauth_token(**kwargs)

    kwargs.update({"numero": 1, 'access_token': token["access_token"]})

    headers = {"Accept": "application/xml",
               "Authorization": "Bearer %s" % kwargs['access_token']}
    r = requests.post(url, headers=headers, data=xml_send)
    print(r.status_code)
    if r.status_code != 200:
        raise Exception(r.text)

    print(r.text)
    response, obj = sanitize_response(r.text)
    return {
        'sent_xml': xml_send,
        'received_xml': response,
        'object': obj
    }


def xml_processar_nota(certificado, **kwargs):
    return _render(certificado, 'processar_nota', **kwargs)


def processar_nota(certificado, **kwargs):
    if "xml" not in kwargs:
        kwargs['xml'] = xml_processar_nota(certificado, **kwargs)
    return _send(certificado, 'processar_nota', **kwargs)


def consultar_nota(certificado, **kwargs):
    url = "https://nfps-e-hml.pmf.sc.gov.br/api/v1/consultas/notas/numero/%s" % (kwargs["numero"])
    url = 'https://nfps-e-hml.pmf.sc.gov.br/api/v1/consultas/notas/prestador/24158233000185?pagina=1'

    headers = {"Accept": "application/json",
               "Authorization": "Bearer %s" % kwargs['access_token']}
    r = requests.get(url, headers=headers)
    print(r.status_code)
    if r.status_code == 200:
        return r.text
    else:
        return r.text