import asyncio
import logging
import ssl
import socket
import sys
import uvloop

CREDENTIAL = ""

URLS = {
    "https://s3.us-west-2.amazonaws.com/archpi.dabase.com/style.css": {
        "method": "get",
        "headers": {
            "User-Agent": "Botocore/1.8.21 Python/3.6.4 Darwin/17.5.0",
            "X-Amz-Date": "20180518T025044Z",
            "X-Amz-Content-SHA256": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
            "Authorization": f"AWS4-HMAC-SHA256 Credential={CREDENTIAL}/20180518/us-west-2/s3/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=ae552641b9aa9a7a267fcb4e36960cd5863e55d91c9b45fd39b30fdcd2e81489",
            "Accept-Encoding": "identity",
        },
    },
    "https://s3.ap-southeast-1.amazonaws.com/archpi.dabase.com/doesnotexist": {
        "method": "GET" if sys.argv[1] == "get_object" else "HEAD",
        "headers": {
            "User-Agent": "Botocore/1.8.21 Python/3.6.4 Darwin/17.5.0",
            "X-Amz-Date": "20180518T025221Z",
            "X-Amz-Content-SHA256": "e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855",
            "Authorization": f"AWS4-HMAC-SHA256 Credential={CREDENTIAL}/20180518/ap-southeast-1/s3/aws4_request, SignedHeaders=host;x-amz-content-sha256;x-amz-date, Signature=7a7675ef6d70cb647ed59e02d532ffa80d437fb03976d8246ea9ef102d118794",
            "Accept-Encoding": "identity",
        },
    },
}

asyncio.set_event_loop_policy(uvloop.EventLoopPolicy())


class HttpClient(asyncio.streams.FlowControlMixin):
    transport = None

    def __init__(self, *args, **kwargs):
        self.__url = kwargs.pop("url")
        self.__logger = logging.getLogger()
        super().__init__()

    def connection_made(self, transport):
        self.transport = transport

        entry = URLS[self.__url]

        body = "HEAD /archpi.dabase.com/doesnotexist HTTP/1.1\r\nAccept: */*\r\nHost: s3.ap-southeast-1.amazonaws.com\r\n"
        for name, value in entry["headers"].items():
            body += f"{name}: {value}\r\n"

        body += "\r\n"
        self.transport.write(body.encode("ascii"))
        self.__logger.info(f'data sent: {body}')

    def data_received(self, data):
        self.__logger.info(f'data received: {data}')

        self.transport.close()
        # asyncio.get_event_loop().call_later(1.0, )

    def eof_received(self):
        self.__logger.info('eof_received')

    def connection_lost(self, exc):
        self.__logger.info(f'connection lost: {exc}')
        super().connection_lost(exc)

    @classmethod
    def create_factory(cls, url: str):
        def factory(*args, **kwargs):
            return cls(*args, url=url, **kwargs)

        return factory


async def test_asyncio(ssl_context):
    loop = asyncio.get_event_loop()

    url = "https://s3.ap-southeast-1.amazonaws.com/archpi.dabase.com/doesnotexist"
    port = 443
    host = "s3.ap-southeast-1.amazonaws.com"
    infos = await loop.getaddrinfo(host, port, family=socket.AF_INET)
    family, type, proto, canonname, sockaddr = infos[0]
    await loop.create_connection(
        HttpClient.create_factory(url),
        sockaddr[0],
        port,
        ssl=ssl_context,
        family=family,
        proto=proto,
        flags=socket.AI_NUMERICHOST,
        server_hostname=host,
        local_addr=None,
    )


async def asyncio_test():
    ssl_context = ssl.create_default_context()

    while True:
        await test_asyncio(ssl_context)


def main():
    print("running")
    loop = asyncio.get_event_loop()
    for i in range(16):
        loop.create_task(asyncio_test())
    loop.run_forever()


main()