import json
import gzip
import sys
import os
import socket

from io import BytesIO

from .utils import timestamp, base64_encode
from .runtime import runtime_info


if runtime_info.PYTHON_2:
    from urllib2 import urlopen
    from urllib2 import Request
    from urllib import urlencode
else:
    from urllib.request import urlopen
    from urllib.request import Request
    from urllib.parse import urlencode


class APIRequest(object):
    def __init__(self, agent):
        self.agent = agent   

    def post(self, endpoint, payload):
        agent_key_64 = base64_encode(self.agent.get_option('agent_key') + ':').replace('\n', '')
        headers = {
            'Accept-Encoding': 'gzip',
            'Authorization': "Basic %s" % agent_key_64,
            'Content-Type': 'application/json',
            'Content-Encoding': 'gzip'
        }

        host_name = 'undefined'
        try:
            host_name = socket.gethostname()
        except Exception:
            self.agent.exception()

        req_body = {
            'runtime_type':    'python',
            'runtime_version': '{0.major}.{0.minor}.{0.micro}'.format(sys.version_info),
            'runtime_path':    sys.prefix,
            'agent_version':   self.agent.AGENT_VERSION,
            'app_name':        self.agent.get_option('app_name'),
            'app_version':     self.agent.get_option('app_version'),
            'app_environment': self.agent.get_option('app_environment'),
            'host_name':       self.agent.get_option('host_name', host_name),
            'process_id':      os.getpid(),
            'run_id':          self.agent.run_id,
            'run_ts':          self.agent.run_ts,
            'sent_at':         timestamp(),
            'payload':         payload,
        }

        gzip_out = BytesIO()
        with gzip.GzipFile(fileobj=gzip_out, mode="w") as out_file:
          out_file.write(json.dumps(req_body).encode('utf-8'))
          out_file.close()

        gzip_out_val = gzip_out.getvalue()
        if isinstance(gzip_out_val, str):
            req_body_gzip = bytearray(gzip_out.getvalue())
        else:
            req_body_gzip = gzip_out.getvalue()

        request = Request(
            url = self.agent.get_option('dashboard_address') + '/agent/v1/' + endpoint,
            data = req_body_gzip,
            headers = headers)

        response = urlopen(request, timeout = 20)

        result_data = response.read()

        if response.info():
            content_type = response.info().get('Content-Encoding')
            if content_type == 'gzip':
                result_data = gzip.GzipFile('', 'r', 0, BytesIO(result_data)).read()

        response.close()

        return json.loads(result_data.decode('utf-8'))


def python_version():
    [sys.version_info.major,'',sys.version_info.minor + sys.version_info.micro]